Possible fix to unit test failure

coverlet-coverage · Aug 19, 2024 · 4dacccc · 4dacccc
1 parent 6d32ae5
commit 4dacccc
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/test/coverlet.core.tests/Coverage/CoverageTests.cs b/test/coverlet.core.tests/Coverage/CoverageTests.cs
@@ -190,6 +190,7 @@ public void TestCoverageUnloadWithParameters()
       File.Copy(pdb, Path.Combine(directory.FullName, Path.GetFileName(pdb)), true);
 
       var mockInstrumentationHelper = new Mock<IInstrumentationHelper>();
+      mockInstrumentationHelper.Setup(x => x.RestoreOriginalModule(It.IsAny<string>(), It.IsAny<string>()))
 
       var parameters = new CoverageParameters
       {
@@ -224,9 +225,10 @@ public void TestCoverageUnloadWithNoParameters()
       File.Copy(pdb, Path.Combine(directory.FullName, Path.GetFileName(pdb)), true);
 
       var mockInstrumentationHelper = new Mock<IInstrumentationHelper>();
-        mockInstrumentationHelper
-          .Setup(x => x.SelectModules(It.IsAny<IEnumerable<string>>(), It.IsAny<string[]>(), It.IsAny<string[]>()))
-          .Returns(new List<string>(){"ModuleX"});
+      mockInstrumentationHelper
+        .Setup(x => x.SelectModules(It.IsAny<IEnumerable<string>>(), It.IsAny<string[]>(), It.IsAny<string[]>()))
+        .Returns(new List<string>(){"ModuleX"});
+      mockInstrumentationHelper.Setup(x => x.RestoreOriginalModule(It.IsAny<string>(), It.IsAny<string>()))
 
       var parameters = new CoverageParameters
       {