Fixed setup save queries within serverless accounts (#1626)

* Fixed setup save queries within serverless accounts

* Fixed format

* ran prettier

src/Common/QueriesClient.ts

@@ -13,6 +13,7 @@ import { createDocument } from "./dataAccess/createDocument";
 import { deleteDocument } from "./dataAccess/deleteDocument";
 import { queryDocuments } from "./dataAccess/queryDocuments";
 import { handleError } from "./ErrorHandlingUtils";
+import { isServerlessAccount } from "Utils/CapabilityUtils";
 
 export class QueriesClient {
   private static readonly PartitionKey: DataModels.PartitionKey = {
@@ -32,25 +33,36 @@ export class QueriesClient {
     }
 
     const clearMessage = NotificationConsoleUtils.logConsoleProgress("Setting up account for saving queries");
-    return createCollection({
+
-      collectionId: SavedQueries.CollectionName,
+    if (isServerlessAccount()) {
-      createNewDatabase: true,
+      return createCollection({
-      databaseId: SavedQueries.DatabaseName,
+        collectionId: SavedQueries.CollectionName,
-      partitionKey: QueriesClient.PartitionKey,
+        createNewDatabase: true,
-      offerThroughput: SavedQueries.OfferThroughput,
+        databaseId: SavedQueries.DatabaseName,
-      databaseLevelThroughput: false,
+        partitionKey: QueriesClient.PartitionKey,
-    })
+        databaseLevelThroughput: false,
-      .then(
+      });
-        (collection: DataModels.Collection) => {
+    } else {
-          NotificationConsoleUtils.logConsoleInfo("Successfully set up account for saving queries");
+      return createCollection({
-          return Promise.resolve(collection);
+        collectionId: SavedQueries.CollectionName,
-        },
+        createNewDatabase: true,
-        (error: any) => {
+        databaseId: SavedQueries.DatabaseName,
-          handleError(error, "setupQueriesCollection", "Failed to set up account for saving queries");
+        partitionKey: QueriesClient.PartitionKey,
-          return Promise.reject(error);
+        offerThroughput: SavedQueries.OfferThroughput,
-        },
+        databaseLevelThroughput: false,
-      )
+      })
-      .finally(() => clearMessage());
+        .then(
+          (collection: DataModels.Collection) => {
+            NotificationConsoleUtils.logConsoleInfo("Successfully set up account for saving queries");
+            return Promise.resolve(collection);
+          },
+          (error: any) => {
+            handleError(error, "setupQueriesCollection", "Failed to set up account for saving queries");
+            return Promise.reject(error);
+          },
+        )
+        .finally(() => clearMessage());
+    }
   }
 
   public async saveQuery(query: DataModels.Query): Promise<void> {

src/Contracts/DataModels.ts

@@ -318,7 +318,7 @@ export interface CreateCollectionParams {
   collectionId: string;
   databaseId: string;
   databaseLevelThroughput: boolean;
-  offerThroughput: number;
+  offerThroughput?: number;
   analyticalStorageTtl?: number;
   autoPilotMaxThroughput?: number;
   indexingPolicy?: IndexingPolicy;

src/Explorer/Panes/SaveQueryPane/SaveQueryPane.tsx

@@ -113,6 +113,7 @@ export const SaveQueryPane: FunctionComponent<SaveQueryPaneProps> = ({
         },
         startKey,
       );
+      closeSidePanel();
     } catch (error) {
       const errorMessage = getErrorMessage(error);
       traceFailure(