def test_get_address(self):
mgmt_command = feeder.Command()
address = mgmt_command._get_address("myself <*****@*****.**>")
self.assertEqual(address, "<*****@*****.**>")
address = mgmt_command._get_address("*****@*****.**")
self.assertEqual(address, "<*****@*****.**>")
with self.assertRaises(CommandError):
mgmt_command._get_address("me <>")
def test_get_server(self, smtp_mock, lmtp_mock):
mgmt_command = feeder.Command()
self.assertEqual(mgmt_command._server, None)
server = mgmt_command._get_server()
self.assertEqual(mgmt_command._server, server)
self.assertTrue(smtp_mock.called)
self.assertFalse(lmtp_mock.called)
smtp_mock.reset_mock()
lmtp_mock.reset_mock()
mgmt_command._server = None
with self.settings(SALMON_SERVER={"type": "lmtp", "path": "/fake/path"}):
server = mgmt_command._get_server()
self.assertEqual(mgmt_command._server, server)
self.assertTrue(lmtp_mock.called)
self.assertFalse(smtp_mock.called)