diff --git a/src/azure-cli/azure/cli/command_modules/acs/_roleassignments.py b/src/azure-cli/azure/cli/command_modules/acs/_roleassignments.py index 492475f7b3a..98d6e51a2a1 100644 --- a/src/azure-cli/azure/cli/command_modules/acs/_roleassignments.py +++ b/src/azure-cli/azure/cli/command_modules/acs/_roleassignments.py @@ -125,8 +125,9 @@ def add_role_assignment(cmd, role, service_principal_msi_id, is_service_principa logger.error(str(ex)) time.sleep(delay + delay * x) else: + hook.end(message="AAD role propagation failed", value=1.0, total_val=1.0) return False - hook.add(message="AAD role propagation done", value=1.0, total_val=1.0) + hook.end(message="AAD role propagation done", value=1.0, total_val=1.0) logger.info("AAD role propagation done") return True @@ -205,8 +206,9 @@ def delete_role_assignments(cli_ctx, role, service_principal, delay=2, scope=Non logger.error(str(ex)) time.sleep(delay + delay * x) else: + hook.end(message="AAD role deletion failed", value=1.0, total_val=1.0) return False - hook.add(message="AAD role deletion done", value=1.0, total_val=1.0) + hook.end(message="AAD role deletion done", value=1.0, total_val=1.0) logger.info("AAD role deletion done") return True