pkg/codegen/modelgenerate/generate_testfile.go

@@ -8,7 +8,6 @@ import (
 	"github.com/jinzhu/inflection"
 
 	pkgschema "git.offline-twitter.com/offline-labs/gas-stack/pkg/schema"
-	"git.offline-twitter.com/offline-labs/gas-stack/pkg/textutils"
 )
 
 // GenerateModelTestAST produces an AST for a starter test file for a given model.
@@ -16,11 +15,9 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam
 	packageName := "db"
 	testpackageName := packageName + "_test"
 
-	makeHelperName := ast.NewIdent("Make" + tbl.GoTypeName)
-
 	// func MakeItem() Item { return Item{} }
 	makeItemFunc := &ast.FuncDecl{
-		Name: makeHelperName,
+		Name: ast.NewIdent("Make" + tbl.GoTypeName),
 		Type: &ast.FuncType{
 			Params: &ast.FieldList{},
 			Results: &ast.FieldList{
@@ -60,8 +57,8 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam
 		},
 	}
 
-	testObj := ast.NewIdent(textutils.CamelToPascal(tbl.GoTypeName))
-	testObj2 := ast.NewIdent(textutils.CamelToPascal(tbl.GoTypeName) + "2")
+	testObj := ast.NewIdent("item")
+	testObj2 := ast.NewIdent("item2")
 	fieldName := ast.NewIdent("Description") // TODO
 	description1 := `"an item"`
 	description2 := `"a big item"`
@@ -97,13 +94,13 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam
 					&ast.AssignStmt{
 						Lhs: []ast.Expr{testObj},
 						Tok: token.DEFINE,
-						Rhs: []ast.Expr{&ast.CallExpr{Fun: makeHelperName, Args: nil}},
+						Rhs: []ast.Expr{&ast.CallExpr{Fun: ast.NewIdent("MakeItem"), Args: nil}},
 					},
 					// item.Description = "an item"
 					&ast.AssignStmt{
 						Lhs: []ast.Expr{
 							&ast.SelectorExpr{
-								X:   testObj,
+								X:   ast.NewIdent("item"),
 								Sel: ast.NewIdent("Description"),
 							},
 						},
@@ -310,11 +307,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 +332,7 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam
 							// err := db.SavePost(&post)
 							&ast.AssignStmt{
 								Lhs: []ast.Expr{ast.NewIdent("err")},
-								Tok: map[bool]token.Token{true: token.DEFINE, false: token.ASSIGN}[shouldDefineErr],
+								Tok: token.DEFINE,
 								Rhs: []ast.Expr{
 									&ast.CallExpr{
 										Fun: &ast.SelectorExpr{
@@ -371,7 +368,6 @@ func GenerateModelTestAST(tbl pkgschema.Table, schema pkgschema.Schema, gomodNam
 								},
 							},
 						}...)
-						shouldDefineErr = false
 					}
 				}
 				return stmts

pkg/schema/table.go

@@ -58,7 +58,8 @@ func (c Column) GoVarName() string {
 
 // LongGoVarName returns a lowercased version of the field name (Pascal => Camel).
 func (c Column) LongGoVarName() string {
-	return textutils.CamelToPascal(c.GoFieldName())
+	fieldname := c.GoFieldName()
+	return strings.ToLower(fieldname)[0:1] + fieldname[1:]
 }
 
 // Table is a single SQLite table.

pkg/textutils/snake.go

@@ -9,7 +9,3 @@ func SnakeToCamel(s string) string {
 	}
 	return strings.Join(parts, "")
 }
-
-func CamelToPascal(s string) string {
-	return strings.ToLower(s)[0:1] + s[1:]
-}