def setUp(self): self._buffered_stream = BufferedStream(MockConnection(self.DATA), 0.1)
class TestBufferedStream(TestCase): DATA = 'foobardiibadaa' def setUp(self): self._buffered_stream = BufferedStream(MockConnection(self.DATA), 0.1) def test_normal_read(self): self.assertEquals(self.DATA, self._buffered_stream.read(len(self.DATA))) def test_empty(self): self._buffered_stream.read(len('foobar')) self._buffered_stream.empty() self.assertRaises(AssertionError, self._buffered_stream.read, len(self.DATA) - len('foobar')) def test_read_all(self): data = self._buffered_stream.read(-1) self.assertEquals(data, self.DATA) def test_read_and_return(self): self._buffered_stream.read(-1) self._buffered_stream.return_data('badaa') data = self._buffered_stream.read(-1) self.assertEquals(data, 'badaa')