def test_rcptto(self): self.sock.sendall(b'RCPT TO:<test>\r\n') self.sock.recv(IsA(int)).AndReturn(b'250 2.0.0 Ok\r\n') self.mox.ReplayAll() client = LmtpClient(self.sock) reply = client.rcptto('test') self.assertEqual('250', reply.code) self.assertEqual('2.0.0 Ok', reply.message) self.assertEqual(b'RCPT', reply.command) self.assertEqual([('test', reply)], client.rcpttos)
def test_rcptto(self): self.sock.sendall(b"RCPT TO:<test>\r\n") self.sock.recv(IsA(int)).AndReturn(b"250 2.0.0 Ok\r\n") self.mox.ReplayAll() client = LmtpClient(self.sock) reply = client.rcptto("test") self.assertEqual("250", reply.code) self.assertEqual("2.0.0 Ok", reply.message) self.assertEqual(b"RCPT", reply.command) self.assertEqual([("test", reply)], client.rcpttos)
def test_rset(self): self.sock.sendall(b"RSET\r\n") self.sock.recv(IsA(int)).AndReturn(b"250 Ok\r\n") self.mox.ReplayAll() client = LmtpClient(self.sock) client.rcpttos = "testing" reply = client.rset() self.assertEqual("250", reply.code) self.assertEqual("2.0.0 Ok", reply.message) self.assertEqual(b"RSET", reply.command) self.assertEqual([], client.rcpttos)
def test_rset(self): self.sock.sendall(b'RSET\r\n') self.sock.recv(IsA(int)).AndReturn(b'250 Ok\r\n') self.mox.ReplayAll() client = LmtpClient(self.sock) client.rcpttos = 'testing' reply = client.rset() self.assertEqual('250', reply.code) self.assertEqual('2.0.0 Ok', reply.message) self.assertEqual(b'RSET', reply.command) self.assertEqual([], client.rcpttos)
def test_lhlo(self): self.sock.sendall(b"LHLO there\r\n") self.sock.recv(IsA(int)).AndReturn(b"250-Hello there\r\n250-TEST arg\r\n") self.sock.recv(IsA(int)).AndReturn(b"250 EXTEN\r\n") self.mox.ReplayAll() client = LmtpClient(self.sock) reply = client.lhlo("there") self.assertEqual("250", reply.code) self.assertEqual("Hello there", reply.message) self.assertEqual(b"LHLO", reply.command) self.assertTrue("TEST" in client.extensions) self.assertTrue("EXTEN" in client.extensions) self.assertEqual("arg", client.extensions.getparam("TEST"))
def test_lhlo(self): self.sock.sendall(b'LHLO there\r\n') self.sock.recv(IsA(int)).AndReturn(b'250-Hello there\r\n250-TEST arg\r\n') self.sock.recv(IsA(int)).AndReturn(b'250 EXTEN\r\n') self.mox.ReplayAll() client = LmtpClient(self.sock) reply = client.lhlo('there') self.assertEqual('250', reply.code) self.assertEqual('Hello there', reply.message) self.assertEqual(b'LHLO', reply.command) self.assertTrue('TEST' in client.extensions) self.assertTrue('EXTEN' in client.extensions) self.assertEqual('arg', client.extensions.getparam('TEST'))
def test_lhlo(self): self.sock.sendall('LHLO there\r\n') self.sock.recv(IsA(int)).AndReturn('250-Hello there\r\n250-TEST arg\r\n') self.sock.recv(IsA(int)).AndReturn('250 EXTEN\r\n') self.mox.ReplayAll() client = LmtpClient(self.sock) reply = client.lhlo('there') self.assertEqual('250', reply.code) self.assertEqual('Hello there', reply.message) self.assertEqual('LHLO', reply.command) self.assertTrue('TEST' in client.extensions) self.assertTrue('EXTEN' in client.extensions) self.assertEqual('arg', client.extensions.getparam('TEST'))
def test_send_empty_data(self): self.sock.sendall(b".\r\n") self.sock.recv(IsA(int)).AndReturn(b"250 2.0.0 Ok\r\n" b"550 5.0.0 Not Ok\r\n") self.mox.ReplayAll() client = LmtpClient(self.sock) client.rcpttos = [("test1", Reply("250")), ("test2", Reply("250")), ("test3", Reply("550"))] replies = client.send_empty_data() self.assertEqual(2, len(replies)) self.assertEqual("test1", replies[0][0]) self.assertEqual("250", replies[0][1].code) self.assertEqual("2.0.0 Ok", replies[0][1].message) self.assertEqual(b"[SEND_DATA]", replies[0][1].command) self.assertEqual("test2", replies[1][0]) self.assertEqual("550", replies[1][1].code) self.assertEqual("5.0.0 Not Ok", replies[1][1].message) self.assertEqual(b"[SEND_DATA]", replies[1][1].command)
def test_send_empty_data(self): self.sock.sendall(b'.\r\n') self.sock.recv(IsA(int)).AndReturn(b'250 2.0.0 Ok\r\n' b'550 5.0.0 Not Ok\r\n') self.mox.ReplayAll() client = LmtpClient(self.sock) client.rcpttos = [('test1', Reply('250')), ('test2', Reply('250')), ('test3', Reply('550'))] replies = client.send_empty_data() self.assertEqual(2, len(replies)) self.assertEqual('test1', replies[0][0]) self.assertEqual('250', replies[0][1].code) self.assertEqual('2.0.0 Ok', replies[0][1].message) self.assertEqual(b'[SEND_DATA]', replies[0][1].command) self.assertEqual('test2', replies[1][0]) self.assertEqual('550', replies[1][1].code) self.assertEqual('5.0.0 Not Ok', replies[1][1].message) self.assertEqual(b'[SEND_DATA]', replies[1][1].command)
def test_helo_invalid(self): client = LmtpClient(self.sock) self.assertRaises(NotImplementedError, client.helo, 'there')