def test_get_domain_by_record_acl_domain_success(self):
account = Account()
account.email = "*****@*****.**"
account.in_global_acl_emails = MagicMock(return_value=True)
dm = MagicMock()
dm.domain = "example.com"
domain = MagicMock()
domain.get = MagicMock(return_value=dm)
account.get_domain_object = MagicMock(return_value=domain)
account.get_global_acl_labels = MagicMock(return_value=["DOMAIN"])
self.assertEquals(
dm, account.get_domain_by_record_acl(1, "example.com", "TXT"))
def test_get_domain_by_record_acl_multiple_label_failure_two(self):
account = Account()
account.email = "*****@*****.**"
account.in_global_acl_emails = MagicMock(return_value=True)
dm = MagicMock()
dm.domain = "example.com"
domain = MagicMock()
domain.get = MagicMock(return_value=dm)
account.get_domain_object = MagicMock(return_value=domain)
account.get_global_acl_labels = MagicMock(
return_value=["_acme-challenge", "DOMAIN"])
self.assertFalse(
account.get_domain_by_record_acl(1, "foo.bar.example.com", "TXT"))
def test_get_domain_by_record_acl_multiple_label_success_no_match(self):
account = Account()
account.email = "*****@*****.**"
account.in_global_acl_emails = MagicMock(return_value=True)
dm = MagicMock()
dm.domain = "example.com"
dm.domain_id = 1
dget = Mock()
dget.side_effect = [dm, peewee.DoesNotExist, peewee.DoesNotExist]
domain = MagicMock()
domain.get = dget
account.get_domain_object = MagicMock(return_value=domain)
account.get_global_acl_labels = MagicMock(
return_value=["_acme-challenge", "DOMAIN"])
self.assertEquals(
dm,
account.get_domain_by_record_acl(
1, "_acme-challenge.foo.bar.example.com", "TXT"))
def test_get_domain_by_record_acl_multiple_label_failure_collision(self):
account = Account()
account.email = "*****@*****.**"
account.in_global_acl_emails = MagicMock(return_value=True)
dm = MagicMock()
dm.domain = "example.com"
dm.domain_id = 1
dm2 = MagicMock()
dm2.domain = "bar.example.com"
dm2.domain_id = 2
dget = Mock()
dget.side_effect = [dm, peewee.DoesNotExist, dm2]
domain = Mock()
domain.get = dget
account.get_domain_object = MagicMock(return_value=domain)
account.get_global_acl_labels = MagicMock(
return_value=["_acme-challenge", "DOMAIN"])
self.assertFalse(
account.get_domain_by_record_acl(
1, "_acme-challenge.foo.bar.example.com", "TXT"))
