Add more tests where PartitionKey.None maps to PartitionKey.Undefined

Author: neildsh

Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/Query/BypassQueryParsingTests.cs

@@ -43,9 +43,10 @@ async Task ImplementationAsync(Container container, IReadOnlyList<CosmosObject>
 
             await this.CreateIngestQueryDeleteAsync(
                 ConnectionModes.Direct | ConnectionModes.Gateway,
-                CollectionTypes.NonPartitioned,
+                CollectionTypes.NonPartitioned | CollectionTypes.SinglePartition | CollectionTypes.MultiPartition,
                 documents,
-                ImplementationAsync);
+                ImplementationAsync,
+                "/undefinedPartitionKey");
         }
 
         private static IReadOnlyList<string> CreateDocuments(int documentCount)

Microsoft.Azure.Cosmos/tests/Microsoft.Azure.Cosmos.EmulatorTests/Query/OptimisticDirectExecutionQueryTests.cs

@@ -44,13 +44,6 @@ public async Task TestPassingOptimisticDirectExecutionQueries()
                     enableOptimisticDirectExecution: true,
                     pageSizeOptions: PageSizeOptions.NonGroupByPageSizeOptions,
                     expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
-                CreateInput(
-                    query: $"SELECT TOP 5 VALUE r.numberField FROM r ORDER BY r.{PartitionKeyField}",
-                    expectedResult: empty,
-                    partitionKey: PartitionKey.None,
-                    enableOptimisticDirectExecution: true,
-                    pageSizeOptions: PageSizeOptions.NonGroupByPageSizeOptions,
-                    expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
                 CreateInput(
                     query: $"SELECT TOP 5 VALUE r.numberField FROM r ORDER BY r.{PartitionKeyField}",
                     expectedResult: first5Integers,
@@ -67,13 +60,6 @@ public async Task TestPassingOptimisticDirectExecutionQueries()
                     enableOptimisticDirectExecution: true,
                     pageSizeOptions: PageSizeOptions.NonGroupByPageSizeOptions,
                     expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
-                CreateInput(
-                    query: $"SELECT VALUE r.numberField FROM r",
-                    expectedResult: empty,
-                    partitionKey: PartitionKey.None,
-                    enableOptimisticDirectExecution: true,
-                    pageSizeOptions: PageSizeOptions.NonGroupByPageSizeOptions,
-                    expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
                 CreateInput(
                     query: $"SELECT VALUE r.numberField FROM r",
                     expectedResult: first7Integers,
@@ -90,13 +76,6 @@ public async Task TestPassingOptimisticDirectExecutionQueries()
                     enableOptimisticDirectExecution: true,
                     pageSizeOptions: PageSizeOptions.NonGroupByPageSizeOptions,
                     expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
-                CreateInput(
-                    query: $"SELECT DISTINCT VALUE r.{NumberField} FROM r ORDER BY r.{NumberField} DESC",
-                    expectedResult: empty,
-                    partitionKey: PartitionKey.None,
-                    enableOptimisticDirectExecution: true,
-                    pageSizeOptions: PageSizeOptions.NonGroupByPageSizeOptions,
-                    expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
                 CreateInput(
                     query: $"SELECT DISTINCT VALUE r.{NumberField} FROM r ORDER BY r.{NumberField} DESC",
                     expectedResult: first7IntegersReversed,
@@ -113,13 +92,6 @@ public async Task TestPassingOptimisticDirectExecutionQueries()
                     enableOptimisticDirectExecution: true,
                     pageSizeOptions: PageSizeOptions.GroupByPageSizeOptions,
                     expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
-                CreateInput(
-                    query: $"SELECT TOP 5 VALUE r.{NumberField} FROM r GROUP BY r.{NumberField}",
-                    expectedResult: empty,
-                    partitionKey: PartitionKey.None,
-                    enableOptimisticDirectExecution: true,
-                    pageSizeOptions: PageSizeOptions.GroupByPageSizeOptions,
-                    expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
                 CreateInput(
                     query: $"SELECT TOP 5 VALUE r.{NumberField} FROM r GROUP BY r.{NumberField}",
                     expectedResult: first5Integers,
@@ -136,13 +108,6 @@ public async Task TestPassingOptimisticDirectExecutionQueries()
                     pageSizeOptions: PageSizeOptions.NonGroupByPageSizeOptions,
                     enableOptimisticDirectExecution: true,
                     expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
-                CreateInput(
-                    query: $"SELECT VALUE r.numberField FROM r WHERE r.{NumberField} BETWEEN 0 AND {NumberOfDocuments} OFFSET 1 LIMIT 1",
-                    expectedResult: empty,
-                    partitionKey: PartitionKey.None,
-                    pageSizeOptions: PageSizeOptions.NonGroupByPageSizeOptions,
-                    enableOptimisticDirectExecution: true,
-                    expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
                 CreateInput(
                     query: $"SELECT VALUE r.numberField FROM r WHERE r.{NumberField} BETWEEN 0 AND {NumberOfDocuments} OFFSET 1 LIMIT 1",
                     expectedResult: new List<int> { 1 },
@@ -221,7 +186,7 @@ await this.CreateIngestQueryDeleteAsync(
         }
 
         [TestMethod]
-        public async Task TestNonPartitionedQueriesWithPartitionKeyNone()
+        public async Task TestQueriesWithPartitionKeyNone()
         {
             int documentCount = 400;
             IReadOnlyList<int> first400Integers = Enumerable.Range(0, documentCount).ToList();
@@ -287,13 +252,19 @@ public async Task TestNonPartitionedQueriesWithPartitionKeyNone()
                     expectedPipelineType: TestInjections.PipelineType.OptimisticDirectExecution),
             };
 
-            IReadOnlyList<string> documents = CreateDocuments(documentCount, PartitionKeyField, NumberField, NullField);
+            List<string> documents = new List<string>(documentCount);
+            for (int i = 0; i < documentCount; ++i)
+            {
+                string document = $@"{{ {NumberField}: {i}, {NullField}: null }}";
+                documents.Add(document);
+            }
 
             await this.CreateIngestQueryDeleteAsync(
                 ConnectionModes.Direct | ConnectionModes.Gateway,
-                CollectionTypes.NonPartitioned,
+                CollectionTypes.NonPartitioned | CollectionTypes.SinglePartition | CollectionTypes.MultiPartition,
                 documents,
-                (container, documents) => RunTests(testCases, container));
+                (container, documents) => RunTests(testCases, container),
+                "/undefinedPartitionKey");
         }
 
         [TestMethod]