def test_banner_quit(self):
self.sock.sendall('220 ESMTP server\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()
def test_tls_immediately_sslerror(self):
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, tls_immediately=True)
s.handle()
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_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('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('QUIT arg\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()
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('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('NOOP\r\n')
self.sock.sendall('250 2.0.0 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.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_tls_immediately_sslerror(self):
sock = self.mox.CreateMockAnything()
sock.fileno = lambda: -1
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_immediately=True,
tls_wrapper=sock.tls_wrapper)
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(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_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_helo(self):
self.sock.sendall('220 ESMTP server\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('QUIT\r\n')
self.sock.sendall('221 2.0.0 Bye\r\n')
self.mox.ReplayAll()
s = Server(self.sock, None)
s.handle()
assert_equal('there', 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_unhandled_error(self):
class TestHandler(object):
def BANNER_(self, reply):
raise Exception('test')
self.sock.sendall('421 4.3.0 Unhandled system error\r\n')
self.mox.ReplayAll()
s = Server(self.sock, TestHandler())
with assert_raises(Exception) as cm:
s.handle()
assert_equal(('test', ), cm.exception.args)
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_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_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_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_banner_command(self):
self.sock.sendall('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('BANNER\r\n')
self.sock.sendall('500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn('BANNER_\r\n')
self.sock.sendall('500 5.5.2 Syntax error, command unrecognized\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()
def test_tls_immediately(self):
sock = self.mox.CreateMockAnything()
sock.fileno = lambda: -1
sock.tls_wrapper(sock, self.tls_args).AndReturn(sock)
sock.sendall('220 ESMTP server\r\n')
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_immediately=True,
tls_wrapper=sock.tls_wrapper)
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_ehlo(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 TEST\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('TEST')
s.handle()
assert_equal('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_mailfrom(self):
self.sock.sendall('220 ESMTP server\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:<test">"addr>\r\n')
self.sock.sendall('250 2.1.0 Sender <test">"addr> 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.handle()
assert_true(s.have_mailfrom)
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_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(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 test_data(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'.\r\nQUIT\r\n')
self.sock.sendall(b'250 2.6.0 Message accepted for delivery\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.have_rcptto = True
s.handle()
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_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 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_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_custom_command(self):
class TestHandlers(object):
def TEST(self2, reply, arg, server):
assert_true(server.have_mailfrom)
reply.code = '250'
reply.message = 'Doing '+arg
self.sock.sendall('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('TEST stuff\r\n')
self.sock.sendall('250 2.0.0 Doing stuff\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, 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_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_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_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('220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn('\r\n')
self.sock.sendall('500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn('BADCMD\r\n')
self.sock.sendall('500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn('STARTTLS\r\n')
self.sock.sendall('500 5.5.2 Syntax error, command unrecognized\r\n')
self.sock.recv(IsA(int)).AndReturn('AUTH\r\n')
self.sock.sendall('500 5.5.2 Syntax error, command unrecognized\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()
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_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 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_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_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_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_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_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):
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_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_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_mailfrom_bad(self):
self.sock.sendall(b'220 ESMTP server\r\n')
self.sock.recv(IsA(int)).AndReturn(b'MAIL FROM:<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'MAIL FROM:<test1> SIZE=5\r\n')
self.sock.sendall(b'504 5.5.4 Command parameter not implemented\r\n')
self.sock.recv(IsA(int)).AndReturn(b'MAIL FRM:<addr>\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:<addr\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>\r\n')
self.sock.sendall(b'250 2.1.0 Sender <test1> Ok\r\n')
self.sock.recv(IsA(int)).AndReturn(b'MAIL FROM:<test2>\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.handle()
self.assertTrue(s.have_mailfrom)
def test_tls_immediately_sslerror(self):
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, tls_immediately=True)
s.handle()
