adding back simulation switch

src/cosmosdb-preview/azext_cosmosdb_preview/_params.py

@@ -325,9 +325,11 @@ def load_arguments(self, _):
         c.argument('account_name', account_name_type, id_part=None, required=True, help='Name of the CosmosDB database account')
         c.argument('database_name', database_name_type, required=True, help='Name of the CosmosDB database name')
         c.argument('container_name', options_list=['--name', '-n'], required=True, help='Name of the CosmosDB collection')
+        c.argument('is-simulation', options_list=['--is_simulation', '-s'], arg_type=get_three_state_flag(), help="This flag is set by default, merge will always be a simulation unless this flag is set to false.")
 
     # mongodb collection partition merge
     with self.argument_context('cosmosdb mongodb collection merge') as c:
         c.argument('account_name', account_name_type, id_part=None, required=True, help='Name of the CosmosDB database account')
         c.argument('database_name', database_name_type, required=True, help='Name of the mongoDB database')
-        c.argument('container_name', options_list=['--name', '-n'], required=True, help='Name of the mongoDB collection')
+        c.argument('container_name', options_list=['--name', '-n'], required=True, help='Name of the mongoDB collection')
+        c.argument('is-simulation', options_list=['--is_simulation', '-s'], arg_type=get_three_state_flag(), help="This flag is set by default, merge will always be a simulation unless this flag is set to false.")

src/cosmosdb-preview/azext_cosmosdb_preview/custom.py

@@ -1154,16 +1154,17 @@ def cli_begin_list_sql_container_partition_merge(client,
                                                  resource_group_name,
                                                  account_name,
                                                  database_name,
-                                                 container_name):
+                                                 container_name,
+                                                 is_simulation=True):
 
     try:
         client.get_sql_container(resource_group_name, account_name, database_name, container_name)
     except Exception as ex:
         if ex.error.code == "NotFound":
-            raise CLIError("(NotFound) Container with name '{container_name}' in database '{database_name}' could not be found.")
+            raise CLIError("(NotFound) Container with name '{}' in database '{}' could not be found.".format(container_name, database_name))
         raise CLIError("{}".format(str(ex)))
 
-    mergeParameters = MergeParameters(is_dry_run=False)
+    mergeParameters = MergeParameters(is_dry_run=is_simulation)
 
     async_partition_merge_result = client.begin_list_sql_container_partition_merge(resource_group_name=resource_group_name,
                                                                                    account_name=account_name,
@@ -1178,16 +1179,17 @@ def cli_begin_list_mongo_db_collection_partition_merge(client,
                                                        resource_group_name,
                                                        account_name,
                                                        database_name,
-                                                       container_name):
+                                                       container_name,
+                                                       is_simulation=True):
 
     try:
         client.get_mongo_db_collection(resource_group_name, account_name, database_name, container_name)
     except Exception as ex:
         if ex.error.code == "NotFound":
-            raise CLIError("(NotFound) collection with name '{container_name}' in mongodb '{database_name}' could not be found.")
+            raise CLIError("(NotFound) collection with name '{}' in mongodb '{}' could not be found.".format(container_name, database_name))
         raise CLIError("{}".format(str(ex)))
 
-    mergeParameters = MergeParameters(is_dry_run=False)
+    mergeParameters = MergeParameters(is_dry_run=is_simulation)
 
     async_partition_merge_result = client.begin_list_mongo_db_collection_partition_merge(resource_group_name=resource_group_name,
                                                                                          account_name=account_name,