def add_txt_record(self, domain, record_content, record_ttl): """ Add a TXT record using the supplied information. :param str domain: The fqdn (typically beginning with '_acme-challenge.'). :param str record_content: The record content (typically the challenge validation). :param int record_ttl: The record TTL (number of seconds that the record may be cached). :raises certbot.errors.PluginError: if an error occurs communicating with the Azure API """ try: zone = self._find_managed_zone(domain) relative_record_name = ".".join(domain.split('.')[0:-len(zone.split('.'))]) try: record = self.dns_client.record_sets.get(self.resource_group, zone, relative_record_name, 'TXT') record.txt_records.append(TxtRecord(value=[record_content])) except azure.core.exceptions.ResourceNotFoundError: record = RecordSet(ttl=record_ttl, txt_records=[TxtRecord(value=[record_content])]) self.dns_client.record_sets.create_or_update( self.resource_group, zone, relative_record_name, 'TXT', record ) except CloudError as e: logger.error('Encountered error adding TXT record: %s', e) raise errors.PluginError('Error communicating with the Azure DNS API: {0}'.format(e))
def test_populate_records(self): provider = self._get_provider() rs = [] rs.append(RecordSet(name='a1', ttl=0, type='A', arecords=[ARecord('1.1.1.1')])) rs.append(RecordSet(name='a2', ttl=1, type='A', arecords=[ARecord('1.1.1.1'), ARecord('2.2.2.2')])) rs.append(RecordSet(name='aaaa1', ttl=2, type='AAAA', aaaa_records=[AaaaRecord('1:1ec:1::1')])) rs.append(RecordSet(name='aaaa2', ttl=3, type='AAAA', aaaa_records=[AaaaRecord('1:1ec:1::1'), AaaaRecord('1:1ec:1::2')])) rs.append(RecordSet(name='cname1', ttl=4, type='CNAME', cname_record=CnameRecord('cname.unit.test.'))) rs.append(RecordSet(name='cname2', ttl=5, type='CNAME', cname_record=None)) rs.append(RecordSet(name='mx1', ttl=6, type='MX', mx_records=[MxRecord(10, 'mx1.unit.test.')])) rs.append(RecordSet(name='mx2', ttl=7, type='MX', mx_records=[MxRecord(10, 'mx1.unit.test.'), MxRecord(11, 'mx2.unit.test.')])) rs.append(RecordSet(name='ns1', ttl=8, type='NS', ns_records=[NsRecord('ns1.unit.test.')])) rs.append(RecordSet(name='ns2', ttl=9, type='NS', ns_records=[NsRecord('ns1.unit.test.'), NsRecord('ns2.unit.test.')])) rs.append(RecordSet(name='ptr1', ttl=10, type='PTR', ptr_records=[PtrRecord('ptr1.unit.test.')])) rs.append(RecordSet(name='ptr2', ttl=11, type='PTR', ptr_records=[PtrRecord(None)])) rs.append(RecordSet(name='_srv1._tcp', ttl=12, type='SRV', srv_records=[SrvRecord(1, 2, 3, '1unit.tests.')])) rs.append(RecordSet(name='_srv2._tcp', ttl=13, type='SRV', srv_records=[SrvRecord(1, 2, 3, '1unit.tests.'), SrvRecord(4, 5, 6, '2unit.tests.')])) rs.append(RecordSet(name='txt1', ttl=14, type='TXT', txt_records=[TxtRecord('sample text1')])) rs.append(RecordSet(name='txt2', ttl=15, type='TXT', txt_records=[TxtRecord('sample text1'), TxtRecord('sample text2')])) rs.append(RecordSet(name='', ttl=16, type='SOA', soa_record=[SoaRecord()])) record_list = provider._dns_client.record_sets.list_by_dns_zone record_list.return_value = rs exists = provider.populate(zone) self.assertTrue(exists) self.assertEquals(len(zone.records), 16)
def test_cleanup_existing(self): self.mock_client.record_sets.get.return_value = RecordSet( txt_records=[TxtRecord(value=['someexistingkey'])]) # Extract zone TXT record name and value zone1_req = SINGLE_DOMAIN[0] zone1_domain_name = zone1_req.validation_domain_name(zone1_req.domain) zone1_relative_record = zone1_domain_name.replace('example.com', '').strip('.') zone1_key = zone1_req.validation(zone1_req.account_key) # _attempt_cleanup | pylint: disable=protected-access self.auth._attempt_cleanup = True self.auth.cleanup(SINGLE_DOMAIN) # Check azure client call counts self.assertEqual(self.mock_client.record_sets.get.call_count, 1) self.assertEqual(self.mock_client.record_sets.delete.call_count, 0) self.assertEqual( self.mock_client.record_sets.create_or_update.call_count, 1) # Check recordset is updated to not include key zone1_call = self.mock_client.record_sets.create_or_update.call_args_list[ 0] self.assertEqual(zone1_call[1]['zone_name'], "example.com") self.assertEqual(zone1_call[1]['record_type'], "TXT") self.assertEqual(zone1_call[1]['relative_record_set_name'], zone1_relative_record) zone1_txt_records = zone1_call[1]['parameters'].txt_records self.assertEqual(len(zone1_txt_records), 1) txt_values = zone1_txt_records[0].value self.assertNotIn(zone1_key, txt_values) self.assertIn('someexistingkey', txt_values)
def test_perform_existing(self): self.mock_client.record_sets.get.return_value = RecordSet( txt_records=[TxtRecord(value=['someexistingkey'])]) # Extract zone TXT record name and value zone1_req = SINGLE_DOMAIN[0] zone1_domain_name = zone1_req.validation_domain_name(zone1_req.domain) zone1_relative_record = zone1_domain_name.replace('example.com', '').strip('.') zone1_key = zone1_req.validation(zone1_req.account_key) self.auth.perform(SINGLE_DOMAIN) # Check azure client call counts self.assertEqual(self.mock_client.record_sets.get.call_count, 1) self.assertEqual( self.mock_client.record_sets.create_or_update.call_count, 1) # expected = [ self.mock_client.record_sets.create_or_update.call( resource_group_name='dns1', zone_name='example.com', relative_record_set_name=zone1_domain_name, parameters=RecordSet(txt_records=TxtRecord( value=[zone1_key, 'someexistingkey']))) ] zone1_call = self.mock_client.record_sets.create_or_update.call_args_list[ 0] self.assertEqual(zone1_call[1]['zone_name'], "example.com") self.assertEqual(zone1_call[1]['record_type'], "TXT") self.assertEqual(zone1_call[1]['relative_record_set_name'], zone1_relative_record) zone1_txt_records = zone1_call[1]['parameters'].txt_records self.assertEqual(len(zone1_txt_records), 1) txt_values = zone1_txt_records[0].value self.assertIn(zone1_key, txt_values) self.assertIn('someexistingkey', txt_values)
def _cleanup(self, domain, validation_name, validation): if self.credential is None: self._setup_credentials() azure_domain, subscription_id, resource_group_name = self._get_ids_for_domain( domain) relative_validation_name = self._get_relative_domain( validation_name, azure_domain) client = self._get_azure_client(subscription_id) txt_value = set() try: existing_rr = client.record_sets.get( resource_group_name=resource_group_name, zone_name=azure_domain, relative_record_set_name=relative_validation_name, record_type='TXT') for record in existing_rr.txt_records: for value in record.value: txt_value.add(value) except CloudError as err: if err.status_code != 404: # Ignore RR not found raise errors.PluginError( 'Failed to check TXT record for domain ' '{}, error: {}'.format(domain, err)) txt_value -= {validation} try: if txt_value: client.record_sets.create_or_update( resource_group_name=resource_group_name, zone_name=azure_domain, relative_record_set_name=relative_validation_name, record_type='TXT', parameters=RecordSet( ttl=self.ttl, txt_records=[TxtRecord(value=list(txt_value))])) else: client.record_sets.delete( resource_group_name=resource_group_name, zone_name=azure_domain, relative_record_set_name=relative_validation_name, record_type='TXT') except CloudError as err: if err.status_code != 404: # Ignore RR not found raise errors.PluginError( 'Failed to remove TXT record for domain ' '{}, error: {}'.format(domain, err))
def _setup_dns_challenge( self, record_set_name: str, value: str, ): self._dns_client.record_sets.create_or_update( resource_group_name=self.options.azure_dns_zone_resource_group, zone_name=self.options.azure_dns_zone, relative_record_set_name=record_set_name, record_type=RecordType.TXT, parameters=RecordSet( ttl=ACME_CHALLENGE_TXT_RECORD_TTL, txt_records=[TxtRecord(value=[value])], ), )
def _perform(self, domain, validation_name, validation): azure_domain, subscription_id, resource_group_name = self._get_ids_for_domain( domain) client = self._get_azure_client(subscription_id) relative_validation_name = self._get_relative_domain( validation_name, azure_domain) # Check to see if there are any existing TXT validation record values txt_value = {validation} try: existing_rr = client.record_sets.get( resource_group_name=resource_group_name, zone_name=azure_domain, relative_record_set_name=relative_validation_name, record_type='TXT') for record in existing_rr.txt_records: for value in record.value: txt_value.add(value) except CloudError as err: if err.status_code != 404: # Ignore RR not found raise errors.PluginError( 'Failed to check TXT record for domain ' '{}, error: {}'.format(domain, err)) try: client.record_sets.create_or_update( resource_group_name=resource_group_name, zone_name=azure_domain, relative_record_set_name=relative_validation_name, record_type='TXT', parameters=RecordSet( ttl=self.ttl, txt_records=[TxtRecord(value=list(txt_value))])) except CloudError as err: raise errors.PluginError('Failed to add TXT record to domain ' '{}, error: {}'.format(domain, err))
azure_records.append(_base9) _base10 = _AzureRecord('TestAzure', octo_records[10]) _base10.zone_name = 'unit.tests' _base10.relative_record_set_name = '_srv2._tcp' _base10.record_type = 'SRV' _base10.params['ttl'] = 7 _base10.params['srv_records'] = [SrvRecord(12, 17, 1, 'srvfoo.unit.tests.')] azure_records.append(_base10) _base11 = _AzureRecord('TestAzure', octo_records[11]) _base11.zone_name = 'unit.tests' _base11.relative_record_set_name = 'txt1' _base11.record_type = 'TXT' _base11.params['ttl'] = 8 _base11.params['txt_records'] = [TxtRecord(['txt singleton test'])] azure_records.append(_base11) _base12 = _AzureRecord('TestAzure', octo_records[12]) _base12.zone_name = 'unit.tests' _base12.relative_record_set_name = 'txt2' _base12.record_type = 'TXT' _base12.params['ttl'] = 9 _base12.params['txt_records'] = [TxtRecord(['txt multiple test']), TxtRecord(['txt multiple test 2'])] azure_records.append(_base12) class Test_AzureRecord(TestCase): def test_azure_record(self): assert(len(azure_records) == len(octo_records))
def test_populate_records(self): provider = self._get_provider() rs = [] recordSet = RecordSet(arecords=[ARecord('1.1.1.1')]) recordSet.name, recordSet.ttl, recordSet.type = 'a1', 0, 'A' rs.append(recordSet) recordSet = RecordSet( arecords=[ARecord('1.1.1.1'), ARecord('2.2.2.2')]) recordSet.name, recordSet.ttl, recordSet.type = 'a2', 1, 'A' rs.append(recordSet) recordSet = RecordSet(aaaa_records=[AaaaRecord('1:1ec:1::1')]) recordSet.name, recordSet.ttl, recordSet.type = 'aaaa1', 2, 'AAAA' rs.append(recordSet) recordSet = RecordSet( aaaa_records=[AaaaRecord('1:1ec:1::1'), AaaaRecord('1:1ec:1::2')]) recordSet.name, recordSet.ttl, recordSet.type = 'aaaa2', 3, 'AAAA' rs.append(recordSet) recordSet = RecordSet(cname_record=CnameRecord('cname.unit.test.')) recordSet.name, recordSet.ttl, recordSet.type = 'cname1', 4, 'CNAME' rs.append(recordSet) recordSet = RecordSet(cname_record=None) recordSet.name, recordSet.ttl, recordSet.type = 'cname2', 5, 'CNAME' rs.append(recordSet) recordSet = RecordSet(mx_records=[MxRecord(10, 'mx1.unit.test.')]) recordSet.name, recordSet.ttl, recordSet.type = 'mx1', 6, 'MX' rs.append(recordSet) recordSet = RecordSet(mx_records=[ MxRecord(10, 'mx1.unit.test.'), MxRecord(11, 'mx2.unit.test.') ]) recordSet.name, recordSet.ttl, recordSet.type = 'mx2', 7, 'MX' rs.append(recordSet) recordSet = RecordSet(ns_records=[NsRecord('ns1.unit.test.')]) recordSet.name, recordSet.ttl, recordSet.type = 'ns1', 8, 'NS' rs.append(recordSet) recordSet = RecordSet(ns_records=[ NsRecord('ns1.unit.test.'), NsRecord('ns2.unit.test.') ]) recordSet.name, recordSet.ttl, recordSet.type = 'ns2', 9, 'NS' rs.append(recordSet) recordSet = RecordSet(ptr_records=[PtrRecord('ptr1.unit.test.')]) recordSet.name, recordSet.ttl, recordSet.type = 'ptr1', 10, 'PTR' rs.append(recordSet) recordSet = RecordSet(ptr_records=[PtrRecord(None)]) recordSet.name, recordSet.ttl, recordSet.type = 'ptr2', 11, 'PTR' rs.append(recordSet) recordSet = RecordSet(srv_records=[SrvRecord(1, 2, 3, '1unit.tests.')]) recordSet.name, recordSet.ttl, recordSet.type = '_srv1._tcp', 12, 'SRV' rs.append(recordSet) recordSet = RecordSet(srv_records=[ SrvRecord(1, 2, 3, '1unit.tests.'), SrvRecord(4, 5, 6, '2unit.tests.') ]) recordSet.name, recordSet.ttl, recordSet.type = '_srv2._tcp', 13, 'SRV' rs.append(recordSet) recordSet = RecordSet(txt_records=[TxtRecord('sample text1')]) recordSet.name, recordSet.ttl, recordSet.type = 'txt1', 14, 'TXT' rs.append(recordSet) recordSet = RecordSet( txt_records=[TxtRecord('sample text1'), TxtRecord('sample text2')]) recordSet.name, recordSet.ttl, recordSet.type = 'txt2', 15, 'TXT' rs.append(recordSet) recordSet = RecordSet(soa_record=[SoaRecord()]) recordSet.name, recordSet.ttl, recordSet.type = '', 16, 'SOA' rs.append(recordSet) record_list = provider._dns_client.record_sets.list_by_dns_zone record_list.return_value = rs exists = provider.populate(zone) self.assertTrue(exists) self.assertEquals(len(zone.records), 16)
def test_populate_records(self): provider = self._get_provider() rs = [] recordSet = RecordSet(arecords=[ARecord(ipv4_address='1.1.1.1')]) recordSet.name, recordSet.ttl, recordSet.type = 'a1', 0, 'A' recordSet.target_resource = SubResource() rs.append(recordSet) recordSet = RecordSet(arecords=[ ARecord(ipv4_address='1.1.1.1'), ARecord(ipv4_address='2.2.2.2') ]) recordSet.name, recordSet.ttl, recordSet.type = 'a2', 1, 'A' recordSet.target_resource = SubResource() rs.append(recordSet) aaaa1 = AaaaRecord(ipv6_address='1:1ec:1::1') recordSet = RecordSet(aaaa_records=[aaaa1]) recordSet.name, recordSet.ttl, recordSet.type = 'aaaa1', 2, 'AAAA' recordSet.target_resource = SubResource() rs.append(recordSet) aaaa2 = AaaaRecord(ipv6_address='1:1ec:1::2') recordSet = RecordSet(aaaa_records=[aaaa1, aaaa2]) recordSet.name, recordSet.ttl, recordSet.type = 'aaaa2', 3, 'AAAA' recordSet.target_resource = SubResource() rs.append(recordSet) recordSet = RecordSet(caa_records=[ CaaRecord(flags=0, tag='issue', value='caa1.unit.tests') ]) recordSet.name, recordSet.ttl, recordSet.type = 'caa1', 4, 'CAA' rs.append(recordSet) recordSet = RecordSet(caa_records=[ CaaRecord(flags=0, tag='issue', value='caa1.unit.tests'), CaaRecord(flags=0, tag='issue', value='caa2.unit.tests') ]) recordSet.name, recordSet.ttl, recordSet.type = 'caa2', 4, 'CAA' rs.append(recordSet) cname1 = CnameRecord(cname='cname.unit.test.') recordSet = RecordSet(cname_record=cname1) recordSet.name, recordSet.ttl, recordSet.type = 'cname1', 5, 'CNAME' recordSet.target_resource = SubResource() rs.append(recordSet) recordSet = RecordSet( mx_records=[MxRecord(preference=10, exchange='mx1.unit.test.')]) recordSet.name, recordSet.ttl, recordSet.type = 'mx1', 7, 'MX' rs.append(recordSet) recordSet = RecordSet(mx_records=[ MxRecord(preference=10, exchange='mx1.unit.test.'), MxRecord(preference=11, exchange='mx2.unit.test.') ]) recordSet.name, recordSet.ttl, recordSet.type = 'mx2', 8, 'MX' rs.append(recordSet) recordSet = RecordSet(ns_records=[NsRecord(nsdname='ns1.unit.test.')]) recordSet.name, recordSet.ttl, recordSet.type = 'ns1', 9, 'NS' rs.append(recordSet) recordSet = RecordSet(ns_records=[ NsRecord(nsdname='ns1.unit.test.'), NsRecord(nsdname='ns2.unit.test.') ]) recordSet.name, recordSet.ttl, recordSet.type = 'ns2', 10, 'NS' rs.append(recordSet) ptr1 = PtrRecord(ptrdname='ptr1.unit.test.') recordSet = RecordSet(ptr_records=[ptr1]) recordSet.name, recordSet.ttl, recordSet.type = 'ptr1', 11, 'PTR' rs.append(recordSet) recordSet = RecordSet(srv_records=[ SrvRecord(priority=1, weight=2, port=3, target='1unit.tests.') ]) recordSet.name, recordSet.ttl, recordSet.type = '_srv1._tcp', 13, 'SRV' rs.append(recordSet) recordSet = RecordSet(srv_records=[ SrvRecord(priority=1, weight=2, port=3, target='1unit.tests.'), SrvRecord(priority=4, weight=5, port=6, target='2unit.tests.') ]) recordSet.name, recordSet.ttl, recordSet.type = '_srv2._tcp', 14, 'SRV' rs.append(recordSet) recordSet = RecordSet(txt_records=[TxtRecord(value='sample text1')]) recordSet.name, recordSet.ttl, recordSet.type = 'txt1', 15, 'TXT' recordSet.target_resource = SubResource() rs.append(recordSet) recordSet = RecordSet(txt_records=[ TxtRecord(value='sample text1'), TxtRecord(value='sample text2') ]) recordSet.name, recordSet.ttl, recordSet.type = 'txt2', 16, 'TXT' recordSet.target_resource = SubResource() rs.append(recordSet) recordSet = RecordSet(soa_record=[SoaRecord()]) recordSet.name, recordSet.ttl, recordSet.type = '', 17, 'SOA' rs.append(recordSet) long_txt = "v=spf1 ip4:10.10.0.0/24 ip4:10.10.1.0/24 ip4:10.10.2.0/24" long_txt += " ip4:10.10.3.0/24 ip4:10.10.4.0/24 ip4:10.10.5.0/24 " long_txt += " 10.6.0/24 ip4:10.10.7.0/24 ip4:10.10.8.0/24 " long_txt += " ip4:10.10.10.0/24 ip4:10.10.11.0/24 ip4:10.10.12.0/24" long_txt += " ip4:10.10.13.0/24 ip4:10.10.14.0/24 ip4:10.10.15.0/24" long_txt += " ip4:10.10.16.0/24 ip4:10.10.17.0/24 ip4:10.10.18.0/24" long_txt += " ip4:10.10.19.0/24 ip4:10.10.20.0/24 ~all" recordSet = RecordSet(txt_records=[ TxtRecord(value='sample value1'), TxtRecord(value=long_txt) ]) recordSet.name, recordSet.ttl, recordSet.type = 'txt3', 18, 'TXT' recordSet.target_resource = SubResource() rs.append(recordSet) record_list = provider._dns_client.record_sets.list_by_dns_zone record_list.return_value = rs zone_list = provider._dns_client.zones.list_by_resource_group zone_list.return_value = [zone] exists = provider.populate(zone) self.assertEquals(len(zone.records), 17) self.assertTrue(exists)
_base15 = _AzureRecord('TestAzure', octo_records[15]) _base15.zone_name = 'unit.tests' _base15.relative_record_set_name = '_srv2._tcp' _base15.record_type = 'SRV' _base15.params['ttl'] = 7 _base15.params['srv_records'] = [ SrvRecord(priority=12, weight=17, port=1, target='srvfoo.unit.tests.') ] azure_records.append(_base15) _base16 = _AzureRecord('TestAzure', octo_records[16]) _base16.zone_name = 'unit.tests' _base16.relative_record_set_name = 'txt1' _base16.record_type = 'TXT' _base16.params['ttl'] = 8 _base16.params['txt_records'] = [TxtRecord(value=['txt singleton test'])] azure_records.append(_base16) _base17 = _AzureRecord('TestAzure', octo_records[17]) _base17.zone_name = 'unit.tests' _base17.relative_record_set_name = 'txt2' _base17.record_type = 'TXT' _base17.params['ttl'] = 9 _base17.params['txt_records'] = [ TxtRecord(value=['txt multiple test']), TxtRecord(value=['txt multiple test 2']) ] azure_records.append(_base17) long_txt_az1 = "v=spf1 ip4:10.10.0.0/24 ip4:10.10.1.0/24 ip4:10.10.2.0/24" long_txt_az1 += " ip4:10.10.3.0/24 ip4:10.10.4.0/24 ip4:10.10.5.0/24 "
def create_current_record(self): # takes in a list of str records, the record type and returns a list of the specific record type retrn_lst = [] if self.record_type == 'A': if len(self.records) == 0: return retrn_lst for each_record in self.records: # we want to append to the final list the object type ARecord where the parameter is a str called each_record representing the user input retrn_lst.append(ARecord(ipv4_address=each_record)) return retrn_lst elif self.record_type == 'AAAA': if len(self.records) == 0: return retrn_lst for each_record_1 in self.records: retrn_lst.append(AaaaRecord(ipv6_address=each_record_1)) return retrn_lst elif self.record_type == 'MX': if len(self.records) == 0: return retrn_lst if len(self.records) != len(self.preference): self.fail( 'You must have an exchange for each preference or vice versa') # if type matches to 'MX', loop through records/exchange and preference, using the same indices to create the object for i in range(len(self.records)): retrn_lst.append( MxRecord(preference=int(self.preference[i]), exchange=self.records[i])) return retrn_lst elif self.record_type == 'NS': if len(self.records) == 0: return retrn_lst for each_record_2 in self.records: retrn_lst.append(NsRecord(nsdname=each_record_2)) return retrn_lst elif self.record_type == 'TXT': if len(self.records) == 0: return retrn_lst for each_record_3 in self.records: retrn_lst.append(TxtRecord(value=each_record_3)) return retrn_lst elif self.record_type == 'PTR': if len(self.records) == 0: return retrn_lst for each_record_4 in self.records: retrn_lst.append(PtrRecord(ptrdname=each_record_4)) return retrn_lst elif self.record_type == 'SRV': if len(self.records) == 0: return retrn_lst count = len(self.records) if count != len(self.priority) or count != len( self.port) or count != len(self.weight): self.fail( 'You must have a weight, a port and a priority for each target' ) for i in range(len(self.records)): retrn_lst.append( SrvRecord(priority=int(self.priority[i]), weight=int(self.weight[i]), port=int(self.port[i]), target=self.records[i])) return retrn_lst elif self.record_type == 'CNAME': if len(self.records) > 1: self.fail( 'You cannot have more than one record in a single CNAME record set' ) elif len(self.records) == 0: return retrn_lst curr = CnameRecord(cname=self.records[0]) return curr return retrn_lst
def test_populate_records(self): provider = self._get_provider() rs = [] recordSet = RecordSet(arecords=[ARecord(ipv4_address='1.1.1.1')]) recordSet.name, recordSet.ttl, recordSet.type = 'a1', 0, 'A' rs.append(recordSet) recordSet = RecordSet(arecords=[ARecord(ipv4_address='1.1.1.1'), ARecord(ipv4_address='2.2.2.2')]) recordSet.name, recordSet.ttl, recordSet.type = 'a2', 1, 'A' rs.append(recordSet) aaaa1 = AaaaRecord(ipv6_address='1:1ec:1::1') recordSet = RecordSet(aaaa_records=[aaaa1]) recordSet.name, recordSet.ttl, recordSet.type = 'aaaa1', 2, 'AAAA' rs.append(recordSet) aaaa2 = AaaaRecord(ipv6_address='1:1ec:1::2') recordSet = RecordSet(aaaa_records=[aaaa1, aaaa2]) recordSet.name, recordSet.ttl, recordSet.type = 'aaaa2', 3, 'AAAA' rs.append(recordSet) recordSet = RecordSet(caa_records=[CaaRecord(flags=0, tag='issue', value='caa1.unit.tests')]) recordSet.name, recordSet.ttl, recordSet.type = 'caa1', 4, 'CAA' rs.append(recordSet) recordSet = RecordSet(caa_records=[CaaRecord(flags=0, tag='issue', value='caa1.unit.tests'), CaaRecord(flags=0, tag='issue', value='caa2.unit.tests')]) recordSet.name, recordSet.ttl, recordSet.type = 'caa2', 4, 'CAA' rs.append(recordSet) cname1 = CnameRecord(cname='cname.unit.test.') recordSet = RecordSet(cname_record=cname1) recordSet.name, recordSet.ttl, recordSet.type = 'cname1', 5, 'CNAME' rs.append(recordSet) recordSet = RecordSet(cname_record=None) recordSet.name, recordSet.ttl, recordSet.type = 'cname2', 6, 'CNAME' rs.append(recordSet) recordSet = RecordSet(mx_records=[MxRecord(preference=10, exchange='mx1.unit.test.')]) recordSet.name, recordSet.ttl, recordSet.type = 'mx1', 7, 'MX' rs.append(recordSet) recordSet = RecordSet(mx_records=[MxRecord(preference=10, exchange='mx1.unit.test.'), MxRecord(preference=11, exchange='mx2.unit.test.')]) recordSet.name, recordSet.ttl, recordSet.type = 'mx2', 8, 'MX' rs.append(recordSet) recordSet = RecordSet(ns_records=[NsRecord(nsdname='ns1.unit.test.')]) recordSet.name, recordSet.ttl, recordSet.type = 'ns1', 9, 'NS' rs.append(recordSet) recordSet = RecordSet(ns_records=[NsRecord(nsdname='ns1.unit.test.'), NsRecord(nsdname='ns2.unit.test.')]) recordSet.name, recordSet.ttl, recordSet.type = 'ns2', 10, 'NS' rs.append(recordSet) ptr1 = PtrRecord(ptrdname='ptr1.unit.test.') recordSet = RecordSet(ptr_records=[ptr1]) recordSet.name, recordSet.ttl, recordSet.type = 'ptr1', 11, 'PTR' rs.append(recordSet) recordSet = RecordSet(ptr_records=[PtrRecord(ptrdname=None)]) recordSet.name, recordSet.ttl, recordSet.type = 'ptr2', 12, 'PTR' rs.append(recordSet) recordSet = RecordSet(srv_records=[SrvRecord(priority=1, weight=2, port=3, target='1unit.tests.')]) recordSet.name, recordSet.ttl, recordSet.type = '_srv1._tcp', 13, 'SRV' rs.append(recordSet) recordSet = RecordSet(srv_records=[SrvRecord(priority=1, weight=2, port=3, target='1unit.tests.'), SrvRecord(priority=4, weight=5, port=6, target='2unit.tests.')]) recordSet.name, recordSet.ttl, recordSet.type = '_srv2._tcp', 14, 'SRV' rs.append(recordSet) recordSet = RecordSet(txt_records=[TxtRecord(value='sample text1')]) recordSet.name, recordSet.ttl, recordSet.type = 'txt1', 15, 'TXT' rs.append(recordSet) recordSet = RecordSet(txt_records=[TxtRecord(value='sample text1'), TxtRecord(value='sample text2')]) recordSet.name, recordSet.ttl, recordSet.type = 'txt2', 16, 'TXT' rs.append(recordSet) recordSet = RecordSet(soa_record=[SoaRecord()]) recordSet.name, recordSet.ttl, recordSet.type = '', 17, 'SOA' rs.append(recordSet) record_list = provider._dns_client.record_sets.list_by_dns_zone record_list.return_value = rs exists = provider.populate(zone) self.assertTrue(exists) self.assertEquals(len(zone.records), 18)
_base15.zone_name = 'unit.tests' _base15.relative_record_set_name = '_srv2._tcp' _base15.record_type = 'SRV' _base15.params['ttl'] = 7 _base15.params['srv_records'] = [SrvRecord(priority=12, weight=17, port=1, target='srvfoo.unit.tests.')] azure_records.append(_base15) _base16 = _AzureRecord('TestAzure', octo_records[16]) _base16.zone_name = 'unit.tests' _base16.relative_record_set_name = 'txt1' _base16.record_type = 'TXT' _base16.params['ttl'] = 8 _base16.params['txt_records'] = [TxtRecord(value=['txt singleton test'])] azure_records.append(_base16) _base17 = _AzureRecord('TestAzure', octo_records[17]) _base17.zone_name = 'unit.tests' _base17.relative_record_set_name = 'txt2' _base17.record_type = 'TXT' _base17.params['ttl'] = 9 _base17.params['txt_records'] = [TxtRecord(value=['txt multiple test']), TxtRecord(value=['txt multiple test 2'])] azure_records.append(_base17) class Test_AzureRecord(TestCase): def test_azure_record(self): assert(len(azure_records) == len(octo_records))