def test_error_string_with_custom_class(self): self.reader = hiredis.Reader(replyError=RuntimeError) self.reader.feed(b'-error\r\n') error = self.reply() self.assertEqual(RuntimeError, type(error)) self.assertEqual(('error', ), error.args)
def test_protocol_error_with_custom_callable(self): class CustomException(Exception): pass self.reader = hiredis.Reader( protocolError=lambda e: CustomException(e)) self.reader.feed(b'x') self.assertRaises(CustomException, self.reply)
def test_error_string_with_custom_callable(self): class CustomException(Exception): pass self.reader = hiredis.Reader(replyError=lambda e: CustomException(e)) self.reader.feed(b'-error\r\n') error = self.reply() self.assertEqual(CustomException, type(error)) self.assertEqual(('error', ), error.args)
def test_protocol_error_with_custom_class(self): self.reader = hiredis.Reader(protocolError=RuntimeError) self.reader.feed(b'x') self.assertRaises(RuntimeError, self.reply)
def test_bulk_string_with_invalid_encoding(self): self.reader = hiredis.Reader(encoding='unknown') self.reader.feed(b'$5\r\nhello\r\n') self.assertRaises(LookupError, self.reply)
def test_bulk_string_with_other_encoding(self): snowman = b'\xe2\x98\x83' self.reader = hiredis.Reader(encoding='utf-32') self.reader.feed(b'$3\r\n' + snowman + b'\r\n') self.assertEqual(snowman, self.reply())
def setUp(self): self.reader = hiredis.Reader()
def test_bulk_string_with_encoding(self): snowman = '\xe2\x98\x83' self.reader = hiredis.Reader(encoding=u'utf-8') self.reader.feed('$3\r\n' + snowman + '\r\n') self.assertEqual(snowman.decode(u'utf-8'), self.reply())