test: add RESET_DATABASE environment variable to reset database

ArkEcosystem · Jun 1, 2019 · 0836a05 · 0836a05
1 parent 0690a2f
commit 0836a05
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 9 deletions.
diff --git a/__tests__/integration/core-api/__support__/setup.ts b/__tests__/integration/core-api/__support__/setup.ts
@@ -35,7 +35,7 @@ const setUp = async () => {
     });
 
     const databaseService = app.resolvePlugin<Database.IDatabaseService>("database");
-    await databaseService.connection.roundsRepository.truncate();
+    await databaseService.reset();
     await databaseService.buildWallets();
     await databaseService.saveRound(round);
 

diff --git a/__tests__/integration/core-blockchain/__support__/setup.ts b/__tests__/integration/core-blockchain/__support__/setup.ts
@@ -3,10 +3,8 @@ import { setUpContainer } from "../../../utils/helpers/container";
 
 jest.setTimeout(60000);
 
-export async function setUp() {
-    return setUpContainer({ exit: "@arkecosystem/core-blockchain" });
-}
+process.env.RESET_DATABASE = "1";
 
-export async function tearDown(): Promise<void> {
-    await app.tearDown();
-}
+export const setUp = async () => setUpContainer({ exit: "@arkecosystem/core-blockchain" });
+
+export const tearDown = async (): Promise<void> => app.tearDown();
diff --git a/__tests__/integration/core-blockchain/blockchain.test.ts b/__tests__/integration/core-blockchain/blockchain.test.ts
@@ -70,8 +70,6 @@ describe("Blockchain", () => {
         // Workaround: Add genesis transactions to the exceptions list, because they have a fee of 0
         // and otherwise don't pass validation.
         configManager.set("exceptions.transactions", genesisBlock.transactions.map(tx => tx.id));
-
-        await blockchain.database.reset();
     });
 
     afterAll(async () => {

diff --git a/__tests__/integration/core-transaction-pool/__support__/setup.ts b/__tests__/integration/core-transaction-pool/__support__/setup.ts
@@ -43,6 +43,8 @@ export const setUp = async () => {
 };
 
 export const setUpFull = async () => {
+    process.env.RESET_DATABASE = "1";
+
     try {
         await setUpContainer({
             exit: "@arkecosystem/core-transaction-pool",

diff --git a/packages/core-database/src/database-service.ts b/packages/core-database/src/database-service.ts
@@ -44,6 +44,10 @@ export class DatabaseService implements Database.IDatabaseService {
     }
 
     public async init(): Promise<void> {
+        if (process.env.RESET_DATABASE) {
+            await this.reset();
+        }
+
         await this.createGenesisBlock();
 
         const lastBlock: Interfaces.IBlock = await this.getLastBlock();