Initial Move from Azure DevOps to GitHub

src/Explorer/Tree/Collection.test.ts

@@ -0,0 +1,114 @@
+import * as DataModels from "../../Contracts/DataModels";
+import * as ko from "knockout";
+import * as ViewModels from "../../Contracts/ViewModels";
+import Collection from "./Collection";
+jest.mock("monaco-editor");
+
+describe("Collection", () => {
+  function generateCollection(
+    container: ViewModels.Explorer,
+    databaseId: string,
+    data: DataModels.Collection,
+    quotaInfo: DataModels.CollectionQuotaInfo,
+    offer: DataModels.Offer
+  ): Collection {
+    return new Collection(container, databaseId, data, quotaInfo, offer);
+  }
+
+  function generateMockCollectionsDataModelWithPartitionKey(
+    partitionKey: DataModels.PartitionKey
+  ): DataModels.Collection {
+    return {
+      defaultTtl: 1,
+      indexingPolicy: {} as DataModels.IndexingPolicy,
+      partitionKey,
+      _rid: "",
+      _self: "",
+      _etag: "",
+      _ts: 1,
+      id: ""
+    };
+  }
+
+  function generateMockCollectionWithDataModel(data: DataModels.Collection): Collection {
+    const mockContainer = {} as ViewModels.Explorer;
+    mockContainer.isPreferredApiMongoDB = ko.computed(() => {
+      return false;
+    });
+    mockContainer.isPreferredApiCassandra = ko.computed(() => {
+      return false;
+    });
+    mockContainer.isDatabaseNodeOrNoneSelected = () => {
+      return false;
+    };
+    mockContainer.isPreferredApiDocumentDB = ko.computed(() => {
+      return true;
+    });
+    mockContainer.isPreferredApiGraph = ko.computed(() => {
+      return false;
+    });
+    mockContainer.deleteCollectionText = ko.computed(() => {
+      return "delete collection";
+    });
+
+    return generateCollection(mockContainer, "abc", data, {} as DataModels.CollectionQuotaInfo, {} as DataModels.Offer);
+  }
+
+  describe("Partition key path parsing", () => {
+    let collection: Collection;
+
+    it("should strip out multiple forward slashes from partition key paths", () => {
+      const collectionsDataModel = generateMockCollectionsDataModelWithPartitionKey({
+        paths: ["/somePartitionKey/anotherPartitionKey"],
+        kind: "Hash",
+        version: 2
+      });
+      collection = generateMockCollectionWithDataModel(collectionsDataModel);
+      expect(collection.partitionKeyProperty).toBe("somePartitionKey.anotherPartitionKey");
+    });
+
+    it("should strip out forward slashes from single partition key paths", () => {
+      const collectionsDataModel = generateMockCollectionsDataModelWithPartitionKey({
+        paths: ["/somePartitionKey"],
+        kind: "Hash",
+        version: 2
+      });
+      collection = generateMockCollectionWithDataModel(collectionsDataModel);
+      expect(collection.partitionKeyProperty).toBe("somePartitionKey");
+    });
+  });
+
+  describe("Partition key path header", () => {
+    let collection: Collection;
+
+    it("should preserve forward slashes on partition keys", () => {
+      const collectionsDataModel = generateMockCollectionsDataModelWithPartitionKey({
+        paths: ["/somePartitionKey/anotherPartitionKey"],
+        kind: "Hash",
+        version: 2
+      });
+      collection = generateMockCollectionWithDataModel(collectionsDataModel);
+      expect(collection.partitionKeyPropertyHeader).toBe("/somePartitionKey/anotherPartitionKey");
+    });
+
+    it("should preserve forward slash on a single partition key", () => {
+      const collectionsDataModel = generateMockCollectionsDataModelWithPartitionKey({
+        paths: ["/somePartitionKey"],
+        kind: "Hash",
+        version: 2
+      });
+      collection = generateMockCollectionWithDataModel(collectionsDataModel);
+      expect(collection.partitionKeyPropertyHeader).toBe("/somePartitionKey");
+    });
+
+    it("should be null if there is no partition key", () => {
+      const collectionsDataModel = generateMockCollectionsDataModelWithPartitionKey({
+        version: 2,
+        paths: [],
+        kind: "Hash"
+      });
+      collection = generateMockCollectionWithDataModel(collectionsDataModel);
+      expect(collection.partitionKeyPropertyHeader).toBeNull;
+    });
+  });
+});