postgresdk 0.6.8 → 0.6.11

package/dist/cli.js

@@ -561,13 +561,6 @@ export default {
    */
   // includeDepthLimit: 3,
   
-  /**
-   * How to handle date/timestamp columns in TypeScript
-   * - "date": Use JavaScript Date objects
-   * - "string": Use ISO 8601 strings
-   * @default "date"
-   */
-  // dateType: "date",
   
   /**
    * Server framework for generated API routes
@@ -1015,17 +1008,18 @@ function emitZod(table, opts) {
     if (pg === "jsonb" || pg === "json")
       return `z.unknown()`;
     if (pg === "date" || pg.startsWith("timestamp"))
-      return opts.dateType === "date" ? `z.date()` : `z.string()`;
+      return `z.string()`;
     if (pg.startsWith("_"))
       return `z.array(${zFor(pg.slice(1))})`;
     return `z.string()`;
   };
   const fields = table.columns.map((c) => {
     let z = zFor(c.pgType);
-    if (c.nullable)
-      z += `.nullable()`;
-    if (c.hasDefault)
+    if (c.nullable) {
+      z += `.nullish()`;
+    } else if (c.hasDefault) {
       z += `.optional()`;
+    }
     return `  ${c.name}: ${z}`;
   }).join(`,
 `);
@@ -1764,7 +1758,7 @@ function tsTypeFor(pgType, opts) {
     return opts.numericMode === "number" ? "number" : "string";
   }
   if (t === "date" || t.startsWith("timestamp"))
-    return opts.dateType === "date" ? "Date" : "string";
+    return "string";
   if (t === "json" || t === "jsonb")
     return "unknown";
   return "string";
@@ -2364,14 +2358,57 @@ export async function deleteRecord(
 }
 
 // src/emit-tests.ts
-function emitTableTest(table, clientPath, framework = "vitest") {
+function emitTableTest(table, model, clientPath, framework = "vitest") {
   const Type = pascal(table.name);
   const tableName = table.name;
   const imports = getFrameworkImports(framework);
+  const isJunctionTable = table.pk.length > 1;
   const hasForeignKeys = table.fks.length > 0;
-  const foreignKeySetup = hasForeignKeys ? generateForeignKeySetup(table, clientPath) : null;
+  const foreignKeySetup = hasForeignKeys ? generateForeignKeySetup(table, model, clientPath) : null;
   const sampleData = generateSampleDataFromSchema(table, hasForeignKeys);
   const updateData = generateUpdateDataFromSchema(table);
+  if (isJunctionTable) {
+    return `${imports}
+import { SDK } from '${clientPath}';
+import type { Insert${Type}, Update${Type}, Select${Type} } from '${clientPath}/types/${tableName}';
+${foreignKeySetup?.imports || ""}
+
+/**
+ * Basic tests for ${tableName} table operations
+ * 
+ * This is a junction table with composite primary key.
+ * Tests are simplified since it represents a many-to-many relationship.
+ */
+describe('${Type} SDK Operations', () => {
+  let sdk: SDK;
+  ${foreignKeySetup?.variables || ""}
+  
+  beforeAll(async () => {
+    sdk = new SDK({ 
+      baseUrl: process.env.API_URL || 'http://localhost:3000',
+      auth: process.env.API_KEY ? { apiKey: process.env.API_KEY } : undefined
+    });
+    ${foreignKeySetup?.setup || ""}
+  });
+  
+  ${foreignKeySetup?.cleanup ? `afterAll(async () => {
+    ${foreignKeySetup.cleanup}
+  });
+  
+  ` : ""}it('should create a ${tableName} relationship', async () => {
+    const data: Insert${Type} = ${sampleData};
+    
+    const created = await sdk.${tableName}.create(data);
+    expect(created).toBeDefined();
+  });
+  
+  it('should list ${tableName} relationships', async () => {
+    const list = await sdk.${tableName}.list({ limit: 10 });
+    expect(Array.isArray(list)).toBe(true);
+  });
+});
+`;
+  }
   return `${imports}
 import { SDK } from '${clientPath}';
 import type { Insert${Type}, Update${Type}, Select${Type} } from '${clientPath}/types/${tableName}';
@@ -2632,33 +2669,56 @@ echo "  - To reset the database: docker-compose -f $SCRIPT_DIR/docker-compose.ym
 exit $TEST_EXIT_CODE
 `;
 }
-function generateForeignKeySetup(table, clientPath) {
+function generateForeignKeySetup(table, model, clientPath) {
   const imports = [];
   const variables = [];
   const setupStatements = [];
   const cleanupStatements = [];
-  const foreignTables = new Set;
+  const foreignTablesData = new Map;
   for (const fk of table.fks) {
-    const foreignTable = fk.toTable;
-    if (!foreignTables.has(foreignTable)) {
-      foreignTables.add(foreignTable);
-      const ForeignType = pascal(foreignTable);
-      imports.push(`import type { Insert${ForeignType} } from '${clientPath}/types/${foreignTable}';`);
-      variables.push(`let ${foreignTable}Id: string;`);
-      setupStatements.push(`
-    // Create parent ${foreignTable} record for foreign key reference
-    const ${foreignTable}Data: Insert${ForeignType} = ${generateMinimalDataForTable(foreignTable)};
-    const created${ForeignType} = await sdk.${foreignTable}.create(${foreignTable}Data);
-    ${foreignTable}Id = created${ForeignType}.id;`);
-      cleanupStatements.push(`
-    // Clean up parent ${foreignTable} record
-    if (${foreignTable}Id) {
+    const foreignTableName = fk.toTable;
+    if (!foreignTablesData.has(foreignTableName)) {
+      const foreignTable = model.tables[foreignTableName];
+      if (!foreignTable)
+        continue;
+      foreignTablesData.set(foreignTableName, foreignTable);
+      const ForeignType = pascal(foreignTableName);
+      imports.push(`import type { Insert${ForeignType} } from '${clientPath}/types/${foreignTableName}';`);
+      if (foreignTable.pk.length === 1) {
+        variables.push(`let ${foreignTableName}Id: string;`);
+        const foreignData = generateSampleDataFromSchema(foreignTable, false);
+        setupStatements.push(`
+    // Create parent ${foreignTableName} record for foreign key reference
+    const ${foreignTableName}Data: Insert${ForeignType} = ${foreignData};
+    const created${ForeignType} = await sdk.${foreignTableName}.create(${foreignTableName}Data);
+    ${foreignTableName}Id = created${ForeignType}.${foreignTable.pk[0]};`);
+        cleanupStatements.push(`
+    // Clean up parent ${foreignTableName} record
+    if (${foreignTableName}Id) {
+      try {
+        await sdk.${foreignTableName}.delete(${foreignTableName}Id);
+      } catch (e) {
+        // Parent might already be deleted due to cascading
+      }
+    }`);
+      } else {
+        variables.push(`let ${foreignTableName}Key: any;`);
+        const foreignData = generateSampleDataFromSchema(foreignTable, false);
+        setupStatements.push(`
+    // Create parent ${foreignTableName} record for foreign key reference
+    const ${foreignTableName}Data: Insert${ForeignType} = ${foreignData};
+    const created${ForeignType} = await sdk.${foreignTableName}.create(${foreignTableName}Data);
+    ${foreignTableName}Key = { ${foreignTable.pk.map((pk) => `${pk}: created${ForeignType}.${pk}`).join(", ")} };`);
+        cleanupStatements.push(`
+    // Clean up parent ${foreignTableName} record
+    if (${foreignTableName}Key) {
       try {
-        await sdk.${foreignTable}.delete(${foreignTable}Id);
+        await sdk.${foreignTableName}.delete(${foreignTableName}Key);
       } catch (e) {
         // Parent might already be deleted due to cascading
       }
     }`);
+      }
     }
   }
   return {
@@ -2670,30 +2730,6 @@ function generateForeignKeySetup(table, clientPath) {
     cleanup: cleanupStatements.join("")
   };
 }
-function generateMinimalDataForTable(tableName) {
-  if (tableName.includes("author")) {
-    return `{ name: 'Test Author' }`;
-  }
-  if (tableName.includes("book")) {
-    return `{ title: 'Test Book' }`;
-  }
-  if (tableName.includes("tag")) {
-    return `{ name: 'Test Tag' }`;
-  }
-  if (tableName.includes("user")) {
-    return `{ name: 'Test User', email: 'test@example.com' }`;
-  }
-  if (tableName.includes("category") || tableName.includes("categories")) {
-    return `{ name: 'Test Category' }`;
-  }
-  if (tableName.includes("product")) {
-    return `{ name: 'Test Product', price: 10.99 }`;
-  }
-  if (tableName.includes("order")) {
-    return `{ total: 100.00, status: 'pending' }`;
-  }
-  return `{ name: 'Test ${pascal(tableName)}' }`;
-}
 function getTestCommand(framework, baseCommand) {
   switch (framework) {
     case "vitest":
@@ -2733,20 +2769,15 @@ function generateSampleDataFromSchema(table, hasForeignKeys = false) {
   }
   for (const col of table.columns) {
     const isAutoGenerated = col.hasDefault && ["id", "created_at", "updated_at", "created", "updated", "modified_at"].includes(col.name.toLowerCase());
-    if (isAutoGenerated) {
+    const isPrimaryKey = table.pk.includes(col.name);
+    if (isPrimaryKey && col.hasDefault) {
       continue;
     }
-    if (col.name === "deleted_at" || col.name === "deleted") {
-      if (col.nullable) {
-        fields.push(`    ${col.name}: null`);
-      } else {
-        const value = generateValueForColumn(col);
-        fields.push(`    ${col.name}: ${value}`);
-      }
+    if (isAutoGenerated) {
       continue;
     }
     const foreignTable = foreignKeyColumns.get(col.name);
-    if (!col.nullable || foreignTable || col.hasDefault && !isAutoGenerated || shouldIncludeNullableColumn(col)) {
+    if (!col.nullable || foreignTable || isPrimaryKey) {
       if (foreignTable) {
         fields.push(`    ${col.name}: ${foreignTable}Id`);
       } else {
@@ -2763,51 +2794,40 @@ ${fields.join(`,
 function generateUpdateDataFromSchema(table) {
   const fields = [];
   for (const col of table.columns) {
-    if (table.pk.includes(col.name) || col.hasDefault) {
-      const autoGenerated = ["id", "created_at", "updated_at", "created", "updated", "modified_at"];
-      if (autoGenerated.includes(col.name.toLowerCase())) {
-        continue;
-      }
+    if (table.pk.includes(col.name)) {
+      continue;
     }
-    if (col.name.endsWith("_id")) {
+    const isAutoGenerated = col.hasDefault && ["id", "created_at", "updated_at", "created", "updated", "modified_at"].includes(col.name.toLowerCase());
+    if (isAutoGenerated) {
       continue;
     }
-    if (col.name === "deleted_at" || col.name === "deleted") {
+    if (col.name.endsWith("_id")) {
       continue;
     }
-    if (!col.nullable || shouldIncludeNullableColumn(col)) {
+    if (!col.nullable) {
       const value = generateValueForColumn(col, true);
       fields.push(`    ${col.name}: ${value}`);
       break;
     }
   }
+  if (fields.length === 0) {
+    for (const col of table.columns) {
+      if (table.pk.includes(col.name) || col.name.endsWith("_id")) {
+        continue;
+      }
+      const isAutoGenerated = col.hasDefault && ["id", "created_at", "updated_at", "created", "updated", "modified_at"].includes(col.name.toLowerCase());
+      if (!isAutoGenerated) {
+        const value = generateValueForColumn(col, true);
+        fields.push(`    ${col.name}: ${value}`);
+        break;
+      }
+    }
+  }
   return fields.length > 0 ? `{
 ${fields.join(`,
 `)}
   }` : "{}";
 }
-function shouldIncludeNullableColumn(col) {
-  const importantPatterns = [
-    "_id",
-    "_by",
-    "email",
-    "name",
-    "title",
-    "description",
-    "phone",
-    "address",
-    "status",
-    "type",
-    "category",
-    "price",
-    "amount",
-    "quantity",
-    "url",
-    "slug"
-  ];
-  const name = col.name.toLowerCase();
-  return importantPatterns.some((pattern) => name.includes(pattern));
-}
 function generateValueForColumn(col, isUpdate = false) {
   const type = col.pgType.toLowerCase();
   switch (type) {
@@ -2841,15 +2861,17 @@ function generateValueForColumn(col, isUpdate = false) {
     case "bool":
       return isUpdate ? "false" : "true";
     case "date":
+      return `'2024-01-01'`;
     case "timestamp":
     case "timestamptz":
     case "timestamp without time zone":
     case "timestamp with time zone":
+      return `'2024-01-01T00:00:00.000Z'`;
     case "time":
     case "timetz":
     case "time without time zone":
     case "time with time zone":
-      return `new Date()`;
+      return `'12:00:00'`;
     case "interval":
       return `'1 day'`;
     case "json":
@@ -2917,26 +2939,14 @@ function generateUUID() {
 function generateTestCases(table, sampleData, updateData, hasForeignKeys = false) {
   const Type = pascal(table.name);
   const hasData = sampleData !== "{}";
-  const isJunctionTable = table.pk.length > 1 && table.columns.every((col) => table.pk.includes(col.name) || col.name.endsWith("_id"));
-  if (isJunctionTable) {
-    return `it('should create a ${table.name} relationship', async () => {
-    // This is a junction table for M:N relationships
-    // Test data depends on parent records created in other tests
-    expect(true).toBe(true);
-  });
-  
-  it('should list ${table.name} relationships', async () => {
-    const list = await sdk.${table.name}.list({ limit: 10 });
-    expect(Array.isArray(list)).toBe(true);
-  });`;
-  }
+  const hasSinglePK = table.pk.length === 1;
   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;
+    ${hasSinglePK ? `expect(created.${table.pk[0]}).toBeDefined();
+    createdId = created.${table.pk[0]};` : ""}
     ` : `
     // Table has only auto-generated columns
     // Skip create test or add your own test data
@@ -2949,7 +2959,7 @@ function generateTestCases(table, sampleData, updateData, hasForeignKeys = false
     expect(Array.isArray(list)).toBe(true);
   });
   
-  ${hasData ? `it('should get ${table.name} by id', async () => {
+  ${hasData && hasSinglePK ? `it('should get ${table.name} by id', async () => {
     if (!createdId) {
       console.warn('No ID from create test, skipping get test');
       return;
@@ -2957,7 +2967,7 @@ function generateTestCases(table, sampleData, updateData, hasForeignKeys = false
     
     const item = await sdk.${table.name}.getByPk(createdId);
     expect(item).toBeDefined();
-    expect(item?.id).toBe(createdId);
+    expect(item?.${table.pk[0]}).toBe(createdId);
   });
   
   ${updateData !== "{}" ? `it('should update ${table.name}', async () => {
@@ -3315,7 +3325,6 @@ async function generate(configPath) {
   if (sameDirectory) {
     clientDir = join(originalClientDir, "sdk");
   }
-  const normDateType = cfg.dateType === "string" ? "string" : "date";
   const serverFramework = cfg.serverFramework || "hono";
   const generateTests = cfg.tests?.generate ?? false;
   const originalTestDir = cfg.tests?.output || "./api/tests";
@@ -3359,12 +3368,12 @@ async function generate(configPath) {
     content: emitCoreOperations()
   });
   for (const table of Object.values(model.tables)) {
-    const typesSrc = emitTypes(table, { dateType: normDateType, numericMode: "string" });
+    const typesSrc = emitTypes(table, { numericMode: "string" });
     files.push({ path: join(serverDir, "types", `${table.name}.ts`), content: typesSrc });
     files.push({ path: join(clientDir, "types", `${table.name}.ts`), content: typesSrc });
     files.push({
       path: join(serverDir, "zod", `${table.name}.ts`),
-      content: emitZod(table, { dateType: normDateType, numericMode: "string" })
+      content: emitZod(table, { numericMode: "string" })
     });
     let routeContent;
     if (serverFramework === "hono") {
@@ -3435,7 +3444,7 @@ async function generate(configPath) {
     for (const table of Object.values(model.tables)) {
       files.push({
         path: join(testDir, `${table.name}.test.ts`),
-        content: emitTableTest(table, relativeClientPath, testFramework)
+        content: emitTableTest(table, model, relativeClientPath, testFramework)
       });
     }
   }

package/dist/emit-tests.d.ts

@@ -1,8 +1,8 @@
-import type { Table } from "./introspect";
+import type { Table, Model } from "./introspect";
 /**
  * Generate basic SDK tests for a table
  */
-export declare function emitTableTest(table: Table, clientPath: string, framework?: "vitest" | "jest" | "bun"): string;
+export declare function emitTableTest(table: Table, model: Model, clientPath: string, framework?: "vitest" | "jest" | "bun"): string;
 /**
  * Generate a test setup file
  */

package/dist/emit-types.d.ts

@@ -1,5 +1,4 @@
 import type { Table } from "./introspect";
 export declare function emitTypes(table: Table, opts: {
-    dateType: "date" | "string";
     numericMode: "string" | "number";
 }): string;

package/dist/emit-zod.d.ts

@@ -1,5 +1,4 @@
 import type { Table } from "./introspect";
 export declare function emitZod(table: Table, opts: {
-    dateType: "date" | "string";
     numericMode: "string" | "number";
 }): string;

package/dist/index.js

@@ -745,17 +745,18 @@ function emitZod(table, opts) {
     if (pg === "jsonb" || pg === "json")
       return `z.unknown()`;
     if (pg === "date" || pg.startsWith("timestamp"))
-      return opts.dateType === "date" ? `z.date()` : `z.string()`;
+      return `z.string()`;
     if (pg.startsWith("_"))
       return `z.array(${zFor(pg.slice(1))})`;
     return `z.string()`;
   };
   const fields = table.columns.map((c) => {
     let z = zFor(c.pgType);
-    if (c.nullable)
-      z += `.nullable()`;
-    if (c.hasDefault)
+    if (c.nullable) {
+      z += `.nullish()`;
+    } else if (c.hasDefault) {
       z += `.optional()`;
+    }
     return `  ${c.name}: ${z}`;
   }).join(`,
 `);
@@ -1494,7 +1495,7 @@ function tsTypeFor(pgType, opts) {
     return opts.numericMode === "number" ? "number" : "string";
   }
   if (t === "date" || t.startsWith("timestamp"))
-    return opts.dateType === "date" ? "Date" : "string";
+    return "string";
   if (t === "json" || t === "jsonb")
     return "unknown";
   return "string";
@@ -2094,14 +2095,57 @@ export async function deleteRecord(
 }
 
 // src/emit-tests.ts
-function emitTableTest(table, clientPath, framework = "vitest") {
+function emitTableTest(table, model, clientPath, framework = "vitest") {
   const Type = pascal(table.name);
   const tableName = table.name;
   const imports = getFrameworkImports(framework);
+  const isJunctionTable = table.pk.length > 1;
   const hasForeignKeys = table.fks.length > 0;
-  const foreignKeySetup = hasForeignKeys ? generateForeignKeySetup(table, clientPath) : null;
+  const foreignKeySetup = hasForeignKeys ? generateForeignKeySetup(table, model, clientPath) : null;
   const sampleData = generateSampleDataFromSchema(table, hasForeignKeys);
   const updateData = generateUpdateDataFromSchema(table);
+  if (isJunctionTable) {
+    return `${imports}
+import { SDK } from '${clientPath}';
+import type { Insert${Type}, Update${Type}, Select${Type} } from '${clientPath}/types/${tableName}';
+${foreignKeySetup?.imports || ""}
+
+/**
+ * Basic tests for ${tableName} table operations
+ * 
+ * This is a junction table with composite primary key.
+ * Tests are simplified since it represents a many-to-many relationship.
+ */
+describe('${Type} SDK Operations', () => {
+  let sdk: SDK;
+  ${foreignKeySetup?.variables || ""}
+  
+  beforeAll(async () => {
+    sdk = new SDK({ 
+      baseUrl: process.env.API_URL || 'http://localhost:3000',
+      auth: process.env.API_KEY ? { apiKey: process.env.API_KEY } : undefined
+    });
+    ${foreignKeySetup?.setup || ""}
+  });
+  
+  ${foreignKeySetup?.cleanup ? `afterAll(async () => {
+    ${foreignKeySetup.cleanup}
+  });
+  
+  ` : ""}it('should create a ${tableName} relationship', async () => {
+    const data: Insert${Type} = ${sampleData};
+    
+    const created = await sdk.${tableName}.create(data);
+    expect(created).toBeDefined();
+  });
+  
+  it('should list ${tableName} relationships', async () => {
+    const list = await sdk.${tableName}.list({ limit: 10 });
+    expect(Array.isArray(list)).toBe(true);
+  });
+});
+`;
+  }
   return `${imports}
 import { SDK } from '${clientPath}';
 import type { Insert${Type}, Update${Type}, Select${Type} } from '${clientPath}/types/${tableName}';
@@ -2362,33 +2406,56 @@ echo "  - To reset the database: docker-compose -f $SCRIPT_DIR/docker-compose.ym
 exit $TEST_EXIT_CODE
 `;
 }
-function generateForeignKeySetup(table, clientPath) {
+function generateForeignKeySetup(table, model, clientPath) {
   const imports = [];
   const variables = [];
   const setupStatements = [];
   const cleanupStatements = [];
-  const foreignTables = new Set;
+  const foreignTablesData = new Map;
   for (const fk of table.fks) {
-    const foreignTable = fk.toTable;
-    if (!foreignTables.has(foreignTable)) {
-      foreignTables.add(foreignTable);
-      const ForeignType = pascal(foreignTable);
-      imports.push(`import type { Insert${ForeignType} } from '${clientPath}/types/${foreignTable}';`);
-      variables.push(`let ${foreignTable}Id: string;`);
-      setupStatements.push(`
-    // Create parent ${foreignTable} record for foreign key reference
-    const ${foreignTable}Data: Insert${ForeignType} = ${generateMinimalDataForTable(foreignTable)};
-    const created${ForeignType} = await sdk.${foreignTable}.create(${foreignTable}Data);
-    ${foreignTable}Id = created${ForeignType}.id;`);
-      cleanupStatements.push(`
-    // Clean up parent ${foreignTable} record
-    if (${foreignTable}Id) {
+    const foreignTableName = fk.toTable;
+    if (!foreignTablesData.has(foreignTableName)) {
+      const foreignTable = model.tables[foreignTableName];
+      if (!foreignTable)
+        continue;
+      foreignTablesData.set(foreignTableName, foreignTable);
+      const ForeignType = pascal(foreignTableName);
+      imports.push(`import type { Insert${ForeignType} } from '${clientPath}/types/${foreignTableName}';`);
+      if (foreignTable.pk.length === 1) {
+        variables.push(`let ${foreignTableName}Id: string;`);
+        const foreignData = generateSampleDataFromSchema(foreignTable, false);
+        setupStatements.push(`
+    // Create parent ${foreignTableName} record for foreign key reference
+    const ${foreignTableName}Data: Insert${ForeignType} = ${foreignData};
+    const created${ForeignType} = await sdk.${foreignTableName}.create(${foreignTableName}Data);
+    ${foreignTableName}Id = created${ForeignType}.${foreignTable.pk[0]};`);
+        cleanupStatements.push(`
+    // Clean up parent ${foreignTableName} record
+    if (${foreignTableName}Id) {
+      try {
+        await sdk.${foreignTableName}.delete(${foreignTableName}Id);
+      } catch (e) {
+        // Parent might already be deleted due to cascading
+      }
+    }`);
+      } else {
+        variables.push(`let ${foreignTableName}Key: any;`);
+        const foreignData = generateSampleDataFromSchema(foreignTable, false);
+        setupStatements.push(`
+    // Create parent ${foreignTableName} record for foreign key reference
+    const ${foreignTableName}Data: Insert${ForeignType} = ${foreignData};
+    const created${ForeignType} = await sdk.${foreignTableName}.create(${foreignTableName}Data);
+    ${foreignTableName}Key = { ${foreignTable.pk.map((pk) => `${pk}: created${ForeignType}.${pk}`).join(", ")} };`);
+        cleanupStatements.push(`
+    // Clean up parent ${foreignTableName} record
+    if (${foreignTableName}Key) {
       try {
-        await sdk.${foreignTable}.delete(${foreignTable}Id);
+        await sdk.${foreignTableName}.delete(${foreignTableName}Key);
       } catch (e) {
         // Parent might already be deleted due to cascading
       }
     }`);
+      }
     }
   }
   return {
@@ -2400,30 +2467,6 @@ function generateForeignKeySetup(table, clientPath) {
     cleanup: cleanupStatements.join("")
   };
 }
-function generateMinimalDataForTable(tableName) {
-  if (tableName.includes("author")) {
-    return `{ name: 'Test Author' }`;
-  }
-  if (tableName.includes("book")) {
-    return `{ title: 'Test Book' }`;
-  }
-  if (tableName.includes("tag")) {
-    return `{ name: 'Test Tag' }`;
-  }
-  if (tableName.includes("user")) {
-    return `{ name: 'Test User', email: 'test@example.com' }`;
-  }
-  if (tableName.includes("category") || tableName.includes("categories")) {
-    return `{ name: 'Test Category' }`;
-  }
-  if (tableName.includes("product")) {
-    return `{ name: 'Test Product', price: 10.99 }`;
-  }
-  if (tableName.includes("order")) {
-    return `{ total: 100.00, status: 'pending' }`;
-  }
-  return `{ name: 'Test ${pascal(tableName)}' }`;
-}
 function getTestCommand(framework, baseCommand) {
   switch (framework) {
     case "vitest":
@@ -2463,20 +2506,15 @@ function generateSampleDataFromSchema(table, hasForeignKeys = false) {
   }
   for (const col of table.columns) {
     const isAutoGenerated = col.hasDefault && ["id", "created_at", "updated_at", "created", "updated", "modified_at"].includes(col.name.toLowerCase());
-    if (isAutoGenerated) {
+    const isPrimaryKey = table.pk.includes(col.name);
+    if (isPrimaryKey && col.hasDefault) {
       continue;
     }
-    if (col.name === "deleted_at" || col.name === "deleted") {
-      if (col.nullable) {
-        fields.push(`    ${col.name}: null`);
-      } else {
-        const value = generateValueForColumn(col);
-        fields.push(`    ${col.name}: ${value}`);
-      }
+    if (isAutoGenerated) {
       continue;
     }
     const foreignTable = foreignKeyColumns.get(col.name);
-    if (!col.nullable || foreignTable || col.hasDefault && !isAutoGenerated || shouldIncludeNullableColumn(col)) {
+    if (!col.nullable || foreignTable || isPrimaryKey) {
       if (foreignTable) {
         fields.push(`    ${col.name}: ${foreignTable}Id`);
       } else {
@@ -2493,51 +2531,40 @@ ${fields.join(`,
 function generateUpdateDataFromSchema(table) {
   const fields = [];
   for (const col of table.columns) {
-    if (table.pk.includes(col.name) || col.hasDefault) {
-      const autoGenerated = ["id", "created_at", "updated_at", "created", "updated", "modified_at"];
-      if (autoGenerated.includes(col.name.toLowerCase())) {
-        continue;
-      }
+    if (table.pk.includes(col.name)) {
+      continue;
     }
-    if (col.name.endsWith("_id")) {
+    const isAutoGenerated = col.hasDefault && ["id", "created_at", "updated_at", "created", "updated", "modified_at"].includes(col.name.toLowerCase());
+    if (isAutoGenerated) {
       continue;
     }
-    if (col.name === "deleted_at" || col.name === "deleted") {
+    if (col.name.endsWith("_id")) {
       continue;
     }
-    if (!col.nullable || shouldIncludeNullableColumn(col)) {
+    if (!col.nullable) {
       const value = generateValueForColumn(col, true);
       fields.push(`    ${col.name}: ${value}`);
       break;
     }
   }
+  if (fields.length === 0) {
+    for (const col of table.columns) {
+      if (table.pk.includes(col.name) || col.name.endsWith("_id")) {
+        continue;
+      }
+      const isAutoGenerated = col.hasDefault && ["id", "created_at", "updated_at", "created", "updated", "modified_at"].includes(col.name.toLowerCase());
+      if (!isAutoGenerated) {
+        const value = generateValueForColumn(col, true);
+        fields.push(`    ${col.name}: ${value}`);
+        break;
+      }
+    }
+  }
   return fields.length > 0 ? `{
 ${fields.join(`,
 `)}
   }` : "{}";
 }
-function shouldIncludeNullableColumn(col) {
-  const importantPatterns = [
-    "_id",
-    "_by",
-    "email",
-    "name",
-    "title",
-    "description",
-    "phone",
-    "address",
-    "status",
-    "type",
-    "category",
-    "price",
-    "amount",
-    "quantity",
-    "url",
-    "slug"
-  ];
-  const name = col.name.toLowerCase();
-  return importantPatterns.some((pattern) => name.includes(pattern));
-}
 function generateValueForColumn(col, isUpdate = false) {
   const type = col.pgType.toLowerCase();
   switch (type) {
@@ -2571,15 +2598,17 @@ function generateValueForColumn(col, isUpdate = false) {
     case "bool":
       return isUpdate ? "false" : "true";
     case "date":
+      return `'2024-01-01'`;
     case "timestamp":
     case "timestamptz":
     case "timestamp without time zone":
     case "timestamp with time zone":
+      return `'2024-01-01T00:00:00.000Z'`;
     case "time":
     case "timetz":
     case "time without time zone":
     case "time with time zone":
-      return `new Date()`;
+      return `'12:00:00'`;
     case "interval":
       return `'1 day'`;
     case "json":
@@ -2647,26 +2676,14 @@ function generateUUID() {
 function generateTestCases(table, sampleData, updateData, hasForeignKeys = false) {
   const Type = pascal(table.name);
   const hasData = sampleData !== "{}";
-  const isJunctionTable = table.pk.length > 1 && table.columns.every((col) => table.pk.includes(col.name) || col.name.endsWith("_id"));
-  if (isJunctionTable) {
-    return `it('should create a ${table.name} relationship', async () => {
-    // This is a junction table for M:N relationships
-    // Test data depends on parent records created in other tests
-    expect(true).toBe(true);
-  });
-  
-  it('should list ${table.name} relationships', async () => {
-    const list = await sdk.${table.name}.list({ limit: 10 });
-    expect(Array.isArray(list)).toBe(true);
-  });`;
-  }
+  const hasSinglePK = table.pk.length === 1;
   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;
+    ${hasSinglePK ? `expect(created.${table.pk[0]}).toBeDefined();
+    createdId = created.${table.pk[0]};` : ""}
     ` : `
     // Table has only auto-generated columns
     // Skip create test or add your own test data
@@ -2679,7 +2696,7 @@ function generateTestCases(table, sampleData, updateData, hasForeignKeys = false
     expect(Array.isArray(list)).toBe(true);
   });
   
-  ${hasData ? `it('should get ${table.name} by id', async () => {
+  ${hasData && hasSinglePK ? `it('should get ${table.name} by id', async () => {
     if (!createdId) {
       console.warn('No ID from create test, skipping get test');
       return;
@@ -2687,7 +2704,7 @@ function generateTestCases(table, sampleData, updateData, hasForeignKeys = false
     
     const item = await sdk.${table.name}.getByPk(createdId);
     expect(item).toBeDefined();
-    expect(item?.id).toBe(createdId);
+    expect(item?.${table.pk[0]}).toBe(createdId);
   });
   
   ${updateData !== "{}" ? `it('should update ${table.name}', async () => {
@@ -3045,7 +3062,6 @@ async function generate(configPath) {
   if (sameDirectory) {
     clientDir = join(originalClientDir, "sdk");
   }
-  const normDateType = cfg.dateType === "string" ? "string" : "date";
   const serverFramework = cfg.serverFramework || "hono";
   const generateTests = cfg.tests?.generate ?? false;
   const originalTestDir = cfg.tests?.output || "./api/tests";
@@ -3089,12 +3105,12 @@ async function generate(configPath) {
     content: emitCoreOperations()
   });
   for (const table of Object.values(model.tables)) {
-    const typesSrc = emitTypes(table, { dateType: normDateType, numericMode: "string" });
+    const typesSrc = emitTypes(table, { numericMode: "string" });
     files.push({ path: join(serverDir, "types", `${table.name}.ts`), content: typesSrc });
     files.push({ path: join(clientDir, "types", `${table.name}.ts`), content: typesSrc });
     files.push({
       path: join(serverDir, "zod", `${table.name}.ts`),
-      content: emitZod(table, { dateType: normDateType, numericMode: "string" })
+      content: emitZod(table, { numericMode: "string" })
     });
     let routeContent;
     if (serverFramework === "hono") {
@@ -3165,7 +3181,7 @@ async function generate(configPath) {
     for (const table of Object.values(model.tables)) {
       files.push({
         path: join(testDir, `${table.name}.test.ts`),
-        content: emitTableTest(table, relativeClientPath, testFramework)
+        content: emitTableTest(table, model, relativeClientPath, testFramework)
       });
     }
   }

package/dist/types.d.ts

@@ -25,7 +25,6 @@ export interface Config {
     outClient?: string;
     softDeleteColumn?: string | null;
     includeDepthLimit?: number;
-    dateType?: "date" | "string";
     serverFramework?: "hono" | "express" | "fastify";
     auth?: AuthConfigInput;
     pull?: PullConfig;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "postgresdk",
-  "version": "0.6.8",
+  "version": "0.6.11",
   "description": "Generate a typed server/client SDK from a Postgres schema (includes, Zod, Hono).",
   "type": "module",
   "bin": {