import * as ko from "knockout"; import * as ViewModels from "../../Contracts/ViewModels"; import * as Constants from "../../Common/Constants"; import DocumentsTab from "./DocumentsTab"; import Explorer from "../Explorer"; import DocumentClientUtilityBase from "../../Common/DocumentClientUtilityBase"; describe("Documents tab", () => { describe("buildQuery", () => { it("should generate the right select query for SQL API", () => { const documentsTab = new DocumentsTab({ partitionKey: null, documentIds: ko.observableArray(), tabKind: ViewModels.CollectionTabKind.Documents, title: "", tabPath: "", documentClientUtility: new DocumentClientUtilityBase(), selfLink: "", hashLocation: "", isActive: ko.observable(false), onUpdateTabsButtons: (buttons: ViewModels.NavbarButtonConfig[]): void => {} }); expect(documentsTab.buildQuery("")).toContain("select"); }); }); describe("showPartitionKey", () => { const explorer = new Explorer({ documentClientUtility: null, notificationsClient: null, isEmulator: false }); const mongoExplorer = new Explorer({ documentClientUtility: null, notificationsClient: null, isEmulator: false }); mongoExplorer.defaultExperience(Constants.DefaultAccountExperience.MongoDB); const collectionWithoutPartitionKey = ({ id: ko.observable("foo"), database: { id: ko.observable("foo") }, container: explorer }); const collectionWithSystemPartitionKey = ({ id: ko.observable("foo"), database: { id: ko.observable("foo") }, partitionKey: { paths: ["/foo"], kind: "Hash", version: 2, systemKey: true }, container: explorer }); const collectionWithNonSystemPartitionKey = ({ id: ko.observable("foo"), database: { id: ko.observable("foo") }, partitionKey: { paths: ["/foo"], kind: "Hash", version: 2, systemKey: false }, container: explorer }); const mongoCollectionWithSystemPartitionKey = ({ id: ko.observable("foo"), database: { id: ko.observable("foo") }, partitionKey: { paths: ["/foo"], kind: "Hash", version: 2, systemKey: true }, container: mongoExplorer }); it("should be false for null or undefined collection", () => { const documentsTab = new DocumentsTab({ partitionKey: null, documentIds: ko.observableArray(), tabKind: ViewModels.CollectionTabKind.Documents, title: "", tabPath: "", documentClientUtility: new DocumentClientUtilityBase(), selfLink: "", hashLocation: "", isActive: ko.observable(false), onUpdateTabsButtons: (buttons: ViewModels.NavbarButtonConfig[]): void => {} }); expect(documentsTab.showPartitionKey).toBe(false); }); it("should be false for null or undefined partitionKey", () => { const documentsTab = new DocumentsTab({ collection: collectionWithoutPartitionKey, partitionKey: null, documentIds: ko.observableArray(), tabKind: ViewModels.CollectionTabKind.Documents, title: "", tabPath: "", documentClientUtility: new DocumentClientUtilityBase(), selfLink: "", hashLocation: "", isActive: ko.observable(false), onUpdateTabsButtons: (buttons: ViewModels.NavbarButtonConfig[]): void => {} }); expect(documentsTab.showPartitionKey).toBe(false); }); it("should be true for non-Mongo accounts with system partitionKey", () => { const documentsTab = new DocumentsTab({ collection: collectionWithSystemPartitionKey, partitionKey: null, documentIds: ko.observableArray(), tabKind: ViewModels.CollectionTabKind.Documents, title: "", tabPath: "", documentClientUtility: new DocumentClientUtilityBase(), selfLink: "", hashLocation: "", isActive: ko.observable(false), onUpdateTabsButtons: (buttons: ViewModels.NavbarButtonConfig[]): void => {} }); expect(documentsTab.showPartitionKey).toBe(true); }); it("should be false for Mongo accounts with system partitionKey", () => { const documentsTab = new DocumentsTab({ collection: mongoCollectionWithSystemPartitionKey, partitionKey: null, documentIds: ko.observableArray(), tabKind: ViewModels.CollectionTabKind.Documents, title: "", tabPath: "", documentClientUtility: new DocumentClientUtilityBase(), selfLink: "", hashLocation: "", isActive: ko.observable(false), onUpdateTabsButtons: (buttons: ViewModels.NavbarButtonConfig[]): void => {} }); expect(documentsTab.showPartitionKey).toBe(false); }); it("should be true for non-system partitionKey", () => { const documentsTab = new DocumentsTab({ collection: collectionWithNonSystemPartitionKey, partitionKey: null, documentIds: ko.observableArray(), tabKind: ViewModels.CollectionTabKind.Documents, title: "", tabPath: "", documentClientUtility: new DocumentClientUtilityBase(), selfLink: "", hashLocation: "", isActive: ko.observable(false), onUpdateTabsButtons: (buttons: ViewModels.NavbarButtonConfig[]): void => {} }); expect(documentsTab.showPartitionKey).toBe(true); }); }); });