codegen: add auto-timestamp checking to generated tests

pkg/codegen/modelgenerate/generate_testfile.go

@@ -113,6 +113,8 @@ func GenerateModelTestAST(tbl schema.Table, gomodName string) *ast.File {
 	description1 := `"an item"`
 	description2 := `"a big item"`
 
+	hasCreatedAt, hasUpdatedAt := tbl.HasAutoTimestamps()
+
 	testCreateUpdateDelete := &ast.FuncDecl{
 		Name: ast.NewIdent("TestCreateUpdateDelete" + tbl.GoTypeName),
 		Type: &ast.FuncType{
@@ -124,7 +126,15 @@ func GenerateModelTestAST(tbl schema.Table, gomodName string) *ast.File {
 			},
 		},
 		Body: &ast.BlockStmt{
-			List: []ast.Stmt{
+			List: func() []ast.Stmt {
+				assertNotZero := func(obj *ast.Ident, field string) *ast.ExprStmt {
+					return &ast.ExprStmt{X: &ast.CallExpr{
+						Fun:  ast.NewIdent("assert.NotZero"),
+						Args: []ast.Expr{ast.NewIdent("t"), &ast.SelectorExpr{X: obj, Sel: ast.NewIdent(field)}},
+					}}
+				}
+
+				stmts := []ast.Stmt{
 					// item := Item{Description: "an item"}
 					&ast.AssignStmt{
 						Lhs: []ast.Expr{testObj},
@@ -151,7 +161,17 @@ func GenerateModelTestAST(tbl schema.Table, gomodName string) *ast.File {
 						Fun:  ast.NewIdent("require.NotZero"),
 						Args: []ast.Expr{ast.NewIdent("t"), &ast.SelectorExpr{X: testObj, Sel: ast.NewIdent("ID")}},
 					}},
+				}
 
+				// After create: assert timestamps are set
+				if hasCreatedAt {
+					stmts = append(stmts, assertNotZero(testObj, "CreatedAt"))
+				}
+				if hasUpdatedAt {
+					stmts = append(stmts, assertNotZero(testObj, "UpdatedAt"))
+				}
+
+				stmts = append(stmts,
 					// item2 := Must(TestDB.GetItemByID(item.ID))
 					&ast.AssignStmt{
 						Lhs: []ast.Expr{testObj2},
@@ -187,7 +207,17 @@ func GenerateModelTestAST(tbl schema.Table, gomodName string) *ast.File {
 						Fun:  ast.NewIdent("TestDB.Save" + tbl.GoTypeName),
 						Args: []ast.Expr{&ast.UnaryExpr{Op: token.AND, X: testObj}},
 					}},
+				)
 
+				// After update: assert timestamps are still set
+				if hasCreatedAt {
+					stmts = append(stmts, assertNotZero(testObj, "CreatedAt"))
+				}
+				if hasUpdatedAt {
+					stmts = append(stmts, assertNotZero(testObj, "UpdatedAt"))
+				}
+
+				stmts = append(stmts,
 					// item2 = Must(TestDB.GetItemByID(item.ID))
 					&ast.AssignStmt{
 						Lhs: []ast.Expr{testObj2},
@@ -236,7 +266,10 @@ func GenerateModelTestAST(tbl schema.Table, gomodName string) *ast.File {
 							ast.NewIdent("ErrNotInDB"),
 						},
 					}},
-			},
+				)
+
+				return stmts
+			}(),
 		},
 	}