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_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)