def test_custom_command(self): self.sock.sendall(b'cmd arg\r\n') self.sock.recv(IsA(int)).AndReturn(b'250 Ok\r\n') self.mox.ReplayAll() client = Client(self.sock) reply = client.custom_command(b'cmd', b'arg') self.assertEqual('250', reply.code) self.assertEqual('2.0.0 Ok', reply.message) self.assertEqual(b'CMD', reply.command)
def test_custom_command(self): self.sock.sendall(b"cmd arg\r\n") self.sock.recv(IsA(int)).AndReturn(b"250 Ok\r\n") self.mox.ReplayAll() client = Client(self.sock) reply = client.custom_command(b"cmd", b"arg") self.assertEqual("250", reply.code) self.assertEqual("2.0.0 Ok", reply.message) self.assertEqual(b"CMD", reply.command)