Made MergeTestUtil.IsFailedFallbackPipelineTest a readonly property

Azure · Jan 11, 2023 · c94db62 · c94db62
1 parent 6179112
commit c94db62
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/...s/tests/Microsoft.Azure.Cosmos.Tests/Query/OptimisticDirectExecutionQueryBaselineTests.cs b/...s/tests/Microsoft.Azure.Cosmos.Tests/Query/OptimisticDirectExecutionQueryBaselineTests.cs
@@ -322,10 +322,7 @@ private static async Task<bool> TestHandlingOfFailedFallbackPipeline(bool isMult
         private static async Task<(MergeTestUtil, IQueryPipelineStage)> CreateFallbackPipelineTestInfrastructure(int numItems, bool isFailedFallbackPipelineTest, bool isMultiPartition, QueryRequestOptions queryRequestOptions)
         {
             List<CosmosElement> documents = new List<CosmosElement>();
-            MergeTestUtil mergeTest = new MergeTestUtil
-            {
-                IsFailedFallbackPipelineTest = isFailedFallbackPipelineTest
-            };
+            MergeTestUtil mergeTest = new MergeTestUtil(isFailedFallbackPipelineTest);
 
             OptimisticDirectExecutionTestInput input = CreateInput(
                     description: @"Single Partition Key and Value Field",
@@ -565,7 +562,7 @@ public override OptimisticDirectExecutionTestOutput ExecuteTest(OptimisticDirect
                 partitionedQueryExecutionInfo: partitionedQueryExecutionInfo,
                 executionEnvironment: null,
                 returnResultsInDeterministicOrder: null,
-                forcePassthrough: true,
+                forcePassthrough: false,
                 testInjections: queryRequestOptions.TestSettings);
 
             string databaseId = "db1234";
@@ -590,11 +587,10 @@ private static QueryRequestOptions GetQueryRequestOptions(bool enableOptimisticD
             {
                 MaxConcurrency = 0,
                 MaxItemCount = 10,
-                MaxBufferedItemCount = 7000,
                 TestSettings = new TestInjections(simulate429s: true, simulateEmptyPages: false, enableOptimisticDirectExecution: enableOptimisticDirectExecution, new TestInjections.ResponseStats()),
                 Properties = new Dictionary<string, object>()
             {
-                { HttpConstants.HttpHeaders.EnumerationDirection, "test" },
+                { HttpConstants.HttpHeaders.EnumerationDirection, ""},
             }
             };
         }
@@ -607,7 +603,12 @@ private class MergeTestUtil
 
             public bool TooManyRequestsFailureCreated { get; private set; }
 
-            public bool IsFailedFallbackPipelineTest { get; set; }
+            public bool IsFailedFallbackPipelineTest { get; }
+
+            public MergeTestUtil(bool isFailedFallbackPipelineTest)
+            {
+                this.IsFailedFallbackPipelineTest = isFailedFallbackPipelineTest;
+            }
 
             public async Task<Exception> ShouldReturnFailure()
             {