Merge pull request 99designs#486 from nicovogelaar/feature/list-of-enums

add list of enums
cgxxv · Jan 5, 2019 · fc94356 · fc94356
2 parents 315315c + 210eede
commit fc94356
Show file tree

Hide file tree

Showing 7 changed files with 72 additions and 1 deletion.
diff --git a/codegen/codegen_test.go b/codegen/codegen_test.go
@@ -18,6 +18,10 @@ func TestGenerateServer(t *testing.T) {
 		id: Int
 		fist_name: String
 	}
+	enum Status {
+		OK
+		ERROR
+	}
 `
 	serverFilename := "gen/" + name + "/server/server.go"
 	cfg := Config{
@@ -42,4 +46,30 @@ func TestGenerateServer(t *testing.T) {
 
 	_, err = conf.Load()
 	require.NoError(t, err)
+
+	t.Run("list of enums", func(t *testing.T) {
+		conf = loader.Config{}
+		conf.CreateFromFilenames("gen/"+name, "gen/"+name+"/model.go")
+
+		program, err := conf.Load()
+		require.NoError(t, err)
+
+		found := false
+
+		for _, c := range program.Created {
+			for ident := range c.Defs {
+				if ident.Name == "AllStatus" {
+					found = true
+					break
+				}
+			}
+			if found {
+				break
+			}
+		}
+
+		if !found {
+			t.Fail()
+		}
+	})
 }
diff --git a/codegen/templates/data.go b/codegen/templates/data.go
diff --git a/codegen/templates/models.gotpl b/codegen/templates/models.gotpl
@@ -59,6 +59,12 @@ import (
 	{{- end }}
 	)
 
+	var All{{.GoType}} = []{{.GoType}}{
+	{{- range $value := .Values}}
+		{{$enum.GoType}}{{ .Name|toCamel }},
+	{{- end }}
+	}
+
 	func (e {{.GoType}}) IsValid() bool {
 		switch e {
 		case {{ range $index, $element := .Values}}{{if $index}},{{end}}{{ $enum.GoType }}{{ $element.Name|toCamel }}{{end}}:

diff --git a/example/starwars/models_gen.go b/example/starwars/models_gen.go
diff --git a/example/todo/models_gen.go b/example/todo/models_gen.go
diff --git a/example/type-system-extension/models_gen.go b/example/type-system-extension/models_gen.go
diff --git a/integration/models-go/generated.go b/integration/models-go/generated.go