test: run db pull e2e test also for postgres

Author: svetch

packages/cli/test/db/pull.test.ts

@@ -2,12 +2,55 @@ import fs from 'node:fs';
 import path from 'node:path';
 import { describe, expect, it } from 'vitest';
 import { createProject, runCli } from '../utils';
+import { getTestDbProvider } from '@zenstackhq/testtools';
+import { createHash } from 'node:crypto';
 
 const getSchema = (workDir: string) => fs.readFileSync(path.join(workDir, 'zenstack/schema.zmodel')).toString();
 
+const TEST_PG_CONFIG = {
+    host: process.env['TEST_PG_HOST'] ?? 'localhost',
+    port: process.env['TEST_PG_PORT'] ? parseInt(process.env['TEST_PG_PORT']) : 5432,
+    user: process.env['TEST_PG_USER'] ?? 'postgres',
+    password: process.env['TEST_PG_PASSWORD'] ?? 'postgres',
+};
+
+function getTestDbName(provider: string) {
+    if (provider === 'sqlite') {
+        return './test.db';
+    }
+    const testName = expect.getState().currentTestName ?? 'unnamed';
+    const testPath = expect.getState().testPath ?? '';
+    // digest test name
+    const digest = createHash('md5')
+        .update(testName + testPath)
+        .digest('hex');
+    // compute a database name based on test name
+    return (
+        'test_' +
+        testName
+            .toLowerCase()
+            .replace(/[^a-z0-9_]/g, '_')
+            .replace(/_+/g, '_')
+            .substring(0, 30) +
+        digest.slice(0, 6)
+    );
+}
+
+const provider = getTestDbProvider() ?? 'sqlite';
+const dbName = getTestDbName(provider);
+const dbUrl =
+    provider === 'sqlite'
+        ? `file:${dbName}`
+        : `postgres://${TEST_PG_CONFIG.user}:${TEST_PG_CONFIG.password}@${TEST_PG_CONFIG.host}:${TEST_PG_CONFIG.port}/${dbName}`;
+
 describe('DB pull', () => {
-    it('sqlite schema', () => {
-        const workDir = createProject(`
+    it('simple schema', () => {
+        const workDir = createProject(
+            `
+datasource db {
+    provider = "${provider}"
+    url      = "${dbUrl}"
+}
 model User {
     id             String   @id @default(cuid())
     email          String   @unique @map("email_address")
@@ -85,7 +128,9 @@ enum Role {
     USER
     ADMIN
     MODERATOR
-}`);
+}`,
+            false,
+        );
         runCli('format', workDir);
         runCli('db push', workDir);