added unit test

Azure · Sep 18, 2024 · e0ad523 · e0ad523
1 parent 8781499
commit e0ad523
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/Utils/QueryUtils.test.ts b/src/Utils/QueryUtils.test.ts
@@ -189,5 +189,18 @@ describe("Query Utils", () => {
       );
       expect(partitionKeyValues.length).toBe(0);
     });
+
+    it("should extract all partition key values for hierarchical and nested partition keys", () => {
+      const mixedPartitionKeyDefinition: PartitionKeyDefinition = {
+        kind: PartitionKeyKind.MultiHash,
+        paths: ["/Country", "/Location/type"],
+      };
+      const partitionKeyValues: PartitionKey[] = extractPartitionKeyValues(
+        documentContent,
+        mixedPartitionKeyDefinition,
+      );
+      expect(partitionKeyValues.length).toBe(2);
+      expect(partitionKeyValues).toEqual(["United States", "Point"]);
+    });
   });
 });