Respond to PR feedback.

packages/server/src/api/routes/tests/queries/mongodb.spec.ts

@@ -6,7 +6,6 @@ import { MongoClient } from "mongodb"
 jest.unmock("mongodb")
 
 describe("/queries", () => {
-  let request = setup.getRequest()
   let config = setup.getConfig()
   let datasource: Datasource
 
@@ -21,18 +20,7 @@ describe("/queries", () => {
       transformer: "return data",
       readable: true,
     }
-
+    return await config.api.query.create({ ...defaultQuery, ...query })
-    const res = await request
-      .post(`/api/queries`)
-      .set(config.defaultHeaders())
-      .send({ ...defaultQuery, ...query })
-      .expect("Content-Type", /json/)
-
-    if (res.status !== 200) {
-      throw new Error(JSON.stringify(res.body))
-    }
-
-    return res.body as Query
   }
 
   afterAll(async () => {

packages/server/src/api/routes/tests/queries/postgres.spec.ts

@@ -25,7 +25,6 @@ DROP TABLE test_table;
 `
 
 describe("/queries", () => {
-  let request = setup.getRequest()
   let config = setup.getConfig()
   let datasource: Datasource
 
@@ -40,18 +39,7 @@ describe("/queries", () => {
       transformer: "return data",
       readable: true,
     }
-
+    return await config.api.query.create({ ...defaultQuery, ...query })
-    const res = await request
-      .post(`/api/queries`)
-      .set(config.defaultHeaders())
-      .send({ ...defaultQuery, ...query })
-      .expect("Content-Type", /json/)
-
-    if (res.status !== 200) {
-      throw new Error(JSON.stringify(res.body))
-    }
-
-    return res.body as Query
   }
 
   afterAll(async () => {

packages/server/src/integrations/tests/utils/mongodb.ts

@@ -4,25 +4,22 @@ import { GenericContainer, Wait, StartedTestContainer } from "testcontainers"
 let container: StartedTestContainer | undefined
 
 export async function start(): Promise<StartedTestContainer> {
-  if (!container) {
+  return await new GenericContainer("mongo:7.0-jammy")
-    container = await new GenericContainer("mongo:7.0-jammy")
     .withExposedPorts(27017)
     .withEnvironment({
       MONGO_INITDB_ROOT_USERNAME: "mongo",
       MONGO_INITDB_ROOT_PASSWORD: "password",
     })
     .withWaitStrategy(
-        Wait.forSuccessfulCommand(
+      Wait.forSuccessfulCommand(`mongosh --eval "db.version()"`)
-          `mongosh --eval "db.version()"`
-        ).withStartupTimeout(10000)
     )
     .start()
 }
-  return container
-}
 
 export async function datasource(): Promise<Datasource> {
-  const container = await start()
+  if (!container) {
+    container = await start()
+  }
   const host = container.getHost()
   const port = container.getMappedPort(27017)
   return {

packages/server/src/integrations/tests/utils/postgres.ts

@@ -4,8 +4,7 @@ import { GenericContainer, Wait, StartedTestContainer } from "testcontainers"
 let container: StartedTestContainer | undefined
 
 export async function start(): Promise<StartedTestContainer> {
-  if (!container) {
+  return await new GenericContainer("postgres:16.1-bullseye")
-    container = await new GenericContainer("postgres:16.1-bullseye")
     .withExposedPorts(5432)
     .withEnvironment({ POSTGRES_PASSWORD: "password" })
     .withWaitStrategy(
@@ -15,11 +14,11 @@ export async function start(): Promise<StartedTestContainer> {
     )
     .start()
 }
-  return container
-}
 
 export async function datasource(): Promise<Datasource> {
-  const container = await start()
+  if (!container) {
+    container = await start()
+  }
   const host = container.getHost()
   const port = container.getMappedPort(5432)