postgresdk 0.5.1-alpha.1 → 0.6.1

package/README.md

@@ -974,6 +974,104 @@ The pulled SDK includes metadata about when it was generated and from where:
 }
 ```
 
+## Generated Tests
+
+postgresdk can generate basic SDK tests to help you get started quickly. These tests demonstrate CRUD operations for each table and include Docker setup for easy testing.
+
+### Enabling Test Generation
+
+Add test configuration to your `postgresdk.config.ts`:
+
+```typescript
+export default {
+  connectionString: process.env.DATABASE_URL,
+  
+  tests: {
+    generate: true,              // Enable test generation
+    output: "./generated/tests", // Output directory
+    framework: "vitest"          // Test framework (vitest, jest, or bun)
+  }
+};
+```
+
+### What Gets Generated
+
+When tests are enabled, postgresdk generates:
+
+1. **Test files for each table** - Basic CRUD operation tests
+2. **setup.ts** - Common test utilities and helpers
+3. **docker-compose.yml** - PostgreSQL test database configuration
+4. **run-tests.sh** - Script to run tests with Docker
+
+### Running Tests with Docker
+
+The generated Docker setup makes it easy to run tests in isolation:
+
+```bash
+# Navigate to test directory
+cd generated/tests
+
+# Start test database
+docker-compose up -d
+
+# Wait for database to be ready
+sleep 3
+
+# Set environment variables
+export TEST_DATABASE_URL="postgres://testuser:testpass@localhost:5432/testdb"
+export TEST_API_URL="http://localhost:3000"
+
+# Run your migrations on test database
+# (your migration command here)
+
+# Start your API server
+npm run dev &
+
+# Run tests
+npm test
+
+# Stop database when done
+docker-compose down
+
+# Or use the generated script
+bash run-tests.sh
+```
+
+### Customizing Tests
+
+The generated tests are basic and meant as a starting point. Create your own test files for:
+
+- Business logic validation
+- Complex query scenarios  
+- Edge cases and error handling
+- Performance testing
+- Integration workflows
+
+Example custom test:
+
+```typescript
+// tests/custom/user-workflow.test.ts
+import { describe, it, expect } from 'vitest';
+import { createTestSDK, randomEmail } from '../generated/tests/setup';
+
+describe('User Registration Workflow', () => {
+  const sdk = createTestSDK();
+  
+  it('should handle complete registration flow', async () => {
+    // Your custom business logic tests
+    const user = await sdk.users.create({
+      email: randomEmail(),
+      name: 'Test User'
+    });
+    
+    // Verify welcome email was sent
+    // Check audit logs
+    // Validate permissions
+    // etc.
+  });
+});
+```
+
 ## CLI Commands
 
 ```bash

package/dist/cli.js

