Move Delete Container call to use ARM when logged in with AAD (#110)

src/Common/CosmosClient.ts

@@ -2,7 +2,7 @@ import * as Cosmos from "@azure/cosmos";
 import { RequestInfo, setAuthorizationTokenHeaderUsingMasterKey } from "@azure/cosmos";
 import { DatabaseAccount } from "../Contracts/DataModels";
 import { HttpHeaders, EmulatorMasterKey } from "./Constants";
-import { NotificationConsoleUtils } from "../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../Explorer/Menus/NotificationConsole/NotificationConsoleComponent";
 import { config, Platform } from "../Config";
 

src/Common/DataAccessUtilityBase.ts

@@ -377,16 +377,6 @@ export function deleteConflict(
   );
 }
 
-export function deleteCollection(collection: ViewModels.Collection, options: any): Q.Promise<any> {
-  return Q(
-    CosmosClient.client()
-      .database(collection.databaseId)
-      .container(collection.id())
-      .delete()
-      .then(() => refreshCachedResources())
-  );
-}
-
 export function deleteDatabase(database: ViewModels.Database, options: any): Q.Promise<any> {
   return Q(
     CosmosClient.client()

src/Common/DocumentClientUtilityBase.ts

@@ -10,7 +10,7 @@ import * as Logger from "./Logger";
 import { sendMessage } from "./MessageHandler";
 import { MessageTypes } from "../Contracts/ExplorerContracts";
 import { MinimalQueryIterator, nextPage } from "./IteratorUtilities";
-import { NotificationConsoleUtils } from "../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../Utils/NotificationConsoleUtils";
 import { RequestOptions } from "@azure/cosmos/dist-esm";
 import StoredProcedure from "../Explorer/Tree/StoredProcedure";
 import ConflictId from "../Explorer/Tree/ConflictId";
@@ -730,39 +730,6 @@ export function deleteConflict(
   return deferred.promise;
 }
 
-export function deleteCollection(collection: ViewModels.Collection, options: any = {}): Q.Promise<any> {
-  var deferred = Q.defer<any>();
-
-  const id = NotificationConsoleUtils.logConsoleMessage(
-    ConsoleDataType.InProgress,
-    `Deleting container ${collection.id()}`
-  );
-  DataAccessUtilityBase.deleteCollection(collection, options)
-    .then(
-      (response: any) => {
-        NotificationConsoleUtils.logConsoleMessage(
-          ConsoleDataType.Info,
-          `Successfully deleted container ${collection.id()}`
-        );
-        deferred.resolve(response);
-      },
-      (error: any) => {
-        NotificationConsoleUtils.logConsoleMessage(
-          ConsoleDataType.Error,
-          `Error while deleting container ${collection.id()}:\n ${JSON.stringify(error)}`
-        );
-        Logger.logError(JSON.stringify(error), "DeleteCollection", error.code);
-        sendNotificationForError(error);
-        deferred.reject(error);
-      }
-    )
-    .finally(() => {
-      NotificationConsoleUtils.clearInProgressMessageWithId(id);
-    });
-
-  return deferred.promise;
-}
-
 export function deleteDatabase(database: ViewModels.Database, options: any = {}): Q.Promise<any> {
   var deferred = Q.defer<any>();
 

src/Common/MongoProxyClient.ts

@@ -13,7 +13,7 @@ import { Constants as CosmosSDKConstants } from "@azure/cosmos";
 import { CosmosClient } from "./CosmosClient";
 import { sendMessage } from "./MessageHandler";
 import { MessageTypes } from "../Contracts/ExplorerContracts";
-import { NotificationConsoleUtils } from "../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../Utils/NotificationConsoleUtils";
 import { ResourceProviderClient } from "../ResourceProvider/ResourceProviderClient";
 import { MinimalQueryIterator } from "./IteratorUtilities";
 import DocumentId from "../Explorer/Tree/DocumentId";

src/Common/QueriesClient.ts

@@ -8,7 +8,7 @@ import { ConsoleDataType } from "../Explorer/Menus/NotificationConsole/Notificat
 import { CosmosClient } from "./CosmosClient";
 import { ItemDefinition, QueryIterator, Resource } from "@azure/cosmos";
 import * as Logger from "./Logger";
-import { NotificationConsoleUtils } from "../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../Utils/NotificationConsoleUtils";
 import { QueryUtils } from "../Utils/QueryUtils";
 import Explorer from "../Explorer/Explorer";
 import {

src/Common/dataAccess/deleteCollection.test.ts

@@ -0,0 +1,13 @@
+jest.mock("../../Utils/arm/request");
+import { deleteCollection } from "./deleteCollection";
+import { armRequest } from "../../Utils/arm/request";
+import { AuthType } from "../../AuthType";
+
+describe("deleteCollection", () => {
+  it("should call ARM if logged in with AAD", async () => {
+    window.authType = AuthType.AAD;
+    await deleteCollection("database", "collection");
+    expect(armRequest).toHaveBeenCalled();
+  });
+  // TODO: Test non-AAD case
+});

src/Common/dataAccess/deleteCollection.ts

@@ -0,0 +1,31 @@
+import { CosmosClient } from "../CosmosClient";
+import { refreshCachedResources } from "../DataAccessUtilityBase";
+import { logConsoleProgress, logConsoleInfo, logConsoleError } from "../../Utils/NotificationConsoleUtils";
+import { AuthType } from "../../AuthType";
+import { deleteSqlContainer } from "../../Utils/arm/generatedClients/2020-04-01/sqlResources";
+
+export async function deleteCollection(databaseId: string, collectionId: string): Promise<void> {
+  const clearMessage = logConsoleProgress(`Deleting container ${collectionId}`);
+  try {
+    if (window.authType === AuthType.AAD) {
+      await deleteSqlContainer(
+        CosmosClient.subscriptionId(),
+        CosmosClient.resourceGroup(),
+        CosmosClient.databaseAccount().name,
+        databaseId,
+        collectionId
+      );
+    } else {
+      await CosmosClient.client()
+        .database(databaseId)
+        .container(collectionId)
+        .delete();
+    }
+  } catch (error) {
+    logConsoleError(`Error while deleting container ${collectionId}:\n ${JSON.stringify(error)}`);
+    throw error;
+  }
+  logConsoleInfo(`Successfully deleted container ${collectionId}`);
+  clearMessage();
+  await refreshCachedResources();
+}

src/Explorer/Controls/Notebook/NotebookTerminalComponent.tsx

@@ -5,7 +5,7 @@
 import * as React from "react";
 import * as DataModels from "../../../Contracts/DataModels";
 import * as Logger from "../../../Common/Logger";
-import { NotificationConsoleUtils } from "../../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../../Menus/NotificationConsole/NotificationConsoleComponent";
 import { StringUtils } from "../../../Utils/StringUtils";
 

src/Explorer/Controls/NotebookGallery/GalleryViewerComponent.tsx

@@ -17,7 +17,7 @@ import * as React from "react";
 import * as Logger from "../../../Common/Logger";
 import { IGalleryItem, JunoClient } from "../../../Juno/JunoClient";
 import * as GalleryUtils from "../../../Utils/GalleryUtils";
-import { NotificationConsoleUtils } from "../../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../../Menus/NotificationConsole/NotificationConsoleComponent";
 import { DialogComponent, DialogProps } from "../DialogReactComponent/DialogComponent";
 import { GalleryCardComponent, GalleryCardComponentProps } from "./Cards/GalleryCardComponent";

src/Explorer/Controls/NotebookViewer/NotebookViewerComponent.tsx

@@ -10,7 +10,7 @@ import * as Logger from "../../../Common/Logger";
 import * as ViewModels from "../../../Contracts/ViewModels";
 import { IGalleryItem, JunoClient } from "../../../Juno/JunoClient";
 import * as GalleryUtils from "../../../Utils/GalleryUtils";
-import { NotificationConsoleUtils } from "../../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../../Menus/NotificationConsole/NotificationConsoleComponent";
 import { NotebookClientV2 } from "../../Notebook/NotebookClientV2";
 import { NotebookComponentBootstrapper } from "../../Notebook/NotebookComponent/NotebookComponentBootstrapper";

src/Explorer/DataSamples/ContainerSampleGenerator.ts

@@ -5,7 +5,7 @@ import GraphTab from ".././Tabs/GraphTab";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import { CosmosClient } from "../../Common/CosmosClient";
 import { GremlinClient } from "../Graph/GraphExplorerComponent/GremlinClient";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import Explorer from "../Explorer";
 import { createDocument, getOrCreateDatabaseAndCollection } from "../../Common/DocumentClientUtilityBase";
 

src/Explorer/DataSamples/DataSamplesUtil.ts

@@ -1,6 +1,6 @@
 import * as ViewModels from "../../Contracts/ViewModels";
 import { ContainerSampleGenerator } from "./ContainerSampleGenerator";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import Explorer from "../Explorer";
 

src/Explorer/Explorer.ts

@@ -56,7 +56,7 @@ import { NotebookContentItem, NotebookContentItemType } from "./Notebook/Noteboo
 import { NotebookUtil } from "./Notebook/NotebookUtil";
 import { NotebookWorkspaceManager } from "../NotebookWorkspaceManager/NotebookWorkspaceManager";
 import { NotificationConsoleComponentAdapter } from "./Menus/NotificationConsole/NotificationConsoleComponentAdapter";
-import { NotificationConsoleUtils } from "../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../Utils/NotificationConsoleUtils";
 import { PlatformType } from "../PlatformType";
 import { QueriesClient } from "../Common/QueriesClient";
 import { QuerySelectPane } from "./Panes/Tables/QuerySelectPane";

src/Explorer/Graph/GraphExplorerComponent/D3ForceGraph.ts

@@ -15,7 +15,7 @@ import { GraphData, D3Node, D3Link } from "./GraphData";
 import { HashMap } from "../../../Common/HashMap";
 import { BaseType } from "d3";
 import { ConsoleDataType } from "../../Menus/NotificationConsole/NotificationConsoleComponent";
-import { NotificationConsoleUtils } from "../../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../../Utils/NotificationConsoleUtils";
 import { GraphConfig } from "../../Tabs/GraphTab";
 import { GraphExplorer } from "./GraphExplorer";
 import * as Constants from "../../../Common/Constants";

src/Explorer/Graph/GraphExplorerComponent/GraphExplorer.tsx

@@ -8,7 +8,7 @@ import * as D3ForceGraph from "./D3ForceGraph";
 import { GraphVizComponentProps } from "./GraphVizComponent";
 import * as GraphData from "./GraphData";
 import { ConsoleDataType } from "../../Menus/NotificationConsole/NotificationConsoleComponent";
-import { NotificationConsoleUtils } from "../../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../../Utils/NotificationConsoleUtils";
 import { GraphUtil } from "./GraphUtil";
 import * as DataModels from "../../../Contracts/DataModels";
 import * as ViewModels from "../../../Contracts/ViewModels";

src/Explorer/Graph/GraphExplorerComponent/GremlinClient.test.ts

@@ -1,6 +1,6 @@
 import * as sinon from "sinon";
 import { GremlinClient, GremlinClientParameters } from "./GremlinClient";
-import { NotificationConsoleUtils } from "../../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../../Utils/NotificationConsoleUtils";
 import * as Logger from "../../../Common/Logger";
 
 describe("Gremlin Client", () => {

src/Explorer/Graph/GraphExplorerComponent/GremlinClient.ts

@@ -4,7 +4,7 @@
 
 import * as Q from "q";
 import { GremlinSimpleClient, Result } from "./GremlinSimpleClient";
-import { NotificationConsoleUtils } from "../../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../../Menus/NotificationConsole/NotificationConsoleComponent";
 import { HashMap } from "../../../Common/HashMap";
 import * as Logger from "../../../Common/Logger";

src/Explorer/Notebook/NotebookComponent/epics.ts

@@ -34,7 +34,7 @@ import { sessions, kernels } from "rx-jupyter";
 import { RecordOf } from "immutable";
 
 import * as Constants from "../../../Common/Constants";
-import { NotificationConsoleUtils } from "../../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../../Menus/NotificationConsole/NotificationConsoleComponent";
 import * as CdbActions from "./actions";
 import TelemetryProcessor from "../../../Shared/Telemetry/TelemetryProcessor";

src/Explorer/Notebook/NotebookContainerClient.ts

@@ -4,7 +4,7 @@
 import * as DataModels from "../../Contracts/DataModels";
 import * as Constants from "../../Common/Constants";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import * as Logger from "../../Common/Logger";
 
 export class NotebookContainerClient {

src/Explorer/Panes/DeleteCollectionConfirmationPane.test.ts

@@ -1,4 +1,4 @@
-jest.mock("../../Common/DocumentClientUtilityBase");
+jest.mock("../../Common/dataAccess/deleteCollection");
 import * as ko from "knockout";
 import * as sinon from "sinon";
 import Q from "q";
@@ -7,10 +7,10 @@ import * as ViewModels from "../../Contracts/ViewModels";
 import { Action, ActionModifiers } from "../../Shared/Telemetry/TelemetryConstants";
 import DeleteCollectionConfirmationPane from "./DeleteCollectionConfirmationPane";
 import DeleteFeedback from "../../Common/DeleteFeedback";
-import { deleteCollection } from "../../Common/DocumentClientUtilityBase";
 import Explorer from "../Explorer";
 import TelemetryProcessor from "../../Shared/Telemetry/TelemetryProcessor";
 import { TreeNode } from "../../Contracts/ViewModels";
+import { deleteCollection } from "../../Common/dataAccess/deleteCollection";
 
 describe("Delete Collection Confirmation Pane", () => {
   describe("Explorer.isLastCollection()", () => {
@@ -84,6 +84,7 @@ describe("Delete Collection Confirmation Pane", () => {
     let telemetryProcessorSpy: sinon.SinonSpy;
 
     beforeEach(() => {
+      (deleteCollection as jest.Mock).mockResolvedValue(undefined);
       telemetryProcessorSpy = sinon.spy(TelemetryProcessor, "trace");
     });
 
@@ -93,7 +94,6 @@ describe("Delete Collection Confirmation Pane", () => {
 
     it("it should log feedback if last collection and database is not shared", () => {
       let selectedCollectionId = "testCol";
-      (deleteCollection as jest.Mock).mockResolvedValue(null);
       let fakeExplorer = {} as Explorer;
       fakeExplorer.findSelectedCollection = () => {
         return {

src/Explorer/Panes/DeleteCollectionConfirmationPane.ts

@@ -9,9 +9,9 @@ import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsol
 import { ContextualPaneBase } from "./ContextualPaneBase";
 import { DefaultExperienceUtility } from "../../Shared/DefaultExperienceUtility";
 import DeleteFeedback from "../../Common/DeleteFeedback";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import TelemetryProcessor from "../../Shared/Telemetry/TelemetryProcessor";
-import { deleteCollection } from "../../Common/DocumentClientUtilityBase";
+import { deleteCollection } from "../../Common/dataAccess/deleteCollection";
 
 export default class DeleteCollectionConfirmationPane extends ContextualPaneBase {
   public collectionIdConfirmationText: ko.Observable<string>;
@@ -29,7 +29,7 @@ export default class DeleteCollectionConfirmationPane extends ContextualPaneBase
     this.resetData();
   }
 
-  public submit(): Q.Promise<any> {
+  public submit(): Promise<any> {
     if (!this._isValid()) {
       const selectedCollection: ViewModels.Collection = this.container.findSelectedCollection();
       this.formErrors("Input collection name does not match the selected collection");
@@ -37,7 +37,7 @@ export default class DeleteCollectionConfirmationPane extends ContextualPaneBase
         ConsoleDataType.Error,
         `Error while deleting collection ${selectedCollection && selectedCollection.id()}: ${this.formErrors()}`
       );
-      return Q.resolve();
+      return Promise.resolve();
     }
 
     this.formErrors("");
@@ -50,16 +50,16 @@ export default class DeleteCollectionConfirmationPane extends ContextualPaneBase
       dataExplorerArea: Constants.Areas.ContextualPane,
       paneTitle: this.title()
     });
-    let promise: Q.Promise<any>;
+    let promise: Promise<any>;
     if (this.container.isPreferredApiCassandra()) {
-      promise = (<CassandraAPIDataClient>this.container.tableDataClient).deleteTableOrKeyspace(
+      promise = ((<CassandraAPIDataClient>this.container.tableDataClient).deleteTableOrKeyspace(
         this.container.databaseAccount().properties.cassandraEndpoint,
         this.container.databaseAccount().id,
         `DROP TABLE ${selectedCollection.databaseId}.${selectedCollection.id()};`,
         this.container
-      );
+      ) as unknown) as Promise<any>;
     } else {
-      promise = deleteCollection(selectedCollection);
+      promise = deleteCollection(selectedCollection.databaseId, selectedCollection.id());
     }
     return promise.then(
       () => {

src/Explorer/Panes/DeleteDatabaseConfirmationPane.ts

@@ -10,7 +10,7 @@ import { ContextualPaneBase } from "./ContextualPaneBase";
 import { DefaultExperienceUtility } from "../../Shared/DefaultExperienceUtility";
 import DeleteFeedback from "../../Common/DeleteFeedback";
 
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import TelemetryProcessor from "../../Shared/Telemetry/TelemetryProcessor";
 import { deleteDatabase } from "../../Common/DocumentClientUtilityBase";
 

src/Explorer/Panes/GitHubReposPane.ts

@@ -8,7 +8,7 @@ import { Action, ActionModifiers } from "../../Shared/Telemetry/TelemetryConstan
 import TelemetryProcessor from "../../Shared/Telemetry/TelemetryProcessor";
 import * as GitHubUtils from "../../Utils/GitHubUtils";
 import { JunoUtils } from "../../Utils/JunoUtils";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { AuthorizeAccessComponent } from "../Controls/GitHub/AuthorizeAccessComponent";
 import { GitHubReposComponent, GitHubReposComponentProps, RepoListItem } from "../Controls/GitHub/GitHubReposComponent";
 import { GitHubReposComponentAdapter } from "../Controls/GitHub/GitHubReposComponentAdapter";

src/Explorer/Panes/LoadQueryPane.ts

@@ -5,7 +5,7 @@ import * as ViewModels from "../../Contracts/ViewModels";
 import { ContextualPaneBase } from "./ContextualPaneBase";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import * as Logger from "../../Common/Logger";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import QueryTab from "../Tabs/QueryTab";
 
 export class LoadQueryPane extends ContextualPaneBase {

src/Explorer/Panes/PublishNotebookPaneAdapter.tsx

@@ -4,7 +4,7 @@ import { ReactAdapter } from "../../Bindings/ReactBindingHandler";
 import * as Logger from "../../Common/Logger";
 import Explorer from "../Explorer";
 import { JunoClient } from "../../Juno/JunoClient";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import { GenericRightPaneComponent, GenericRightPaneProps } from "./GenericRightPaneComponent";
 import { PublishNotebookPaneComponent, PublishNotebookPaneProps } from "./PublishNotebookPaneComponent";

src/Explorer/Panes/RenewAdHocAccessPane.ts

@@ -6,7 +6,7 @@ import { ConnectionStringParser } from "../../Platform/Hosted/Helpers/Connection
 import { ContextualPaneBase } from "./ContextualPaneBase";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import { DefaultExperienceUtility } from "../../Shared/DefaultExperienceUtility";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 
 export class RenewAdHocAccessPane extends ContextualPaneBase {
   public accessKey: ko.Observable<string>;

src/Explorer/Panes/SaveQueryPane.ts

@@ -5,7 +5,7 @@ import * as ViewModels from "../../Contracts/ViewModels";
 import { Action } from "../../Shared/Telemetry/TelemetryConstants";
 import { ContextualPaneBase } from "./ContextualPaneBase";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import TelemetryProcessor from "../../Shared/Telemetry/TelemetryProcessor";
 import QueryTab from "../Tabs/QueryTab";
 

src/Explorer/Panes/SettingsPane.ts

@@ -4,7 +4,7 @@ import * as ViewModels from "../../Contracts/ViewModels";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import { ContextualPaneBase } from "./ContextualPaneBase";
 import { LocalStorageUtility, StorageKey } from "../../Shared/StorageUtility";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { StringUtility } from "../../Shared/StringUtility";
 import { config } from "../../Config";
 

src/Explorer/Panes/SetupNotebooksPane.ts

@@ -3,7 +3,7 @@ import { Action } from "../../Shared/Telemetry/TelemetryConstants";
 import { Areas, KeyCodes } from "../../Common/Constants";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import { ContextualPaneBase } from "./ContextualPaneBase";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import TelemetryProcessor from "../../Shared/Telemetry/TelemetryProcessor";
 import * as ko from "knockout";
 

src/Explorer/Panes/StringInputPane.ts

@@ -3,7 +3,7 @@ import Q from "q";
 import * as ViewModels from "../../Contracts/ViewModels";
 import { ContextualPaneBase } from "./ContextualPaneBase";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 
 export interface StringInputPaneOpenOptions {
   paneTitle: string;

src/Explorer/Panes/UploadFilePane.ts

@@ -3,7 +3,7 @@ import * as Constants from "../../Common/Constants";
 import * as ViewModels from "../../Contracts/ViewModels";
 import { ContextualPaneBase } from "./ContextualPaneBase";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 
 export interface UploadFilePaneOpenOptions {
   paneTitle: string;

src/Explorer/Panes/UploadItemsPane.ts

@@ -3,7 +3,7 @@ import * as Constants from "../../Common/Constants";
 import * as ViewModels from "../../Contracts/ViewModels";
 import { ContextualPaneBase } from "./ContextualPaneBase";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import * as ErrorParserUtility from "../../Common/ErrorParserUtility";
 import { UploadDetailsRecord, UploadDetails } from "../../workers/upload/definitions";
 

src/Explorer/Panes/UploadItemsPaneAdapter.tsx

@@ -6,7 +6,7 @@ import * as ViewModels from "../../Contracts/ViewModels";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import { IconButton } from "office-ui-fabric-react/lib/Button";
 import { GenericRightPaneComponent, GenericRightPaneProps } from "./GenericRightPaneComponent";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { ReactAdapter } from "../../Bindings/ReactBindingHandler";
 import { UploadDetailsRecord, UploadDetails } from "../../workers/upload/definitions";
 import InfoBubbleIcon from "../../../images/info-bubble.svg";

src/Explorer/Tables/TableDataClient.ts

@@ -9,7 +9,7 @@ import * as Entities from "./Entities";
 import EnvironmentUtility from "../../Common/EnvironmentUtility";
 import * as HeadersUtility from "../../Common/HeadersUtility";
 import * as Logger from "../../Common/Logger";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import * as TableConstants from "./Constants";
 import * as TableEntityProcessor from "./TableEntityProcessor";
 import * as ViewModels from "../../Contracts/ViewModels";

src/Explorer/Tabs/DocumentsTab.ts

@@ -22,7 +22,7 @@ import DeleteDocumentIcon from "../../../images/DeleteDocument.svg";
 import UploadIcon from "../../../images/Upload_16x16.svg";
 import { extractPartitionKey, PartitionKeyDefinition, QueryIterator, ItemDefinition, Resource } from "@azure/cosmos";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import Explorer from "../Explorer";
 import {
   readDocument,

src/Explorer/Tabs/MongoShellTab.ts

@@ -11,7 +11,7 @@ import { Action, ActionModifiers } from "../../Shared/Telemetry/TelemetryConstan
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import { CosmosClient } from "../../Common/CosmosClient";
 import { HashMap } from "../../Common/HashMap";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { PlatformType } from "../../PlatformType";
 import Explorer from "../Explorer";
 

src/Explorer/Tabs/NotebookV2Tab.ts

@@ -22,7 +22,7 @@ import { Action, ActionModifiers } from "../../Shared/Telemetry/TelemetryConstan
 import { Areas, ArmApiVersions } from "../../Common/Constants";
 import { CommandBarComponentButtonFactory } from "../Menus/CommandBar/CommandBarComponentButtonFactory";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { NotebookComponentAdapter } from "../Notebook/NotebookComponent/NotebookComponentAdapter";
 import { NotebookConfigurationUtils } from "../../Utils/NotebookConfigurationUtils";
 import { KernelSpecsDisplay, NotebookClientV2 } from "../Notebook/NotebookClientV2";

src/Explorer/Tree/Collection.ts

@@ -11,7 +11,7 @@ import * as ViewModels from "../../Contracts/ViewModels";
 import { PlatformType } from "../../PlatformType";
 import { Action, ActionModifiers } from "../../Shared/Telemetry/TelemetryConstants";
 import TelemetryProcessor from "../../Shared/Telemetry/TelemetryProcessor";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { OfferUtils } from "../../Utils/OfferUtils";
 import { StartUploadMessageParams, UploadDetails, UploadDetailsRecord } from "../../workers/upload/definitions";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";

src/Explorer/Tree/Database.ts

@@ -8,7 +8,7 @@ import { Action, ActionModifiers } from "../../Shared/Telemetry/TelemetryConstan
 import DatabaseSettingsTab from "../Tabs/DatabaseSettingsTab";
 import Collection from "./Collection";
 import TelemetryProcessor from "../../Shared/Telemetry/TelemetryProcessor";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../Menus/NotificationConsole/NotificationConsoleComponent";
 import * as Logger from "../../Common/Logger";
 import Explorer from "../Explorer";

src/GitHub/GitHubOAuthService.ts

@@ -6,7 +6,7 @@ import { AuthorizeAccessComponent } from "../Explorer/Controls/GitHub/AuthorizeA
 import { ConsoleDataType } from "../Explorer/Menus/NotificationConsole/NotificationConsoleComponent";
 import { JunoClient } from "../Juno/JunoClient";
 import { isInvalidParentFrameOrigin } from "../Utils/MessageValidation";
-import { NotificationConsoleUtils } from "../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../Utils/NotificationConsoleUtils";
 import { GitHubConnectorMsgType, IGitHubConnectorParams } from "./GitHubConnector";
 
 window.addEventListener("message", (event: MessageEvent) => {

src/Platform/Hosted/Authorization.ts

@@ -5,7 +5,7 @@ import * as Constants from "../../Common/Constants";
 import * as DataModels from "../../Contracts/DataModels";
 import * as ViewModels from "../../Contracts/ViewModels";
 import { AuthType } from "../../AuthType";
-import { NotificationConsoleUtils } from "../../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../../Utils/NotificationConsoleUtils";
 import { ConsoleDataType } from "../../Explorer/Menus/NotificationConsole/NotificationConsoleComponent";
 import { DefaultExperienceUtility } from "../../Shared/DefaultExperienceUtility";
 import { CosmosClient } from "../../Common/CosmosClient";

src/Shared/AddCollectionUtility.ts

@@ -9,7 +9,7 @@ import { CosmosClient } from "../Common/CosmosClient";
 import { HttpStatusCodes } from "../Common/Constants";
 import { sendMessage } from "../Common/MessageHandler";
 import { MessageTypes } from "../Contracts/ExplorerContracts";
-import { NotificationConsoleUtils } from "../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../Utils/NotificationConsoleUtils";
 import { ResourceProviderClient } from "../ResourceProvider/ResourceProviderClient";
 import Explorer from "../Explorer/Explorer";
 

src/Shared/AddDatabaseUtility.ts

@@ -6,7 +6,7 @@ import { CosmosClient } from "../Common/CosmosClient";
 import { HttpStatusCodes } from "../Common/Constants";
 import { sendMessage } from "../Common/MessageHandler";
 import { MessageTypes } from "../Contracts/ExplorerContracts";
-import { NotificationConsoleUtils } from "../Utils/NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "../Utils/NotificationConsoleUtils";
 import { ResourceProviderClient } from "../ResourceProvider/ResourceProviderClient";
 
 export class AddDbUtilities {

src/Utils/GalleryUtils.ts

@@ -1,5 +1,5 @@
 import { IGalleryItem, JunoClient } from "../Juno/JunoClient";
-import { NotificationConsoleUtils } from "./NotificationConsoleUtils";
+import * as NotificationConsoleUtils from "./NotificationConsoleUtils";
 import { ConsoleDataType } from "../Explorer/Menus/NotificationConsole/NotificationConsoleComponent";
 import * as Logger from "../Common/Logger";
 import {

src/Utils/NotificationConsoleUtils.ts

@@ -3,26 +3,77 @@ import { ConsoleDataType } from "../Explorer/Menus/NotificationConsole/Notificat
 
 const _global = typeof self === "undefined" ? window : self;
 
-export class NotificationConsoleUtils {
-  public static logConsoleMessage(type: ConsoleDataType, message: string, id?: string): string {
-    const dataExplorer = _global.dataExplorer;
-    if (dataExplorer != null) {
-      const date = new Date();
-      const formattedDate: string = new Intl.DateTimeFormat("en-EN", {
-        hour12: true,
-        hour: "numeric",
-        minute: "numeric"
-      }).format(date);
-      if (!id) {
-        id = _.uniqueId();
-      }
-      dataExplorer.logConsoleData({ type: type, date: formattedDate, message: message, id: id });
+// DEPRECATED: Use logConsoleInfo, logConsoleError, logConsoleProgress instead
+export function logConsoleMessage(type: ConsoleDataType, message: string, id?: string): string {
+  const dataExplorer = _global.dataExplorer;
+  if (dataExplorer) {
+    const date = new Date();
+    const formattedDate: string = new Intl.DateTimeFormat("en-EN", {
+      hour12: true,
+      hour: "numeric",
+      minute: "numeric"
+    }).format(date);
+    if (!id) {
+      id = _.uniqueId();
     }
-    return id;
+    dataExplorer.logConsoleData({ type: type, date: formattedDate, message: message, id: id });
   }
+  return id;
+}
 
-  public static clearInProgressMessageWithId(id: string) {
-    const dataExplorer = _global.dataExplorer;
-    dataExplorer && (dataExplorer as any).deleteInProgressConsoleDataWithId(id);
+export function clearInProgressMessageWithId(id: string): void {
+  const dataExplorer = _global.dataExplorer;
+  dataExplorer && dataExplorer.deleteInProgressConsoleDataWithId(id);
+}
+
+export function logConsoleProgress(message: string): () => void {
+  const type = ConsoleDataType.InProgress;
+  const dataExplorer = _global.dataExplorer;
+  if (dataExplorer) {
+    const id = _.uniqueId();
+    const date = new Date();
+    const formattedDate: string = new Intl.DateTimeFormat("en-EN", {
+      hour12: true,
+      hour: "numeric",
+      minute: "numeric"
+    }).format(date);
+    dataExplorer.logConsoleData({ type, date: formattedDate, message, id });
+    return () => {
+      dataExplorer.deleteInProgressConsoleDataWithId(id);
+    };
+  } else {
+    return () => {
+      return;
+    };
+  }
+}
+
+export function logConsoleError(message: string): void {
+  const type = ConsoleDataType.Error;
+  const dataExplorer = _global.dataExplorer;
+  if (dataExplorer) {
+    const id = _.uniqueId();
+    const date = new Date();
+    const formattedDate: string = new Intl.DateTimeFormat("en-EN", {
+      hour12: true,
+      hour: "numeric",
+      minute: "numeric"
+    }).format(date);
+    dataExplorer.logConsoleData({ type, date: formattedDate, message, id });
+  }
+}
+
+export function logConsoleInfo(message: string): void {
+  const type = ConsoleDataType.Info;
+  const dataExplorer = _global.dataExplorer;
+  if (dataExplorer) {
+    const id = _.uniqueId();
+    const date = new Date();
+    const formattedDate: string = new Intl.DateTimeFormat("en-EN", {
+      hour12: true,
+      hour: "numeric",
+      minute: "numeric"
+    }).format(date);
+    dataExplorer.logConsoleData({ type, date: formattedDate, message, id });
   }
 }

src/Utils/arm/generatedClients/2020-04-01/cassandraResources.ts

@@ -0,0 +1,196 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Lists the Cassandra keyspaces under an existing Azure Cosmos DB database account. */
+export async function listCassandraKeyspaces(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.CassandraKeyspaceListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the Cassandra keyspaces under an existing Azure Cosmos DB database account with the provided name. */
+export async function getCassandraKeyspace(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string
+): Promise<Types.CassandraKeyspaceGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB Cassandra keyspace */
+export async function createUpdateCassandraKeyspace(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string,
+  body: Types.CassandraKeyspaceCreateUpdateParameters
+): Promise<Types.CassandraKeyspaceGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB Cassandra keyspace. */
+export async function deleteCassandraKeyspace(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Gets the RUs per second of the Cassandra Keyspace under an existing Azure Cosmos DB database account with the provided name. */
+export async function getCassandraKeyspaceThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string
+): Promise<Types.ThroughputSettingsGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Update RUs per second of an Azure Cosmos DB Cassandra Keyspace */
+export async function updateCassandraKeyspaceThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string,
+  body: Types.ThroughputSettingsUpdateParameters
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Migrate an Azure Cosmos DB Cassandra Keyspace from manual throughput to autoscale */
+export async function migrateCassandraKeyspaceToAutoscale(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/throughputSettings/default/migrateToAutoscale`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Migrate an Azure Cosmos DB Cassandra Keyspace from autoscale to manual throughput */
+export async function migrateCassandraKeyspaceToManualThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/throughputSettings/default/migrateToManualThroughput`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Lists the Cassandra table under an existing Azure Cosmos DB database account. */
+export async function listCassandraTables(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string
+): Promise<Types.CassandraTableListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/tables`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the Cassandra table under an existing Azure Cosmos DB database account. */
+export async function getCassandraTable(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string,
+  tableName: string
+): Promise<Types.CassandraTableGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/tables/${tableName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB Cassandra Table */
+export async function createUpdateCassandraTable(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string,
+  tableName: string,
+  body: Types.CassandraTableCreateUpdateParameters
+): Promise<Types.CassandraTableGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/tables/${tableName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB Cassandra table. */
+export async function deleteCassandraTable(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string,
+  tableName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/tables/${tableName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Gets the RUs per second of the Cassandra table under an existing Azure Cosmos DB database account with the provided name. */
+export async function getCassandraTableThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string,
+  tableName: string
+): Promise<Types.ThroughputSettingsGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/tables/${tableName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Update RUs per second of an Azure Cosmos DB Cassandra table */
+export async function updateCassandraTableThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string,
+  tableName: string,
+  body: Types.ThroughputSettingsUpdateParameters
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/tables/${tableName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Migrate an Azure Cosmos DB Cassandra table from manual throughput to autoscale */
+export async function migrateCassandraTableToAutoscale(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string,
+  tableName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/tables/${tableName}/throughputSettings/default/migrateToAutoscale`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Migrate an Azure Cosmos DB Cassandra table from autoscale to manual throughput */
+export async function migrateCassandraTableToManualThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  keyspaceName: string,
+  tableName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/cassandraKeyspaces/${keyspaceName}/tables/${tableName}/throughputSettings/default/migrateToManualThroughput`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/collection.ts

@@ -0,0 +1,46 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given database account and collection. */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseRid: string,
+  collectionRid: string
+): Promise<Types.MetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/databases/${databaseRid}/collections/${collectionRid}/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Retrieves the usages (most recent storage data) for the given collection. */
+export async function listUsages(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseRid: string,
+  collectionRid: string
+): Promise<Types.UsagesResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/databases/${databaseRid}/collections/${collectionRid}/usages`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Retrieves metric definitions for the given collection. */
+export async function listMetricDefinitions(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseRid: string,
+  collectionRid: string
+): Promise<Types.MetricDefinitionsListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/databases/${databaseRid}/collections/${collectionRid}/metricDefinitions`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/collectionPartition.ts

@@ -0,0 +1,34 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given collection, split by partition. */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseRid: string,
+  collectionRid: string
+): Promise<Types.PartitionMetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/databases/${databaseRid}/collections/${collectionRid}/partitions/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Retrieves the usages (most recent storage data) for the given collection, split by partition. */
+export async function listUsages(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseRid: string,
+  collectionRid: string
+): Promise<Types.PartitionUsagesResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/databases/${databaseRid}/collections/${collectionRid}/partitions/usages`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/collectionPartitionRegion.ts

@@ -0,0 +1,23 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given collection and region, split by partition. */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  region: string,
+  databaseRid: string,
+  collectionRid: string
+): Promise<Types.PartitionMetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/region/${region}/databases/${databaseRid}/collections/${collectionRid}/partitions/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/collectionRegion.ts

@@ -0,0 +1,23 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given database account, collection and region. */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  region: string,
+  databaseRid: string,
+  collectionRid: string
+): Promise<Types.MetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/region/${region}/databases/${databaseRid}/collections/${collectionRid}/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/database.ts

@@ -0,0 +1,43 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given database account and database. */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseRid: string
+): Promise<Types.MetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/databases/${databaseRid}/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Retrieves the usages (most recent data) for the given database. */
+export async function listUsages(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseRid: string
+): Promise<Types.UsagesResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/databases/${databaseRid}/usages`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Retrieves metric definitions for the given database. */
+export async function listMetricDefinitions(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseRid: string
+): Promise<Types.MetricDefinitionsListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/databases/${databaseRid}/metricDefinitions`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/databaseAccountRegion.ts

@@ -0,0 +1,21 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given database account and region. */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  region: string
+): Promise<Types.MetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/region/${region}/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/databaseAccounts.ts

@@ -0,0 +1,183 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the properties of an existing Azure Cosmos DB database account. */
+export async function get(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.DatabaseAccountGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Updates the properties of an existing Azure Cosmos DB database account. */
+export async function update(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  body: Types.DatabaseAccountUpdateParameters
+): Promise<Types.DatabaseAccountGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PATCH", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Creates or updates an Azure Cosmos DB database account. The "Update" method is preferred when performing updates on an account. */
+export async function createOrUpdate(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  body: Types.DatabaseAccountCreateUpdateParameters
+): Promise<Types.DatabaseAccountGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB database account. */
+export async function destroy(subscriptionId: string, resourceGroupName: string, accountName: string): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Changes the failover priority for the Azure Cosmos DB database account. A failover priority of 0 indicates a write region. The maximum value for a failover priority = (total number of regions - 1). Failover priority values must be unique for each of the regions in which the database account exists. */
+export async function failoverPriorityChange(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  body: Types.FailoverPolicies
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/failoverPriorityChange`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Lists all the Azure Cosmos DB database accounts available under the subscription. */
+export async function list(subscriptionId: string): Promise<Types.DatabaseAccountsListResult> {
+  const path = `/subscriptions/${subscriptionId}/providers/Microsoft.DocumentDB/databaseAccounts`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Lists all the Azure Cosmos DB database accounts available under the given resource group. */
+export async function listByResourceGroup(
+  subscriptionId: string,
+  resourceGroupName: string
+): Promise<Types.DatabaseAccountsListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Lists the access keys for the specified Azure Cosmos DB database account. */
+export async function listKeys(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.DatabaseAccountListKeysResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/listKeys`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Lists the connection strings for the specified Azure Cosmos DB database account. */
+export async function listConnectionStrings(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.DatabaseAccountListConnectionStringsResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/listConnectionStrings`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Offline the specified region for the specified Azure Cosmos DB database account. */
+export async function offlineRegion(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  body: Types.RegionForOnlineOffline
+): Promise<void | Types.ErrorResponse> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/offlineRegion`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Online the specified region for the specified Azure Cosmos DB database account. */
+export async function onlineRegion(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  body: Types.RegionForOnlineOffline
+): Promise<void | Types.ErrorResponse> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/onlineRegion`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Lists the read-only access keys for the specified Azure Cosmos DB database account. */
+export async function getReadOnlyKeys(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.DatabaseAccountListReadOnlyKeysResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/readonlykeys`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Lists the read-only access keys for the specified Azure Cosmos DB database account. */
+export async function listReadOnlyKeys(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.DatabaseAccountListReadOnlyKeysResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/readonlykeys`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Regenerates an access key for the specified Azure Cosmos DB database account. */
+export async function regenerateKey(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  body: Types.DatabaseAccountRegenerateKeyParameters
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/regenerateKey`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Checks that the Azure Cosmos DB account name already exists. A valid account name may contain only lowercase letters, numbers, and the '-' character, and must be between 3 and 50 characters. */
+export async function checkNameExists(accountName: string): Promise<void> {
+  const path = `/providers/Microsoft.DocumentDB/databaseAccountNames/${accountName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "HEAD", apiVersion });
+}
+
+/* Retrieves the metrics determined by the given filter for the given database account. */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.MetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Retrieves the usages (most recent data) for the given database account. */
+export async function listUsages(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.UsagesResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/usages`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Retrieves metric definitions for the given database account. */
+export async function listMetricDefinitions(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.MetricDefinitionsListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/metricDefinitions`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/gremlinResources.ts

@@ -0,0 +1,196 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Lists the Gremlin databases under an existing Azure Cosmos DB database account. */
+export async function listGremlinDatabases(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.GremlinDatabaseListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the Gremlin databases under an existing Azure Cosmos DB database account with the provided name. */
+export async function getGremlinDatabase(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.GremlinDatabaseGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB Gremlin database */
+export async function createUpdateGremlinDatabase(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  body: Types.GremlinDatabaseCreateUpdateParameters
+): Promise<Types.GremlinDatabaseGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB Gremlin database. */
+export async function deleteGremlinDatabase(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Gets the RUs per second of the Gremlin database under an existing Azure Cosmos DB database account with the provided name. */
+export async function getGremlinDatabaseThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.ThroughputSettingsGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Update RUs per second of an Azure Cosmos DB Gremlin database */
+export async function updateGremlinDatabaseThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  body: Types.ThroughputSettingsUpdateParameters
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Migrate an Azure Cosmos DB Gremlin database from manual throughput to autoscale */
+export async function migrateGremlinDatabaseToAutoscale(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/throughputSettings/default/migrateToAutoscale`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Migrate an Azure Cosmos DB Gremlin database from autoscale to manual throughput */
+export async function migrateGremlinDatabaseToManualThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/throughputSettings/default/migrateToManualThroughput`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Lists the Gremlin graph under an existing Azure Cosmos DB database account. */
+export async function listGremlinGraphs(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.GremlinGraphListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/graphs`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the Gremlin graph under an existing Azure Cosmos DB database account. */
+export async function getGremlinGraph(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  graphName: string
+): Promise<Types.GremlinGraphGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/graphs/${graphName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB Gremlin graph */
+export async function createUpdateGremlinGraph(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  graphName: string,
+  body: Types.GremlinGraphCreateUpdateParameters
+): Promise<Types.GremlinGraphGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/graphs/${graphName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB Gremlin graph. */
+export async function deleteGremlinGraph(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  graphName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/graphs/${graphName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Gets the Gremlin graph throughput under an existing Azure Cosmos DB database account with the provided name. */
+export async function getGremlinGraphThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  graphName: string
+): Promise<Types.ThroughputSettingsGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/graphs/${graphName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Update RUs per second of an Azure Cosmos DB Gremlin graph */
+export async function updateGremlinGraphThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  graphName: string,
+  body: Types.ThroughputSettingsUpdateParameters
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/graphs/${graphName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Migrate an Azure Cosmos DB Gremlin graph from manual throughput to autoscale */
+export async function migrateGremlinGraphToAutoscale(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  graphName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/graphs/${graphName}/throughputSettings/default/migrateToAutoscale`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Migrate an Azure Cosmos DB Gremlin graph from autoscale to manual throughput */
+export async function migrateGremlinGraphToManualThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  graphName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/gremlinDatabases/${databaseName}/graphs/${graphName}/throughputSettings/default/migrateToManualThroughput`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/mongoDBResources.ts

@@ -0,0 +1,196 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Lists the MongoDB databases under an existing Azure Cosmos DB database account. */
+export async function listMongoDBDatabases(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.MongoDBDatabaseListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the MongoDB databases under an existing Azure Cosmos DB database account with the provided name. */
+export async function getMongoDBDatabase(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.MongoDBDatabaseGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or updates Azure Cosmos DB MongoDB database */
+export async function createUpdateMongoDBDatabase(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  body: Types.MongoDBDatabaseCreateUpdateParameters
+): Promise<Types.MongoDBDatabaseGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB MongoDB database. */
+export async function deleteMongoDBDatabase(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Gets the RUs per second of the MongoDB database under an existing Azure Cosmos DB database account with the provided name. */
+export async function getMongoDBDatabaseThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.ThroughputSettingsGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Update RUs per second of the an Azure Cosmos DB MongoDB database */
+export async function updateMongoDBDatabaseThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  body: Types.ThroughputSettingsUpdateParameters
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Migrate an Azure Cosmos DB MongoDB database from manual throughput to autoscale */
+export async function migrateMongoDBDatabaseToAutoscale(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/throughputSettings/default/migrateToAutoscale`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Migrate an Azure Cosmos DB MongoDB database from autoscale to manual throughput */
+export async function migrateMongoDBDatabaseToManualThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/throughputSettings/default/migrateToManualThroughput`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Lists the MongoDB collection under an existing Azure Cosmos DB database account. */
+export async function listMongoDBCollections(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.MongoDBCollectionListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/collections`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the MongoDB collection under an existing Azure Cosmos DB database account. */
+export async function getMongoDBCollection(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  collectionName: string
+): Promise<Types.MongoDBCollectionGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/collections/${collectionName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB MongoDB Collection */
+export async function createUpdateMongoDBCollection(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  collectionName: string,
+  body: Types.MongoDBCollectionCreateUpdateParameters
+): Promise<Types.MongoDBCollectionGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/collections/${collectionName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB MongoDB Collection. */
+export async function deleteMongoDBCollection(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  collectionName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/collections/${collectionName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Gets the RUs per second of the MongoDB collection under an existing Azure Cosmos DB database account with the provided name. */
+export async function getMongoDBCollectionThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  collectionName: string
+): Promise<Types.ThroughputSettingsGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/collections/${collectionName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Update the RUs per second of an Azure Cosmos DB MongoDB collection */
+export async function updateMongoDBCollectionThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  collectionName: string,
+  body: Types.ThroughputSettingsUpdateParameters
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/collections/${collectionName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Migrate an Azure Cosmos DB MongoDB collection from manual throughput to autoscale */
+export async function migrateMongoDBCollectionToAutoscale(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  collectionName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/collections/${collectionName}/throughputSettings/default/migrateToAutoscale`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Migrate an Azure Cosmos DB MongoDB collection from autoscale to manual throughput */
+export async function migrateMongoDBCollectionToManualThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  collectionName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/mongodbDatabases/${databaseName}/collections/${collectionName}/throughputSettings/default/migrateToManualThroughput`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/operations.ts

@@ -0,0 +1,16 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Lists all of the available Cosmos DB Resource Provider operations. */
+export async function list(): Promise<Types.OperationListResult> {
+  const path = `/providers/Microsoft.DocumentDB/operations`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/partitionKeyRangeId.ts

@@ -0,0 +1,23 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given partition key range id. */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseRid: string,
+  collectionRid: string,
+  partitionKeyRangeId: string
+): Promise<Types.PartitionMetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/databases/${databaseRid}/collections/${collectionRid}/partitionKeyRangeId/${partitionKeyRangeId}/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/partitionKeyRangeIdRegion.ts

@@ -0,0 +1,24 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given partition key range id and region. */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  region: string,
+  databaseRid: string,
+  collectionRid: string,
+  partitionKeyRangeId: string
+): Promise<Types.PartitionMetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/region/${region}/databases/${databaseRid}/collections/${collectionRid}/partitionKeyRangeId/${partitionKeyRangeId}/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/percentile.ts

@@ -0,0 +1,20 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given database account. This url is only for PBS and Replication Latency data */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.PercentileMetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/percentile/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/percentileSourceTarget.ts

@@ -0,0 +1,22 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given account, source and target region. This url is only for PBS and Replication Latency data */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  sourceRegion: string,
+  targetRegion: string
+): Promise<Types.PercentileMetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sourceRegion/${sourceRegion}/targetRegion/${targetRegion}/percentile/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/percentileTarget.ts

@@ -0,0 +1,21 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Retrieves the metrics determined by the given filter for the given account target region. This url is only for PBS and Replication Latency data */
+export async function listMetrics(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  targetRegion: string
+): Promise<Types.PercentileMetricListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/targetRegion/${targetRegion}/percentile/metrics`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/sqlResources.ts

@@ -0,0 +1,352 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Lists the SQL databases under an existing Azure Cosmos DB database account. */
+export async function listSqlDatabases(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.SqlDatabaseListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the SQL database under an existing Azure Cosmos DB database account with the provided name. */
+export async function getSqlDatabase(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.SqlDatabaseGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB SQL database */
+export async function createUpdateSqlDatabase(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  body: Types.SqlDatabaseCreateUpdateParameters
+): Promise<Types.SqlDatabaseGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB SQL database. */
+export async function deleteSqlDatabase(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Gets the RUs per second of the SQL database under an existing Azure Cosmos DB database account with the provided name. */
+export async function getSqlDatabaseThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.ThroughputSettingsGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Update RUs per second of an Azure Cosmos DB SQL database */
+export async function updateSqlDatabaseThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  body: Types.ThroughputSettingsUpdateParameters
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Migrate an Azure Cosmos DB SQL database from manual throughput to autoscale */
+export async function migrateSqlDatabaseToAutoscale(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/throughputSettings/default/migrateToAutoscale`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Migrate an Azure Cosmos DB SQL database from autoscale to manual throughput */
+export async function migrateSqlDatabaseToManualThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/throughputSettings/default/migrateToManualThroughput`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Lists the SQL container under an existing Azure Cosmos DB database account. */
+export async function listSqlContainers(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string
+): Promise<Types.SqlContainerListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the SQL container under an existing Azure Cosmos DB database account. */
+export async function getSqlContainer(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string
+): Promise<Types.SqlContainerGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB SQL container */
+export async function createUpdateSqlContainer(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  body: Types.SqlContainerCreateUpdateParameters
+): Promise<Types.SqlContainerGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB SQL container. */
+export async function deleteSqlContainer(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Gets the RUs per second of the SQL container under an existing Azure Cosmos DB database account. */
+export async function getSqlContainerThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string
+): Promise<Types.ThroughputSettingsGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Update RUs per second of an Azure Cosmos DB SQL container */
+export async function updateSqlContainerThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  body: Types.ThroughputSettingsUpdateParameters
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Migrate an Azure Cosmos DB SQL container from manual throughput to autoscale */
+export async function migrateSqlContainerToAutoscale(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/throughputSettings/default/migrateToAutoscale`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Migrate an Azure Cosmos DB SQL container from autoscale to manual throughput */
+export async function migrateSqlContainerToManualThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/throughputSettings/default/migrateToManualThroughput`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Lists the SQL storedProcedure under an existing Azure Cosmos DB database account. */
+export async function listSqlStoredProcedures(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string
+): Promise<Types.SqlStoredProcedureListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/storedProcedures`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the SQL storedProcedure under an existing Azure Cosmos DB database account. */
+export async function getSqlStoredProcedure(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  storedProcedureName: string
+): Promise<Types.SqlStoredProcedureGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/storedProcedures/${storedProcedureName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB SQL storedProcedure */
+export async function createUpdateSqlStoredProcedure(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  storedProcedureName: string,
+  body: Types.SqlStoredProcedureCreateUpdateParameters
+): Promise<Types.SqlStoredProcedureGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/storedProcedures/${storedProcedureName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB SQL storedProcedure. */
+export async function deleteSqlStoredProcedure(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  storedProcedureName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/storedProcedures/${storedProcedureName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Lists the SQL userDefinedFunction under an existing Azure Cosmos DB database account. */
+export async function listSqlUserDefinedFunctions(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string
+): Promise<Types.SqlUserDefinedFunctionListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/userDefinedFunctions`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the SQL userDefinedFunction under an existing Azure Cosmos DB database account. */
+export async function getSqlUserDefinedFunction(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  userDefinedFunctionName: string
+): Promise<Types.SqlUserDefinedFunctionGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/userDefinedFunctions/${userDefinedFunctionName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB SQL userDefinedFunction */
+export async function createUpdateSqlUserDefinedFunction(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  userDefinedFunctionName: string,
+  body: Types.SqlUserDefinedFunctionCreateUpdateParameters
+): Promise<Types.SqlUserDefinedFunctionGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/userDefinedFunctions/${userDefinedFunctionName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB SQL userDefinedFunction. */
+export async function deleteSqlUserDefinedFunction(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  userDefinedFunctionName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/userDefinedFunctions/${userDefinedFunctionName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Lists the SQL trigger under an existing Azure Cosmos DB database account. */
+export async function listSqlTriggers(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string
+): Promise<Types.SqlTriggerListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/triggers`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the SQL trigger under an existing Azure Cosmos DB database account. */
+export async function getSqlTrigger(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  triggerName: string
+): Promise<Types.SqlTriggerGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/triggers/${triggerName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB SQL trigger */
+export async function createUpdateSqlTrigger(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  triggerName: string,
+  body: Types.SqlTriggerCreateUpdateParameters
+): Promise<Types.SqlTriggerGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/triggers/${triggerName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB SQL trigger. */
+export async function deleteSqlTrigger(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  databaseName: string,
+  containerName: string,
+  triggerName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/sqlDatabases/${databaseName}/containers/${containerName}/triggers/${triggerName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}

src/Utils/arm/generatedClients/2020-04-01/tableResources.ts

@@ -0,0 +1,99 @@
+/* 
+  AUTOGENERATED FILE
+  Do not manually edit
+  Run "npm run generateARMClients" to regenerate
+*/
+
+import { armRequest } from "../../request";
+import * as Types from "./types";
+import { config } from "../../../../Config";
+const apiVersion = "2020-04-01";
+
+/* Lists the Tables under an existing Azure Cosmos DB database account. */
+export async function listTables(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string
+): Promise<Types.TableListResult> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/tables`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Gets the Tables under an existing Azure Cosmos DB database account with the provided name. */
+export async function getTable(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  tableName: string
+): Promise<Types.TableGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/tables/${tableName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Create or update an Azure Cosmos DB Table */
+export async function createUpdateTable(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  tableName: string,
+  body: Types.TableCreateUpdateParameters
+): Promise<Types.TableGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/tables/${tableName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Deletes an existing Azure Cosmos DB Table. */
+export async function deleteTable(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  tableName: string
+): Promise<void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/tables/${tableName}`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "DELETE", apiVersion });
+}
+
+/* Gets the RUs per second of the Table under an existing Azure Cosmos DB database account with the provided name. */
+export async function getTableThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  tableName: string
+): Promise<Types.ThroughputSettingsGetResults> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/tables/${tableName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "GET", apiVersion });
+}
+
+/* Update RUs per second of an Azure Cosmos DB Table */
+export async function updateTableThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  tableName: string,
+  body: Types.ThroughputSettingsUpdateParameters
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/tables/${tableName}/throughputSettings/default`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "PUT", apiVersion, body: JSON.stringify(body) });
+}
+
+/* Migrate an Azure Cosmos DB Table from manual throughput to autoscale */
+export async function migrateTableToAutoscale(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  tableName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/tables/${tableName}/throughputSettings/default/migrateToAutoscale`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}
+
+/* Migrate an Azure Cosmos DB Table from autoscale to manual throughput */
+export async function migrateTableToManualThroughput(
+  subscriptionId: string,
+  resourceGroupName: string,
+  accountName: string,
+  tableName: string
+): Promise<Types.ThroughputSettingsGetResults | void> {
+  const path = `/subscriptions/${subscriptionId}/resourceGroups/${resourceGroupName}/providers/Microsoft.DocumentDB/databaseAccounts/${accountName}/tables/${tableName}/throughputSettings/default/migrateToManualThroughput`;
+  return armRequest({ host: config.ARM_ENDPOINT, path, method: "POST", apiVersion });
+}

src/Utils/arm/request.test.ts

@@ -0,0 +1,44 @@
+import { armRequest } from "./request";
+
+describe("ARM request", () => {
+  it("should call window.fetch", async () => {
+    window.fetch = jest.fn().mockResolvedValue({
+      ok: true,
+      json: async () => {
+        return {};
+      }
+    });
+    await armRequest({ apiVersion: "2001-01-01", host: "https://foo.com", path: "foo", method: "GET" });
+    expect(window.fetch).toHaveBeenCalled();
+  });
+
+  it("should poll for async operations", async () => {
+    const headers = new Headers();
+    headers.set("azure-asyncoperation", "https://foo.com/operationStatus");
+    window.fetch = jest.fn().mockResolvedValue({
+      ok: true,
+      headers,
+      json: async () => {
+        return { status: "Succeeded" };
+      }
+    });
+    await armRequest({ apiVersion: "2001-01-01", host: "https://foo.com", path: "foo", method: "GET" });
+    expect(window.fetch).toHaveBeenCalledTimes(2);
+  });
+
+  it("should throw for failed async operations", async () => {
+    const headers = new Headers();
+    headers.set("azure-asyncoperation", "https://foo.com/operationStatus");
+    window.fetch = jest.fn().mockResolvedValue({
+      ok: true,
+      headers,
+      json: async () => {
+        return { status: "Failed" };
+      }
+    });
+    await expect(() =>
+      armRequest({ apiVersion: "2001-01-01", host: "https://foo.com", path: "foo", method: "GET" })
+    ).rejects.toThrow();
+    expect(window.fetch).toHaveBeenCalledTimes(2);
+  });
+});

src/Utils/arm/request.ts

@@ -0,0 +1,102 @@
+/* 
+
+A general purpose fetch function for ARM resources. Not designed to be used directly
+Instead, generate ARM clients that consume this function with stricter typing.
+
+*/
+
+import { CosmosClient } from "../../Common/CosmosClient";
+import promiseRetry, { AbortError } from "p-retry";
+
+interface ErrorResponse {
+  error: {
+    code: string;
+    message: string;
+  };
+}
+
+interface ARMError extends Error {
+  code: string;
+}
+
+interface Options {
+  host: string;
+  path: string;
+  apiVersion: string;
+  method: "GET" | "POST" | "PUT" | "PATCH" | "DELETE" | "HEAD";
+  body?: unknown;
+}
+
+// TODO: This is very similar to what is happening in ResourceProviderClient.ts. Should probably merge them.
+export async function armRequest<T>({ host, path, apiVersion, method, body: requestBody }: Options): Promise<T> {
+  const url = new URL(path, host);
+  url.searchParams.append("api-version", apiVersion);
+  const response = await window.fetch(url.href, {
+    method,
+    headers: {
+      Authorization: CosmosClient.authorizationToken()
+    },
+    body: requestBody ? JSON.stringify(requestBody) : undefined
+  });
+  if (!response.ok) {
+    const errorResponse = (await response.json()) as ErrorResponse;
+    const error = new Error(errorResponse.error?.message) as ARMError;
+    error.code = errorResponse.error.code;
+    throw error;
+  }
+
+  const operationStatusUrl = response.headers && response.headers.get("azure-asyncoperation");
+  if (operationStatusUrl) {
+    await promiseRetry(() => getOperationStatus(operationStatusUrl));
+    // TODO: ARM is supposed to return a resourceLocation property, but it does not https://github.com/microsoft/api-guidelines/blob/vNext/Guidelines.md#target-resource-location
+    // When Cosmos RP adds resourceLocation, we should use it instead
+    // For now manually execute a GET if the operation was a mutation and not a deletion
+    if (method === "POST" || method === "PATCH" || method === "PUT") {
+      return armRequest({
+        host,
+        path,
+        apiVersion,
+        method: "GET"
+      });
+    }
+  }
+
+  const responseBody = (await response.json()) as T;
+  return responseBody;
+}
+
+const SUCCEEDED = "Succeeded" as const;
+const FAILED = "Failed" as const;
+const CANCELED = "Canceled" as const;
+
+type Status = typeof SUCCEEDED | typeof FAILED | typeof CANCELED;
+
+interface OperationResponse {
+  status: Status;
+  error: unknown;
+}
+
+async function getOperationStatus(operationStatusUrl: string) {
+  const response = await window.fetch(operationStatusUrl, {
+    headers: {
+      Authorization: CosmosClient.authorizationToken()
+    }
+  });
+  if (!response.ok) {
+    const errorResponse = (await response.json()) as ErrorResponse;
+    const error = new Error(errorResponse.error?.message) as ARMError;
+    error.code = errorResponse.error.code;
+    throw new AbortError(error);
+  }
+  const body = (await response.json()) as OperationResponse;
+  const status = body.status;
+  if (status === SUCCEEDED) {
+    return;
+  }
+  if (status === CANCELED || status === FAILED) {
+    const errorMessage = body.error ? JSON.stringify(body.error) : "Operation could not be completed";
+    const error = new Error(errorMessage);
+    throw new AbortError(error);
+  }
+  throw new Error(`Operation Response: ${JSON.stringify(body)}. Retrying.`);
+}