From 150947e4e78981c12fc252f0a8cca34bbcce712a Mon Sep 17 00:00:00 2001
From: sundowndev <raphael@crvx.fr>
Date: Thu, 10 Dec 2020 17:16:16 +0400
Subject: [PATCH] test(worker): consumers

Put mocks into mocks folder
---
 worker/consumers_test.go     | 49 +++++-------------------------------
 worker/mocks/acknowledger.go | 22 ++++++++++++++++
 worker/mocks/logger.go       | 25 ++++++++++++++++++
 3 files changed, 53 insertions(+), 43 deletions(-)
 create mode 100644 worker/mocks/acknowledger.go
 create mode 100644 worker/mocks/logger.go

diff --git a/worker/consumers_test.go b/worker/consumers_test.go
index 9e280611..7501bd95 100644
--- a/worker/consumers_test.go
+++ b/worker/consumers_test.go
@@ -3,6 +3,7 @@ package worker
 import (
 	"encoding/json"
 	"errors"
+	"github.com/dreamvo/gilfoyle/worker/mocks"
 	"github.com/google/uuid"
 	"github.com/streadway/amqp"
 	"github.com/stretchr/testify/mock"
@@ -11,44 +12,6 @@ import (
 	"time"
 )
 
-type MockedLogger struct {
-	mock.Mock
-}
-
-func (m *MockedLogger) Error(msg string, fields ...zap.Field) {
-	m.Called(msg, fields)
-	return
-}
-
-func (m *MockedLogger) Info(msg string, fields ...zap.Field) {
-	m.Called(msg, fields)
-	return
-}
-
-func (m *MockedLogger) Fatal(msg string, fields ...zap.Field) {
-	m.Called(msg, fields)
-	return
-}
-
-type MockedAcknowledger struct {
-	mock.Mock
-}
-
-func (m *MockedAcknowledger) Ack(tag uint64, multiple bool) error {
-	args := m.Called(tag, multiple)
-	return args.Error(0)
-}
-
-func (m *MockedAcknowledger) Nack(tag uint64, multiple bool, requeue bool) error {
-	args := m.Called(tag, multiple, requeue)
-	return args.Error(0)
-}
-
-func (m *MockedAcknowledger) Reject(tag uint64, requeue bool) error {
-	args := m.Called(tag, requeue)
-	return args.Error(0)
-}
-
 func TestConsumers(t *testing.T) {
 	t.Run("videoTranscodingQueueConsumer", func(t *testing.T) {
 		t.Run("should receive one message and succeed", func(t *testing.T) {
@@ -59,8 +22,8 @@ func TestConsumers(t *testing.T) {
 
 			body, _ := json.Marshal(params)
 
-			loggerMock := new(MockedLogger)
-			AckMock := new(MockedAcknowledger)
+			loggerMock := new(mocks.MockedLogger)
+			AckMock := new(mocks.MockedAcknowledger)
 
 			w := &Worker{
 				Logger: loggerMock,
@@ -89,7 +52,7 @@ func TestConsumers(t *testing.T) {
 		})
 
 		t.Run("should fail to unmarshall json", func(t *testing.T) {
-			loggerMock := new(MockedLogger)
+			loggerMock := new(mocks.MockedLogger)
 
 			w := &Worker{
 				Logger: loggerMock,
@@ -119,8 +82,8 @@ func TestConsumers(t *testing.T) {
 
 			body, _ := json.Marshal(params)
 
-			loggerMock := new(MockedLogger)
-			AckMock := new(MockedAcknowledger)
+			loggerMock := new(mocks.MockedLogger)
+			AckMock := new(mocks.MockedAcknowledger)
 
 			w := &Worker{
 				Logger: loggerMock,
diff --git a/worker/mocks/acknowledger.go b/worker/mocks/acknowledger.go
new file mode 100644
index 00000000..18c416a3
--- /dev/null
+++ b/worker/mocks/acknowledger.go
@@ -0,0 +1,22 @@
+package mocks
+
+import "github.com/stretchr/testify/mock"
+
+type MockedAcknowledger struct {
+	mock.Mock
+}
+
+func (m *MockedAcknowledger) Ack(tag uint64, multiple bool) error {
+	args := m.Called(tag, multiple)
+	return args.Error(0)
+}
+
+func (m *MockedAcknowledger) Nack(tag uint64, multiple bool, requeue bool) error {
+	args := m.Called(tag, multiple, requeue)
+	return args.Error(0)
+}
+
+func (m *MockedAcknowledger) Reject(tag uint64, requeue bool) error {
+	args := m.Called(tag, requeue)
+	return args.Error(0)
+}
diff --git a/worker/mocks/logger.go b/worker/mocks/logger.go
new file mode 100644
index 00000000..f6f06e14
--- /dev/null
+++ b/worker/mocks/logger.go
@@ -0,0 +1,25 @@
+package mocks
+
+import (
+	"github.com/stretchr/testify/mock"
+	"go.uber.org/zap"
+)
+
+type MockedLogger struct {
+	mock.Mock
+}
+
+func (m *MockedLogger) Error(msg string, fields ...zap.Field) {
+	m.Called(msg, fields)
+	return
+}
+
+func (m *MockedLogger) Info(msg string, fields ...zap.Field) {
+	m.Called(msg, fields)
+	return
+}
+
+func (m *MockedLogger) Fatal(msg string, fields ...zap.Field) {
+	m.Called(msg, fields)
+	return
+}