1 files changed, 42 insertions, 0 deletions

diff --git a/language-web/src/test/java/tools/refinery/language/web/tests/AwaitTerminationExecutorServiceProvider.java b/language-web/src/test/java/tools/refinery/language/web/tests/AwaitTerminationExecutorServiceProvider.java
new file mode 100644
index 00000000..b70d0ed5
--- /dev/null
+++ b/language-web/src/test/java/tools/refinery/language/web/tests/AwaitTerminationExecutorServiceProvider.java
@@ -0,0 +1,42 @@
+package tools.refinery.language.web.tests;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.ExecutorService;
+
+import org.eclipse.xtext.ide.ExecutorServiceProvider;
+
+import com.google.inject.Singleton;
+
+@Singleton
+public class AwaitTerminationExecutorServiceProvider extends ExecutorServiceProvider {
+        private List<RestartableCachedThreadPool> servicesToShutDown = new ArrayList<>();
+
+        @Override
+        protected ExecutorService createInstance(String key) {
+                var instance = new RestartableCachedThreadPool();
+                synchronized (servicesToShutDown) {
+                        servicesToShutDown.add(instance);
+                }
+                return instance;
+        }
+
+        public void waitForAllTasksToFinish() {
+                synchronized (servicesToShutDown) {
+                        for (var executorService : servicesToShutDown) {
+                                executorService.waitForAllTasksToFinish();
+                        }
+                }
+        }
+
+        @Override
+        public void dispose() {
+                super.dispose();
+                synchronized (servicesToShutDown) {
+                        for (var executorService : servicesToShutDown) {
+                                executorService.waitForTermination();
+                        }
+                        servicesToShutDown.clear();
+                }
+        }
+}