def test_banner_quit(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
def test_gather_params(self):
s = Server(None, None)
self.assertEqual({b'ONE': b'1'}, s._gather_params(b' ONE=1'))
self.assertEqual({b'TWO': True}, s._gather_params(b'TWO'))
self.assertEqual({b'THREE': b'foo', b'FOUR': b'bar'},
s._gather_params(b' THREE=foo FOUR=bar'))
self.assertEqual({b'FIVE': True}, s._gather_params(b'five'))
def test_recv_command(self):
self.sock.recv(IsA(int)).AndReturn(b'cmd ARG\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
cmd, arg = s._recv_command()
self.assertEqual(b'CMD', cmd)
self.assertEqual(b'ARG', arg)
def test_tls_immediately(self):
self.context.wrap_socket(self.sock, server_side=True).AndReturn(self.sock)
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None, context=self.context, tls_immediately=True)
s.handle()
def test_quit_bad(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT arg\r\n')
self.sock.sendall(b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
def test_tls_immediately_sslerror(self):
sock = self.mox.CreateMockAnything()
sock.fileno = lambda: -1
sock.tls_wrapper(sock, self.tls_args).AndRaise(SSLError())
sock.sendall(b'421 4.7.0 TLS negotiation failed\r\n')
self.mox.ReplayAll()
s = Server(sock, None, tls=self.tls_args, tls_immediately=True,
tls_wrapper=sock.tls_wrapper)
s.handle()
def test_noop(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'NOOP\r\n')
self.sock.sendall(b'250 2.0.0 Ok\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
def test_helo_empty(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'HELO\r\n')
self.sock.sendall(b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
self.assertEqual(None, s.ehlo_as)
def test_unhandled_error(self):
class TestHandler(object):
def BANNER_(self, reply):
raise Exception('test')
self.sock.sendall(b'421 4.3.0 Unhandled system error\r\n')
self.mox.ReplayAll()
s = Server(self.sock, TestHandler())
with self.assertRaises(Exception) as cm:
s.handle()
self.assertEqual(('test', ), cm.exception.args)
def test_helo(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'HELO there\r\n')
self.sock.sendall(b'250 Hello there\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
self.assertEqual('there', s.ehlo_as)
def test_get_message_data(self):
expected_reply = b'250 2.6.0 Message accepted for delivery\r\n'
self.sock.recv(IsA(int)).AndReturn(b'one\r\n')
self.sock.recv(IsA(int)).AndReturn(b'.\r\n')
self.sock.sendall(expected_reply)
self.mox.ReplayAll()
s = Server(self.sock, None)
s._get_message_data()
self.assertFalse(s.have_mailfrom)
self.assertFalse(s.have_rcptto)
def test_data_connectionlost(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'DATA\r\n')
self.sock.sendall(b'354 Start mail input; end with <CRLF>.<CRLF>\r\n')
self.sock.recv(IsA(int)).AndReturn(b'')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.ehlo_as = b'test'
s.have_mailfrom = True
s.have_rcptto = True
self.assertRaises(ConnectionLost, s.handle)
def test_tls_immediately(self):
sock = self.mox.CreateMockAnything()
sock.fileno = lambda: -1
sock.tls_wrapper(sock, self.tls_args).AndReturn(sock)
sock.sendall(b'220 ESMTP server\r\n')
sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(sock, None, tls=self.tls_args, tls_immediately=True,
tls_wrapper=sock.tls_wrapper)
s.handle()
def test_banner_command(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'BANNER\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'BANNER_\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
def test_ehlo(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
self.sock.sendall(b'250-Hello there\r\n250 TEST\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.extensions.reset()
s.extensions.add('TEST')
s.handle()
self.assertEqual('there', s.ehlo_as)
def test_mailfrom(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'HELO there\r\n')
self.sock.sendall(b'250 Hello there\r\n')
self.sock.recv(IsA(int)).AndReturn(b'MAIL FROM:<test">"addr>\r\n')
self.sock.sendall(b'250 2.1.0 Sender <test">"addr> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
self.assertTrue(s.have_mailfrom)
def test_data_bad(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'DATA arg\r\n')
self.sock.sendall(b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'DATA\r\n')
self.sock.sendall(b'503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.ehlo_as = b'test'
s.have_mailfrom = True
s.handle()
def handle(self, socket, address):
try:
handlers = SmtpSession(address, self.validator_class, self.handoff)
smtp_server = Server(socket, handlers, self.auth_class,
self.tls, self.tls_immediately,
command_timeout=self.command_timeout,
data_timeout=self.data_timeout)
if self.max_size:
smtp_server.extensions.add('SIZE', self.max_size)
smtp_server.handle()
except ConnectionLost:
pass
finally:
smtp_server.io.close()
def test_auth(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
self.sock.sendall(b'250-Hello there\r\n250 AUTH PLAIN\r\n')
self.sock.recv(IsA(int)).AndReturn(b'AUTH PLAIN dGVzdHppZAB0ZXN0dXNlcgB0ZXN0cGFzc3dvcmQ=\r\n')
self.sock.sendall(b'235 2.7.0 Authentication successful\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.extensions.reset()
s.extensions.add('AUTH', AuthSession(SASLAuth([b'PLAIN']), s.io))
s.handle()
self.assertTrue(s.authed)
def test_rcptto(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'RCPT TO:<test">"addr>\r\n')
self.sock.sendall(b'250 2.1.5 Recipient <test">"addr> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn(b'RCPT TO:<test2>\r\n')
self.sock.sendall(b'250 2.1.5 Recipient <test2> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.ehlo_as = b'test'
s.have_mailfrom = True
s.handle()
self.assertTrue(s.have_rcptto)
def test_auth(self):
self.sock.sendall('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('EHLO there\r\n')
self.sock.sendall('250-Hello there\r\n250 AUTH PLAIN\r\n')
self.sock.recv(IsA(int)).AndReturn('AUTH PLAIN dGVzdHppZAB0ZXN0dXNlcgB0ZXN0cGFzc3dvcmQ=\r\n')
self.sock.sendall('235 2.7.0 Authentication successful\r\n')
self.sock.recv(IsA(int)).AndReturn('QUIT\r\n')
self.sock.sendall('221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.extensions.reset()
s.extensions.add('AUTH', FakeAuth(s))
s.handle()
assert_equal(('testuser', 'testzid'), s.auth_result)
def test_helo_empty_with_ehlo(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'HELO\r\n')
self.sock.sendall(b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
self.sock.sendall(b'250-Hello there\r\n250 TEST\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.extensions.reset()
s.extensions.add('TEST')
s.handle()
self.assertEqual('there', s.ehlo_as)
def test_starttls(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
self.sock.sendall(b'250-Hello there\r\n250 STARTTLS\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
self.sock.sendall(b'220 2.7.0 Go ahead\r\n')
self.context.wrap_socket(self.sock, server_side=True).AndReturn(self.sock)
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None, context=self.context)
s.extensions.reset()
s.extensions.add('STARTTLS')
s.handle()
self.assertEqual(None, s.ehlo_as)
def test_auth(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
self.sock.sendall(b'250-Hello there\r\n250 AUTH PLAIN\r\n')
self.sock.recv(IsA(int)).AndReturn(
b'AUTH PLAIN dGVzdHppZAB0ZXN0dXNlcgB0ZXN0cGFzc3dvcmQ=\r\n')
self.sock.sendall(b'235 2.7.0 Authentication successful\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.extensions.reset()
s.extensions.add('AUTH', AuthSession(SASLAuth([b'PLAIN']), s.io))
s.handle()
self.assertTrue(s.authed)
def test_bad_commands(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'BADCMD\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'AUTH\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
def test_custom_command(self):
class TestHandlers(object):
def TEST(self2, reply, arg, server):
self.assertTrue(server.have_mailfrom)
reply.code = '250'
reply.message = 'Doing '+arg.decode()
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'TEST stuff\r\n')
self.sock.sendall(b'250 2.0.0 Doing stuff\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, TestHandlers())
s.have_mailfrom = True
s.handle()
def test_bad_commands(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'BADCMD\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'AUTH\r\n')
self.sock.sendall(b'500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
def test_rset(self):
class TestHandlers(object):
server = None
def NOOP(self2, reply):
self.assertEqual(b'test', self2.server.ehlo_as)
self.assertFalse(self2.server.have_mailfrom)
self.assertFalse(self2.server.have_rcptto)
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'RSET arg\r\n')
self.sock.sendall(
b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'RSET\r\n')
self.sock.sendall(b'250 2.0.0 Ok\r\n')
self.sock.recv(IsA(int)).AndReturn(b'NOOP\r\n')
self.sock.sendall(b'250 2.0.0 Ok\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
h = TestHandlers()
s = h.server = Server(self.sock, h)
s.ehlo_as = b'test'
s.have_mailfrom = True
s.have_rcptto = True
s.handle()
def test_starttls(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
self.sock.sendall(b'250-Hello there\r\n250 STARTTLS\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
self.sock.sendall(b'220 2.7.0 Go ahead\r\n')
self.context.wrap_socket(self.sock,
server_side=True).AndReturn(self.sock)
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None, context=self.context)
s.extensions.reset()
s.extensions.add('STARTTLS')
s.handle()
self.assertEqual(None, s.ehlo_as)
def test_custom_command(self):
class TestHandlers(object):
def TEST(self2, reply, arg, server):
self.assertTrue(server.have_mailfrom)
reply.code = '250'
reply.message = 'Doing ' + arg.decode()
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'TEST stuff\r\n')
self.sock.sendall(b'250 2.0.0 Doing stuff\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, TestHandlers())
s.have_mailfrom = True
s.handle()
def test_starttls(self):
sock = self.mox.CreateMockAnything()
sock.fileno = lambda: -1
sock.sendall('220 ESMTP server\r\n')
sock.recv(IsA(int)).AndReturn('EHLO there\r\n')
sock.sendall('250-Hello there\r\n250 STARTTLS\r\n')
sock.recv(IsA(int)).AndReturn('STARTTLS\r\n')
sock.sendall('220 2.7.0 Go ahead\r\n')
sock.tls_wrapper(sock, self.tls_args).AndReturn(sock)
sock.recv(IsA(int)).AndReturn('QUIT\r\n')
sock.sendall('221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(sock, None, tls=self.tls_args, tls_wrapper=sock.tls_wrapper)
s.extensions.reset()
s.extensions.add('STARTTLS')
s.handle()
assert_equal(None, s.ehlo_as)
def test_starttls(self):
sock = self.mox.CreateMockAnything()
sock.fileno = lambda: -1
sock.sendall(b'220 ESMTP server\r\n')
sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
sock.sendall(b'250-Hello there\r\n250 STARTTLS\r\n')
sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
sock.sendall(b'220 2.7.0 Go ahead\r\n')
sock.tls_wrapper(sock, self.tls_args).AndReturn(sock)
sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(sock, None, tls=self.tls_args, tls_wrapper=sock.tls_wrapper)
s.extensions.reset()
s.extensions.add('STARTTLS')
s.handle()
self.assertEqual(None, s.ehlo_as)
def handle(self, socket, address):
try:
handlers = SmtpSession(address, self.validator_class, self.handoff)
smtp_server = Server(socket,
handlers,
self.auth_class,
self.tls,
self.tls_immediately,
command_timeout=self.command_timeout,
data_timeout=self.data_timeout)
if self.max_size:
smtp_server.extensions.add('SIZE', self.max_size)
smtp_server.handle()
except ConnectionLost:
pass
finally:
smtp_server.io.close()
def test_starttls_bad(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
self.sock.sendall(b'503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS badarg\r\n')
self.sock.sendall(b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
self.sock.sendall(b'250-Hello there\r\n250 STARTTLS\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
self.sock.sendall(b'220 2.7.0 Go ahead\r\n')
self.context.wrap_socket(self.sock, server_side=True).AndRaise(SSLError())
self.sock.sendall(b'421 4.7.0 TLS negotiation failed\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None, context=self.context)
s.extensions.reset()
s.extensions.add('STARTTLS')
s.handle()
self.assertEqual('there', s.ehlo_as)
def test_mailfrom_send_extension(self):
self.sock.sendall('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('EHLO there\r\n')
self.sock.sendall('250-Hello there\r\n250 SIZE 10\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FROM:<test1> SIZE=ASDF\r\n')
self.sock.sendall('501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FROM:<test1> SIZE=20\r\n')
self.sock.sendall('552 5.3.4 Message size exceeds 10 limit\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FROM:<test1> SIZE=5\r\n')
self.sock.sendall('250 2.1.0 Sender <test1> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn('QUIT\r\n')
self.sock.sendall('221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.extensions.reset()
s.extensions.add('SIZE', 10)
s.handle()
assert_true(s.have_mailfrom)
def test_mailfrom_send_extension(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
self.sock.sendall(b'250-Hello there\r\n250 SIZE 10\r\n')
self.sock.recv(IsA(int)).AndReturn(b'MAIL FROM:<test1> SIZE=ASDF\r\n')
self.sock.sendall(b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'MAIL FROM:<test1> SIZE=20\r\n')
self.sock.sendall(b'552 5.3.4 Message size exceeds 10 limit\r\n')
self.sock.recv(IsA(int)).AndReturn(b'MAIL FROM:<test1> SIZE=5\r\n')
self.sock.sendall(b'250 2.1.0 Sender <test1> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.extensions.reset()
s.extensions.add('SIZE', 10)
s.handle()
self.assertTrue(s.have_mailfrom)
def test_rcptto_bad(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'RCPT TO:<test>\r\n')
self.sock.sendall(b'503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn(b'HELO there\r\n')
self.sock.sendall(b'250 Hello there\r\n')
self.sock.recv(IsA(int)).AndReturn(b'RCPT TO:<test>\r\n')
self.sock.sendall(b'503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn(b'MAIL FROM:<test1>\r\n')
self.sock.sendall(b'250 2.1.0 Sender <test1> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn(b'RCPT T:<test1>\r\n')
self.sock.sendall(b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'RCPT TO:<test1\r\n')
self.sock.sendall(b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'QUIT\r\n')
self.sock.sendall(b'221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
self.assertFalse(s.have_rcptto)
def test_rcptto_bad(self):
self.sock.sendall('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('RCPT TO:<test>\r\n')
self.sock.sendall('503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn('HELO there\r\n')
self.sock.sendall('250 Hello there\r\n')
self.sock.recv(IsA(int)).AndReturn('RCPT TO:<test>\r\n')
self.sock.sendall('503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FROM:<test1>\r\n')
self.sock.sendall('250 2.1.0 Sender <test1> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn('RCPT T:<test1>\r\n')
self.sock.sendall('501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn('RCPT TO:<test1\r\n')
self.sock.sendall('501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn('QUIT\r\n')
self.sock.sendall('221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
assert_false(s.have_rcptto)
def test_starttls_bad(self):
sock = self.mox.CreateMockAnything()
sock.fileno = lambda: -1
sock.sendall(b'220 ESMTP server\r\n')
sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
sock.sendall(b'503 5.5.1 Bad sequence of commands\r\n')
sock.recv(IsA(int)).AndReturn(b'STARTTLS badarg\r\n')
sock.sendall(b'501 5.5.4 Syntax error in parameters or arguments\r\n')
sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
sock.sendall(b'250-Hello there\r\n250 STARTTLS\r\n')
sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
sock.sendall(b'220 2.7.0 Go ahead\r\n')
sock.tls_wrapper(sock, self.tls_args).AndRaise(SSLError())
sock.sendall(b'421 4.7.0 TLS negotiation failed\r\n')
self.mox.ReplayAll()
s = Server(sock, None, tls=self.tls_args, tls_wrapper=sock.tls_wrapper)
s.extensions.reset()
s.extensions.add('STARTTLS')
s.handle()
self.assertEqual('there', s.ehlo_as)
def test_starttls_bad(self):
sock = self.mox.CreateMockAnything()
sock.fileno = lambda: -1
sock.sendall('220 ESMTP server\r\n')
sock.recv(IsA(int)).AndReturn('STARTTLS\r\n')
sock.sendall('503 5.5.1 Bad sequence of commands\r\n')
sock.recv(IsA(int)).AndReturn('STARTTLS badarg\r\n')
sock.sendall('501 5.5.4 Syntax error in parameters or arguments\r\n')
sock.recv(IsA(int)).AndReturn('EHLO there\r\n')
sock.sendall('250-Hello there\r\n250 STARTTLS\r\n')
sock.recv(IsA(int)).AndReturn('STARTTLS\r\n')
sock.sendall('220 2.7.0 Go ahead\r\n')
sock.tls_wrapper(sock, self.tls_args).AndRaise(SSLError())
sock.sendall('421 4.7.0 TLS negotiation failed\r\n')
self.mox.ReplayAll()
s = Server(sock, None, tls=self.tls_args, tls_wrapper=sock.tls_wrapper)
s.extensions.reset()
s.extensions.add('STARTTLS')
s.handle()
assert_equal('there', s.ehlo_as)
def test_starttls_bad(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
self.sock.sendall(b'503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS badarg\r\n')
self.sock.sendall(
b'501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn(b'EHLO there\r\n')
self.sock.sendall(b'250-Hello there\r\n250 STARTTLS\r\n')
self.sock.recv(IsA(int)).AndReturn(b'STARTTLS\r\n')
self.sock.sendall(b'220 2.7.0 Go ahead\r\n')
self.context.wrap_socket(self.sock,
server_side=True).AndRaise(SSLError())
self.sock.sendall(b'421 4.7.0 TLS negotiation failed\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None, context=self.context)
s.extensions.reset()
s.extensions.add('STARTTLS')
s.handle()
self.assertEqual('there', s.ehlo_as)
def test_mailfrom_bad(self):
self.sock.sendall('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FROM:<test>\r\n')
self.sock.sendall('503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn('HELO there\r\n')
self.sock.sendall('250 Hello there\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FROM:<test1> SIZE=5\r\n')
self.sock.sendall('504 5.5.4 Command parameter not implemented\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FRM:<addr>\r\n')
self.sock.sendall('501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FROM:<addr\r\n')
self.sock.sendall('501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FROM:<test1>\r\n')
self.sock.sendall('250 2.1.0 Sender <test1> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn('MAIL FROM:<test2>\r\n')
self.sock.sendall('503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn('QUIT\r\n')
self.sock.sendall('221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
assert_true(s.have_mailfrom)
def test_gather_params(self):
s = Server(None, None)
self.assertEqual({b'ONE': b'1'}, s._gather_params(b' ONE=1'))
self.assertEqual({b'TWO': True}, s._gather_params(b'TWO'))
self.assertEqual({
b'THREE': b'foo',
b'FOUR': b'bar'
}, s._gather_params(b' THREE=foo FOUR=bar'))
self.assertEqual({b'FIVE': True}, s._gather_params(b'five'))
def test_data(self):
self.sock.sendall('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('DATA\r\n')
self.sock.sendall('354 Start mail input; end with <CRLF>.<CRLF>\r\n')
self.sock.recv(IsA(int)).AndReturn('.\r\nQUIT\r\n')
self.sock.sendall('250 2.6.0 Message Accepted for Delivery\r\n')
self.sock.sendall('221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.ehlo_as = 'test'
s.have_mailfrom = True
s.have_rcptto = True
s.handle()
def test_data_bad(self):
self.sock.sendall('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('DATA arg\r\n')
self.sock.sendall('501 5.5.4 Syntax error in parameters or arguments\r\n')
self.sock.recv(IsA(int)).AndReturn('DATA\r\n')
self.sock.sendall('503 5.5.1 Bad sequence of commands\r\n')
self.sock.recv(IsA(int)).AndReturn('QUIT\r\n')
self.sock.sendall('221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.ehlo_as = 'test'
s.have_mailfrom = True
s.handle()
def test_rcptto(self):
self.sock.sendall('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('RCPT TO:<test">"addr>\r\n')
self.sock.sendall('250 2.1.5 Recipient <test">"addr> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn('RCPT TO:<test2>\r\n')
self.sock.sendall('250 2.1.5 Recipient <test2> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn('QUIT\r\n')
self.sock.sendall('221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.ehlo_as = 'test'
s.have_mailfrom = True
s.handle()
assert_true(s.have_rcptto)
def test_call_custom_handler(self):
class TestHandler(object):
def TEST(self, arg):
return arg.lower()
s = Server(None, TestHandler())
assert_equal('stuff', s._call_custom_handler('TEST', 'STUFF'))
def test_gather_params(self):
s = Server(None, None)
assert_equal({'ONE': '1'}, s._gather_params(' ONE=1'))
assert_equal({'TWO': True}, s._gather_params('TWO'))
assert_equal({'THREE': 'foo', 'FOUR': 'bar'}, s._gather_params(' THREE=foo FOUR=bar'))
assert_equal({'FIVE': True}, s._gather_params('five'))
