diff --git a/pkg/codegen/modelgenerate/generate_testfile.go b/pkg/codegen/modelgenerate/generate_testfile.go index 94fc996..c3e3f94 100644 --- a/pkg/codegen/modelgenerate/generate_testfile.go +++ b/pkg/codegen/modelgenerate/generate_testfile.go @@ -108,7 +108,7 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam Rhs: []ast.Expr{ &ast.BasicLit{ Kind: token.STRING, - Value: `"an item"`, + Value: fmt.Sprintf("%q", description1), }, }, }, @@ -148,12 +148,12 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam })}, }, - // assert.Equal(t, "an item", item2.Description) + // assert.Equal(t, item.Description, item2.Description) &ast.ExprStmt{X: &ast.CallExpr{ Fun: &ast.SelectorExpr{X: ast.NewIdent("assert"), Sel: ast.NewIdent("Equal")}, Args: []ast.Expr{ ast.NewIdent("t"), - &ast.BasicLit{Kind: token.STRING, Value: description1}, + &ast.SelectorExpr{X: testObj, Sel: fieldName}, &ast.SelectorExpr{X: testObj2, Sel: fieldName}, }, }},