Add migration in DO tests

tests/bindings.d.ts

@@ -1,5 +1,6 @@
 export type Env = {
   DB: D1Database
+  TEST_DO: DurableObjectNamespace
 }
 
 declare module 'cloudflare:test' {

tests/index.ts

@@ -0,0 +1,10 @@
+import { DurableObject } from 'cloudflare:workers'
+import { Env } from './bindings'
+
+export class TestDO extends DurableObject {}
+
+export default {
+  async fetch(request: Request, env: Env) {
+    return new Response('test')
+  },
+}

tests/integration/migrations-do.test.ts

@@ -0,0 +1,152 @@
+import { env, runInDurableObject } from 'cloudflare:test'
+import { describe, expect, it } from 'vitest'
+import { D1QB, DOQB, Migration } from '../../src'
+
+export const migrations: Migration[] = [
+  {
+    name: '100000000000000_add_logs_table.sql',
+    sql: `
+      create table logs
+      (
+        id   INTEGER PRIMARY KEY AUTOINCREMENT,
+        name TEXT NOT NULL
+      );`,
+  },
+]
+
+describe('Migrations', () => {
+  it('initialize', async () => {
+    const id = env.TEST_DO.idFromName('test')
+    const stub = env.TEST_DO.get(id)
+
+    await runInDurableObject(stub, async (_instance, state) => {
+      const qb = new DOQB(state.storage.sql)
+
+      expect(
+        Array.from(
+          state.storage.sql.exec(`SELECT name
+                                                FROM sqlite_master
+                                                WHERE type = 'table'
+                                                  AND name not in ('_cf_KV', 'sqlite_sequence', '_cf_METADATA')`)
+        )
+      ).toEqual([])
+
+      qb.migrations({ migrations }).initialize()
+
+      expect(
+        Array.from(
+          state.storage.sql.exec(`SELECT name
+                                                FROM sqlite_master
+                                                WHERE type = 'table'
+                                                  AND name not in ('_cf_KV', 'sqlite_sequence', '_cf_METADATA')`)
+        )
+      ).toEqual([
+        {
+          name: 'migrations',
+        },
+      ])
+    })
+  })
+
+  it('apply', async () => {
+    const id = env.TEST_DO.idFromName('test')
+    const stub = env.TEST_DO.get(id)
+
+    await runInDurableObject(stub, async (_instance, state) => {
+      const qb = new DOQB(state.storage.sql)
+
+      expect(
+        Array.from(
+          state.storage.sql.exec(`SELECT name
+                                                FROM sqlite_master
+                                                WHERE type = 'table'
+                                                  AND name not in ('_cf_KV', 'sqlite_sequence', '_cf_METADATA')`)
+        )
+      ).toEqual([])
+
+      const applyResp = qb.migrations({ migrations }).apply()
+
+      expect(applyResp.length).toEqual(1)
+      expect(applyResp[0]?.name).toEqual('100000000000000_add_logs_table.sql')
+
+      expect(
+        Array.from(
+          state.storage.sql.exec(`SELECT name
+                                                FROM sqlite_master
+                                                WHERE type = 'table'
+                                                  AND name not in ('_cf_KV', 'sqlite_sequence', '_cf_METADATA')`)
+        )
+      ).toEqual([
+        {
+          name: 'migrations',
+        },
+        {
+          name: 'logs',
+        },
+      ])
+
+      const applyResp2 = qb.migrations({ migrations }).apply()
+      expect(applyResp2.length).toEqual(0)
+    })
+  })
+
+  it('incremental migrations', async () => {
+    const id = env.TEST_DO.idFromName('test')
+    const stub = env.TEST_DO.get(id)
+
+    await runInDurableObject(stub, async (_instance, state) => {
+      const qb = new DOQB(state.storage.sql)
+
+      expect(
+        Array.from(
+          state.storage.sql.exec(`SELECT name
+                                                FROM sqlite_master
+                                                WHERE type = 'table'
+                                                  AND name not in ('_cf_KV', 'sqlite_sequence', '_cf_METADATA')`)
+        )
+      ).toEqual([])
+
+      const applyResp = qb.migrations({ migrations }).apply()
+
+      expect(applyResp.length).toEqual(1)
+      expect(applyResp[0]?.name).toEqual('100000000000000_add_logs_table.sql')
+
+      expect(
+        Array.from(
+          state.storage.sql.exec(`SELECT name
+                                                FROM sqlite_master
+                                                WHERE type = 'table'
+                                                  AND name not in ('_cf_KV', 'sqlite_sequence', '_cf_METADATA')`)
+        )
+      ).toEqual([
+        {
+          name: 'migrations',
+        },
+        {
+          name: 'logs',
+        },
+      ])
+
+      const updatedMigrations = [
+        ...migrations,
+        {
+          name: '100000000000001_add_second_table.sql',
+          sql: `
+          create table logs_two
+          (
+            id   INTEGER PRIMARY KEY AUTOINCREMENT,
+            name TEXT NOT NULL
+          );`,
+        },
+      ]
+
+      const applyResp2 = qb.migrations({ migrations: updatedMigrations }).apply()
+
+      expect(applyResp2.length).toEqual(1)
+      expect(applyResp2[0]?.name).toEqual('100000000000001_add_second_table.sql')
+
+      const applyResp3 = qb.migrations({ migrations: updatedMigrations }).apply()
+      expect(applyResp3.length).toEqual(0)
+    })
+  })
+})

tests/vitest.config.ts

@@ -4,6 +4,9 @@ export default defineWorkersConfig({
   test: {
     poolOptions: {
       workers: {
+        wrangler: {
+          configPath: './wrangler.toml',
+        },
         miniflare: {
           compatibilityFlags: ['nodejs_compat'],
           d1Databases: {

tests/wrangler.toml

@@ -0,0 +1,11 @@
+name = "test"
+main = "index.ts"
+compatibility_date = "2024-11-09"
+
+[[durable_objects.bindings]]
+name = "TEST_DO"
+class_name = "TestDO"
+
+[[migrations]]
+tag = "v1"
+new_sqlite_classes = ["TestDO"]