Extract constants for directorieswinh-test-file-coverage

1 files changed, 19 insertions, 10 deletions

diff --git a/scripts/frontend/file_test_coverage.js b/scripts/frontend/file_test_coverage.js
index e7626b22d96..7d1eb45d4bc 100755
--- a/scripts/frontend/file_test_coverage.js
+++ b/scripts/frontend/file_test_coverage.js
@@ -13,6 +13,19 @@
 const fs = require('fs');
 const path = require('path');
 
+const sourceDirectories = ['app/assets/javascripts'];
+const testDirectories = ['spec/javascripts', 'spec/frontend'];
+
+if (fs.existsSync('ee')) {
+  sourceDirectories.forEach(dir => {
+    sourceDirectories.push(`ee/${dir}`);
+  });
+
+  testDirectories.forEach(dir => {
+    testDirectories.push(`ee/${dir}`);
+  });
+}
+
 let numSourceFiles = 0;
 let numTestFiles = 0;
 
@@ -40,16 +53,6 @@ const countTestFiles = path =>
     }
   });
 
-countSourceFiles('app/assets/javascripts');
-countTestFiles('spec/javascripts');
-countTestFiles('spec/frontend');
-
-if (fs.existsSync('ee')) {
-  countSourceFiles('ee/app/assets/javascripts');
-  countTestFiles('ee/spec/javascripts');
-  countTestFiles('ee/spec/frontend');
-}
-
 function forEachFileIn(dirPath, callback) {
   fs.readdir(dirPath, (err, files) => {
     if (err) {
@@ -72,6 +75,12 @@ function forEachFileIn(dirPath, callback) {
   });
 }
 
+console.log(`Source directories: ${sourceDirectories.join(', ')}`);
+console.log(`Test directories: ${testDirectories.join(', ')}`);
+
+sourceDirectories.forEach(countSourceFiles);
+testDirectories.forEach(countTestFiles);
+
 process.on('exit', () => {
   console.log(`Source files: ${numSourceFiles}`);
   console.log(`Test files: ${numTestFiles}`);