def test_process_source_progress():
with ThreadPoolExecutor(2) as executor:
source = Source(path.join(here, 'files/1234'))
sink = Mock()
progress = Mock(spec=Progress)
with source:
process_source(executor, source, [sink], block_size=3, progress=progress)
progress.set.assert_has_calls([call(3), call(4)])
def test_process_source():
with ThreadPoolExecutor(2) as executor:
source = Source(path.join(here, 'files/empty'))
sink = Mock()
with source:
process_source(executor, source, [sink])
assert not sink.called
source = Source(path.join(here, 'files/1234'))
with source:
process_source(executor, source, [sink])
sink.process.assert_called_once_with(b'\x01\x02\x03\x04')
sink.reset_mock()
with source:
process_source(executor, source, [sink], block_size=1)
assert sink.process.call_count == 4