def test_rename_domain(self):
"""Test domain rename."""
domain = admin_factories.DomainFactory(name="domain.test")
domain.name = "domain1.test"
domain.save()
name = "@{}".format(domain.name)
self.assertTrue(models.Users.objects.filter(email=name).exists())
self.assertTrue(
models.Policy.objects.filter(policy_name=name).exists())
# Now from form
self.client.force_login(self.admin)
rdomain = admin_factories.DomainFactory(name="domain.relay",
type="relaydomain")
rdomain.transport = tr_factories.TransportFactory(
pattern=rdomain.name,
service="relay",
_settings={
"relay_target_host": "external.host.tld",
"relay_target_port": "25",
"relay_verify_recipients": False
})
rdomain.save()
values = {
"name": "domain2.relay",
"quota": rdomain.quota,
"default_mailbox_quota": rdomain.default_mailbox_quota,
"type": "relaydomain",
"enabled": rdomain.enabled,
"service": rdomain.transport.service,
"relay_target_host": "127.0.0.1",
"relay_target_port": 25,
}
self.ajax_post(reverse("admin:domain_change", args=[rdomain.pk]),
values)
def test_create_domain(self):
"""Test domain creation."""
domain = admin_factories.DomainFactory(name="domain.test")
name = "@{}".format(domain.name)
policy = models.Policy.objects.get(policy_name=name)
user = models.Users.objects.filter(policy=policy).first()
self.assertIsNot(user, None)
self.assertEqual(user.email, name)
# Create a domain alias
self.client.force_login(self.admin)
data = {
"name": domain.name,
"type": "domain",
"enabled": domain.enabled,
"quota": domain.quota,
"default_mailbox_quota": domain.default_mailbox_quota,
"aliases_1": "dalias.test"
}
self.ajax_post(
reverse("admin:domain_change", args=[domain.pk]), data)
name = "@dalias.test"
self.assertFalse(
models.Policy.objects.filter(policy_name=name).exists())
user = models.Users.objects.get(email=name)
self.assertEqual(user.policy, policy)
# Delete domain alias
del data["aliases_1"]
self.ajax_post(
reverse("admin:domain_change", args=[domain.pk]), data)
self.assertFalse(
models.Users.objects.filter(email=name).exists())
def test_delete_domain(self):
"""Test domain removal."""
domain = admin_factories.DomainFactory(name="domain.test")
domain.delete(None)
name = "@{}".format(domain.name)
self.assertFalse(models.Users.objects.filter(email=name).exists())
self.assertFalse(
models.Policy.objects.filter(policy_name=name).exists())
def test_rename_domain(self):
"""Test domain rename."""
domain = admin_factories.DomainFactory(name="domain.test")
domain.name = "domain1.test"
domain.save()
name = "@{}".format(domain.name)
self.assertTrue(models.Users.objects.filter(email=name).exists())
self.assertTrue(
models.Policy.objects.filter(policy_name=name).exists())
def setUpTestData(cls): # NOQA:N802
super(DataMixin, cls).setUpTestData()
transport = tr_factories.TransportFactory(
pattern="test.com", _settings={
"relay_target_host": "external.host.tld",
"relay_target_port": 25,
"relay_verify_recipients": False
}
)
cls.domain1 = admin_factories.DomainFactory(
name="test.com", type="relaydomain", transport=transport)
transport = tr_factories.TransportFactory(
pattern="domain2.test", _settings={
"relay_target_host": "external.host.tld",
"relay_target_port": 25,
"relay_verify_recipients": True
}
)
cls.domain2 = admin_factories.DomainFactory(
name="test2.com", type="relaydomain", transport=transport)
def setUpTestData(cls):
"""Create test data."""
super(RelayDomainsTestCase, cls).setUpTestData()
admin_factories.populate_database()
dom = admin_factories.DomainFactory(name="relaydomain.tld",
type="relaydomain")
dom.relaydomain.target_host = "external.host.tld"
dom.relaydomain.save()
cls.rdom = dom.relaydomain
admin_factories.DomainAliasFactory(name="relaydomainalias.tld",
target=cls.rdom.domain)
admin_factories.MailboxFactory(domain=cls.rdom.domain,
address="local",
user__username="******",
user__groups=("SimpleUsers", ))
def setUpTestData(cls):
"""Create test data."""
super(RelayDomainsTestCase, cls).setUpTestData()
admin_factories.populate_database()
cls.dom = admin_factories.DomainFactory(name="relaydomain.tld",
type="relaydomain")
cls.transport = tr_factories.TransportFactory(
pattern=cls.dom.name,
_settings={
"relay_target_host": "external.host.tld",
"relay_target_port": "25",
"relay_verify_recipients": False
})
admin_factories.DomainAliasFactory(name="relaydomainalias.tld",
target=cls.dom)
admin_factories.MailboxFactory(domain=cls.dom,
address="local",
user__username="******",
user__groups=("SimpleUsers", ))
def test_update_domain_policy(self):
"""Check domain policy edition."""
self.client.force_login(self.admin)
domain = admin_factories.DomainFactory(name="domain.test")
url = reverse("admin:domain_change", args=[domain.pk])
# response = self.client.get(url)
# self.assertContains(response, "Content filter")
custom_title = "This is SPAM!"
data = {
"name": domain.name,
"type": "domain",
"enabled": domain.enabled,
"quota": domain.quota,
"default_mailbox_quota": domain.default_mailbox_quota,
"bypass_virus_checks": "Y",
"spam_subject_tag2_act": False,
"spam_subject_tag2": custom_title
}
self.ajax_post(url, data)
name = "@{}".format(domain.name)
policy = models.Policy.objects.get(policy_name=name)
self.assertEqual(policy.spam_subject_tag2, custom_title)
def setUpTestData(cls):
super(ManagementCommandTestCase, cls).setUpTestData()
cls.domain = admin_factories.DomainFactory(name="ngyn.org")
def setUpTestData(cls):
"""Create some data."""
super(EventsTestCase, cls).setUpTestData()
admin_factories.DomainFactory(name="test.com")
def setUpTestData(cls):
"""Create test data."""
super(DMARCViewsTestCase, cls).setUpTestData()
cls.domain = admin_factories.DomainFactory(name="ngyn.org")
