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)