refactor: use orchestrator.clearDatabase() in tests

joseallef · Sep 14, 2024 · 9a24eb4 · 9a24eb4
1 parent b9dd19c
commit 9a24eb4
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/tests/integration/api/v1/migrations/get.test.js b/tests/integration/api/v1/migrations/get.test.js
@@ -1,10 +1,9 @@
 /* eslint-disable jest/no-commented-out-tests */
-import database from "infra/database.js";
 import orchestrator from "tests/orchestrator.js";
 
 beforeAll(async () => {
   await orchestrator.waitForAllServices();
-  await database.query("drop schema public cascade; create schema public");
+  await orchestrator.clearDatabase();
 });
 
 describe("GET /api/v1/migrations", () => {

diff --git a/tests/integration/api/v1/migrations/post.test.js b/tests/integration/api/v1/migrations/post.test.js
@@ -1,10 +1,9 @@
 /* eslint-disable jest/no-commented-out-tests */
-import database from "infra/database.js";
 import orchestrator from "tests/orchestrator.js";
 
 beforeAll(async () => {
   await orchestrator.waitForAllServices();
-  await database.query("drop schema public cascade; create schema public");
+  await orchestrator.clearDatabase();
 });
 
 describe("POST /api/v1/migrations", () => {

diff --git a/tests/orchestrator.js b/tests/orchestrator.js
@@ -1,4 +1,5 @@
 import retry from "async-retry";
+import database from "infra/database.js";
 
 async function waitForAllServices() {
   await waitForWebServer();
@@ -18,7 +19,13 @@ async function waitForAllServices() {
     }
   }
 }
+
+async function clearDatabase() {
+  await database.query("drop schema public cascade; create schema public");
+}
+
 const orchestrator = {
   waitForAllServices,
+  clearDatabase
 };
 export default orchestrator;