fix: tests

Author: pregnantboy

packages/backend/src/graphql/__tests__/mutations/tiles/create-table.itest.ts

@@ -43,13 +43,13 @@ describe('create table mutation', () => {
   it('should be able create tables with the same name', async () => {
     const table = await createTable(
       null,
-      { input: { name: 'Test Table', isBlank: false } },
+      { input: { name: 'Test Table', isBlank: false, databaseType: 'ddb' } },
       context,
     )
 
     const table2 = await createTable(
       null,
-      { input: { name: 'Test Table', isBlank: false } },
+      { input: { name: 'Test Table', isBlank: false, databaseType: 'ddb' } },
       context,
     )
     expect(table.name).toBe('Test Table')
@@ -58,7 +58,11 @@ describe('create table mutation', () => {
 
   it('should throw an error when table name is empty', async () => {
     await expect(
-      createTable(null, { input: { name: '', isBlank: false } }, context),
+      createTable(
+        null,
+        { input: { name: '', isBlank: false, databaseType: 'ddb' } },
+        context,
+      ),
     ).rejects.toThrow()
   })
 })

packages/backend/test/pg-global-setup.ts

@@ -24,8 +24,8 @@ export async function setup() {
     `PostgreSQL container started at port ${process.env.POSTGRES_PORT}`,
   )
 
-  const config = await import('../knexfile')
-  const client = knex(config.default)
+  const config = await (await import('../knexfile')).default
+  const client = knex(config)
 
   // manually running migrations since the programmatic API doesn't work
   // see issue here: https://github.com/knex/knex/issues/5323

packages/backend/test/pg-reset-db-setup.ts

@@ -8,19 +8,22 @@ import { afterEach, beforeEach } from 'vitest'
 
 import config from '../knexfile'
 
-const client = knex(config as Knex.Config)
-
 beforeEach(async () => {
+  const client = knex(config as Knex.Config)
+
   // manually running seeds for the same reasons
   const seedsToRun = readdirSync(config.seeds.directory)
   for (const seedFile of seedsToRun) {
     const { seed } = await import(join(config.seeds.directory, seedFile))
     await seed(client)
   }
+  await client.destroy()
   console.info(`vite: PostgreSQL seeds run`)
 })
 
 afterEach(async () => {
+  const client = knex(config as Knex.Config)
+
   // truncate all tables
   const tables = await client('pg_catalog.pg_tables')
     .select('tablename')
@@ -31,4 +34,7 @@ afterEach(async () => {
     await client.raw(`TRUNCATE TABLE "${table}" CASCADE`)
   }
   console.info(`vite: PostgreSQL tables truncated`)
+
+  // Close the database connection
+  await client.destroy()
 })