@@ -590,6 +590,18 @@ export default {
    */
   // useJsExtensionsClient: false,
   
+  // ========== TEST GENERATION ==========
+  
+  /**
+   * Generate basic SDK tests
+   * Uncomment to enable test generation with Docker setup
+   */
+  // tests: {
+  //   generate: true,
+  //   output: "./generated/tests",
+  //   framework: "vitest"  // or "jest" or "bun"
+  // },
+  
   // ========== AUTHENTICATION ==========
   
   /**
@@ -2327,6 +2339,301 @@ export async function deleteRecord(
 }`;
 }
 
+// src/emit-tests.ts
+function emitTableTest(table, framework = "vitest") {
+  const Type = pascal(table.name);
+  const tableName = table.name;
+  const imports = getFrameworkImports(framework);
+  const sampleData = generateSampleData(table);
+  const updateData = generateUpdateData(table);
+  return `${imports}
+import { SDK } from '../client';
+import type { Insert${Type}, Update${Type}, Select${Type} } from '../client/types/${tableName}';
+
+/**
+ * Basic tests for ${tableName} table operations
+ * 
+ * These tests demonstrate basic CRUD operations.
+ * Add your own business logic tests in separate files.
+ */
+describe('${Type} SDK Operations', () => {
+  let sdk: SDK;
+  let createdId: string;
+  
+  beforeAll(() => {
+    sdk = new SDK({ 
+      baseUrl: process.env.API_URL || 'http://localhost:3000',
+      auth: process.env.API_KEY ? { apiKey: process.env.API_KEY } : undefined
+    });
+  });
+  
+  ${generateTestCases(table, sampleData, updateData)}
+});
+`;
+}
+function emitTestSetup(framework = "vitest") {
+  return `/**
+ * Test Setup and Utilities
+ * 
+ * This file provides common test utilities and configuration.
+ * Extend this with your own helpers as needed.
+ */
+
+${getFrameworkImports(framework)}
+
+// Test database connection
+export const TEST_DATABASE_URL = process.env.TEST_DATABASE_URL || process.env.DATABASE_URL;
+
+// API configuration
+export const TEST_API_URL = process.env.TEST_API_URL || 'http://localhost:3000';
+export const TEST_API_KEY = process.env.TEST_API_KEY;
+
+// Utility to create SDK instance
+export function createTestSDK() {
+  const { SDK } = require('../client');
+  return new SDK({
+    baseUrl: TEST_API_URL,
+    auth: TEST_API_KEY ? { apiKey: TEST_API_KEY } : undefined
+  });
+}
+
+// Utility to generate random test data
+export function randomString(prefix = 'test'): string {
+  return \`\${prefix}_\${Date.now()}_\${Math.random().toString(36).substr(2, 9)}\`;
+}
+
+export function randomEmail(): string {
+  return \`\${randomString('user')}@example.com\`;
+}
+
+export function randomInt(min = 1, max = 1000): number {
+  return Math.floor(Math.random() * (max - min + 1)) + min;
+}
+
+export function randomDate(): Date {
+  const start = new Date(2020, 0, 1);
+  const end = new Date();
+  return new Date(start.getTime() + Math.random() * (end.getTime() - start.getTime()));
+}
+`;
+}
+function emitDockerCompose() {
+  return `# Docker Compose for Test Database
+# 
+# Start: docker-compose up -d
+# Stop: docker-compose down
+# Reset: docker-compose down -v && docker-compose up -d
+
+version: '3.8'
+
+services:
+  postgres:
+    image: postgres:16-alpine
+    environment:
+      POSTGRES_USER: testuser
+      POSTGRES_PASSWORD: testpass
+      POSTGRES_DB: testdb
+    ports:
+      - "5432:5432"
+    volumes:
+      - postgres_data:/var/lib/postgresql/data
+    healthcheck:
+      test: ["CMD-SHELL", "pg_isready -U testuser"]
+      interval: 5s
+      timeout: 5s
+      retries: 5
+
+volumes:
+  postgres_data:
+`;
+}
+function emitTestScript(framework = "vitest") {
+  const runCommand = framework === "bun" ? "bun test" : framework;
+  return `#!/bin/bash
+# Test Runner Script
+# 
+# This script sets up and runs tests with a Docker PostgreSQL database
+
+set -e
+
+echo "\uD83D\uDC33 Starting test database..."
+docker-compose up -d --wait
+
+# Export test database URL
+export TEST_DATABASE_URL="postgres://testuser:testpass@localhost:5432/testdb"
+export TEST_API_URL="http://localhost:3000"
+
+# Wait for database to be ready
+echo "⏳ Waiting for database..."
+sleep 2
+
+# Run migrations if needed (customize this)
+# npm run migrate
+
+echo "\uD83D\uDE80 Starting API server..."
+# Start your API server in the background
+# npm run dev &
+# SERVER_PID=$!
+# sleep 3
+
+echo "\uD83E\uDDEA Running tests..."
+${runCommand} $@
+
+# Cleanup
+# kill $SERVER_PID 2>/dev/null || true
+
+echo "✅ Tests completed!"
+`;
+}
+function getFrameworkImports(framework) {
+  switch (framework) {
+    case "vitest":
+      return "import { describe, it, expect, beforeAll, afterAll } from 'vitest';";
+    case "jest":
+      return "// Jest is configured globally, no imports needed";
+    case "bun":
+      return "import { describe, it, expect, beforeAll, afterAll } from 'bun:test';";
+    default:
+      return "import { describe, it, expect, beforeAll, afterAll } from 'vitest';";
+  }
+}
+function generateSampleData(table) {
+  const fields = [];
+  for (const col of table.columns) {
+    if (col.hasDefault || col.name === "id" || col.name === "created_at" || col.name === "updated_at") {
+      continue;
+    }
+    if (col.nullable) {
+      continue;
+    }
+    const value = getSampleValue(col.pgType, col.name);
+    fields.push(`    ${col.name}: ${value}`);
+  }
+  return fields.length > 0 ? `{
+${fields.join(`,
+`)}
+  }` : "{}";
+}
+function generateUpdateData(table) {
+  const fields = [];
+  for (const col of table.columns) {
+    if (col.hasDefault || col.name === "id" || col.name === "created_at" || col.name === "updated_at") {
+      continue;
+    }
+    if (!col.nullable && fields.length === 0) {
+      const value = getSampleValue(col.pgType, col.name, true);
+      fields.push(`    ${col.name}: ${value}`);
+      break;
+    }
+  }
+  return fields.length > 0 ? `{
+${fields.join(`,
+`)}
+  }` : "{}";
+}
+function getSampleValue(type, name, isUpdate = false) {
+  const suffix = isUpdate ? ' + " (updated)"' : "";
+  if (name.includes("email")) {
+    return `'test${isUpdate ? ".updated" : ""}@example.com'`;
+  }
+  if (name.includes("name") || name.includes("title")) {
+    return `'Test ${pascal(name)}'${suffix}`;
+  }
+  if (name.includes("description") || name.includes("bio") || name.includes("content")) {
+    return `'Test description'${suffix}`;
+  }
+  switch (type) {
+    case "text":
+    case "varchar":
+    case "char":
+      return `'test_value'${suffix}`;
+    case "int":
+    case "integer":
+    case "smallint":
+    case "bigint":
+      return isUpdate ? "42" : "1";
+    case "decimal":
+    case "numeric":
+    case "real":
+    case "double precision":
+    case "float":
+      return isUpdate ? "99.99" : "10.50";
+    case "boolean":
+    case "bool":
+      return isUpdate ? "false" : "true";
+    case "date":
+      return `'2024-01-01'`;
+    case "timestamp":
+    case "timestamptz":
+      return `new Date().toISOString()`;
+    case "json":
+    case "jsonb":
+      return `{ key: 'value' }`;
+    case "uuid":
+      return `'${isUpdate ? "b" : "a"}0e0e0e0-e0e0-e0e0-e0e0-e0e0e0e0e0e0'`;
+    default:
+      return `'test'`;
+  }
+}
+function generateTestCases(table, sampleData, updateData) {
+  const Type = pascal(table.name);
+  const hasData = sampleData !== "{}";
+  return `it('should create a ${table.name}', async () => {
+    const data: Insert${Type} = ${sampleData};
+    ${hasData ? `
+    const created = await sdk.${table.name}.create(data);
+    expect(created).toBeDefined();
+    expect(created.id).toBeDefined();
+    createdId = created.id;
+    ` : `
+    // Table has only auto-generated columns
+    // Skip create test or add your own test data
+    expect(true).toBe(true);
+    `}
+  });
+  
+  it('should list ${table.name}', async () => {
+    const list = await sdk.${table.name}.list({ limit: 10 });
+    expect(Array.isArray(list)).toBe(true);
+  });
+  
+  ${hasData ? `it('should get ${table.name} by id', async () => {
+    if (!createdId) {
+      console.warn('No ID from create test, skipping get test');
+      return;
+    }
+    
+    const item = await sdk.${table.name}.getByPk(createdId);
+    expect(item).toBeDefined();
+    expect(item?.id).toBe(createdId);
+  });
+  
+  ${updateData !== "{}" ? `it('should update ${table.name}', async () => {
+    if (!createdId) {
+      console.warn('No ID from create test, skipping update test');
+      return;
+    }
+    
+    const updateData: Update${Type} = ${updateData};
+    const updated = await sdk.${table.name}.update(createdId, updateData);
+    expect(updated).toBeDefined();
+  });` : ""}
+  
+  it('should delete ${table.name}', async () => {
+    if (!createdId) {
+      console.warn('No ID from create test, skipping delete test');
+      return;
+    }
+    
+    const deleted = await sdk.${table.name}.delete(createdId);
+    expect(deleted).toBeDefined();
+    
+    // Verify deletion
+    const item = await sdk.${table.name}.getByPk(createdId);
+    expect(item).toBeNull();
+  });` : ""}`;
+}
+
 // src/types.ts
 function normalizeAuthConfig(input) {
   if (!input)
@@ -2384,15 +2691,22 @@ async function generate(configPath) {
   }
   const normDateType = cfg.dateType === "string" ? "string" : "date";
   const serverFramework = cfg.serverFramework || "hono";
+  const generateTests = cfg.tests?.generate ?? false;
+  const testDir = cfg.tests?.output || "./generated/tests";
+  const testFramework = cfg.tests?.framework || "vitest";
   console.log("\uD83D\uDCC1 Creating directories...");
-  await ensureDirs([
+  const dirs = [
     serverDir,
     join(serverDir, "types"),
     join(serverDir, "zod"),
     join(serverDir, "routes"),
     clientDir,
     join(clientDir, "types")
-  ]);
+  ];
+  if (generateTests) {
+    dirs.push(testDir);
+  }
+  await ensureDirs(dirs);
   const files = [];
   const includeSpec = emitIncludeSpec(graph);
   files.push({ path: join(serverDir, "include-spec.ts"), content: includeSpec });
@@ -2459,11 +2773,37 @@ async function generate(configPath) {
     path: join(serverDir, "sdk-bundle.ts"),
     content: emitSdkBundle(clientFiles, clientDir)
   });
+  if (generateTests) {
+    console.log("\uD83E\uDDEA Generating tests...");
+    files.push({
+      path: join(testDir, "setup.ts"),
+      content: emitTestSetup(testFramework)
+    });
+    files.push({
+      path: join(testDir, "docker-compose.yml"),
+      content: emitDockerCompose()
+    });
+    files.push({
+      path: join(testDir, "run-tests.sh"),
+      content: emitTestScript(testFramework)
+    });
+    for (const table of Object.values(model.tables)) {
+      files.push({
+        path: join(testDir, `${table.name}.test.ts`),
+        content: emitTableTest(table, testFramework)
+      });
+    }
+  }
   console.log("✍️  Writing files...");
   await writeFiles(files);
   console.log(`✅ Generated ${files.length} files`);
   console.log(`  Server: ${serverDir}`);
   console.log(`  Client: ${sameDirectory ? clientDir + " (in sdk subdir due to same output dir)" : clientDir}`);
+  if (generateTests) {
+    console.log(`  Tests: ${testDir}`);
+    console.log(`  \uD83D\uDC33 Run 'cd ${testDir} && docker-compose up -d' to start test database`);
+    console.log(`  \uD83E\uDDEA Run 'bash ${testDir}/run-tests.sh' to execute tests`);
+  }
 }
 
 // src/cli.ts

