Add import, if source pkg is not destination pkg

Fixes #86
matryer · Jan 22, 2019 · 81c463c · 81c463c
1 parent 39f6db0
commit 81c463c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pkg/moq/moq.go b/pkg/moq/moq.go
@@ -174,6 +174,7 @@ func (m *Mocker) Mock(w io.Writer, name ...string) error {
 
 	if tpkg.Name() != m.pkgName {
 		doc.SourcePackagePrefix = tpkg.Name() + "."
+		doc.Imports = append(doc.Imports, tpkg.Path())
 	}
 
 	var buf bytes.Buffer