def test_call_validator(self):
mock = self.mox.CreateMockAnything()
mock.__call__(IsA(SmtpSession)).AndReturn(mock)
mock.handle_test('arg')
self.mox.ReplayAll()
h = SmtpSession(None, mock, None)
h._call_validator('test', 'arg')
def test_call_validator(self):
mock = self.mox.CreateMockAnything()
mock.__call__(IsA(SmtpSession)).AndReturn(mock)
mock.handle_test('arg')
self.mox.ReplayAll()
h = SmtpSession(None, mock, None)
h._call_validator('test', 'arg')
def test_have_data_errors(self):
h = SmtpSession(None, None, None)
reply = Reply('250')
h.HAVE_DATA(reply, None, MessageTooBig())
self.assertEqual('552', reply.code)
with self.assertRaises(ValueError):
h.HAVE_DATA(reply, None, ValueError())
def test_protocol_attribute(self):
h = SmtpSession(None, None, None)
self.assertEqual('SMTP', h.protocol)
h.extended_smtp = True
self.assertEqual('ESMTP', h.protocol)
h.security = 'TLS'
self.assertEqual('ESMTPS', h.protocol)
h.auth = 'test'
self.assertEqual('ESMTPSA', h.protocol)
def test_protocol_attribute(self):
h = SmtpSession(None, None, None)
self.assertEqual("SMTP", h.protocol)
h.extended_smtp = True
self.assertEqual("ESMTP", h.protocol)
h.security = "TLS"
self.assertEqual("ESMTPS", h.protocol)
h.auth_result = "test"
self.assertEqual("ESMTPSA", h.protocol)
def test_protocol_attribute(self):
h = SmtpSession(None, None, None)
assert_equal('SMTP', h.protocol)
h.extended_smtp = True
assert_equal('ESMTP', h.protocol)
h.security = 'TLS'
assert_equal('ESMTPS', h.protocol)
h.auth_result = 'test'
assert_equal('ESMTPSA', h.protocol)
def test_protocol_attribute(self):
h = SmtpSession(None, None, None)
self.assertEqual('SMTP', h.protocol)
h.extended_smtp = True
self.assertEqual('ESMTP', h.protocol)
h.security = 'TLS'
self.assertEqual('ESMTPS', h.protocol)
h.auth = 'test'
self.assertEqual('ESMTPSA', h.protocol)
def test_have_data(self):
env = Envelope()
handoff = self.mox.CreateMockAnything()
handoff(env).AndReturn([(env, 'testid')])
self.mox.ReplayAll()
h = SmtpSession(('127.0.0.1', 0), None, handoff)
h.envelope = env
reply = Reply('250')
h.HAVE_DATA(reply, b'', None)
self.assertEqual('250', reply.code)
self.assertEqual('2.6.0 Message accepted for delivery', reply.message)
def test_have_data(self):
env = Envelope()
handoff = self.mox.CreateMockAnything()
handoff(env).AndReturn([(env, 'testid')])
self.mox.ReplayAll()
h = SmtpSession(('127.0.0.1', 0), None, handoff)
h.envelope = env
reply = Reply('250')
h.HAVE_DATA(reply, '', None)
assert_equal('250', reply.code)
assert_equal('2.6.0 Message accepted for delivery', reply.message)
def test_ptr_lookup(self):
self.mox.StubOutWithMock(resolver, 'query')
resolver.query(IgnoreArg(), 'PTR').AndRaise(DNSException)
resolver.query(IgnoreArg(), 'PTR').AndReturn(['example.com'])
self.mox.ReplayAll()
h = SmtpSession(('1.2.3.4', None), None, None)
with self.assertRaises(DNSException):
h._ptr_lookup()
self.assertIsNone(h.reverse_address)
h._ptr_lookup()
self.assertEqual('example.com', h.reverse_address)
def test_have_data(self):
env = Envelope()
handoff = self.mox.CreateMockAnything()
handoff(env).AndReturn([(env, "testid")])
self.mox.ReplayAll()
h = SmtpSession(("127.0.0.1", 0), None, handoff)
h.envelope = env
reply = Reply("250")
h.HAVE_DATA(reply, "", None)
self.assertEqual("250", reply.code)
self.assertEqual("2.6.0 Message accepted for delivery", reply.message)
def test_simple_handshake(self):
mock = self.mox.CreateMockAnything()
mock.__call__(IsA(SmtpSession)).AndReturn(mock)
mock.handle_banner(IsA(Reply), ('127.0.0.1', 0))
mock.handle_helo(IsA(Reply), 'there')
self.mox.ReplayAll()
h = SmtpSession(('127.0.0.1', 0), mock, None)
h.BANNER_(Reply('220'))
h.HELO(Reply('250'), 'there')
self.assertEqual('there', h.ehlo_as)
self.assertFalse(h.extended_smtp)
def test_have_data_queueerror(self):
env = Envelope()
handoff = self.mox.CreateMockAnything()
handoff(env).AndReturn([(env, QueueError())])
self.mox.ReplayAll()
h = SmtpSession(('127.0.0.1', 0), None, handoff)
h.envelope = env
reply = Reply('250')
h.HAVE_DATA(reply, b'', None)
self.assertEqual('451', reply.code)
self.assertEqual('4.3.0 Error queuing message', reply.message)
def test_have_data_queueerror(self):
env = Envelope()
handoff = self.mox.CreateMockAnything()
handoff(env).AndReturn([(env, QueueError())])
self.mox.ReplayAll()
h = SmtpSession(("127.0.0.1", 0), None, handoff)
h.envelope = env
reply = Reply("250")
h.HAVE_DATA(reply, "", None)
self.assertEqual("550", reply.code)
self.assertEqual("5.6.0 Error queuing message", reply.message)
def test_have_data_queueerror(self):
env = Envelope()
handoff = self.mox.CreateMockAnything()
handoff(env).AndReturn([(env, QueueError())])
self.mox.ReplayAll()
h = SmtpSession(('127.0.0.1', 0), None, handoff)
h.envelope = env
reply = Reply('250')
h.HAVE_DATA(reply, b'', None)
self.assertEqual('451', reply.code)
self.assertEqual('4.3.0 Error queuing message', reply.message)
def test_have_data(self):
class PtrLookup(object):
def finish(self, *args):
return 'localhost'
env = Envelope()
handoff = self.mox.CreateMockAnything()
handoff(env).AndReturn([(env, 'testid')])
self.mox.ReplayAll()
h = SmtpSession(('127.0.0.1', 0), None, handoff)
h.envelope = env
h._ptr_lookup = PtrLookup()
reply = Reply('250')
h.HAVE_DATA(reply, b'', None)
self.assertEqual('250', reply.code)
self.assertEqual('2.6.0 Message accepted for delivery', reply.message)
self.assertEqual('localhost', env.client['host'])
def test_have_data(self):
class PtrLookup(object):
def finish(self, *args):
return 'localhost'
env = Envelope()
handoff = self.mox.CreateMockAnything()
handoff(env).AndReturn([(env, 'testid')])
self.mox.ReplayAll()
h = SmtpSession(('127.0.0.1', 0), None, handoff)
h.envelope = env
h._ptr_lookup = PtrLookup()
reply = Reply('250')
h.HAVE_DATA(reply, b'', None)
self.assertEqual('250', reply.code)
self.assertEqual('2.6.0 Message accepted for delivery', reply.message)
self.assertEqual('localhost', env.client['host'])
def test_mail_rcpt_data_rset(self):
mock = self.mox.CreateMockAnything()
mock.__call__(IsA(SmtpSession)).AndReturn(mock)
mock.handle_mail(IsA(Reply), '*****@*****.**', {})
mock.handle_rcpt(IsA(Reply), '*****@*****.**', {})
mock.handle_data(IsA(Reply))
self.mox.ReplayAll()
h = SmtpSession(None, mock, None)
h.MAIL(Reply('250'), '*****@*****.**', {})
h.RCPT(Reply('250'), '*****@*****.**', {})
self.assertEqual('*****@*****.**', h.envelope.sender)
self.assertEqual(['*****@*****.**'], h.envelope.recipients)
h.DATA(Reply('550'))
h.RSET(Reply('250'))
self.assertFalse(h.envelope)
def test_extended_handshake(self):
mock = self.mox.CreateMockAnything()
mock.__call__(IsA(SmtpSession)).AndReturn(mock)
mock.handle_banner(IsA(Reply), ('127.0.0.1', 0))
mock.handle_ehlo(IsA(Reply), 'there')
mock.handle_tls()
self.mox.ReplayAll()
h = SmtpSession(('127.0.0.1', 0), mock, None)
h.BANNER_(Reply('220'))
h.EHLO(Reply('250'), 'there')
h.TLSHANDSHAKE()
h.AUTH(Reply('250'), 'testauth')
assert_equal('there', h.ehlo_as)
assert_true(h.extended_smtp)
assert_equal('TLS', h.security)
assert_equal('testauth', h.auth_result)
assert_equal('ESMTPSA', h.protocol)
def test_extended_handshake(self):
creds = self.mox.CreateMockAnything()
creds.authcid = 'testuser'
creds.authzid = 'testzid'
mock = self.mox.CreateMockAnything()
mock.__call__(IsA(SmtpSession)).AndReturn(mock)
mock.handle_banner(IsA(Reply), ('127.0.0.1', 0))
mock.handle_ehlo(IsA(Reply), 'there')
mock.handle_tls()
mock.handle_auth(IsA(Reply), creds)
self.mox.ReplayAll()
h = SmtpSession(('127.0.0.1', 0), mock, None)
h.BANNER_(Reply('220'))
h.EHLO(Reply('250'), 'there')
h.TLSHANDSHAKE()
h.AUTH(Reply('235'), creds)
self.assertEqual('there', h.ehlo_as)
self.assertTrue(h.extended_smtp)
self.assertEqual('TLS', h.security)
self.assertEqual(('testuser', 'testzid'), h.auth)
self.assertEqual('ESMTPSA', h.protocol)