package/dist/emit-tests.d.ts

@@ -0,0 +1,17 @@
+import type { Table } from "./introspect";
+/**
+ * Generate basic SDK tests for a table
+ */
+export declare function emitTableTest(table: Table, framework?: "vitest" | "jest" | "bun"): string;
+/**
+ * Generate a test setup file
+ */
+export declare function emitTestSetup(framework?: "vitest" | "jest" | "bun"): string;
+/**
+ * Generate docker-compose.yml for test database
+ */
+export declare function emitDockerCompose(): string;
+/**
+ * Generate test runner script
+ */
+export declare function emitTestScript(framework?: "vitest" | "jest" | "bun"): string;

package/dist/index.js

@@ -2069,6 +2069,301 @@ export async function deleteRecord(
 }`;
 }
 
+// src/emit-tests.ts
+function emitTableTest(table, framework = "vitest") {
+  const Type = pascal(table.name);
+  const tableName = table.name;
+  const imports = getFrameworkImports(framework);
+  const sampleData = generateSampleData(table);
+  const updateData = generateUpdateData(table);
+  return `${imports}
+import { SDK } from '../client';
+import type { Insert${Type}, Update${Type}, Select${Type} } from '../client/types/${tableName}';
+
+/**
+ * Basic tests for ${tableName} table operations
+ * 
+ * These tests demonstrate basic CRUD operations.
+ * Add your own business logic tests in separate files.
+ */
+describe('${Type} SDK Operations', () => {
+  let sdk: SDK;
+  let createdId: string;
+  
+  beforeAll(() => {
+    sdk = new SDK({ 
+      baseUrl: process.env.API_URL || 'http://localhost:3000',
+      auth: process.env.API_KEY ? { apiKey: process.env.API_KEY } : undefined
+    });
+  });
+  
+  ${generateTestCases(table, sampleData, updateData)}
+});
+`;
+}
+function emitTestSetup(framework = "vitest") {
+  return `/**
+ * Test Setup and Utilities
+ * 
+ * This file provides common test utilities and configuration.
+ * Extend this with your own helpers as needed.
+ */
+
+${getFrameworkImports(framework)}
+
+// Test database connection
+export const TEST_DATABASE_URL = process.env.TEST_DATABASE_URL || process.env.DATABASE_URL;
+
+// API configuration
+export const TEST_API_URL = process.env.TEST_API_URL || 'http://localhost:3000';
+export const TEST_API_KEY = process.env.TEST_API_KEY;
+
+// Utility to create SDK instance
+export function createTestSDK() {
+  const { SDK } = require('../client');
+  return new SDK({
+    baseUrl: TEST_API_URL,
+    auth: TEST_API_KEY ? { apiKey: TEST_API_KEY } : undefined
+  });
+}
+
+// Utility to generate random test data
+export function randomString(prefix = 'test'): string {
+  return \`\${prefix}_\${Date.now()}_\${Math.random().toString(36).substr(2, 9)}\`;
+}
+
+export function randomEmail(): string {
+  return \`\${randomString('user')}@example.com\`;
+}
+
+export function randomInt(min = 1, max = 1000): number {
+  return Math.floor(Math.random() * (max - min + 1)) + min;
+}
+
+export function randomDate(): Date {
+  const start = new Date(2020, 0, 1);
+  const end = new Date();
+  return new Date(start.getTime() + Math.random() * (end.getTime() - start.getTime()));
+}
+`;
+}
+function emitDockerCompose() {
+  return `# Docker Compose for Test Database
+# 
+# Start: docker-compose up -d
+# Stop: docker-compose down
+# Reset: docker-compose down -v && docker-compose up -d
+
+version: '3.8'
+
+services:
+  postgres:
+    image: postgres:16-alpine
+    environment:
+      POSTGRES_USER: testuser
+      POSTGRES_PASSWORD: testpass
+      POSTGRES_DB: testdb
+    ports:
+      - "5432:5432"
+    volumes:
+      - postgres_data:/var/lib/postgresql/data
+    healthcheck:
+      test: ["CMD-SHELL", "pg_isready -U testuser"]
+      interval: 5s
+      timeout: 5s
+      retries: 5
+
+volumes:
+  postgres_data:
+`;
+}
+function emitTestScript(framework = "vitest") {
+  const runCommand = framework === "bun" ? "bun test" : framework;
+  return `#!/bin/bash
+# Test Runner Script
+# 
+# This script sets up and runs tests with a Docker PostgreSQL database
+
+set -e
+
+echo "\uD83D\uDC33 Starting test database..."
+docker-compose up -d --wait
+
+# Export test database URL
+export TEST_DATABASE_URL="postgres://testuser:testpass@localhost:5432/testdb"
+export TEST_API_URL="http://localhost:3000"
+
+# Wait for database to be ready
+echo "⏳ Waiting for database..."
+sleep 2
+
+# Run migrations if needed (customize this)
+# npm run migrate
+
+echo "\uD83D\uDE80 Starting API server..."
+# Start your API server in the background
+# npm run dev &
+# SERVER_PID=$!
+# sleep 3
+
+echo "\uD83E\uDDEA Running tests..."
+${runCommand} $@
+
+# Cleanup
+# kill $SERVER_PID 2>/dev/null || true
+
+echo "✅ Tests completed!"
+`;
+}
+function getFrameworkImports(framework) {
+  switch (framework) {
+    case "vitest":
+      return "import { describe, it, expect, beforeAll, afterAll } from 'vitest';";
+    case "jest":
+      return "// Jest is configured globally, no imports needed";
+    case "bun":
+      return "import { describe, it, expect, beforeAll, afterAll } from 'bun:test';";
+    default:
+      return "import { describe, it, expect, beforeAll, afterAll } from 'vitest';";
+  }
+}
+function generateSampleData(table) {
+  const fields = [];
+  for (const col of table.columns) {
+    if (col.hasDefault || col.name === "id" || col.name === "created_at" || col.name === "updated_at") {
+      continue;
+    }
+    if (col.nullable) {
+      continue;
+    }
+    const value = getSampleValue(col.pgType, col.name);
+    fields.push(`    ${col.name}: ${value}`);
+  }
+  return fields.length > 0 ? `{
+${fields.join(`,
+`)}
+  }` : "{}";
+}
+function generateUpdateData(table) {
+  const fields = [];
+  for (const col of table.columns) {
+    if (col.hasDefault || col.name === "id" || col.name === "created_at" || col.name === "updated_at") {
+      continue;
+    }
+    if (!col.nullable && fields.length === 0) {
+      const value = getSampleValue(col.pgType, col.name, true);
+      fields.push(`    ${col.name}: ${value}`);
+      break;
+    }
+  }
+  return fields.length > 0 ? `{
+${fields.join(`,
+`)}
+  }` : "{}";
+}
+function getSampleValue(type, name, isUpdate = false) {
+  const suffix = isUpdate ? ' + " (updated)"' : "";
+  if (name.includes("email")) {
+    return `'test${isUpdate ? ".updated" : ""}@example.com'`;
+  }
+  if (name.includes("name") || name.includes("title")) {
+    return `'Test ${pascal(name)}'${suffix}`;
+  }
+  if (name.includes("description") || name.includes("bio") || name.includes("content")) {
+    return `'Test description'${suffix}`;
+  }
+  switch (type) {
+    case "text":
+    case "varchar":
+    case "char":
+      return `'test_value'${suffix}`;
+    case "int":
+    case "integer":
+    case "smallint":
+    case "bigint":
+      return isUpdate ? "42" : "1";
+    case "decimal":
+    case "numeric":
+    case "real":
+    case "double precision":
+    case "float":
+      return isUpdate ? "99.99" : "10.50";
+    case "boolean":
+    case "bool":
+      return isUpdate ? "false" : "true";
+    case "date":
+      return `'2024-01-01'`;
+    case "timestamp":
+    case "timestamptz":
+      return `new Date().toISOString()`;
+    case "json":
+    case "jsonb":
+      return `{ key: 'value' }`;
+    case "uuid":
+      return `'${isUpdate ? "b" : "a"}0e0e0e0-e0e0-e0e0-e0e0-e0e0e0e0e0e0'`;
+    default:
+      return `'test'`;
+  }
+}
+function generateTestCases(table, sampleData, updateData) {
+  const Type = pascal(table.name);
+  const hasData = sampleData !== "{}";
+  return `it('should create a ${table.name}', async () => {
+    const data: Insert${Type} = ${sampleData};
+    ${hasData ? `
+    const created = await sdk.${table.name}.create(data);
+    expect(created).toBeDefined();
+    expect(created.id).toBeDefined();
+    createdId = created.id;
+    ` : `
+    // Table has only auto-generated columns
+    // Skip create test or add your own test data
+    expect(true).toBe(true);
+    `}
+  });
+  
+  it('should list ${table.name}', async () => {
+    const list = await sdk.${table.name}.list({ limit: 10 });
+    expect(Array.isArray(list)).toBe(true);
+  });
+  
+  ${hasData ? `it('should get ${table.name} by id', async () => {
+    if (!createdId) {
+      console.warn('No ID from create test, skipping get test');
+      return;
+    }
+    
+    const item = await sdk.${table.name}.getByPk(createdId);
+    expect(item).toBeDefined();
+    expect(item?.id).toBe(createdId);
+  });
+  
+  ${updateData !== "{}" ? `it('should update ${table.name}', async () => {
+    if (!createdId) {
+      console.warn('No ID from create test, skipping update test');
+      return;
+    }
+    
+    const updateData: Update${Type} = ${updateData};
+    const updated = await sdk.${table.name}.update(createdId, updateData);
+    expect(updated).toBeDefined();
+  });` : ""}
+  
+  it('should delete ${table.name}', async () => {
+    if (!createdId) {
+      console.warn('No ID from create test, skipping delete test');
+      return;
+    }
+    
+    const deleted = await sdk.${table.name}.delete(createdId);
+    expect(deleted).toBeDefined();
+    
+    // Verify deletion
+    const item = await sdk.${table.name}.getByPk(createdId);
+    expect(item).toBeNull();
+  });` : ""}`;
+}
+
 // src/types.ts
 function normalizeAuthConfig(input) {
   if (!input)
@@ -2126,15 +2421,22 @@ async function generate(configPath) {
   }
   const normDateType = cfg.dateType === "string" ? "string" : "date";
   const serverFramework = cfg.serverFramework || "hono";
+  const generateTests = cfg.tests?.generate ?? false;
+  const testDir = cfg.tests?.output || "./generated/tests";
+  const testFramework = cfg.tests?.framework || "vitest";
   console.log("\uD83D\uDCC1 Creating directories...");
-  await ensureDirs([
+  const dirs = [
     serverDir,
     join(serverDir, "types"),
     join(serverDir, "zod"),
     join(serverDir, "routes"),
     clientDir,
     join(clientDir, "types")
-  ]);
+  ];
+  if (generateTests) {
+    dirs.push(testDir);
+  }
+  await ensureDirs(dirs);
   const files = [];
   const includeSpec = emitIncludeSpec(graph);
   files.push({ path: join(serverDir, "include-spec.ts"), content: includeSpec });
@@ -2201,11 +2503,37 @@ async function generate(configPath) {
     path: join(serverDir, "sdk-bundle.ts"),
     content: emitSdkBundle(clientFiles, clientDir)
   });
