Test ResetHits()

Author: Mohammad Dehghan

SG.CodeCoverage.Tests.NetFx/InstrumenterTester.cs

@@ -126,17 +126,28 @@ public void RunIsPrimeInApp(int number)
             WaitForAppIdle();
         }
 
-        private void CheckAppStarted()
+        public CoverageResult GetCoverageResult()
         {
-            if (_process == null)
-                throw new InvalidOperationException("Sample application is not started.");
+            CheckInstrumented();
+            return RecordingController.CollectResultAndReset();
         }
 
-        public CoverageResult GetCoverageResult()
+        public void ResetHits()
+        {
+            CheckInstrumented();
+            RecordingController.ResetHits();
+        }
+
+        private void CheckInstrumented()
         {
             if (InstrumentedAssemblyPath == null || RecordingController == null)
                 throw new InvalidOperationException("Sample assembly is not instrumented.");
-            return RecordingController.CollectResultAndReset();
+        }
+
+        private void CheckAppStarted()
+        {
+            if (_process == null)
+                throw new InvalidOperationException("Sample application is not started.");
         }
     }
 }

SG.CodeCoverage.Tests.NetFx/TestInstrumentation.cs

@@ -68,6 +68,37 @@ public void Coverage_IsPrime7()
                     .Concat(Methods.SampleStruct));
         }
 
+        [TestMethod]
+        public void Coverage_ResetHits_IsPrime2()
+        {
+            _tester.RunApp();
+            _tester.ResetHits();
+            _tester.RunIsPrimeInApp(2);
+            AssertVisitedFilesAndMethods(
+                new string[] { Files.App }
+                    .Append(Files.PrimeCalculator),
+                Methods.RunCommand
+                    .Concat(Methods.PrimeCalculator.IsPrimeAndIsLessThan2)
+                    .Append(Methods.get_Commands));
+        }
+
+        [TestMethod]
+        public void Coverage_ResetHits_IsPrime7()
+        {
+            _tester.RunApp();
+            _tester.ResetHits();
+            _tester.RunIsPrimeInApp(7);
+            AssertVisitedFilesAndMethods(
+                new string[] { Files.App }
+                    .Append(Files.PrimeCalculator)
+                    .Append(Files.SampleStruct),
+                Methods.RunCommand
+                    .Concat(Methods.PrimeCalculator.IsPrimeAndIsLessThan2)
+                    .Concat(Methods.PrimeCalculator.GetUpperBound)
+                    .Concat(Methods.SampleStruct)
+                    .Append(Methods.get_Commands));
+        }
+
 
         [TestCleanup]
         public void ExitApp()
@@ -90,12 +121,12 @@ private void AssertVisitedFilesAndMethods(
         private static void ShouldVisit(IReadOnlyList<string> expectedNames, IReadOnlyList<string> actualNames, string what)
         {
             Assert.AreEqual(expectedNames.Count, actualNames.Count, $"visited {what}s");
-            foreach(var expected in expectedNames)
+            foreach (var expected in expectedNames)
             {
                 bool found = false;
-                foreach(var actual in actualNames)
+                foreach (var actual in actualNames)
                 {
-                    if(actual.EndsWith(expected))
+                    if (actual.EndsWith(expected))
                     {
                         found = true;
                         break;
@@ -123,16 +154,19 @@ private static class Files
 
         private static class Methods
         {
+            public static string get_Commands = "App::get_Commands()";
+
             public static readonly IEnumerable<string> Startup = new string[]
             {
                 "Program::Main(System.String[])",
                 "App::.ctor()",
-                "App::get_Commands()",
+                get_Commands,
                 "App::Run(System.String[])",
                 "Command::.ctor(System.String,System.String,System.Action`1<System.String>)",
                 "Command::get_Help()",
             };
 
+
             public static readonly IEnumerable<string> RunCommand = new string[]
             {
                 "App::RunCommand(System.String[])",