Beispiel #1
0
    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))
    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))