Minor unit test tweak

newrelic · Jun 22, 2023 · dfec477 · dfec477
1 parent 36ffe22
commit dfec477
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/Agent/UnitTests/Core.UnitTest/DataTransport/HttpCollectorWireTests.cs b/tests/Agent/UnitTests/Core.UnitTest/DataTransport/HttpCollectorWireTests.cs
@@ -334,17 +334,17 @@ public void SendData_ShouldNotCallAuditLog_UnlessAuditLogIsEnabled(bool isEnable
 
             var collectorWire = CreateHttpCollectorWire();
 
-            // ensure that .ForContext() just returns the mock logger instance
+            var mockForContextLogger = Mock.Create<ILogger>();
             Mock.Arrange(() => _mockILogger.ForContext(Arg.AnyString, Arg.AnyObject, false))
-                .Returns(() => _mockILogger);
+                .Returns(() => mockForContextLogger);
 
             AuditLog.IsAuditLogEnabled = isEnabled;
 
             // Act
             var response = collectorWire.SendData("test_method", connectionInfo, serializedData, Guid.NewGuid());
 
             // Assert
-            Mock.Assert(() => _mockILogger.Fatal(Arg.AnyString), isEnabled ? Occurs.Exactly(3) : Occurs.Never());
+            Mock.Assert(() => mockForContextLogger.Fatal(Arg.AnyString), isEnabled ? Occurs.Exactly(3) : Occurs.Never());
         }
     }
 

diff --git a/tests/Agent/UnitTests/Core.UnitTest/Logging/AuditLogTests.cs b/tests/Agent/UnitTests/Core.UnitTest/Logging/AuditLogTests.cs
@@ -48,17 +48,17 @@ public void ExcludeAuditLog_DisablesAuditLog()
         [TestCase(true)]
         public void Log_OnlyLogsWhenAuditLogEnabled(bool logEnabled)
         {
-            // ensure that .ForContext() just returns the mock logger instance
+            var mockForContextLogger = Mock.Create<ILogger>();
             Mock.Arrange(() => _mockILogger.ForContext(Arg.AnyString, Arg.AnyObject, false))
-                .Returns(() => _mockILogger);
+                .Returns(() => mockForContextLogger);
 
             AuditLog.IsAuditLogEnabled = logEnabled;
 
             var message = "This is an audit message";
 
             AuditLog.Log(message);
 
-            Mock.Assert(() => _mockILogger.Fatal(message), logEnabled ? Occurs.Once() : Occurs.Never());
+            Mock.Assert(() => mockForContextLogger.Fatal(message), logEnabled ? Occurs.Once() : Occurs.Never());
         }
     }
 }