class TestGetDnsRecords(TestCase):
@patch.object(DNSaaS, '_get_oauth_token')
def setUp(self, mocked):
mocked.return_value = 'token'
self.dnsaas = DNSaaS()
@patch.object(DNSaaS, 'get_api_result')
def test_return_empty_when_api_returns_empty(self, mocked):
mocked.return_value = []
found_dns = self.dnsaas.get_dns_records(['192.168.0.1'])
self.assertEqual(found_dns, [])
def test_return_empty_when_no_ipaddress(self):
found_dns = self.dnsaas.get_dns_records([])
self.assertEqual(found_dns, [])
@patch.object(DNSaaS, 'get_api_result')
def test_return_dns_records_when_api_returns_records(self, mocked):
data = {
'content': '127.0.0.3',
'name': '1.test.pl',
'type': 'A',
'id': 1
}
mocked.return_value = [data]
found_dns = self.dnsaas.get_dns_records(['192.168.0.1'])
self.assertEqual(len(found_dns), 1)
self.assertEqual(found_dns[0]['content'], data['content'])
self.assertEqual(found_dns[0]['name'], data['name'])
self.assertEqual(found_dns[0]['type'], RecordType.a)
@override_settings(DNSAAS_URL='http://dnsaas.com/')
def test_build_url(self):
self.assertEqual(self.dnsaas.build_url('domains'),
'http://dnsaas.com/api/domains/')
@override_settings(DNSAAS_URL='http://dnsaas.com/')
def test_build_url_with_version(self):
self.assertEqual(self.dnsaas.build_url('domains'),
'http://dnsaas.com/api/domains/')
@override_settings(DNSAAS_URL='http://dnsaas.com/')
def test_build_url_with_id(self):
self.assertEqual(self.dnsaas.build_url('domains', id=1),
'http://dnsaas.com/api/domains/1/')
@override_settings(DNSAAS_URL='http://dnsaas.com/')
def test_build_url_with_get_params(self):
self.assertEqual(
self.dnsaas.build_url('domains', get_params=[('name', 'ralph')]),
'http://dnsaas.com/api/domains/?name=ralph')
@override_settings(DNSAAS_URL='http://dnsaas.com/')
def test_build_url_with_id_and_get_params(self):
self.assertEqual(
self.dnsaas.build_url('domains',
id=1,
get_params=[('name', 'ralph')]),
'http://dnsaas.com/api/domains/1/?name=ralph')
def clean_dns(cls, instances, **kwargs):
"""
Clean DNS entries for each instance if DNSaaS integration is enabled.
"""
if not settings.ENABLE_DNSAAS_INTEGRATION:
raise DNSaaSIntegrationNotEnabledError()
dnsaas = DNSaaS()
# TODO: transaction?
for instance in instances:
ips = list(
instance.ipaddresses.exclude(is_management=True).values_list(
'address', flat=True))
if not ips:
logger.info('No IPs for %s - skipping cleaning DNS entries',
instance)
continue
records = dnsaas.get_dns_records(ips)
if len(records) > settings.DEPLOYMENT_MAX_DNS_ENTRIES_TO_CLEAN:
raise Exception(
'Cannot clean {} entries for {} - clean it manually'.format(
len(records), instance))
for record in records:
logger.warning('Deleting %s (%s / %s / %s) DNS record',
record['pk'], record['type'], record['name'],
record['content'])
if dnsaas.delete_dns_record(record['pk']):
raise Exception() # TODO
class TestGetDnsRecords(TestCase):
def setUp(self):
self.dnsaas = DNSaaS()
@patch.object(DNSaaS, 'get_api_result')
def test_return_empty_when_api_returns_empty(self, mocked):
mocked.return_value = []
found_dns = self.dnsaas.get_dns_records(['192.168.0.1'])
self.assertEqual(found_dns, [])
@patch.object(DNSaaS, 'get_api_result')
def test_return_dns_records_when_api_returns_records(self, mocked):
data = {
'content': '127.0.0.3',
'name': '1.test.pl',
'type': 'A',
'id': 1
}
mocked.return_value = [data]
found_dns = self.dnsaas.get_dns_records(['192.168.0.1'])
self.assertEqual(len(found_dns), 1)
self.assertEqual(found_dns[0]['content'], data['content'])
self.assertEqual(found_dns[0]['name'], data['name'])
self.assertEqual(found_dns[0]['type'], RecordType.a)
class DNSView(RalphDetailView):
icon = 'chain-broken'
name = 'dns_edit'
label = 'DNS'
url_name = 'dns_edit'
template_name = 'dns/dns_edit.html'
def __init__(self, *args, **kwargs):
if not settings.ENABLE_DNSAAS_INTEGRATION:
raise DNSaaSIntegrationNotEnabledError()
self.dnsaas = DNSaaS()
return super().__init__(*args, **kwargs)
def get_forms(self):
forms = []
ipaddresses = self.object.ipaddress_set.all().values_list('address',
flat=True)
if not ipaddresses:
# If ipaddresses is empty return empty form list because we can not
# identify the records do not have any IP address
return forms
initial = self.dnsaas.get_dns_records(ipaddresses)
for item in initial:
forms.append(DNSRecordForm(item))
if initial and initial[0]['type'] == RecordType.a.id:
# from API "A" record is always first
empty_form = DNSRecordForm(initial={'name': initial[0]['name']})
else:
empty_form = DNSRecordForm()
forms.append(empty_form)
return forms
def get(self, request, *args, **kwargs):
if 'forms' not in kwargs:
kwargs['forms'] = self.get_forms()
return super().get(request, *kwargs, **kwargs)
def post(self, request, *args, **kwargs):
forms = self.get_forms()
posted_form = DNSRecordForm(request.POST)
# Find form which request's data belongs to
for i, form in enumerate(forms):
if (str(form.data.get('pk',
'')) == str(posted_form.data.get('pk', ''))):
forms[i] = posted_form
break
if posted_form.is_valid():
if posted_form.data.get('delete'):
errors = self.dnsaas.delete_dns_record(form.data['pk'])
elif posted_form.cleaned_data.get('pk'):
errors = self.dnsaas.update_dns_record(
posted_form.cleaned_data)
else:
errors = self.dnsaas.create_dns_record(
posted_form.cleaned_data)
if errors:
for field_name, field_errors in errors.items():
for field_error in field_errors:
if field_name == 'non_field_errors':
field_name = None
posted_form.add_error(field_name, field_error)
else:
return HttpResponseRedirect('.')
kwargs['forms'] = forms
return self.get(request, *args, **kwargs)