fix integration tests

Author: n1ru4l

.github/workflows/ci.yml

@@ -103,6 +103,9 @@ jobs:
       - name: Build Packages
         run: pnpm build
 
+      - name: Start Docker
+        run: docker compose -f examples/hackernews/docker-compose.yml up -d --wait
+
       - name: Run Tests
         uses: nick-fields/retry@v3
         with:

examples/hackernews/__integration-tests__/hackernews.spec.ts

@@ -1,57 +1,51 @@
-import path from 'node:path';
+import { resolve } from 'node:path';
+import { sql } from 'drizzle-orm';
+import { drizzle, NodePgDatabase } from 'drizzle-orm/node-postgres';
+import { migrate } from 'drizzle-orm/postgres-js/migrator';
 import { createYoga, YogaServerInstance } from 'graphql-yoga';
-import { PrismaClient } from '@prisma/client';
-import { DbDrop, MigrateDev } from '@prisma/migrate';
-import type { GraphQLContext } from '../src/context';
-import { schema } from '../src/schema';
+import pg from 'pg';
+import type { GraphQLContext } from '../src/context.js';
+import { schema } from '../src/schema.js';
+
+const connectionString =
+  process.env['PG_CONNECTION_STRING'] ??
+  'postgres://postgres:postgres@localhost:5432/postgres?currentSchema=integration-tests';
+
+export async function resetDatabase(db: NodePgDatabase, schema: string) {
+  // sql query for resetting the database
+  const query = sql`
+    DROP SCHEMA IF EXISTS ${sql.raw(schema)} CASCADE;
+    CREATE SCHEMA ${sql.raw(schema)};
+  `;
+  await db.execute(query);
+}
 
 describe('hackernews example integration', () => {
   let yoga: YogaServerInstance<GraphQLContext, GraphQLContext>;
+  let db: NodePgDatabase;
+  let client: pg.Client;
+  let currentSchema: string;
   beforeAll(async () => {
-    const { createContext } = await import('../src/context');
-    yoga = createYoga({ schema, context: createContext });
-
-    // migrate
-    await MigrateDev.new().parse([
-      `--schema=${path.resolve(__dirname, '..', 'prisma', 'schema.prisma')}`,
-    ]);
-
-    // seed
-    const client = new PrismaClient();
-    await client.link.create({
-      data: {
-        url: 'https://www.prisma.io',
-        description: 'Prisma replaces traditional ORMs',
-      },
+    const url = new URL(connectionString);
+    const cs = url.searchParams.get('currentSchema');
+    if (!cs) {
+      throw new Error("Must provide 'currentSchema' in the connection string");
+    }
+    currentSchema = cs;
+    client = new pg.Client(connectionString);
+    await client.connect();
+    db = drizzle(client);
+    await resetDatabase(db, currentSchema);
+    await migrate(drizzle(client), {
+      migrationsSchema: currentSchema,
+      migrationsFolder: resolve(__dirname, '../src/drizzle'),
     });
-    await client.$disconnect();
+    yoga = createYoga({ schema, context: { db } });
   });
 
   afterAll(async () => {
-    // drop
-    await DbDrop.new().parse([
-      `--schema=${path.resolve(__dirname, '..', 'prisma', 'schema.prisma')}`,
-      '--preview-feature', // DbDrop is an experimental feature
-      '--force',
-    ]);
-  });
-
-  it('should get posts from feed', async () => {
-    const response = await yoga.fetch('http://yoga/graphql?query={feed{url,description}}');
-
-    const body = await response.json();
-    expect(body).toMatchInlineSnapshot(`
-      {
-        "data": {
-          "feed": [
-            {
-              "description": "Prisma replaces traditional ORMs",
-              "url": "https://www.prisma.io",
-            },
-          ],
-        },
-      }
-    `);
+    await resetDatabase(db, currentSchema);
+    await client.end();
   });
 
   it('should create a new post', async () => {

examples/hackernews/__integration-tests__/prisma__migrate.d.ts

@@ -1,14 +1,14 @@
 import { createServer } from 'node:http';
 import { drizzle } from 'drizzle-orm/node-postgres';
-import { migrate } from 'drizzle-orm/postgres-js/migrator';
 import { createYoga } from 'graphql-yoga';
 import pg from 'pg';
+import { migrateDatabase } from './migrate-database.js';
 import { schema } from './schema.js';
 
 async function main() {
   const client = new pg.Client(process.env['PG_CONNECTION_STRING']);
   await client.connect();
-  await migrate(drizzle(client), { migrationsFolder: './src/drizzle' });
+  await migrateDatabase(client);
   const db = drizzle(client);
   const yoga = createYoga({ schema, context: { db } });
   const server = createServer(yoga);

examples/hackernews/src/main.ts

@@ -0,0 +1,11 @@
+import { resolve } from 'node:path';
+import { drizzle } from 'drizzle-orm/node-postgres';
+import { migrate } from 'drizzle-orm/postgres-js/migrator';
+import type pg from 'pg';
+
+// get current directory in esm
+const __dirname = new URL('.', import.meta.url).pathname;
+
+export async function migrateDatabase(client: pg.Client) {
+  await migrate(drizzle(client), { migrationsFolder: resolve(__dirname, '/drizzle') });
+}