add skip-ensure flag to avoid import cycle (#140)

For mocks generated outside of the tested package with tests lives inside the same package as the tested code (i.e. pkg_test not used) --skip-ensure suppresses import of the source pkg

https://github.com/matryer/moq/issues/139

fix typo in readme

README.md

@@ -33,6 +33,9 @@ moq [flags] source-dir interface [interface2 [interface3 [...]]]
     	package name (default will infer)
   -stub
     	return zero values when no mock implementation is provided, do not panic
+  -skip-ensure
+		suppress mock implementation check, avoid import cycle if mocks generated outside of the tested package
+
 Specifying an alias for the mock is also supported with the format 'interface:alias'
 Ex: moq -pkg different . MyInterface:MyMock
 ```

main.go

@@ -21,6 +21,7 @@ type userFlags struct {
 	pkgName    string
 	formatter  string
 	stubImpl   bool
+	skipEnsure bool
 	args       []string
 }
 
@@ -31,6 +32,8 @@ func main() {
 	flag.StringVar(&flags.formatter, "fmt", "", "go pretty-printer: gofmt, goimports or noop (default gofmt)")
 	flag.BoolVar(&flags.stubImpl, "stub", false,
 		"return zero values when no mock implementation is provided, do not panic")
+	flag.BoolVar(&flags.skipEnsure, "skip-ensure", false,
+		"suppress mock implementation check, avoid import cycle if mocks generated outside of the tested package")
 
 	flag.Usage = func() {
 		fmt.Println(`moq [flags] source-dir interface [interface2 [interface3 [...]]]`)
@@ -66,6 +69,7 @@ func run(flags userFlags) error {
 		PkgName:    flags.pkgName,
 		Formatter:  flags.formatter,
 		StubImpl:   flags.stubImpl,
+		SkipEnsure: flags.skipEnsure,
 	})
 	if err != nil {
 		return err

pkg/moq/moq.go

@@ -25,6 +25,7 @@ type Mocker struct {
 	pkgPath    string
 	fmter      func(src []byte) ([]byte, error)
 	stubImpl   bool
+	skipEnsure bool
 
 	imports map[string]bool
 }
@@ -36,6 +37,7 @@ type Config struct {
 	PkgName    string
 	Formatter  string
 	StubImpl   bool
+	SkipEnsure bool
 }
 
 // New makes a new Mocker for the specified package directory.
@@ -75,6 +77,7 @@ func New(conf Config) (*Mocker, error) {
 		pkgPath:    pkgPath,
 		fmter:      fmter,
 		stubImpl:   conf.StubImpl,
+		skipEnsure: conf.SkipEnsure,
 		imports:    make(map[string]bool),
 	}, nil
 }
@@ -114,6 +117,7 @@ func (m *Mocker) Mock(w io.Writer, names ...string) error {
 		PackageName: m.pkgName,
 		Imports:     moqImports,
 		StubImpl:    m.stubImpl,
+		SkipEnsure:  m.skipEnsure,
 	}
 
 	mocksMethods := false
@@ -254,6 +258,7 @@ type doc struct {
 	Objects             []obj
 	Imports             []string
 	StubImpl            bool
+	SkipEnsure          bool
 }
 
 type obj struct {

pkg/moq/moq_test.go

@@ -172,6 +172,33 @@ func TestMoqExplicitPackageWithStaticCheck(t *testing.T) {
 	}
 }
 
+func TestMoqSkipEnsure(t *testing.T) {
+	m, err := New(Config{SrcDir: "testpackages/example", PkgName: "different", SkipEnsure: true})
+	if err != nil {
+		t.Fatalf("moq.New: %s", err)
+	}
+	var buf bytes.Buffer
+	err = m.Mock(&buf, "PersonStore")
+	if err != nil {
+		t.Errorf("m.Mock: %s", err)
+	}
+	s := buf.String()
+	// assertions of things that should be mentioned
+	var strs = []string{
+		"package different",
+		"type PersonStoreMock struct",
+		"CreateFunc func(ctx context.Context, person *example.Person, confirm bool) error",
+		"GetFunc func(ctx context.Context, id string) (*example.Person, error)",
+		"func (mock *PersonStoreMock) Create(ctx context.Context, person *example.Person, confirm bool) error",
+		"func (mock *PersonStoreMock) Get(ctx context.Context, id string) (*example.Person, error)",
+	}
+	for _, str := range strs {
+		if !strings.Contains(s, str) {
+			t.Errorf("expected but missing: \"%s\"", str)
+		}
+	}
+}
+
 func TestNotCreatingEmptyDirWhenPkgIsGiven(t *testing.T) {
 	m, err := New(Config{SrcDir: "testpackages/example", PkgName: "different"})
 	if err != nil {

pkg/moq/template.go

@@ -11,6 +11,7 @@ var moqTemplate = `// Code generated by moq; DO NOT EDIT.
 package {{.PackageName}}
 {{- $sourcePackagePrefix := .SourcePackagePrefix}}
 {{- $stubImpl := .StubImpl}}
+{{- $skipEnsure := .SkipEnsure}}
 
 import (
 {{- range .Imports }}
@@ -20,9 +21,11 @@ import (
 
 {{ range $i, $obj := .Objects -}}
 
+{{- if not $skipEnsure -}}
 // Ensure, that {{.MockName}} does implement {{$sourcePackagePrefix}}{{.InterfaceName}}.
 // If this is not the case, regenerate this file with moq.
 var _ {{$sourcePackagePrefix}}{{.InterfaceName}} = &{{.MockName}}{}
+{{- end }}
 
 // {{.MockName}} is a mock implementation of {{$sourcePackagePrefix}}{{.InterfaceName}}.
 //

pkg/moq/testpackages/gogenvendoring/user/user_moq_test.go

@@ -8,10 +8,6 @@ import (
 	"sync"
 )
 
-var (
-	lockServiceMockDoSomething sync.RWMutex
-)
-
 // Ensure, that ServiceMock does implement Service.
 // If this is not the case, regenerate this file with moq.
 var _ Service = &ServiceMock{}
@@ -43,6 +39,7 @@ type ServiceMock struct {
 			In1 somerepo.SomeType
 		}
 	}
+	lockDoSomething sync.RWMutex
 }
 
 // DoSomething calls DoSomethingFunc.
@@ -55,9 +52,9 @@ func (mock *ServiceMock) DoSomething(in1 somerepo.SomeType) error {
 	}{
 		In1: in1,
 	}
-	lockServiceMockDoSomething.Lock()
+	mock.lockDoSomething.Lock()
 	mock.calls.DoSomething = append(mock.calls.DoSomething, callInfo)
-	lockServiceMockDoSomething.Unlock()
+	mock.lockDoSomething.Unlock()
 	return mock.DoSomethingFunc(in1)
 }
 
@@ -70,8 +67,8 @@ func (mock *ServiceMock) DoSomethingCalls() []struct {
 	var calls []struct {
 		In1 somerepo.SomeType
 	}
-	lockServiceMockDoSomething.RLock()
+	mock.lockDoSomething.RLock()
 	calls = mock.calls.DoSomething
-	lockServiceMockDoSomething.RUnlock()
+	mock.lockDoSomething.RUnlock()
 	return calls
 }