Merge pull request #114 from heetch/sixstone-003/conditional-import

Fix Go generation code when schema just contains an enum

cmd/avrogo/generate.go

@@ -42,7 +42,13 @@ func generate(w io.Writer, pkg string, ns *parser.Namespace, definitions []schem
 		imports:  make(map[string]string),
 		extTypes: extTypes,
 	}
-	gc.addImport("github.com/heetch/avro/avrotypegen")
+	// Add avrotypegen package conditionally when there is a RecordDefinition in the namespace.
+	for _, def := range ns.Definitions {
+		if _, ok := def.(*schema.RecordDefinition); ok {
+			gc.addImport("github.com/heetch/avro/avrotypegen")
+			break
+		}
+	}
 	var body bytes.Buffer
 	if err := bodyTemplate.Execute(&body, bodyTemplateParams{
 		Definitions: localDefinitions,

cmd/avrogo/internal/generated_tests/justEnum/schema.avsc

@@ -0,0 +1,9 @@
+{
+                "name": "MyEnum",
+                "type": "enum",
+                "symbols": [
+                    "a",
+                    "b",
+                    "c"
+                ]
+            }

cmd/avrogo/testdata/enum.cue

@@ -18,6 +18,17 @@ tests: simpleEnum: {
 	outData: inData
 }
 
+tests: justEnum: {
+       inSchema: {
+                type: "enum"
+                name: "MyEnum"
+		symbols: ["a", "b", "c"]
+       }
+       outSchema: inSchema
+       inData: "b"
+       outData: inData
+}
+
 tests: simpleEnum: otherTests: """
 	package simpleEnum
 	import (