Add import, if source pkg is not destination pkg

Fixes #86
2019-01-22 21:40:29 +01:00 · 2019-01-22 21:40:29 +01:00 · 81c463c9b9
commit 81c463c9b9
parent 39f6db0ed3
1 changed files with 1 additions and 0 deletions
--- 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