def test_get_delim(self): buffer = DataBuffer() self.assertEqual(b'', buffer.get_until_delim(b'2')) buffer.put(b'12345') buffer.put(b'678') self.assertEqual(b'12345', buffer.get_until_delim(b'5')) self.assertEqual(b'678', buffer.get_bytes(5)) self.assertEqual(b'', buffer.get_bytes(5)) self.assertEqual(b'', buffer.get_until_delim(b'2')) buffer.put(b'12345') buffer.put(b'678') self.assertEqual(b'123456', buffer.get_until_delim(b'56')) self.assertEqual(b'78', buffer.get_bytes(5))