jatin 9 лет назад
Родитель
Сommit
97d47fd849

+ 4 - 2
cloudbridge/cloud/providers/azure/mock_azure_client.py

@@ -171,8 +171,10 @@ class MockAzureClient:
     def create_security_group_rule(self, security_group,
                                    rule_name, parameters):
         new_sg_rule = SecurityRule(protocol=parameters['protocol'],
-                                   source_address_prefix=parameters['source_address_prefix'],
-                                   destination_address_prefix=parameters['destination_address_prefix'],
+                                   source_address_prefix=parameters
+                                   ['source_address_prefix'],
+                                   destination_address_prefix=parameters
+                                   ['destination_address_prefix'],
                                    access="Allow", direction="Inbound")
         new_sg_rule.name = "rule1"
         new_sg_rule.id = "r1"

+ 4 - 3
cloudbridge/cloud/providers/azure/services.py

@@ -15,8 +15,8 @@ from msrestazure.azure_exceptions import CloudError
 from .resources import AzureBucket, AzureSecurityGroup, \
     AzureSnapshot, AzureVolume, \
     NETWORK_SECURITY_GROUP_RESOURCE_ID, SECURITY_GROUP_NAME, \
-    VOLUME_NAME, VOLUME_RESOURCE_ID, AzureSnapshot, \
-    SNAPSHOT_RESOURCE_ID, SNAPSHOT_NAME
+    SNAPSHOT_NAME, SNAPSHOT_RESOURCE_ID, VOLUME_NAME, \
+    VOLUME_RESOURCE_ID
 
 log = logging.getLogger(__name__)
 
@@ -213,7 +213,8 @@ class AzureSnapshotService(BaseSnapshotService):
     def get(self, ss_id):
         try:
             params = azure_helpers.parse_url(SNAPSHOT_RESOURCE_ID, ss_id)
-            snapshot = self.provider.azure_client.get_snapshot(params.get(SNAPSHOT_NAME))
+            snapshot = self.provider.azure_client. \
+                get_snapshot(params.get(SNAPSHOT_NAME))
             return AzureSnapshot(self.provider, snapshot)
         except CloudError as cloudError:
             log.exception(cloudError.message)

+ 0 - 1
test/test_azure_security_service.py

@@ -108,7 +108,6 @@ class AzureSecurityServiceTestCase(ProviderTestBase):
         second_rule = cb.add_rule('*', '25', '100')
         self.assertEqual(first_rule, second_rule)
 
-
     @helpers.skipIfNoService(['security.security_groups'])
     def test_azure_security_group_rule_delete(self):
         list = self.provider.security.security_groups.list()