Add running result cache

lib/run-tests/worker.js

@@ -1,10 +1,18 @@
 const path = require("path");
+const fs = require("fs");
+const crypto = require('crypto');
+const lmdb = require("lmdb");
+const strip = require('strip-comments');
 
 const BabelAgent = require("./babel-agent");
 const transpile = require("./transpile");
 
 const TEST_TIMEOUT = 60 * 1000; // 1 minute
 
+function sha1(str) {
+  return crypto.createHash('sha1').update(str, 'utf8').digest('hex');
+}
+
 function timeout(file, waitMs = TEST_TIMEOUT) {
   return new Promise((_resolve, reject) =>
     setTimeout(
@@ -14,23 +22,37 @@ function timeout(file, waitMs = TEST_TIMEOUT) {
   );
 }
 
-let agent, testRoot;
+let agent, testRoot, cacheDB;
 
 exports.setup = function (opts) {
   agent = new BabelAgent(opts);
   testRoot = opts.testRoot;
+  cacheDB = new lmdb.open("cache.lmdb", {
+    compression: true,
+  });
 };
 
 exports.runTest = async function (test) {
   let { attrs, contents, file } = test;
   const isModule = attrs.flags.module;
 
+  const cacheKey = sha1(JSON.stringify({ attrs, file }));
+  const fileContents = strip(fs.readFileSync(path.join(testRoot, file), "utf8"));
+  const allowCache = ["eval(", "require(", "import", "evalScript("].every((keyword) => !fileContents.includes(keyword));
+
   try {
     contents = transpile(contents, { features: attrs.features, isModule, isStrict: false });
   } catch (error) {
     return { result: "parser error", error };
   }
 
+  const hash = sha1(contents);
+
+  const cacheData = allowCache ? cacheDB.get(cacheKey) : null;
+  if (cacheData?.hash === hash) {
+    return cacheData.ret;
+  }
+
   let result;
   let ret;
   try {
@@ -42,16 +64,20 @@ exports.runTest = async function (test) {
       }),
       timeout(file),
     ]);
+
+    if (result.error) {
+      ret = { result: "runtime error", error: result.error };
+    } else {
+      ret = { result: "success", output: result.stdout };
+    }
   } catch (error) {
     agent.stop(); // kill process avoid 100% cpu usage
 
-    return { result: "timeout error", error };
+    ret = { result: "timeout error", error };
   }
 
-  if (result.error) {
-    ret = { result: "runtime error", error: result.error };
-  } else {
-    ret = { result: "success", output: result.stdout };
+  if (allowCache) {
+    cacheDB.put(cacheKey, { hash, ret });
   }
 
   return ret;