Resolve differences in Karma test bundle between CE and EE

spec/javascripts/test_bundle.js

@@ -122,10 +122,15 @@ afterEach(() => {
 const axiosDefaultAdapter = getDefaultAdapter();
 
 // render all of our tests
-const testsContext = require.context('.', true, /_spec$/);
-testsContext.keys().forEach(function(path) {
+const testContexts = [
+  require.context('spec', true, /_spec$/),
+  require.context('ee_spec', true, /_spec$/),
+];
+
+testContexts.forEach(context => {
+  context.keys().forEach(path => {
     try {
-    testsContext(path);
+      context(path);
     } catch (err) {
       console.log(err);
       console.error('[GL SPEC RUNNER ERROR] Unable to load spec: ', path);
@@ -135,6 +140,7 @@ testsContext.keys().forEach(function(path) {
         });
       });
     }
+  });
 });
 
 describe('test errors', () => {
@@ -204,19 +210,27 @@ if (process.env.BABEL_ENV === 'coverage') {
   ];
 
   describe('Uncovered files', function() {
-    const sourceFiles = require.context('~', true, /\.(js|vue)$/);
+    const sourceFilesContexts = [
+      require.context('~', true, /\.(js|vue)$/),
+      require.context('ee', true, /\.(js|vue)$/),
+    ];
+    const allTestFiles = testContexts.reduce(
+      (accumulator, context) => accumulator.concat(context.keys()),
+      [],
+    );
 
     $.holdReady(true);
 
-    sourceFiles.keys().forEach(function(path) {
+    sourceFilesContexts.forEach(context => {
+      context.keys().forEach(path => {
         // ignore if there is a matching spec file
-      if (testsContext.keys().indexOf(`${path.replace(/\.(js|vue)$/, '')}_spec`) > -1) {
+        if (allTestFiles.indexOf(`${path.replace(/\.(js|vue)$/, '')}_spec`) > -1) {
           return;
         }
 
         it(`includes '${path}'`, function() {
           try {
-          sourceFiles(path);
+            context(path);
           } catch (err) {
             if (troubleMakers.indexOf(path) === -1) {
               expect(err).toBeNull();
@@ -225,4 +239,5 @@ if (process.env.BABEL_ENV === 'coverage') {
         });
       });
     });
+  });
 }