codegen: fix defining the 'err' variable multiple times in foreign key checking test

pkg/codegen/modelgenerate/generate_testfile.go

@@ -310,11 +310,11 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam
 						},
 					},
 				}
+				shouldDefineErr := true
 				for _, col := range tbl.Columns {
 					if col.IsForeignKey {
 						shouldIncludeTestFkCheck = true
 						stmts = append(stmts, []ast.Stmt{
-
 							// post.QuotedPostID = 94354538969386985
 							&ast.AssignStmt{
 								Lhs: []ast.Expr{
@@ -335,7 +335,7 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam
 							// err := db.SavePost(&post)
 							&ast.AssignStmt{
 								Lhs: []ast.Expr{ast.NewIdent("err")},
-								Tok: token.DEFINE,
+								Tok: map[bool]token.Token{true: token.DEFINE, false: token.ASSIGN}[shouldDefineErr],
 								Rhs: []ast.Expr{
 									&ast.CallExpr{
 										Fun: &ast.SelectorExpr{
@@ -371,6 +371,7 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam
 								},
 							},
 						}...)
+						shouldDefineErr = false
 					}
 				}
 				return stmts