Makes sure to generate Choice and All elements, in addition to Sequences

hooklift · Aug 25, 2014 · 2fd755c · 2fd755c
1 parent 34caa81
commit 2fd755c
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/generator/types_tmpl.go b/generator/types_tmpl.go
@@ -40,6 +40,8 @@ var typesTmpl = `
 			{{template "SimpleContent" .SimpleContent}}
 		{{ else }}
 			{{template "Elements" .Sequence.Elements}}
+			{{template "Elements" .Choice}}
+			{{template "Elements" .All}}
 			{{template "Attributes" .Attributes}}
 		{{end}}
 	}
@@ -55,6 +57,8 @@ var typesTmpl = `
 				{{template "SimpleContent" .SimpleContent}}
 			{{ else }}
 				{{template "Elements" .Sequence.Elements}}
+				{{template "Elements" .Choice}}
+				{{template "Elements" .All}}
 				{{template "Attributes" .Attributes}}
 			{{end}}
 		}