Respect suppressNotification when deleting CosmosDBAccount (#2181)

microsoft · Sep 21, 2023 · 266299d · 266299d
1 parent 9bbf104
commit 266299d
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/src/commands/deleteDatabaseAccount/deleteCosmosDBAccount.ts b/src/commands/deleteDatabaseAccount/deleteCosmosDBAccount.ts
@@ -17,11 +17,16 @@ export async function deleteCosmosDBAccount(context: IDeleteWizardContext, node:
     const client: CosmosDBManagementClient = await createCosmosDBClient([context, node.subscription]);
     const resourceGroup: string = getResourceGroupFromId(node.fullId);
     const accountName: string = getDatabaseAccountNameFromId(node.fullId);
-    const deletingMessage: string = `Deleting account "${accountName}"...`;
-    await vscode.window.withProgress({ location: vscode.ProgressLocation.Notification, title: deletingMessage }, async () => {
-        await client.databaseAccounts.beginDeleteAndWait(resourceGroup, accountName);
-        const deleteMessage: string = localize("deleteAccountMsg", `Successfully deleted account "{0}".`, accountName);
-        void vscode.window.showInformationMessage(deleteMessage);
-        ext.outputChannel.appendLog(deleteMessage);
-    });
+    const deletePromise = client.databaseAccounts.beginDeleteAndWait(resourceGroup, accountName);
+    if (!context.suppressNotification) {
+        const deletingMessage: string = `Deleting account "${accountName}"...`;
+        await vscode.window.withProgress({ location: vscode.ProgressLocation.Notification, title: deletingMessage }, async () => {
+            await deletePromise;
+            const deleteMessage: string = localize("deleteAccountMsg", `Successfully deleted account "{0}".`, accountName);
+            void vscode.window.showInformationMessage(deleteMessage);
+            ext.outputChannel.appendLog(deleteMessage);
+        });
+    } else {
+        await deletePromise;
+    }
 }