class TestVideoStreamProcessor(object):
SAMPLE_IMAGE = object()
@mock.patch('services.video_streaming.stream_writers.InMemoryBufferedStreamWriter.InMemoryBufferedStreamWriter', autospec = False)
def setup(self, stream_writer_mock):
self._stream_writer_mock = stream_writer_mock
self._video_stream_processor = VideoStreamProcessor(self._stream_writer_mock)
self._video_stream_processor.start_async()
def teardown(self):
self._video_stream_processor.stop()
def test_when_processing_image_then_image_is_dispatched_to_stream_writer(self):
self._video_stream_processor.process(self.SAMPLE_IMAGE)
time.sleep(0.25)
self._stream_writer_mock.write.assert_called_with(self.SAMPLE_IMAGE)
def _init_video_processors(self):
for stream_writer in self._stream_writers:
video_processor = VideoStreamProcessor(stream_writer)
self._video_processors.append(video_processor)
video_processor.start_async()
def setup(self, stream_writer_mock):
self._stream_writer_mock = stream_writer_mock
self._video_stream_processor = VideoStreamProcessor(self._stream_writer_mock)
self._video_stream_processor.start_async()
