more coverage

Author: hjgraca

libraries/tests/AWS.Lambda.Powertools.Common.Tests/ConsoleWrapperTests.cs

@@ -109,7 +109,8 @@ public void StaticWriteLine_FormatsLogMessageCorrectly()
 
         // Act - Using reflection to call internal static method
         typeof(ConsoleWrapper)
-            .GetMethod("WriteLine", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Static, null, new[] { typeof(string), typeof(string) }, null)
+            .GetMethod("WriteLine", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Static,
+                null, new[] { typeof(string), typeof(string) }, null)
             ?.Invoke(null, new object[] { "INFO", "Test log message" });
 
         // Assert
@@ -123,16 +124,73 @@ public void ClearOutputResetFlag_ResetsFlag()
         // Arrange
         var consoleWrapper = new ConsoleWrapper();
         ConsoleWrapper.SetOut(_writer);
-        
+
         // Act
         consoleWrapper.WriteLine("First message"); // Should set the reset flag
         ConsoleWrapper.ClearOutputResetFlag();
         consoleWrapper.WriteLine("Second message"); // Should set it again
-        
+
         // Assert
         Assert.Equal($"First message{Environment.NewLine}Second message{Environment.NewLine}", _writer.ToString());
     }
 
+    [Fact]
+    public void Debug_InTestMode_WritesToTestOutputStream()
+    {
+        // Arrange
+        var consoleWrapper = new ConsoleWrapper();
+        ConsoleWrapper.SetOut(_writer);
+
+        // Act
+        consoleWrapper.Debug("debug message");
+
+        // Assert
+        Assert.Equal($"debug message{Environment.NewLine}", _writer.ToString());
+    }
+
+    [Fact]
+    public void Debug_NotInTestMode_WritesToDebugConsole()
+    {
+        // Since capturing Debug output is difficult in a unit test
+        // We'll use a mock or just verify the path doesn't throw
+    
+        // Arrange
+        var consoleWrapper = new ConsoleWrapper();
+        ConsoleWrapper.ResetForTest(); // Ensure we're not in test mode
+
+        // Act & Assert - Just verify it doesn't throw
+        var exception = Record.Exception(() => consoleWrapper.Debug("debug message"));
+        Assert.Null(exception);
+    }
+
+    [Fact]
+    public void Error_DoesNotThrowWhenNotOverridden()
+    {
+        // Arrange
+        var consoleWrapper = new ConsoleWrapper();
+        ConsoleWrapper.ResetForTest(); // Reset to ensure _override is false
+    
+        // Act & Assert - Just verify it doesn't throw
+        var exception = Record.Exception(() => consoleWrapper.Error("error without override"));
+        Assert.Null(exception);
+    }
+
+    [Fact]
+    public void Error_UsesTestOutputStreamWhenInTestMode()
+    {
+        // Arrange
+        var consoleWrapper = new ConsoleWrapper();
+    
+        // Set test mode
+        ConsoleWrapper.SetOut(_writer);
+    
+        // Act
+        consoleWrapper.Error("error in test mode");
+
+        // Assert
+        Assert.Contains("error in test mode", _writer.ToString());
+    }
+
     public void Dispose()
     {
         ConsoleWrapper.ResetForTest();