Merge pull request #387 from safaci2000/feature/enums

Adding support for enum model AllValues
go-jet · Sep 23, 2024 · 6a0798e · 6a0798e
2 parents 1c42ff1 + d3ce39f
commit 6a0798e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/generator/template/file_templates.go b/generator/template/file_templates.go
diff --git a/tests/postgres/generator_template_test.go b/tests/postgres/generator_template_test.go
@@ -171,6 +171,7 @@ func TestGeneratorTemplate_Model_RenameFilesAndTypes(t *testing.T) {
 
 	mpaaRating := file2.Exists(t, defaultModelPath, "mpaa_rating_enum.go")
 	require.Contains(t, mpaaRating, "type MpaaRatingEnum string")
+	require.Contains(t, mpaaRating, "MpaaRatingEnumAllValues")
 }
 
 func TestGeneratorTemplate_Model_SkipTableAndEnum(t *testing.T) {
@@ -268,7 +269,6 @@ func UseSchema(schema string) {
 	FilmList = FilmList.FromSchema(schema)
 }
 `)
-
 }
 
 func TestGeneratorTemplate_SQLBuilder_ChangeTypeAndFileName(t *testing.T) {
@@ -366,7 +366,6 @@ func TestGeneratorTemplate_SQLBuilder_DefaultAlias(t *testing.T) {
 }
 
 func TestGeneratorTemplate_Model_AddTags(t *testing.T) {
-
 	err := postgres.Generate(
 		tempTestDir,
 		dbConnection,