cache sqlite db for testing

DeNA · Aug 26, 2021 · 7111fb7 · 7111fb7
1 parent 1a6bc20
commit 7111fb7
Show file tree

Hide file tree

Showing 8 changed files with 40 additions and 37 deletions.
diff --git a/packages/nota-server/package.json b/packages/nota-server/package.json
@@ -79,9 +79,14 @@
     "collectCoverage": true,
     "transform": {},
     "maxConcurrency": 1,
+    "maxWorkers": 1,
     "setupFilesAfterEnv": [
       "./testSetup.js"
     ],
-    "clearMocks": true
+    "globalSetup": "./testGlobalSetup.js",
+    "clearMocks": true,
+    "roots": [
+      "./src"
+    ]
   }
 }
diff --git a/packages/nota-server/src/lib/testUtils.js b/packages/nota-server/src/lib/testUtils.js
@@ -14,22 +14,34 @@ const {
 } = require("../models");
 const authUtils = require("./authUtils");
 const { spawn } = require("child-process-promise");
-// const fs = require("fs");
-// const config = require("../config");
+const fs = require("fs");
+const config = require("../config");
+const DB_FILE = config.db.storage;
+const DB_CLEAN_FILE = DB_FILE + ".tmp";
+const DB_DATA_JSON = DB_FILE + ".tmp.json";
 
-const resetTestDb = async () => {
+const setupDb = async () => {
   const options = { stdio: "inherit" };
   const sequelize = "node_modules/.bin/sequelize";
   try {
     // fs.unlinkSync("./" + config.db.storage);
     await spawn(sequelize, ["db:migrate:undo:all"]);
     await spawn(sequelize, ["db:migrate"]);
+    const data = await generateTestData();
+    fs.writeFileSync(DB_DATA_JSON, JSON.stringify(data));
+    fs.copyFileSync(DB_FILE, DB_CLEAN_FILE);
   } catch (error) {
     console.error(error);
     process.exit(1);
   }
 };
 
+const resetTestDb = async () => {
+  fs.copyFileSync(DB_CLEAN_FILE, DB_FILE);
+  const data = require(DB_DATA_JSON);
+  return data;
+};
+
 const createProject = async ({
   name = "sample project",
   status = Project.STATUS.READY,
@@ -734,19 +746,12 @@ const generateTestData = async function() {
 
 module.exports = {
   db: {
+    setupDb,
     resetTestDb,
     createUser,
     createProject,
     createTaskTemplate,
     createTask,
-    createMediaSource,
-    generateTestData: async () => {
-      try {
-        const data = await generateTestData();
-        return data;
-      } catch (error) {
-        console.error(error);
-      }
-    }
+    createMediaSource
   }
 };
diff --git a/packages/nota-server/src/routes/projects.test.js b/packages/nota-server/src/routes/projects.test.js
@@ -8,8 +8,7 @@ const { Project } = require("../models");
 let data;
 
 beforeAll(async () => {
-  await db.resetTestDb();
-  data = await db.generateTestData();
+  data = await db.resetTestDb();
 });
 
 describe("/api/projects", () => {
@@ -347,8 +346,7 @@ describe("/api/projects", () => {
       };
     });
     afterAll(async () => {
-      await db.resetTestDb();
-      data = await db.generateTestData();
+      data = await db.resetTestDb();
     });
 
     test("Should create project (app admin)", async () => {
@@ -441,8 +439,7 @@ describe("/api/projects", () => {
 
   describe("PUT /:id", () => {
     beforeEach(async () => {
-      await db.resetTestDb();
-      data = await db.generateTestData();
+      data = await db.resetTestDb();
     });
     test("Admin should be able to update project name", async () => {
       const response = await request(app)

diff --git a/packages/nota-server/src/routes/tasks.test.js b/packages/nota-server/src/routes/tasks.test.js
@@ -12,8 +12,7 @@ const taskExportService = require("../services/taskExportService");
 let data;
 
 beforeAll(async () => {
-  await db.resetTestDb();
-  data = await db.generateTestData();
+  data = await db.resetTestDb();
 });
 
 describe("/api/projects/:projectId/tasks", () => {
@@ -375,8 +374,7 @@ describe("/api/projects/:projectId/tasks", () => {
       };
     });
     afterAll(async () => {
-      await db.resetTestDb();
-      data = await db.generateTestData();
+      data = await db.resetTestDb();
     });
 
     test("Should queue task creation job (super admin)", async () => {
@@ -536,8 +534,7 @@ describe("/api/projects/:projectId/tasks", () => {
   });
   describe("PUT /:id", () => {
     afterAll(async () => {
-      await db.resetTestDb();
-      data = await db.generateTestData();
+      data = await db.resetTestDb();
     });
 
     test("Should be able to update name/description/status (super admin)", async () => {
@@ -768,8 +765,7 @@ describe("/api/projects/:projectId/tasks", () => {
   });
   describe("DELETE /:id", () => {
     afterAll(async () => {
-      await db.resetTestDb();
-      data = await db.generateTestData();
+      data = await db.resetTestDb();
     });
 
     test("Should be able to delete task (super admin)", async () => {
@@ -924,8 +920,7 @@ describe("/api/projects/:projectId/tasks", () => {
   });
   describe("POST /:id/refreshTaskItems", () => {
     afterAll(async () => {
-      await db.resetTestDb();
-      data = await db.generateTestData();
+      data = await db.resetTestDb();
     });
 
     test("Should queue task refresh job (super admin)", async () => {
@@ -1064,8 +1059,7 @@ describe("/api/projects/:projectId/tasks", () => {
 
   describe("POST /:id/export", () => {
     afterAll(async () => {
-      await db.resetTestDb();
-      data = await db.generateTestData();
+      data = await db.resetTestDb();
     });
 
     test("Should queue task export job (super admin)", async () => {

diff --git a/packages/nota-server/src/routes/templates.test.js b/packages/nota-server/src/routes/templates.test.js
@@ -8,8 +8,7 @@ const { TaskTemplate } = require("../models");
 let data;
 
 beforeAll(async () => {
-  await db.resetTestDb();
-  data = await db.generateTestData();
+  data = await db.resetTestDb();
 });
 
 describe("/api/projects/:projectId/templates", () => {

diff --git a/packages/nota-server/src/services/mediaSourceFetch.test.js b/packages/nota-server/src/services/mediaSourceFetch.test.js
@@ -34,8 +34,7 @@ const createMediaSource = async function() {
 };
 
 beforeAll(async () => {
-  await db.resetTestDb();
-  data = await db.generateTestData();
+  data = await db.resetTestDb();
   // @ts-ignore
   ds()._resetFilesForTest();
   mediaSource = await createMediaSource();

diff --git a/packages/nota-server/src/services/taskInitialize.test.js b/packages/nota-server/src/services/taskInitialize.test.js
@@ -25,8 +25,7 @@ const createTask = async function({
 };
 
 beforeAll(async () => {
-  await db.resetTestDb();
-  data = await db.generateTestData();
+  data = await db.resetTestDb();
   // @ts-ignore
   ds()._resetFilesForTest();
   mediaSource = await db.createMediaSource({

diff --git a/packages/nota-server/testGlobalSetup.js b/packages/nota-server/testGlobalSetup.js
@@ -0,0 +1,5 @@
+const { db } = require("./src/lib/testUtils");
+
+module.exports = async () => {
+  await db.setupDb();
+};