Add complete flag to delete_role

azure/servicemanagement/servicemanagementservice.py

@@ -1517,7 +1517,7 @@ def update_role(self, service_name, deployment_name, role_name,
                 provision_guest_agent),
             async=True)
 
-    def delete_role(self, service_name, deployment_name, role_name):
+    def delete_role(self, service_name, deployment_name, role_name, complete = False):
         '''
         Deletes the specified virtual machine.
 
@@ -1527,13 +1527,21 @@ def delete_role(self, service_name, deployment_name, role_name):
             The name of the deployment.
         role_name:
             The name of the role.
+        complete:
+            True if all OS/data disks and the source blobs for the disks should
+            also be deleted from storage.
         '''
         _validate_not_none('service_name', service_name)
         _validate_not_none('deployment_name', deployment_name)
         _validate_not_none('role_name', role_name)
-        return self._perform_delete(
-            self._get_role_path(service_name, deployment_name, role_name),
-            async=True)
+
+        path = self._get_role_path(service_name, deployment_name, role_name)
+
+        if complete == True:
+            path = path +'?comp=media'
+
+        return self._perform_delete(path,
+                                    async=True)
 
     def capture_role(self, service_name, deployment_name, role_name,
                      post_capture_action, target_image_name,

tests/test_servicemanagementservice.py

@@ -1854,7 +1854,7 @@ def test_delete_role(self):
         self._add_role_windows(service_name, deployment_name, role_name2, '59914')
 
         # Act
-        result = self.sms.delete_role(service_name, deployment_name, role_name2)
+        result = self.sms.delete_role(service_name, deployment_name, role_name2, True)
         self._wait_for_async(result.request_id)
 
         # Assert