class WarnExpireTestCase(MyBaseTest):
def setUp(self):
MyBaseTest.setUp(self)
self.cmd = WarnExpireCommand()
@patch("fwadmin.management.commands.warnexpire.send_renew_mail")
def test_no_send_renew_mail_when_still_active(self, mock_f):
"""Ensure we do not send mails if there is enough delta"""
today = datetime.date.today()
delta = self.host.active_until - today
self.assertEqual(delta, datetime.timedelta(days=360))
self.cmd.handle()
self.assertEqual(mock_f.mock_calls, [])
@patch("fwadmin.management.commands.warnexpire.send_renew_mail")
def test_send_renew_mail_from_cmd(self, mock_f):
"""Ensure we do send mails"""
tomorrow = datetime.date.today() + datetime.timedelta(days=1)
self.host.active_until = tomorrow
self.host.save()
self.cmd.handle()
mock_f.assert_called_with(self.host)
@patch("fwadmin.management.commands.warnexpire.send_mail")
def test_send_renew_mail(self, mock_f):
"""Ensure we do send mails"""
tomorrow = datetime.date.today() + datetime.timedelta(days=1)
self.host.active_until = tomorrow
self.host.save()
send_renew_mail(self.host)
subject = u"Firewall config for 'test'"
body = u"""Dear user1,
The firewall config for machine: 'test' (192.168.1.1) will expire at
'%s'.
Please click on https://fwadmin.uni-trier.de%s to renew.
""" % (self.host.active_until,
reverse("fwadmin:edit_host", args=(self.host.id,)))
mock_f.assert_called_with(
subject, body, FWADMIN_EMAIL_FROM,
[self.host.owner.email])
class WarnExpireTestCase(MyBaseTest):
def setUp(self):
MyBaseTest.setUp(self)
self.cmd = WarnExpireCommand()
@patch("fwadmin.management.commands.warnexpire.send_renew_mail")
def test_no_send_renew_mail_when_still_active(self, mock_f):
"""Ensure we do not send mails if there is enough delta"""
today = datetime.date.today()
delta = self.host.active_until - today
self.assertEqual(delta, datetime.timedelta(days=360))
self.cmd.handle()
self.assertEqual(mock_f.mock_calls, [])
@patch("fwadmin.management.commands.warnexpire.send_renew_mail")
def test_send_renew_mail_from_cmd(self, mock_f):
"""Ensure we do send mails"""
tomorrow = datetime.date.today() + datetime.timedelta(days=1)
self.host.active_until = tomorrow
self.host.save()
self.cmd.handle()
mock_f.assert_called_with(self.host)
@patch("fwadmin.management.commands.warnexpire.send_mail")
def test_send_renew_mail(self, mock_f):
"""Ensure we do send mails"""
tomorrow = datetime.date.today() + datetime.timedelta(days=1)
self.host.active_until = tomorrow
self.host.save()
send_renew_mail(self.host)
subject = u"Firewall config for 'test'"
body = u"""Dear user1,
The firewall config for machine: 'test' (192.168.1.1) will expire at
'%s'.
Please click on https://fwadmin.uni-trier.de%s to renew.
""" % (self.host.active_until,
reverse("fwadmin:edit_host", args=(self.host.id, )))
mock_f.assert_called_with(subject, body, FWADMIN_EMAIL_FROM,
[self.host.owner.email])
def setUp(self):
MyBaseTest.setUp(self)
self.cmd = WarnExpireCommand()
def setUp(self):
MyBaseTest.setUp(self)
self.cmd = WarnExpireCommand()
