Example #1
0
 def setUp(self):
     self._buffered_stream = BufferedStream(MockConnection(self.DATA), 0.1)
Example #2
0
 def setUp(self):
     self._buffered_stream = BufferedStream(MockConnection(self.DATA), 0.1)
Example #3
0
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')
Example #4
0
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')