Do no import index package when no index used. Fixes #36

generator/generator_test.go

@@ -3,6 +3,7 @@ package generator
 //go:generate go test -run=TestGenerate -update
 
 import (
+	"bufio"
 	"bytes"
 	"flag"
 	"fmt"
@@ -119,6 +120,51 @@ func TestGenerate(t *testing.T) {
 		})
 		require.Error(t, err)
 	})
+
+	// Make sure imports are correct when using no index
+	// https://github.com/asdine/genji/issues/36
+	t.Run("No index", func(t *testing.T) {
+		src := `
+			package s
+
+			type S struct {
+				X,Y,Z string
+			}
+		`
+
+		expectedHeader := `// Code generated by genji.
+// DO NOT EDIT!
+
+package s
+
+import (
+	"errors"
+
+	"github.com/asdine/genji/field"
+	"github.com/asdine/genji/query"
+	"github.com/asdine/genji/record"
+)
+`
+
+		var buf bytes.Buffer
+		err := Generate(&buf, Config{
+			Sources: []io.Reader{strings.NewReader(src)},
+			Structs: []Struct{{"S"}},
+		})
+		require.NoError(t, err)
+
+		rd := bufio.NewReader(&buf)
+		var res bytes.Buffer
+
+		for i := 0; i < 12; i++ {
+			l, err := rd.ReadString('\n')
+			require.NoError(t, err)
+			res.WriteString(l)
+		}
+
+		require.Equal(t, expectedHeader, res.String())
+	})
+
 }
 
 func TestGeneratedRecords(t *testing.T) {