Add mocks with mockery for mqtt

Signed-off-by: JeffMboya <[email protected]>

mqtt/events/streams.go

@@ -12,6 +12,7 @@ import (
 
 const streamID = "magistrala.mqtt"
 
+//go:generate mockery --name EventStore --output=../mocks --filename redis.go --quiet --note "Copyright (c) Abstract Machines"
 type EventStore interface {
 	Connect(ctx context.Context, clientID string) error
 	Disconnect(ctx context.Context, clientID string) error

mqtt/handler_test.go

@@ -64,7 +64,7 @@ var (
 )
 
 func TestAuthConnect(t *testing.T) {
-	handler, _ := newHandler()
+	handler, _ := newHandler(t)
 
 	cases := []struct {
 		desc    string
@@ -117,7 +117,7 @@ func TestAuthConnect(t *testing.T) {
 }
 
 func TestAuthPublish(t *testing.T) {
-	handler, auth := newHandler()
+	handler, auth := newHandler(t)
 
 	cases := []struct {
 		desc    string
@@ -169,7 +169,7 @@ func TestAuthPublish(t *testing.T) {
 }
 
 func TestAuthSubscribe(t *testing.T) {
-	handler, auth := newHandler()
+	handler, auth := newHandler(t)
 
 	cases := []struct {
 		desc    string
@@ -222,7 +222,7 @@ func TestAuthSubscribe(t *testing.T) {
 }
 
 func TestConnect(t *testing.T) {
-	handler, _ := newHandler()
+	handler, _ := newHandler(t)
 	logBuffer.Reset()
 
 	cases := []struct {
@@ -256,7 +256,7 @@ func TestConnect(t *testing.T) {
 }
 
 func TestPublish(t *testing.T) {
-	handler, _ := newHandler()
+	handler, _ := newHandler(t)
 	logBuffer.Reset()
 
 	malformedSubtopics := topic + "/" + subtopic + "%"
@@ -335,7 +335,7 @@ func TestPublish(t *testing.T) {
 }
 
 func TestSubscribe(t *testing.T) {
-	handler, _ := newHandler()
+	handler, _ := newHandler(t)
 	logBuffer.Reset()
 
 	cases := []struct {
@@ -371,7 +371,7 @@ func TestSubscribe(t *testing.T) {
 }
 
 func TestUnsubscribe(t *testing.T) {
-	handler, _ := newHandler()
+	handler, _ := newHandler(t)
 	logBuffer.Reset()
 
 	cases := []struct {
@@ -407,7 +407,7 @@ func TestUnsubscribe(t *testing.T) {
 }
 
 func TestDisconnect(t *testing.T) {
-	handler, _ := newHandler()
+	handler, _ := newHandler(t)
 	logBuffer.Reset()
 
 	cases := []struct {
@@ -442,12 +442,12 @@ func TestDisconnect(t *testing.T) {
 	}
 }
 
-func newHandler() (session.Handler, *authmocks.AuthClient) {
+func newHandler(t *testing.T) (session.Handler, *authmocks.AuthClient) {
 	logger, err := mglog.New(&logBuffer, "debug")
 	if err != nil {
 		log.Fatalf("failed to create logger: %s", err)
 	}
 	auth := new(authmocks.AuthClient)
-	eventStore := mocks.NewEventStore()
+	eventStore := mocks.NewEventStore(t)
 	return mqtt.NewHandler(mocks.NewPublisher(), eventStore, logger, auth), auth
 }