def test_extra_tabs(self):
self.assertEqual(BytesProtocol().write(b'foo', b'bar\tbaz'),
b'foo\tbar\tbaz')
self.assertEqual(BytesProtocol().write(b'foo\tbar', b'baz'),
b'foo\tbar\tbaz')
self.assertEqual(BytesProtocol().read(b'foo\tbar\tbaz'),
(b'foo', b'bar\tbaz'))
def test_no_tabs(self):
self.assertEqual(BytesProtocol().write(b'foo', None), b'foo')
self.assertEqual(BytesProtocol().write(None, b'foo'), b'foo')
self.assertEqual(BytesProtocol().read(b'foo'), (b'foo', None))
self.assertEqual(BytesProtocol().write(b'', None), b'')
self.assertEqual(BytesProtocol().write(None, None), b'')
self.assertEqual(BytesProtocol().write(None, b''), b'')
self.assertEqual(BytesProtocol().read(b''), (b'', None))
def test_no_strip(self):
self.assertEqual(BytesProtocol().read(b'foo\t \n\n'),
(b'foo', b' \n\n'))
def test_round_trip(self):
self.assertRoundTripOK(BytesProtocol(), b'foo', b'bar')
self.assertRoundTripOK(BytesProtocol(), b'foo', None)
self.assertRoundTripOK(BytesProtocol(), b'foo', b'')
self.assertRoundTripOK(BytesProtocol(), b'caf\xe9', b'\xe90\c1a')
