Only compile unittests in root modules

Author: marler8997

src/dmd/dmodule.d

@@ -292,8 +292,8 @@ extern (C++) final class Module : Package
     extern (C++) static __gshared Dsymbols deferred3;   // deferred Dsymbol's needing semantic3() run on them
     extern (C++) static __gshared uint dprogress;       // progress resolving the deferred list
     /**
-     * A callback function that is called once an imported module is
-     * parsed. If the callback returns true, then it tells the
+     * A callback function that is called when a module is loaded as soon as its
+     * module declaration has been parsed. If the callback returns true, then it tells the
      * frontend that the driver intends on compiling the import.
      */
     extern (C++) static __gshared bool function(Module mod) onImport;
@@ -377,6 +377,7 @@ extern (C++) final class Module : Package
      * ownership of template instantiation.
      */
     Module importedFrom;
+    bool isCompiledImport;
 
     Dsymbols* decldefs;         // top level declarations for this Module
 
@@ -515,41 +516,6 @@ extern (C++) final class Module : Package
             message("import    %s", buf.peekString());
         }
         m = m.parse();
-
-        // Call onImport here because if the module is going to be compiled then we
-        // need to determine it early because it affects semantic analysis. This is
-        // being done after parsing the module so the full module name can be taken
-        // from whatever was declared in the file.
-
-        //!!!!!!!!!!!!!!!!!!!!!!!
-        // Workaround for bug in dmd version 2.068.2 platform Darwin_64_32.
-        // This is the compiler version that the autotester uses, and this code
-        // has been carefully crafted using trial and error to prevent a seg fault
-        // bug that occurs with that version of the compiler.  Note, this segfault
-        // does not occur on the next version of dmd, namely, version 2.069.0. If
-        // the autotester upgrades to that version, then this workaround can be removed.
-        //!!!!!!!!!!!!!!!!!!!!!!!
-        version(OSX)
-        {
-            if (!m.isRoot() && onImport)
-            {
-                auto onImportResult = onImport(m);
-                if(onImportResult)
-                {
-                    m.importedFrom = m;
-                    assert(m.isRoot());
-                }
-            }
-        }
-        else
-        {
-            if (!m.isRoot() && onImport && onImport(m))
-            {
-                m.importedFrom = m;
-                assert(m.isRoot());
-            }
-        }
-
         Target.loadModule(m);
         return m;
     }
@@ -1288,9 +1254,9 @@ extern (C++) final class Module : Package
         return false;
     }
 
-    bool isRoot()
+    bool isRoot() const
     {
-        return this.importedFrom == this;
+        return this.importedFrom == this || isCompiledImport;
     }
 
     // true if the module source file is directly

src/dmd/glue.d

@@ -752,8 +752,17 @@ void FuncDeclaration_toObjFile(FuncDeclaration fd, bool multiobj)
     if (!fd.fbody)
         return;
 
+    // Find module m for this function
+    Module m = null;
+    for (Dsymbol p = fd.parent; p; p = p.parent)
+    {
+        m = p.isModule();
+        if (m)
+            break;
+    }
+
     UnitTestDeclaration ud = fd.isUnitTestDeclaration();
-    if (ud && !global.params.useUnitTests)
+    if (ud && (!global.params.useUnitTests || !m.isRoot))
         return;
 
     if (multiobj && !fd.isStaticDtorDeclaration() && !fd.isStaticCtorDeclaration())
@@ -886,15 +895,6 @@ void FuncDeclaration_toObjFile(FuncDeclaration fd, bool multiobj)
     symtab_t *symtabsave = cstate.CSpsymtab;
     cstate.CSpsymtab = &f.Flocsym;
 
-    // Find module m for this function
-    Module m = null;
-    for (Dsymbol p = fd.parent; p; p = p.parent)
-    {
-        m = p.isModule();
-        if (m)
-            break;
-    }
-
     Dsymbols deferToObj;                   // write these to OBJ file later
     Array!(elem*) varsInScope;
     Label*[void*] labels = null;

src/dmd/module.h

@@ -101,6 +101,7 @@ class Module : public Package
     // i.e. a module that will be taken all the
     // way to an object file
     Module *importedFrom;
+    bool isCompiledImport;
 
     Dsymbols *decldefs;         // top level declarations for this Module
 
@@ -144,7 +145,7 @@ class Module : public Package
     static void clearCache();
     int imports(Module *m);
 
-    bool isRoot() { return this->importedFrom == this; }
+    bool isRoot() const { return this->importedFrom == this || this->isCompiledImport; }
     // true if the module source file is directly
     // listed in command line.
     bool isCoreModule(Identifier *ident);

src/dmd/parse.d

@@ -377,6 +377,16 @@ final class Parser(AST) : Lexer
             }
         }
 
+        // Call onImport here because if the module is going to be compiled then we
+        // need to determine it early because it affects whether unittests will be parsed
+        // and semantic analysis. This is being done right after parsing the module delclaration
+        // so the full module name can be taken from whatever was declared in the file.
+        if (!mod.isRoot() && AST.Module.onImport && AST.Module.onImport(mod))
+        {
+            mod.isCompiledImport = true;
+            assert(mod.isRoot());
+        }
+
         decldefs = parseDeclDefs(0, &lastDecl);
         if (token.value != TOK.endOfFile)
         {
@@ -547,7 +557,7 @@ final class Parser(AST) : Lexer
                     break;
                 }
             case TOK.unittest_:
-                if (global.params.useUnitTests || global.params.doDocComments || global.params.doHdrGeneration)
+                if ( (global.params.useUnitTests || global.params.doDocComments || global.params.doHdrGeneration) && mod.isRoot)
                 {
                     s = parseUnitTest(pAttrs);
                     if (*pLastDecl)