refactor: standardize console output handling in test classes

Author: magic5644

CodeLineCounter.Tests/CodeAnalyzerTests.cs

@@ -6,26 +6,37 @@ namespace CodeLineCounter.Tests
 {
     public class CodeAnalyzerTests
     {
+        private readonly TextWriter _originalConsoleOut;
+
+        public CodeAnalyzerTests()
+        {
+            _originalConsoleOut = Console.Out;
+        }
+
         [Fact]
         public void TestAnalyzeSolution()
         {
-            using StringWriter consoleOutput = new();
-            Console.SetOut(consoleOutput);
-
-            string basePath = FileUtils.GetBasePath();
-            var solutionPath = Path.GetFullPath(Path.Combine(basePath, "..", "..", "..", "..", "CodeLineCounter.sln"));
-
-            // Act
-            var (metrics, projectTotals, totalLines, totalFiles, duplicationMap, dependencies) = CodeMetricsAnalyzer.AnalyzeSolution(solutionPath);
-
-            // Assert
-            Assert.NotNull(metrics);
-            Assert.NotEmpty(metrics);
-            Assert.NotEmpty(projectTotals);
-            Assert.NotEqual(0, totalLines);
-            Assert.NotEqual(0, totalFiles);
-            Assert.NotNull(duplicationMap);
-            Assert.NotNull(dependencies);
+            using (StringWriter consoleOutput = new())
+            {
+                Console.SetOut(consoleOutput);
+
+                string basePath = FileUtils.GetBasePath();
+                var solutionPath = Path.GetFullPath(Path.Combine(basePath, "..", "..", "..", "..", "CodeLineCounter.sln"));
+
+                // Act
+                var (metrics, projectTotals, totalLines, totalFiles, duplicationMap, dependencies) = CodeMetricsAnalyzer.AnalyzeSolution(solutionPath);
+
+                // Assert
+                Assert.NotNull(metrics);
+                Assert.NotEmpty(metrics);
+                Assert.NotEmpty(projectTotals);
+                Assert.NotEqual(0, totalLines);
+                Assert.NotEqual(0, totalFiles);
+                Assert.NotNull(duplicationMap);
+                Assert.NotNull(dependencies);
+            }
+            // Reset console output
+            Console.SetOut(_originalConsoleOut);
         }
 
         [Fact]
@@ -52,6 +63,8 @@ public void AnalyzeSourceCode_Should_Set_CurrentNamespace()
                 Assert.Equal("MyNamespace", currentNamespace);
 
             }
+            // Reset console output
+            Console.SetOut(_originalConsoleOut);
 
         }
 
@@ -78,6 +91,8 @@ public void AnalyzeSourceCode_Should_Set_FileLineCount()
                 // Assert - 3 lines only because comment lines are ignored
                 Assert.Equal(3, fileLineCount);
             }
+            // Reset console output
+            Console.SetOut(_originalConsoleOut);
 
         }
 
@@ -104,6 +119,8 @@ public void AnalyzeSourceCode_Should_Set_FileCyclomaticComplexity()
                 // Assert
                 Assert.Equal(1, fileCyclomaticComplexity);
             }
+            // Reset console output
+            Console.SetOut(_originalConsoleOut);
 
         }
 

CodeLineCounter.Tests/CodeDuplicationCheckerTests.cs

@@ -7,23 +7,27 @@ public class CodeDuplicationCheckerTests : IDisposable
         private readonly string _testDirectory;
         private bool _disposed;
 
+        private readonly TextWriter _originalConsoleOut;
+
         public CodeDuplicationCheckerTests()
         {
             _testDirectory = Path.Combine(Path.GetTempPath(), "CodeDuplicationCheckerTests");
             Directory.CreateDirectory(_testDirectory);
+            _originalConsoleOut = Console.Out;
         }
 
         [Fact]
         public void DetectCodeDuplicationInFiles_ShouldDetectDuplicates()
         {
-            using StringWriter consoleOutput = new();
-            Console.SetOut(consoleOutput);
+            using (StringWriter consoleOutput = new())
+            {
+                Console.SetOut(consoleOutput);
 
-            // Arrange
-            var file1 = Path.Combine(_testDirectory, "TestFile1.cs");
-            var file2 = Path.Combine(_testDirectory, "TestFile2.cs");
+                // Arrange
+                var file1 = Path.Combine(_testDirectory, "TestFile1.cs");
+                var file2 = Path.Combine(_testDirectory, "TestFile2.cs");
 
-            var code1 = @"
+                var code1 = @"
                 public class TestClass
                 {
                     public void TestMethod()
@@ -35,7 +39,7 @@ public void TestMethod()
                     }
                 }";
 
-            var code2 = @"
+                var code2 = @"
                 public class AnotherTestClass
                 {
                     public void AnotherTestMethod()
@@ -47,24 +51,27 @@ public void AnotherTestMethod()
                     }
                 }";
 
-            File.WriteAllText(file1, code1);
-            File.WriteAllText(file2, code2);
+                File.WriteAllText(file1, code1);
+                File.WriteAllText(file2, code2);
 
-            var files = new List<string> { file1, file2 };
-            var checker = new CodeDuplicationChecker();
+                var files = new List<string> { file1, file2 };
+                var checker = new CodeDuplicationChecker();
 
-            // Act
-            checker.DetectCodeDuplicationInFiles(files);
-            var result = checker.GetCodeDuplicationMap();
+                // Act
+                checker.DetectCodeDuplicationInFiles(files);
+                var result = checker.GetCodeDuplicationMap();
 
-            // Assert
-            Assert.NotEmpty(result);
-            var duplicateEntry = result.First();
-            Assert.Equal(2, duplicateEntry.Value.Count); // Both files should be detected as duplicates
+                // Assert
+                Assert.NotEmpty(result);
+                var duplicateEntry = result.First();
+                Assert.Equal(2, duplicateEntry.Value.Count); // Both files should be detected as duplicates
 
-            // Clean up
-            File.Delete(file1);
-            File.Delete(file2);
+                // Clean up
+                File.Delete(file1);
+                File.Delete(file2);
+            }
+            // Reset console output
+            Console.SetOut(_originalConsoleOut);
         }
 
         [Fact]
@@ -114,6 +121,8 @@ public void AnotherTestMethod()
                 var duplicateEntry = result.First();
                 Assert.Equal(2, duplicateEntry.Value.Count); // Both methods should be detected as duplicates
             }
+            // Reset console output
+            Console.SetOut(_originalConsoleOut);
 
         }
 
@@ -159,6 +168,8 @@ public void AnotherTestMethod()
                 // Assert
                 Assert.Empty(result); // No duplicates should be detected
             }
+            // Reset console output
+            Console.SetOut(_originalConsoleOut);
 
         }