Merge pull request #35 from inaciogu/fix/set-logger

fix: test set logger method
inaciogu · Jan 23, 2024 · d7337da · d7337da
2 parents 4dc212b + 7816bc9
commit d7337da
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/consumer/sqsclient_test.go b/consumer/sqsclient_test.go
@@ -16,6 +16,14 @@ import (
 	"github.com/stretchr/testify/suite"
 )
 
+type MockLogger struct {
+	mock.Mock
+}
+
+func (m *MockLogger) Log(message string, v ...interface{}) {
+	m.Called(message, v)
+}
+
 type UnitTest struct {
 	suite.Suite
 	mockSQSService *mocks.SQSService
@@ -51,6 +59,18 @@ func (ut *UnitTest) TestNewWithoutQueueName() {
 	})
 }
 
+func (ut *UnitTest) TestSetLogger() {
+	logger := new(MockLogger)
+
+	client := client.New(nil, client.SQSClientOptions{
+		QueueName: "fake-queue-name",
+	})
+
+	client.SetLogger(logger)
+
+	assert.NotNil(ut.T(), client)
+}
+
 func (ut *UnitTest) TestGetQueueUrl() {
 	expectedOutput := &sqs.GetQueueUrlOutput{
 		QueueUrl: aws.String("https://fake-queue-url"),