fix: add proper checks for Hierarchical PartionKeys

Author: manekinekko

lib/cosmos-db/cosmos-db-core.module.ts

@@ -47,7 +47,7 @@ export class AzureCosmosDbCoreModule {
     const connectionProvider = {
       provide: cosmosConnectionName,
       useFactory: async (cosmosModuleOptions: AzureCosmosDbOptions): Promise<any> => {
-        const { dbName, retryAttempts, retryDelay, connectionName, ...cosmosOptions } = cosmosModuleOptions;
+        const { dbName, retryAttempts, retryDelay, ...cosmosOptions } = cosmosModuleOptions;
 
         return await defer(async () => {
           cosmosOptions.userAgentSuffix = await getuserAgentSuffix();

lib/cosmos-db/cosmos-db.decorators.spec.ts

@@ -2,10 +2,7 @@ import { AZURE_COSMOS_DB_ENTITY, CosmosPartitionKey } from './cosmos-db.decorato
 import { PartitionKeyDefinitionVersion, PartitionKeyKind } from '@azure/cosmos';
 
 describe('Azure CosmosDB Decorators', () => {
-  beforeEach(() => {
-    // tslint:disable-next-line: no-empty
-    function MockEntity() {}
-  });
+  beforeEach(() => {});
 
   describe('@CosmosPartitionKey()', () => {
     it('should add a PartitionKey ', () => {

lib/cosmos-db/cosmos-db.providers.ts

@@ -52,9 +52,17 @@ export function createAzureCosmosDbProviders(
       }
 
       if (partitionKey != null) {
-        containerOptions.partitionKey = {
-          paths: [`/${partitionKey}`],
-        };
+        if (typeof partitionKey === 'string') {
+          containerOptions.partitionKey = {
+            paths: [`/${partitionKey}`],
+          };
+        }
+        else if (typeof partitionKey === 'object') {
+          containerOptions.partitionKey = partitionKey;
+        } else {
+          throw new Error(`Invalid PartitionKey definition for model ${model.dto.name}`);
+        }
+
       }
       const coResponse = await database.containers.createIfNotExists(containerOptions);
 

package-lock.json

@@ -34,8 +34,8 @@
   "dependencies": {
     "@azure/cosmos": "^4.0.0",
     "@azure/data-tables": "^13.2.2",
-    "@nestjs/common": "^11.0.0",
-    "@nestjs/core": "^11.0.0"
+    "@nestjs/common": "^10.0.0 || ^11.0.0",
+    "@nestjs/core": "^10.0.0 || ^11.0.0"
   },
   "devDependencies": {
     "@commitlint/cli": "19.8.1",
@@ -45,7 +45,7 @@
     "@nestjs/testing": "11.1.3",
     "@types/jest": "29.5.14",
     "@types/node": "22.13.8",
-    "azurite": "3.34.0",
+    "azurite": "^3.34.0",
     "dotenv": "16.5.0",
     "eslint": "9.28.0",
     "eslint-config-prettier": "10.1.5",