diff --git a/example/example.go b/example/example.go index 312652d..6ef1b04 100644 --- a/example/example.go +++ b/example/example.go @@ -2,7 +2,7 @@ package example import "context" -//go:generate moq PersonStore -out mockpersonstore_test.go +//go:generate moq -out mockpersonstore_test.go PersonStore type Person struct { ID string diff --git a/package/moq/moq_test.go b/package/moq/moq_test.go index 6a5d74b..2b85c3a 100644 --- a/package/moq/moq_test.go +++ b/package/moq/moq_test.go @@ -7,7 +7,7 @@ import ( ) func TestMoq(t *testing.T) { - m, err := New("../../example", "") + m, err := New("testdata/example", "") if err != nil { t.Errorf("moq.New: %s", err) } @@ -34,7 +34,7 @@ func TestMoq(t *testing.T) { } func TestMoqExplicitPackage(t *testing.T) { - m, err := New("../../example", "different") + m, err := New("testdata/example", "different") if err != nil { t.Errorf("moq.New: %s", err) } @@ -59,3 +59,30 @@ func TestMoqExplicitPackage(t *testing.T) { } } } + +// TestVeradicArguments tests to ensure variadic work as +// expected. +// see https://github.com/matryer/moq/issues/5 +func TestVariadicArguments(t *testing.T) { + m, err := New("testdata/variadic", "") + if err != nil { + t.Errorf("moq.New: %s", err) + } + var buf bytes.Buffer + err = m.Mock(&buf, "Greeter") + if err != nil { + t.Errorf("m.Mock: %s", err) + } + s := buf.String() + // assertions of things that should be mentioned + var strs = []string{ + "package variadic", + "type GreeterMock struct", + "GreetFunc func(ctx context.Context, names ...string) string", + } + for _, str := range strs { + if !strings.Contains(s, str) { + t.Errorf("expected but missing: \"%s\"", str) + } + } +} diff --git a/package/moq/testdata/example/example.go b/package/moq/testdata/example/example.go new file mode 100644 index 0000000..3adcbb8 --- /dev/null +++ b/package/moq/testdata/example/example.go @@ -0,0 +1,15 @@ +package example + +import "context" + +type Person struct { + ID string + Name string + Company string + Website string +} + +type PersonStore interface { + Get(ctx context.Context, id string) (*Person, error) + Create(ctx context.Context, person *Person, confirm bool) error +} diff --git a/package/moq/testdata/variadic/greeter.go b/package/moq/testdata/variadic/greeter.go new file mode 100644 index 0000000..44f746f --- /dev/null +++ b/package/moq/testdata/variadic/greeter.go @@ -0,0 +1,8 @@ +package variadic + +import "context" + +// Greeter greets people. +type Greeter interface { + Greet(ctx context.Context, names ...string) string +}