Merge pull request #87 from breml/fix-imports

Add import, if source pkg is not destination pkg
2019-02-11 10:46:50 +00:00 · 2019-02-11 10:46:50 +00:00 · 70d7959eb2
commit 70d7959eb2
parent 39f6db0ed3 81c463c9b9
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