Add producer mock (#28)

m3db · Jun 7, 2018 · 830ac18 · 830ac18
1 parent 907e311
commit 830ac18
Show file tree

Hide file tree

Showing 2 changed files with 81 additions and 2 deletions.
diff --git a/src/msg/generated/mocks/generate.go b/src/msg/generated/mocks/generate.go
@@ -19,7 +19,7 @@
 // THE SOFTWARE.
 
 // mockgen rules for generating mocks for exported interfaces (reflection mode).
-//go:generate sh -c "mockgen -package=producer $PACKAGE/producer Message | mockclean -pkg $PACKAGE/producer -out $GOPATH/src/$PACKAGE/producer/message_mock.go"
+//go:generate sh -c "mockgen -package=producer $PACKAGE/producer Message,Producer | mockclean -pkg $PACKAGE/producer -out $GOPATH/src/$PACKAGE/producer/producer_mock.go"
 //go:generate sh -c "mockgen -package=proto $PACKAGE/protocol/proto EncodeDecoder | mockclean -pkg $PACKAGE/protocol/proto -out $GOPATH/src/$PACKAGE/protocol/proto/encode_decoder_mock.go"
 
 // mockgen rules for generating mocks for unexported interfaces (file mode).

diff --git a/src/msg/producer/message_mock.go → src/msg/producer/producer_mock.go b/src/msg/producer/message_mock.go → src/msg/producer/producer_mock.go