| 250 | } |
| 251 | |
| 252 | func TestLogger_LogLevels(t *testing.T) { |
| 253 | var buf bytes.Buffer |
| 254 | config := &Config{ |
| 255 | Level: LevelInfo, // Set to Info level |
| 256 | Output: &buf, |
| 257 | } |
| 258 | |
| 259 | logger, err := NewLogger(config) |
| 260 | require.NoError(t, err) |
| 261 | defer logger.Close() |
| 262 | |
| 263 | // Test that debug messages are filtered out |
| 264 | logger.Debug("debug message") |
| 265 | assert.Empty(t, buf.String()) |
| 266 | |
| 267 | // Test that info messages are logged |
| 268 | logger.Info("info message") |
| 269 | |
| 270 | output := buf.String() |
| 271 | assert.Contains(t, output, "[INFO]") |
| 272 | assert.Contains(t, output, "info message") |
| 273 | |
| 274 | // Reset buffer |
| 275 | buf.Reset() |
| 276 | |
| 277 | // Test that error messages are logged |
| 278 | logger.Error("error message") |
| 279 | |
| 280 | output = buf.String() |
| 281 | assert.Contains(t, output, "[ERROR]") |
| 282 | assert.Contains(t, output, "error message") |
| 283 | } |
| 284 | |
| 285 | func TestLogger_DebugLevel(t *testing.T) { |
| 286 | var buf bytes.Buffer |