chore: update test schema generation (#105)

* chore: update test schema generation

* break down large test suite into smaller ones

Author: ymc9

packages/runtime/package.json

@@ -8,7 +8,7 @@
         "watch": "tsup-node --watch",
         "lint": "eslint src --ext ts",
         "test": "vitest run && pnpm test:generate && pnpm test:typecheck",
-        "test:generate": "tsx test/typing/generate.ts",
+        "test:generate": "tsx test/scripts/generate.ts",
         "test:typecheck": "tsc --project tsconfig.test.json",
         "pack": "pnpm pack"
     },

packages/runtime/test/scripts/generate.ts

@@ -0,0 +1,25 @@
+import { glob } from 'glob';
+import { TsSchemaGenerator } from '@zenstackhq/sdk';
+import path from 'node:path';
+import fs from 'node:fs';
+import { fileURLToPath } from 'node:url';
+
+const dir = path.dirname(fileURLToPath(import.meta.url));
+
+async function main() {
+    await generate(path.resolve(dir, '../typing/typing-test.zmodel'));
+    await generate(path.resolve(dir, '../test-schema/schema.zmodel'));
+}
+
+async function generate(schemaPath: string) {
+    const generator = new TsSchemaGenerator();
+    const outputDir = path.dirname(schemaPath);
+    const tsPath = path.join(outputDir, 'schema.ts');
+    const pluginModelFiles = glob.sync(path.resolve(dir, '../../dist/**/plugin.zmodel'));
+    await generator.generate(schemaPath, pluginModelFiles, outputDir);
+    const content = fs.readFileSync(tsPath, 'utf-8');
+    fs.writeFileSync(tsPath, content.replace(/@zenstackhq\/runtime/g, '../../dist'));
+    console.log('TS schema generated at:', outputDir);
+}
+
+main();

packages/runtime/test/test-schema.ts

@@ -0,0 +1,11 @@
+import type { DataSourceProviderType } from '@zenstackhq/sdk/schema';
+import { schema } from './schema';
+
+export function getSchema<ProviderType extends DataSourceProviderType>(type: ProviderType) {
+    return {
+        ...schema,
+        provider: {
+            type,
+        },
+    };
+}

packages/runtime/test/test-schema/helper.ts

@@ -0,0 +1,2 @@
+export { getSchema } from './helper';
+export { schema } from './schema';