Add getVmContext and clean up vm context in teardown

lh0x00 · Feb 21, 2020 · 7ff4bfc · 7ff4bfc
1 parent 2f7b6d1
commit 7ff4bfc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/packages/jest-environment-jsdom/src/index.ts b/packages/jest-environment-jsdom/src/index.ts
@@ -127,6 +127,7 @@ class JSDOMEnvironment implements JestEnvironment {
     // @ts-ignore
     this.global = null;
     this.dom = null;
+    this.vm = null;
     this.fakeTimers = null;
     this.fakeTimersLolex = null;
   }
@@ -135,8 +136,13 @@ class JSDOMEnvironment implements JestEnvironment {
     if (this.vm) {
       return script.runInContext(this.vm);
     }
+
     return null;
   }
+
+  getVmContext() {
+    return this.vm;
+  }
 }
 
 export = JSDOMEnvironment;