コード例 #1
0
class TestInMemoryBufferedStreamWriter(object):

    SAMPLE_IMAGE = object()
    ANOTHER_SAMPLE_IMAGE = object()
    SAMPLE_TIMEOUT = 0
    SAMPLE_BUFFER_SIZE = 2

    def setup(self):
        self._stream_writer = InMemoryBufferedStreamWriter(self.SAMPLE_BUFFER_SIZE)

    def test_given_image_written_to_buffer_when_getting_image_then_image_is_returned(self):
        self._stream_writer.write(self.SAMPLE_IMAGE)
        image = self._stream_writer.get()
        assert_is(image, self.SAMPLE_IMAGE)

    def test_given_images_written_to_buffer_when_getting_images_then_images_are_returned_in_order(self):
        self._stream_writer.write(self.SAMPLE_IMAGE)
        self._stream_writer.write(self.ANOTHER_SAMPLE_IMAGE)

        image_1 = self._stream_writer.get()
        image_2 = self._stream_writer.get()

        assert_is(image_1, self.SAMPLE_IMAGE)
        assert_is(image_2, self.ANOTHER_SAMPLE_IMAGE)

    @raises(Empty)
    def test_given_empty_buffer_when_getting_image_then_nothing_is_returned(self):
        image = self._stream_writer.get(self.SAMPLE_TIMEOUT)
コード例 #2
0
 def setup(self):
     self._stream_writer = InMemoryBufferedStreamWriter(self.SAMPLE_BUFFER_SIZE)