def test_read(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
self.assertEqual(stream.read(3), b'foo')
stream.buffer.feed(b'foo bar')
self.assertEqual(stream.read(3), b'foo')
self.assertEqual(stream.read(4), b' bar')
def test_read(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
self.assertEqual(stream.read(3), b'foo')
stream.buffer.feed(b'foo bar')
self.assertEqual(stream.read(3), b'foo')
self.assertEqual(stream.read(4), b' bar')
def test_read_eof_and_error(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
stream.buffer.feed_eof()
stream.buffer.feed_error(RuntimeError)
self.assertEqual(stream.read(), b'foo')
self.assertEqual(stream.read(), b'')
def test_read_incr(self):
stream = Stream(None)
buf = b'foobar'
for i in range(len(buf)):
stream.buffer.feed(buf[i:i+1])
stream.buffer.feed_eof()
self.assertEqual(stream.read(), b'foobar')
def test_read_eof_and_error(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
stream.buffer.feed_eof()
stream.buffer.feed_error(RuntimeError)
self.assertEqual(stream.read(), b'foo')
self.assertEqual(stream.read(), b'')
def test_read_incr(self):
stream = Stream(None)
buf = b'foobar'
for i in range(len(buf)):
stream.buffer.feed(buf[i:i + 1])
stream.buffer.feed_eof()
self.assertEqual(stream.read(), b'foobar')
def test_read_wait_error(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
def write_more():
gruvi.sleep(0.01)
stream.buffer.feed(b'bar')
gruvi.sleep(0.01)
stream.buffer.feed_error(RuntimeError)
gruvi.spawn(write_more)
self.assertEqual(stream.read(), b'foobar')
self.assertRaises(RuntimeError, stream.read)
def test_read_wait(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
def write_more():
gruvi.sleep(0.01)
stream.buffer.feed(b'bar')
gruvi.sleep(0.01)
stream.buffer.feed_eof()
gruvi.spawn(write_more)
self.assertEqual(stream.read(), b'foobar')
self.assertEqual(stream.read(), b'')
def test_read_wait_error(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
def write_more():
gruvi.sleep(0.01)
stream.buffer.feed(b'bar')
gruvi.sleep(0.01)
stream.buffer.feed_error(RuntimeError)
gruvi.spawn(write_more)
self.assertEqual(stream.read(), b'foobar')
self.assertRaises(RuntimeError, stream.read)
def test_read_wait(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
def write_more():
gruvi.sleep(0.01)
stream.buffer.feed(b'bar')
gruvi.sleep(0.01)
stream.buffer.feed_eof()
gruvi.spawn(write_more)
self.assertEqual(stream.read(), b'foobar')
self.assertEqual(stream.read(), b'')
def test_read_eof(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
stream.buffer.feed_eof()
self.assertEqual(stream.read(), b'foo')
self.assertEqual(stream.read(), b'')
def test_read_eof(self):
stream = Stream(None)
stream.buffer.feed(b'foo')
stream.buffer.feed_eof()
self.assertEqual(stream.read(), b'foo')
self.assertEqual(stream.read(), b'')