def test_attempt(self):
env = Envelope()
m = MaildropRelay()
self.mox.StubOutWithMock(m, "_exec_maildrop")
m._exec_maildrop(env).AndReturn((0, None))
self.mox.ReplayAll()
m.attempt(env, 0)
def test_attempt_permanentfail(self):
env = Envelope()
m = MaildropRelay()
self.mox.StubOutWithMock(m, "_exec_maildrop")
m._exec_maildrop(env).AndReturn((13, "permanent failure"))
self.mox.ReplayAll()
with self.assertRaises(PermanentRelayError):
m.attempt(env, 0)
def test_attempt_timeout(self):
env = Envelope()
m = MaildropRelay()
self.mox.StubOutWithMock(m, "_exec_maildrop")
m._exec_maildrop(env).AndRaise(Timeout)
self.mox.ReplayAll()
with self.assertRaises(TransientRelayError):
m.attempt(env, 0)
def test_attempt_transientfail(self):
env = Envelope()
m = MaildropRelay()
self.mox.StubOutWithMock(m, "_exec_maildrop")
m._exec_maildrop(env).AndReturn((MaildropRelay.EX_TEMPFAIL, "transient failure"))
self.mox.ReplayAll()
with self.assertRaises(TransientRelayError):
m.attempt(env, 0)
def test_exec_maildrop_error(self):
pmock = self.mox.CreateMock(gevent_subprocess.Popen)
self.mox.StubOutWithMock(gevent_subprocess, "Popen")
env = Envelope("*****@*****.**", ["*****@*****.**"])
env.parse("From: [email protected]\r\n\r\ntest test\r\n")
gevent_subprocess.Popen(
["maildrop", "-f", "*****@*****.**"],
executable="/path/to/maildrop",
stdin=gevent_subprocess.PIPE,
stdout=gevent_subprocess.PIPE,
stderr=gevent_subprocess.PIPE,
).AndReturn(pmock)
pmock.communicate("From: [email protected]\r\n\r\ntest test\r\n").AndReturn(("", "maildrop: error msg"))
pmock.pid = -1
pmock.returncode = MaildropRelay.EX_TEMPFAIL
self.mox.ReplayAll()
m = MaildropRelay("/path/to/maildrop")
status, msg = m._exec_maildrop(env)
self.assertEqual(MaildropRelay.EX_TEMPFAIL, status)
self.assertEqual("error msg", msg)
