def test_read(self): buf = StringBuffer() buf.feed(b'hello world') data = buf.read(5) self.assertEqual(data, b'hello') data = buf.read(6) self.assertEqual(data, b' world') data = buf.read(100) self.assertEqual(data, None)
def test_read_until(self): buf = StringBuffer() buf.feed(b'hello\nworld\n') data = buf.read_until(b'\n') self.assertEqual(data, b'hello\n') data = buf.read_until(b'\n') self.assertEqual(data, b'world\n') data = buf.read_until(b'\n') self.assertEqual(data, None)
def test_read_until_regex(self): regex = re.compile(b'~~') buf = StringBuffer() buf.feed(b'hello~~world~~') data = buf.read_until_regex(regex) self.assertEqual(data, b'hello~~') data = buf.read_until_regex(regex) self.assertEqual(data, b'world~~') data = buf.read_until_regex(regex) self.assertEqual(data, None)
def test_close(self): buf = StringBuffer() buf.feed(b'hello world') buf.close() self.assertTrue(buf.closed) self.assertRaises(ValueError, buf.read, 5)
def test_clear(self): buf = StringBuffer() buf.feed(b'hello world') buf.clear() data = buf.read(5) self.assertEqual(data, None)