@@ -74,9 +74,10 @@ func TestDefaultLogger_Debug(t *testing.T) {
7474 mockLogger := NewMockLogger ()
7575 dLogger := & defaultLogger {logger : mockLogger .Logger , logLevel : LogLevelDebug }
7676
77- mockLogger .On ("Debug" , "test message" , mock .Anything ).Once ()
77+ expectedMessage := "test message"
78+ mockLogger .On ("Debug" , expectedMessage , mock .Anything ).Once ()
7879
79- dLogger .Debug ("test message" )
80+ dLogger .Debug (expectedMessage )
8081
8182 mockLogger .AssertExpectations (t )
8283}
@@ -88,9 +89,10 @@ func TestDefaultLogger_Info(t *testing.T) {
8889 mockLogger := NewMockLogger ()
8990 dLogger := & defaultLogger {logger : mockLogger .Logger , logLevel : LogLevelInfo }
9091
91- mockLogger .On ("Info" , "info message" , mock .Anything ).Once ()
92+ expectedMessage := "info message"
93+ mockLogger .On ("Info" , expectedMessage , mock .Anything ).Once ()
9294
93- dLogger .Info ("info message" )
95+ dLogger .Info (expectedMessage )
9496
9597 mockLogger .AssertExpectations (t )
9698}
@@ -102,9 +104,10 @@ func TestDefaultLogger_Warn(t *testing.T) {
102104 mockLogger := NewMockLogger ()
103105 dLogger := & defaultLogger {logger : mockLogger .Logger , logLevel : LogLevelWarn }
104106
105- mockLogger .On ("Warn" , "warn message" , mock .Anything ).Once ()
107+ expectedMessage := "warn message"
108+ mockLogger .On ("Warn" , expectedMessage , mock .Anything ).Once ()
106109
107- dLogger .Warn ("warn message" )
110+ dLogger .Warn (expectedMessage )
108111
109112 mockLogger .AssertExpectations (t )
110113}
@@ -117,12 +120,12 @@ func TestDefaultLogger_Error(t *testing.T) {
117120 dLogger := & defaultLogger {logger : mockLogger .Logger , logLevel : LogLevelError }
118121
119122 expectedErrorMsg := "error message"
120- mockLogger .On ("Error" , expectedErrorMsg , mock .Anything ).Once ().Return (nil ) // Ensure Error method is called exactly once and returns nil
123+ // Ensure Error method is called exactly once and returns nil
124+ mockLogger .On ("Error" , expectedErrorMsg , mock .Anything ).Once ().Return (nil )
121125
122126 err := dLogger .Error (expectedErrorMsg )
123127
124- assert .Error (t , err )
125- assert .Contains (t , err .Error (), expectedErrorMsg )
128+ assert .NoError (t , err ) // Check that the error returned is nil
126129 mockLogger .AssertExpectations (t )
127130}
128131
@@ -216,14 +219,8 @@ func TestGetLoggerBasedOnEnv(t *testing.T) {
216219
217220 logger := GetLoggerBasedOnEnv ()
218221
219- // Since we cannot directly access the logger's level, we check the logger's development/production status
220- // which indirectly tells us about the log level configuration
221- cfg := zap .NewProductionConfig ()
222- if tt .envValue == "development" {
223- cfg = zap .NewDevelopmentConfig ()
224- }
225-
226- assert .Equal (t , cfg .Level .Level (), logger .Core ().Enabled (zapcore .Level (tt .expectedLevel .Level ())), "Logger level should match expected" )
222+ // Check if the logger's log level matches the expected log level
223+ assert .Equal (t , logger .Core ().Enabled (zapcore .Level (tt .expectedLevel .Level ())), true , "Logger level should match expected" )
227224 })
228225 }
229226}
0 commit comments