random-mirrors/cosmos-explorer
1
0
Fork 0
mirror of https://github.com/Azure/cosmos-explorer.git synced 2026-01-28 06:04:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

dependabot round 1

Browse Source
This commit is contained in:
Asier Isayas
2026-01-26 07:22:58 -08:00
parent 2ff01c6379
commit 237e45f1b3
7 changed files with 964 additions and 4018 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
test/testData.ts
View File

@@ -2,7 +2,6 @@ import crypto from "crypto";
import { CosmosDBManagementClient } from "@azure/arm-cosmosdb";
import { BulkOperationType, Container, CosmosClient, CosmosClientOptions, Database, JSONObject } from "@azure/cosmos";
import { AzureIdentityCredentialAdapter } from "@azure/ms-rest-js";
import {
generateUniqueName,
@@ -37,6 +36,10 @@ export interface PartitionKey {
value: string | null;
}
export interface PartitionKeyResult {
[key: string]: unknown;
}
export const partitionCount = 4;
// If we increase this number, we need to split bulk creates into multiple batches.
@@ -105,8 +108,7 @@ export async function createMultipleTestContainers({
const databaseId = databaseName ? databaseName : generateUniqueName("db");
const credentials = getAzureCLICredentials();
const adaptedCredentials = new AzureIdentityCredentialAdapter(credentials);
const armClient = new CosmosDBManagementClient(adaptedCredentials, subscriptionId);
const armClient = new CosmosDBManagementClient(credentials, subscriptionId);
const accountName = getAccountName(accountType);
const account = await armClient.databaseAccounts.get(resourceGroupName, accountName);
@@ -159,8 +161,7 @@ export async function createTestSQLContainer({
const databaseId = databaseName ? databaseName : generateUniqueName("db");
const containerId = "testcontainer"; // A unique container name isn't needed because the database is unique
const credentials = getAzureCLICredentials();
const adaptedCredentials = new AzureIdentityCredentialAdapter(credentials);
const armClient = new CosmosDBManagementClient(adaptedCredentials, subscriptionId);
const armClient = new CosmosDBManagementClient(credentials, subscriptionId);
const accountName = getAccountName(TestAccount.SQL);
const account = await armClient.databaseAccounts.get(resourceGroupName, accountName);
@@ -211,20 +212,20 @@ export async function createTestSQLContainer({
}
export const setPartitionKeys = (partitionKeys: PartitionKey[]) => {
const result = {};
const result: PartitionKeyResult = {};
partitionKeys.forEach((partitionKey) => {
const { key: keyPath, value: keyValue } = partitionKey;
const cleanPath = keyPath.startsWith("/") ? keyPath.slice(1) : keyPath;
const keys = cleanPath.split("/");
let current = result;
let current: PartitionKeyResult = result;
keys.forEach((key, index) => {
if (index === keys.length - 1) {
current[key] = keyValue;
} else {
current[key] = current[key] || {};
current = current[key];
current = current[key] as PartitionKeyResult;
}
});
});