test: update cli tests

Author: svetch

packages/cli/test/check.test.ts

@@ -83,17 +83,12 @@ describe('CLI validate command test', () => {
 
     it('should validate schema with syntax errors', () => {
         const modelWithSyntaxError = `
-datasource db {
-    provider = "sqlite"
-    url = "file:./dev.db"
-}
-
 model User {
     id String @id @default(cuid())
     email String @unique
     // Missing closing brace - syntax error
         `;
-        const workDir = createProject(modelWithSyntaxError, false);
+        const workDir = createProject(modelWithSyntaxError);
 
         // Should throw an error due to syntax error
         expect(() => runCli('check', workDir)).toThrow();

packages/cli/test/db.test.ts

@@ -11,13 +11,13 @@ model User {
 
 describe('CLI db commands test', () => {
     it('should generate a database with db push', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         runCli('db push', workDir);
-        expect(fs.existsSync(path.join(workDir, 'zenstack/dev.db'))).toBe(true);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/test.db'))).toBe(true);
     });
 
     it('should seed the database with db seed with seed script', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         const pkgJson = JSON.parse(fs.readFileSync(path.join(workDir, 'package.json'), 'utf8'));
         pkgJson.zenstack = {
             seed: 'node seed.js',
@@ -36,7 +36,7 @@ fs.writeFileSync('seed.txt', 'success');
     });
 
     it('should seed the database after migrate reset', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         const pkgJson = JSON.parse(fs.readFileSync(path.join(workDir, 'package.json'), 'utf8'));
         pkgJson.zenstack = {
             seed: 'node seed.js',
@@ -55,7 +55,7 @@ fs.writeFileSync('seed.txt', 'success');
     });
 
     it('should skip seeding the database without seed script', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         runCli('db seed', workDir);
     });
 });

packages/cli/test/db/push.test.ts

@@ -11,8 +11,8 @@ model User {
 
 describe('CLI db commands test', () => {
     it('should generate a database with db push', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         runCli('db push', workDir);
-        expect(fs.existsSync(path.join(workDir, 'zenstack/dev.db'))).toBe(true);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/test.db'))).toBe(true);
     });
 });

packages/cli/test/migrate.test.ts

@@ -11,36 +11,36 @@ model User {
 
 describe('CLI migrate commands test', () => {
     it('should generate a database with migrate dev', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         runCli('migrate dev --name init', workDir);
-        expect(fs.existsSync(path.join(workDir, 'zenstack/dev.db'))).toBe(true);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/test.db'))).toBe(true);
         expect(fs.existsSync(path.join(workDir, 'zenstack/migrations'))).toBe(true);
     });
 
     it('should reset the database with migrate reset', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         runCli('db push', workDir);
-        expect(fs.existsSync(path.join(workDir, 'zenstack/dev.db'))).toBe(true);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/test.db'))).toBe(true);
         runCli('migrate reset --force', workDir);
-        expect(fs.existsSync(path.join(workDir, 'zenstack/dev.db'))).toBe(true);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/test.db'))).toBe(true);
     });
 
     it('should reset the database with migrate deploy', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         runCli('migrate dev --name init', workDir);
-        fs.rmSync(path.join(workDir, 'zenstack/dev.db'));
+        fs.rmSync(path.join(workDir, 'zenstack/test.db'));
         runCli('migrate deploy', workDir);
-        expect(fs.existsSync(path.join(workDir, 'zenstack/dev.db'))).toBe(true);
+        expect(fs.existsSync(path.join(workDir, 'zenstack/test.db'))).toBe(true);
     });
 
     it('supports migrate status', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         runCli('migrate dev --name init', workDir);
         runCli('migrate status', workDir);
     });
 
     it('supports migrate resolve', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         runCli('migrate dev --name init', workDir);
 
         // find the migration record "timestamp_init"
@@ -51,7 +51,7 @@ describe('CLI migrate commands test', () => {
         fs.writeFileSync(path.join(workDir, 'zenstack/migrations', migration!, 'migration.sql'), 'invalid content');
 
         // redeploy the migration, which will fail
-        fs.rmSync(path.join(workDir, 'zenstack/dev.db'), { force: true });
+        fs.rmSync(path.join(workDir, 'zenstack/test.db'), { force: true });
         try {
             runCli('migrate deploy', workDir);
         } catch {
@@ -66,7 +66,7 @@ describe('CLI migrate commands test', () => {
     });
 
     it('should throw error when neither applied nor rolled-back is provided', () => {
-        const workDir = createProject(model);
+        const workDir = createProject(model, { provider: 'sqlite' });
         expect(() => runCli('migrate resolve', workDir)).toThrow();
     });
 });

packages/cli/test/utils.ts

@@ -34,8 +34,11 @@ function getTestDbName(provider: string) {
     );
 }
 
-export function createProject(zmodel: string, addPrelude = true) {
-    const provider = getTestDbProvider() ?? 'sqlite';
+export function createProject(
+    zmodel: string,
+    options?: { customPrelude?: boolean; provider?: 'sqlite' | 'postgresql' },
+) {
+    const provider = (options?.provider || getTestDbProvider()) ?? 'sqlite';
     const dbName = getTestDbName(provider);
     const dbUrl =
         provider === 'sqlite'
@@ -50,7 +53,7 @@ export function createProject(zmodel: string, addPrelude = true) {
     const workDir = createTestProject();
     fs.mkdirSync(path.join(workDir, 'zenstack'), { recursive: true });
     const schemaPath = path.join(workDir, 'zenstack/schema.zmodel');
-    fs.writeFileSync(schemaPath, addPrelude ? `${ZMODEL_PRELUDE}\n${zmodel}` : zmodel);
+    fs.writeFileSync(schemaPath, !options?.customPrelude ? `${ZMODEL_PRELUDE}\n${zmodel}` : zmodel);
     return workDir;
 }