def test_domain_records_perms(self):
"""
Test common domain record perms (cname, mx, txt, srv, ns)
"""
self.setup_request()
perm_table = {
'cyder_admin': ['all'],
'admin': ['all'],
'user': ['all'],
'guest': [cy.ACTION_VIEW],
}
# Initialize objs into ctnrs.
domain = Domain(id=None, name='foo')
domain.save()
self.ctnr_admin.domains.add(domain)
self.ctnr_user.domains.add(domain)
self.ctnr_guest.domains.add(domain)
self.save_all_ctnrs()
domain_records = []
domain_records.append(AddressRecord(domain=domain))
domain_records.append(CNAME(domain=domain))
domain_records.append(MX(domain=domain))
domain_records.append(SRV(domain=domain))
domain_records.append(TXT(domain=domain))
domain_records.append(Nameserver(domain=domain))
for obj in domain_records:
self.check_perms_each_user(obj, perm_table)
def do_generic_add(self, data):
mx = MX(**data)
mx.__repr__()
mx.save()
self.assertTrue(mx.details())
rmx = MX.objects.filter(**data)
self.assertTrue(len(rmx) == 1)
return mx
def test_basic_add_remove5(self):
# Make sure all record types block
c = Domain(name='foo22')
c.save()
self.assertFalse(c.purgeable)
f_c = Domain(name='foo.foo22')
s, _ = SOA.objects.get_or_create(primary="foo",
contact="foo",
comment="foo.foo22")
f_c.soa = s
f_c.save()
self.assertFalse(f_c.purgeable)
fqdn = "bar.x.y.z.foo.foo22"
label, the_domain = ensure_label_domain(fqdn)
txt = TXT(label=label, domain=the_domain, txt_data="Nthing")
txt.save()
self.assertFalse(prune_tree(the_domain))
txt.delete()
label, the_domain = ensure_label_domain(fqdn)
addr = AddressRecord(label=label,
domain=the_domain,
ip_type='4',
ip_str="10.2.3.4")
addr.save()
self.assertFalse(prune_tree(the_domain))
addr.delete()
label, the_domain = ensure_label_domain(fqdn)
mx = MX(label=label, domain=the_domain, server="foo", priority=4)
mx.save()
self.assertFalse(prune_tree(the_domain))
mx.delete()
label, the_domain = ensure_label_domain(fqdn)
ns = Nameserver(domain=the_domain, server="asdfasffoo")
ns.save()
self.assertFalse(prune_tree(the_domain))
ns.delete()
label, the_domain = ensure_label_domain(fqdn)
srv = SRV(label='_' + label,
domain=the_domain,
target="foo",
priority=4,
weight=4,
port=34)
srv.save()
self.assertFalse(prune_tree(the_domain))
srv.delete()
def test_cleanup_mx(self):
self.assertFalse(Domain.objects.filter(name="x.y.z.foo.poo"))
self.assertFalse(Domain.objects.filter(name="y.z.foo.poo"))
self.assertFalse(Domain.objects.filter(name="z.foo.poo"))
self.assertTrue(Domain.objects.filter(name="foo.poo"))
fqdn = "bar.x.y.z.foo.poo"
label, the_domain = ensure_label_domain(fqdn)
mx = MX(label=label, domain=the_domain, server="foo", priority=4)
mx.save()
self.assertFalse(prune_tree(the_domain))
mx.delete()
self.assertFalse(Domain.objects.filter(name="x.y.z.foo.poo"))
self.assertFalse(Domain.objects.filter(name="y.z.foo.poo"))
self.assertFalse(Domain.objects.filter(name="z.foo.poo"))
self.assertTrue(Domain.objects.filter(name="foo.poo"))
def test_add_with_cname(self):
label = "cnamederp"
domain = self.o_e
data = "foo.com"
cn = CNAME(label=label, domain=domain, target=data)
cn.full_clean()
cn.save()
data = {
'label': '',
'domain': self.o_e,
'server': 'cnamederp.oregonstate.org',
'priority': 2,
'ttl': 2222
}
mx = MX(**data)
self.assertRaises(ValidationError, mx.save)
def test_add_mx_with_cname(self):
label = "cnamederp1"
domain = self.c_g
data = "foo.com"
fqdn = label + '.' + domain.name
mx_data = {
'label': '',
'domain': self.c_g,
'server': fqdn,
'priority': 2,
'ttl': 2222
}
mx = MX(**mx_data)
mx.save()
cn = CNAME(label=label, domain=domain, target=data)
self.assertRaises(ValidationError, cn.full_clean)
def test_mx_cname_exists(self):
# Duplicate test?
label = "testyfoo"
data = "wat"
dom, _ = Domain.objects.get_or_create(name="cd")
dom, _ = Domain.objects.get_or_create(name="what.cd")
cn, _ = CNAME.objects.get_or_create(label=label,
domain=dom,
target=data)
cn.full_clean()
cn.save()
rec = MX(label=label,
domain=dom,
server="asdf1",
priority=123,
ttl=123)
self.assertRaises(ValidationError, rec.save)
def test_basic_add_remove5(self):
# Make sure all record types block
f_c = create_fake_zone("foo.foo22", suffix="")
self.assertFalse(f_c.purgeable)
fqdn = "bar.x.y.z.foo.foo22"
label, the_domain = ensure_label_domain(fqdn)
txt = TXT(label=label, domain=the_domain, txt_data="Nthing")
txt.save()
self.assertFalse(prune_tree(the_domain))
txt.delete()
label, the_domain = ensure_label_domain(fqdn)
addr = AddressRecord(label=label, domain=the_domain,
ip_type='4', ip_str="10.2.3.4")
addr.save()
self.assertFalse(prune_tree(the_domain))
addr.delete()
label, the_domain = ensure_label_domain(fqdn)
mx = MX(label=label, domain=the_domain, server="foo", priority=4)
mx.save()
self.assertFalse(prune_tree(the_domain))
mx.delete()
label, the_domain = ensure_label_domain(fqdn)
ns = Nameserver(domain=the_domain, server="asdfasffoo")
ns.save()
self.assertFalse(prune_tree(the_domain))
ns.delete()
label, the_domain = ensure_label_domain(fqdn)
srv = SRV(
label='_' + label, domain=the_domain, target="foo", priority=4,
weight=4, port=34)
srv.save()
self.assertFalse(prune_tree(the_domain))
srv.delete()
def ensure_domain(name):
parts = list(reversed(name.split('.')))
domain_name = ''
for i in range(len(parts)):
domain_name = parts[i] + '.' + domain_name
domain_name = domain_name.strip('.')
clobber_objects = [] # Objects that have the same name as a domain
# need to be deleted and then recreated
try:
mxs = MX.objects.filter(fqdn=domain_name)
# It got here. It exists
for mx in mxs:
label = mx.label
mxdomain = mx.domain
server = mx.server
prio = mx.priority
ttl = mx.ttl
mxviews = [view.name for view in mx.views.all()]
mx.delete()
mx = MX(label='', server=server, priority=prio, ttl=ttl)
clobber_objects.append((mx, mxviews))
except ObjectDoesNotExist, e:
pass
try:
addrs = AddressRecord.objects.filter(fqdn=domain_name)
for exists_a in addrs:
# It got here. It exists
need_to_recreate_a = True
ip_str = exists_a.ip_str
ip_type = exists_a.ip_type
aviews = [view.name for view in exists_a.views.all()]
exists_a.delete(check_cname=False)
a = AddressRecord(label='', ip_str=ip_str, ip_type=ip_type)
clobber_objects.append((a, aviews))
except ObjectDoesNotExist, e:
pass
def test_mx_update(self):
mx = MX(server="foo", priority=4, ctnr=self.ctnr)
self.generic_check(mx)
