Fix test files for HMR projects

Author: Ives van Hoorne

packages/app/src/sandbox/eval/manager.js

@@ -273,12 +273,13 @@ export default class Manager {
    * Will transpile this module and all eventual children (requires) that go with it
    * @param {*} entry
    */
-  async transpileModules(entry: Module, isEntry: boolean = true) {
+  async transpileModules(entry: Module, isTestFile: boolean = false) {
     this.hmrStatus = 'check';
     this.setEnvironmentVariables();
     const transpiledModule = this.getTranspiledModule(entry);
 
-    transpiledModule.setIsEntry(isEntry);
+    transpiledModule.setIsEntry(true);
+    transpiledModule.setIsTestFile(isTestFile);
 
     const result = await transpiledModule.transpile(this);
     this.getTranspiledModules().forEach(t => t.postTranspile(this));

packages/app/src/sandbox/eval/tests/jest-lite.js

@@ -90,7 +90,7 @@ export default class TestRunner {
 
   async transpileTests() {
     for (let t of this.tests) {
-      await this.manager.transpileModules(t, false);
+      await this.manager.transpileModules(t, true);
     }
   }
 

packages/app/src/sandbox/eval/transpiled-module.js

@@ -125,6 +125,8 @@ export default class TranspiledModule {
   // Unique identifier
   hash: string;
 
+  isTestFile: boolean = false;
+
   /**
    * Set how this module handles HMR. The default is undefined, which means
    * that we handle the HMR like CodeSandbox does.
@@ -189,6 +191,7 @@ export default class TranspiledModule {
     this.resetTranspilation();
 
     this.setIsEntry(false);
+    this.setIsTestFile(false);
   }
 
   resetTranspilation() {
@@ -389,6 +392,15 @@ export default class TranspiledModule {
     this.isEntry = isEntry;
   }
 
+  /**
+   * Mark if this is a test file. If this is a test file we know that we don't
+   * need to do any refresh or fixing when an error is thrown by the module. It's
+   * not a vital module after all.
+   */
+  setIsTestFile(isTestFile: boolean) {
+    this.isTestFile = isTestFile;
+  }
+
   /**
    * Transpile the module, it takes in all loaders from the default loaders +
    * query string and passes the result from loader to loader. During transpilation
@@ -542,7 +554,7 @@ export default class TranspiledModule {
       if (!this.compilation) {
         const shouldReloadPage = this.hmrConfig
           ? this.hmrConfig.isDeclined(this.isEntry)
-          : this.isEntry;
+          : this.isEntry && !this.isTestFile;
 
         if (shouldReloadPage) {
           location.reload();