def test_recv_nonutf8(self):
self.sock.recv(IsA(int)).AndReturn(b'250 \xff\r\n')
self.mox.ReplayAll()
io = IO(self.sock)
code, message = io.recv_reply()
self.assertEqual('250', code)
self.assertEqual('�', message)
def test_recv_reply_multiline(self):
self.sock.recv(IsA(int)).AndReturn(b'250-One\r\n250 Two\r\n')
self.mox.ReplayAll()
io = IO(self.sock)
code, message = io.recv_reply()
self.assertEqual('250', code)
self.assertEqual('One\r\nTwo', message)
def test_recv_reply(self):
self.sock.recv(IsA(int)).AndReturn('250 Ok\r\n')
self.mox.ReplayAll()
io = IO(self.sock)
code, message = io.recv_reply()
self.assertEqual('250', code)
self.assertEqual('Ok', message)
def test_recv_reply_multiline(self):
self.sock.recv(IsA(int)).AndReturn(b'250-One\r\n250 Two\r\n')
self.mox.ReplayAll()
io = IO(self.sock)
code, message = io.recv_reply()
self.assertEqual('250', code)
self.assertEqual('One\r\nTwo', message)
def test_recv_utf8(self):
self.sock.recv(IsA(int)).AndReturn(b'250 \xc3\xbf\r\n')
self.mox.ReplayAll()
io = IO(self.sock)
code, message = io.recv_reply()
self.assertEqual('250', code)
self.assertEqual(u'\xff', message)
def test_recv_reply(self):
self.sock.recv(IsA(int)).AndReturn(b'250 Ok\r\n')
self.mox.ReplayAll()
io = IO(self.sock)
code, message = io.recv_reply()
self.assertEqual('250', code)
self.assertEqual('Ok', message)
def test_recv_reply_multipart(self):
self.sock.recv(IsA(int)).AndReturn('250 ')
self.sock.recv(IsA(int)).AndReturn('Ok\r\n')
self.mox.ReplayAll()
io = IO(self.sock)
code, message = io.recv_reply()
assert_equal('250', code)
assert_equal('Ok', message)
