def execute(self, args):
manager = DNSManager(self.client)
manager.create_zone(args['<zone>'])
class DNSTests(unittest.TestCase):
def setUp(self):
self.client = MagicMock()
self.dns_client = DNSManager(self.client)
def test_init_exercise(self):
self.assertTrue(hasattr(self.dns_client, 'service'))
self.assertTrue(hasattr(self.dns_client, 'record'))
def test_list_zones(self):
zone_list = ['test']
self.client['Account'].getDomains.return_value = zone_list
zones = self.dns_client.list_zones()
self.assertEqual(zones, zone_list)
def test_get_zone(self):
zone_list = [
{'id': 98765, 'name': 'test-example.com'},
{'id': 12345, 'name': 'example.com', "resourceRecords": ["test"]},
]
# match, with defaults
self.client['Account'].getObject.return_value = zone_list[1]
res = self.dns_client.get_zone(12345)
self.assertEqual(res, zone_list[1])
self.client['Account'].getObject.assert_called_once_with(
id=12345,
mask='resourceRecords')
# No records masked in
self.client['Account'].getObject.reset_mock()
self.client['Account'].getObject.return_value = [zone_list[1]]
self.dns_client.get_zone(12345, records=False)
self.client['Account'].getObject.assert_called_once_with(
id=12345,
mask=None)
def test_resolve_zone_name(self):
zone_list = [{'name': 'example.com', 'id': 12345}]
# matching domain
self.client['Account'].getDomains.return_value = zone_list
res = self.dns_client._get_zone_id_from_name('example.com')
self.assertEqual([12345], res)
self.client['Account'].getDomains.assert_called_once_with(
filter={"domains": {"name": {"operation": "_= example.com"}}})
# no matches
self.client['Account'].getDomains.reset_mock()
self.client['Account'].getDomains.return_value = []
res = self.dns_client._get_zone_id_from_name('example.com')
self.assertEqual([], res)
self.client['Account'].getDomains.assert_called_once_with(
filter={"domains": {"name": {"operation": "_= example.com"}}})
def test_create_zone(self):
call = self.client['Dns_Domain'].createObject
call.return_value = {'name': 'example.com'}
res = self.dns_client.create_zone('example.com')
call.assert_called_once_with({
'name': 'example.com', "resourceRecords": {}, "serial": ANY
})
self.assertEqual(res, {'name': 'example.com'})
def test_delete_zone(self):
self.dns_client.delete_zone(1)
self.client['Dns_Domain'].deleteObject.assert_called_once_with(id=1)
def test_edit_zone(self):
self.dns_client.edit_zone('example.com')
self.client['Dns_Domain'].editObject.assert_called_once_with(
'example.com')
def test_create_record(self):
self.dns_client.create_record(1, 'test', 'TXT', 'testing', ttl=1200)
f = self.client['Dns_Domain_ResourceRecord'].createObject
f.assert_called_once_with(
{
'domainId': 1,
'ttl': 1200,
'host': 'test',
'type': 'TXT',
'data': 'testing'
})
def test_delete_record(self):
self.dns_client.delete_record(1)
f = self.client['Dns_Domain_ResourceRecord'].deleteObject
f.assert_called_once_with(id=1)
def test_edit_record(self):
self.dns_client.edit_record({'id': 1, 'name': 'test', 'ttl': '1800'})
f = self.client['Dns_Domain_ResourceRecord'].editObject
f.assert_called_once_with(
{'id': 1, 'name': 'test', 'ttl': '1800'},
id=1
)
def test_dump_zone(self):
f = self.client['Dns_Domain'].getZoneFileContents
f.return_value = 'lots of text'
self.dns_client.dump_zone(1)
f.assert_called_once_with(id=1)
def test_get_record(self):
records = [
{'ttl': 7200, 'data': 'd', 'host': 'a', 'type': 'cname'},
{'ttl': 900, 'data': '1', 'host': 'b', 'type': 'a'},
{'ttl': 900, 'data': 'x', 'host': 'c', 'type': 'ptr'},
{'ttl': 86400, 'data': 'b', 'host': 'd', 'type': 'txt'},
{'ttl': 86400, 'data': 'b', 'host': 'e', 'type': 'txt'},
{'ttl': 600, 'data': 'b', 'host': 'f', 'type': 'txt'},
]
D = self.client['Dns_Domain'].getResourceRecords
# maybe valid domain, but no records matching
D.return_value = []
self.assertEqual(self.dns_client.get_records(12345),
[])
D.reset_mock()
D.return_value = [records[1]]
self.dns_client.get_records(12345, type='a')
D.assert_called_once_with(
id=12345,
filter={'resourceRecords': {'type': {"operation": "_= a"}}},
mask=ANY)
D.reset_mock()
D.return_value = [records[0]]
self.dns_client.get_records(12345, host='a')
D.assert_called_once_with(
id=12345,
filter={'resourceRecords': {'host': {"operation": "_= a"}}},
mask=ANY)
D.reset_mock()
D.return_value = records[3:5]
self.dns_client.get_records(12345, data='a')
D.assert_called_once_with(
id=12345,
filter={'resourceRecords': {'data': {"operation": "_= a"}}},
mask=ANY)
D.reset_mock()
D.return_value = records[3:5]
self.dns_client.get_records(12345, ttl='86400')
D.assert_called_once_with(
id=12345,
filter={'resourceRecords': {'ttl': {"operation": 86400}}},
mask=ANY)
