Cleanup

Author: codetheweb

src/tests/cleanup/does-database-exist.ts

@@ -0,0 +1,35 @@
+import test from "ava"
+import pRetry from "p-retry"
+import { getTestPostgresDatabaseFactory } from "~/index"
+
+const getTestServer = getTestPostgresDatabaseFactory({
+  postgresVersion: process.env.POSTGRES_VERSION,
+})
+
+const NUM_OF_DEFAULT_POSTGRES_DATABASES = 1
+
+test("database that first test worker created should have been dropped", async (t) => {
+  const { pool } = await getTestServer(t)
+
+  await pRetry(
+    async () => {
+      const {
+        rows: [{ count }],
+      } = await pool.query(
+        'SELECT COUNT(*) FROM "pg_database" WHERE "datistemplate" = false'
+      )
+
+      // (Add one since we create a database in this test)
+      if (Number(count) !== NUM_OF_DEFAULT_POSTGRES_DATABASES + 1) {
+        throw new Error("Database was not dropped")
+      }
+    },
+    {
+      minTimeout: 100,
+      factor: 1,
+      maxRetryTime: 30_000,
+    }
+  )
+
+  t.pass()
+})

src/tests/hooks.test.ts

@@ -1,7 +1,6 @@
 import test from "ava"
 import { getTestPostgresDatabaseFactory } from "~/index"
 import { countDatabaseTemplates } from "./utils/count-database-templates"
-import { doesDatabaseExist } from "./utils/does-database-exist"
 
 test("beforeTemplateIsBaked", async (t) => {
   let wasHookCalled = false