@dbos-inc/drizzle-datasource 4.12.8-preview → 4.13.4-preview

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dbos-inc/drizzle-datasource",
-  "version": "4.12.8-preview",
+  "version": "4.13.4-preview",
   "description": "DBOS DataSource library for Drizzle ORM with PostgreSQL support",
   "license": "MIT",
   "main": "dist/index.js",
@@ -26,7 +26,6 @@
   "devDependencies": {
     "@types/jest": "^29.5.14",
     "@types/pg": "^8.15.2",
-    "drizzle-kit": "^0.31.5",
     "drizzle-orm": "^0.44.5",
     "jest": "^29.7.0",
     "typescript": "^5.4.5"

package/tests/custom-pool.test.ts

@@ -5,8 +5,6 @@ import { dropDB, ensureDB } from './test-helpers';
 import { randomUUID } from 'crypto';
 import SuperJSON from 'superjson';
 import { pgTable, text, integer } from 'drizzle-orm/pg-core';
-import { drizzle } from 'drizzle-orm/node-postgres';
-import { pushSchema } from 'drizzle-kit/api';
 import { sql } from 'drizzle-orm';
 
 const config = { user: 'postgres', database: 'drizzle_pool_test_userdb' };
@@ -69,14 +67,9 @@ describe('DrizzleDataSource with custom Pool', () => {
       poolClient.release();
     }
 
-    const drizzlePool = new Pool(config);
-    const db = drizzle(drizzlePool);
-    try {
-      const res = await pushSchema({ greetingsTable }, db);
-      await res.apply();
-    } finally {
-      await drizzlePool.end();
-    }
+    await customPool.query(
+      `CREATE TABLE IF NOT EXISTS greetings (name TEXT PRIMARY KEY NOT NULL, greet_count INTEGER DEFAULT 0)`,
+    );
   });
 
   afterAll(async () => {

package/tests/datasource.test.ts

@@ -5,8 +5,6 @@ import { dropDB, ensureDB } from './test-helpers';
 import { randomUUID } from 'crypto';
 import SuperJSON from 'superjson';
 import { pgTable, text, integer } from 'drizzle-orm/pg-core';
-import { drizzle } from 'drizzle-orm/node-postgres';
-import { pushSchema } from 'drizzle-kit/api';
 import { eq, sql } from 'drizzle-orm';
 
 const config = { user: 'postgres', database: 'drizzle_ds_test_userdb' };
@@ -19,14 +17,14 @@ interface transaction_completion {
   error: string | null;
 }
 
-async function createSchema(config: PoolConfig, entities: { [key: string]: object }) {
-  const drizzlePool = new Pool(config);
-  const db = drizzle(drizzlePool);
+async function createSchema(config: PoolConfig) {
+  const pool = new Pool(config);
   try {
-    const res = await pushSchema(entities, db);
-    await res.apply();
+    await pool.query(
+      `CREATE TABLE IF NOT EXISTS greetings (name TEXT PRIMARY KEY NOT NULL, greet_count INTEGER DEFAULT 0)`,
+    );
   } finally {
-    await drizzlePool.end();
+    await pool.end();
   }
 }
 
@@ -344,7 +342,7 @@ async function createDatabases(createTxCompletions: boolean) {
   if (createTxCompletions) {
     await DrizzleDataSource.initializeDBOSSchema(config);
   }
-  await createSchema(config, { greetingsTable });
+  await createSchema(config);
 }
 
 async function insertFunction(user: string) {