Updates to Unit test

microsoft · Aug 18, 2021 · b8048ef · b8048ef
1 parent 02b878b
commit b8048ef
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/...Microsoft.Health.Fhir.CosmosDb.UnitTests/Features/Storage/CosmosResponseProcessorTests.cs b/...Microsoft.Health.Fhir.CosmosDb.UnitTests/Features/Storage/CosmosResponseProcessorTests.cs
@@ -17,6 +17,7 @@
 using Microsoft.Health.Core.Configs;
 using Microsoft.Health.Core.Features.Context;
 using Microsoft.Health.Fhir.Core.Exceptions;
+using Microsoft.Health.Fhir.Core.Features;
 using Microsoft.Health.Fhir.Core.Features.Context;
 using Microsoft.Health.Fhir.CosmosDb.Features.Metrics;
 using Microsoft.Health.Fhir.CosmosDb.Features.Queries;
@@ -28,11 +29,14 @@ namespace Microsoft.Health.Fhir.CosmosDb.UnitTests.Features.Storage
 {
     public class CosmosResponseProcessorTests
     {
+        private static readonly string _customAuditHeaderPrefix = KnownHeaders.CustomAuditHeaderPrefix;
+
         private readonly Dictionary<string, StringValues> _requestHeaders = new Dictionary<string, StringValues>();
         private readonly Dictionary<string, StringValues> _responseHeaders = new Dictionary<string, StringValues>();
         private readonly CosmosResponseProcessor _cosmosResponseProcessor;
         private readonly IMediator _mediator;
         private readonly RequestContextAccessor<IFhirRequestContext> _fhirRequestContextAccessor;
+        private readonly IOptions<AuditConfiguration> _optionsAuditConfiguration;
 
         public CosmosResponseProcessorTests()
         {
@@ -44,8 +48,14 @@ public CosmosResponseProcessorTests()
 
             _mediator = Substitute.For<IMediator>();
             var nullLogger = NullLogger<CosmosResponseProcessor>.Instance;
+            var auditConfiguration = new AuditConfiguration()
+            {
+                CustomAuditHeaderPrefix = _customAuditHeaderPrefix,
+            };
 
-            _cosmosResponseProcessor = new CosmosResponseProcessor(_fhirRequestContextAccessor, _mediator, Substitute.For<ICosmosQueryLogger>(), Substitute.For<IOptions<AuditConfiguration>>(), nullLogger);
+            _optionsAuditConfiguration = Substitute.For<IOptions<AuditConfiguration>>();
+            _optionsAuditConfiguration.Value.Returns(auditConfiguration);
+            _cosmosResponseProcessor = new CosmosResponseProcessor(_fhirRequestContextAccessor, _mediator, Substitute.For<ICosmosQueryLogger>(), _optionsAuditConfiguration, nullLogger);
         }
 
         [Fact]