diff --git a/tests/test_octodns_provider_azuredns.py b/tests/test_octodns_provider_azuredns.py index 030bf9f..8f840f0 100644 --- a/tests/test_octodns_provider_azuredns.py +++ b/tests/test_octodns_provider_azuredns.py @@ -375,7 +375,7 @@ class Test_CheckAzureAlias(TestCase): class TestAzureDnsProvider(TestCase): def _provider(self): return self._get_provider('mock_spc', 'mock_dns_client') - + @patch('octodns.provider.azuredns.DnsManagementClient') @patch('octodns.provider.azuredns.ServicePrincipalCredentials') def _get_provider(self, mock_spc, mock_dns_client): @@ -393,9 +393,7 @@ class TestAzureDnsProvider(TestCase): # Fetch the client to force it to load the creds client = provider._dns_client return provider - - @patch('octodns.provider.azuredns.DnsManagementClient') - @patch('octodns.provider.azuredns.ServicePrincipalCredentials') + def test_populate_records(self): provider = self._get_provider() @@ -506,9 +504,7 @@ class TestAzureDnsProvider(TestCase): exists = provider.populate(zone) self.assertTrue(exists) self.assertEquals(len(zone.records), 17) - - @patch('octodns.provider.azuredns.DnsManagementClient') - @patch('octodns.provider.azuredns.ServicePrincipalCredentials') + def test_populate_zone(self): provider = self._get_provider() @@ -519,9 +515,7 @@ class TestAzureDnsProvider(TestCase): provider._populate_zones() self.assertEquals(len(provider._azure_zones), 1) - - @patch('octodns.provider.azuredns.DnsManagementClient') - @patch('octodns.provider.azuredns.ServicePrincipalCredentials') + def test_bad_zone_response(self): provider = self._get_provider() @@ -548,8 +542,6 @@ class TestAzureDnsProvider(TestCase): self.assertEquals(19, provider.apply(Plan(zone, zone, deletes, True))) - @patch('octodns.provider.azuredns.DnsManagementClient') - @patch('octodns.provider.azuredns.ServicePrincipalCredentials') def test_create_zone(self): provider = self._get_provider() @@ -566,8 +558,6 @@ class TestAzureDnsProvider(TestCase): self.assertEquals(19, provider.apply(Plan(None, desired, changes, True))) - @patch('octodns.provider.azuredns.DnsManagementClient') - @patch('octodns.provider.azuredns.ServicePrincipalCredentials') def test_check_zone_no_create(self): provider = self._get_provider()