1 files changed, 36 insertions, 0 deletions
diff --git a/test/dotty/tools/ContextEscapeDetection.java b/test/dotty/tools/ContextEscapeDetection.java
new file mode 100644
index 000000000..6a47de5c6
--- /dev/null
+++ b/test/dotty/tools/ContextEscapeDetection.java
@@ -0,0 +1,36 @@
+package dotty.tools;
+
+import dotty.tools.dotc.core.Contexts;
+import org.junit.*;
+
+import java.lang.ref.WeakReference;
+import java.util.LinkedList;
+import java.util.List;
+
+public abstract class ContextEscapeDetection {
+    public static class TestContext{
+        public TestContext(WeakReference<Contexts.Context> context, String testName) {
+            this.context = context;
+            this.testName = testName;
+        }
+
+        public final WeakReference<Contexts.Context> context;
+        public final String testName;
+
+    }
+    public static final List<TestContext> contexts = new LinkedList<TestContext>();
+
+    public abstract Contexts.Context getCtx();
+
+    public abstract void clearCtx();
+
+    @Before
+    public synchronized void stealContext() {
+        contexts.add(new TestContext(new WeakReference<Contexts.Context>(this.getCtx()), this.getClass().getName()));
+    }
+
+    @After
+    public synchronized void clearContext() {
+        this.clearCtx();
+    }
+}