test: add tests for console output handling and exception scenarios in Lambda environment

Author: hjgraca

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

@@ -299,4 +299,64 @@ public void ClearOutputResetFlag_GivenMultipleCalls_WhenCalled_ThenAllowsRepeate
         // Then
         Assert.Equal($"First message{Environment.NewLine}Second message{Environment.NewLine}", _testWriter.ToString());
     }
+    
+    [Fact]
+    public void HasLambdaReInterceptedConsole_GivenConsoleOutThrowsException_WhenCalled_ThenReturnsTrue()
+    {
+        // Given
+        Environment.SetEnvironmentVariable("AWS_LAMBDA_FUNCTION_NAME", "test-function");
+    
+        // Create a mock TextWriter that throws when accessing its type
+        var mockWriter = new ThrowingTextWriter();
+        Console.SetOut(mockWriter);
+    
+        var wrapper = new ConsoleWrapper();
+    
+        // When - This will trigger HasLambdaReInterceptedConsole which should catch the exception
+        var exception = Record.Exception(() => wrapper.WriteLine("test message"));
+    
+        // Then - Should not throw, the catch block should handle it
+        Assert.Null(exception);
+    }
+
+    [Fact]
+    public void OverrideLambdaLogger_GivenConsoleOpenStandardOutputThrows_WhenCalled_ThenDoesNotThrow()
+    {
+        // Given
+        Environment.SetEnvironmentVariable("AWS_LAMBDA_FUNCTION_NAME", "test-function");
+    
+        // Create a scenario where Console.OpenStandardOutput might fail
+        // We'll simulate this by creating conditions that trigger the catch block
+        var wrapper = new ConsoleWrapper();
+    
+        // When - Multiple calls to trigger the override logic and potential failures
+        var exception = Record.Exception(() =>
+        {
+            for (int i = 0; i < 5; i++)
+            {
+                wrapper.WriteLine($"test message {i}");
+            }
+        });
+    
+        // Then - Should not throw even if StreamWriter creation fails
+        Assert.Null(exception);
+    }
+
+    // Helper class to simulate exceptions in Console.Out.GetType()
+    private class ThrowingTextWriter : TextWriter
+    {
+        public override System.Text.Encoding Encoding => System.Text.Encoding.UTF8;
+
+        public override void Write(char value) 
+        {
+            // Simulate the scenario where accessing type information fails
+            // by throwing when any operation is performed
+            throw new InvalidOperationException("Simulated exception during console operation");
+        }
+    
+        public override void WriteLine(string value) 
+        {
+            throw new InvalidOperationException("Simulated exception during console operation");
+        }
+    }
 }