Merge pull request #87 from breml/fix-imports

Add import, if source pkg is not destination pkg
This commit is contained in:
Mat Ryer 2019-02-11 10:46:50 +00:00 committed by GitHub
commit 70d7959eb2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -174,6 +174,7 @@ func (m *Mocker) Mock(w io.Writer, name ...string) error {
if tpkg.Name() != m.pkgName { if tpkg.Name() != m.pkgName {
doc.SourcePackagePrefix = tpkg.Name() + "." doc.SourcePackagePrefix = tpkg.Name() + "."
doc.Imports = append(doc.Imports, tpkg.Path())
} }
var buf bytes.Buffer var buf bytes.Buffer