+  if (generateTests) {
+    console.log("\uD83E\uDDEA Generating tests...");
+    files.push({
+      path: join(testDir, "setup.ts"),
+      content: emitTestSetup(testFramework)
+    });
+    files.push({
+      path: join(testDir, "docker-compose.yml"),
+      content: emitDockerCompose()
+    });
+    files.push({
+      path: join(testDir, "run-tests.sh"),
+      content: emitTestScript(testFramework)
+    });
+    for (const table of Object.values(model.tables)) {
+      files.push({
+        path: join(testDir, `${table.name}.test.ts`),
+        content: emitTableTest(table, testFramework)
+      });
+    }
+  }
   console.log("✍️  Writing files...");
   await writeFiles(files);
   console.log(`✅ Generated ${files.length} files`);
   console.log(`  Server: ${serverDir}`);
   console.log(`  Client: ${sameDirectory ? clientDir + " (in sdk subdir due to same output dir)" : clientDir}`);
+  if (generateTests) {
+    console.log(`  Tests: ${testDir}`);
+    console.log(`  \uD83D\uDC33 Run 'cd ${testDir} && docker-compose up -d' to start test database`);
+    console.log(`  \uD83E\uDDEA Run 'bash ${testDir}/run-tests.sh' to execute tests`);
+  }
 }
 export {
   generate

package/dist/types.d.ts

@@ -31,6 +31,11 @@ export interface Config {
     pull?: PullConfig;
     useJsExtensions?: boolean;
     useJsExtensionsClient?: boolean;
+    tests?: {
+        generate?: boolean;
+        output?: string;
+        framework?: "vitest" | "jest" | "bun";
+    };
 }
 export interface PullConfig {
     from: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "postgresdk",
-  "version": "0.5.1-alpha.1",
+  "version": "0.6.1",
   "description": "Generate a typed server/client SDK from a Postgres schema (includes, Zod, Hono).",
   "type": "module",
   "bin": {
@@ -22,9 +22,10 @@
   },
   "scripts": {
     "build": "bun build src/cli.ts src/index.ts --outdir dist --target node --format esm --external=pg --external=zod --external=hono --external=node:* && tsc -p tsconfig.build.json --emitDeclarationOnly",
-    "test": "bun test:init && bun test:gen && bun test:pull && bun test:typecheck",
+    "test": "bun test:init && bun test:gen && bun test:gen-with-tests && bun test:pull && bun test:typecheck",
     "test:init": "bun test/test-init.ts",
     "test:gen": "bun test/test-gen.ts",
+    "test:gen-with-tests": "bun src/cli.ts generate -c test/test-with-tests.config.ts",
     "test:pull": "bun test/test-pull.ts",
     "test:typecheck": "bun test/test-typecheck.ts",
     "prepublishOnly": "npm run build",