def test_handle_list(self, get_files):
get_files.return_value = (f for f in self.files)
processor = BaseListProcessor(remove_processed_files=False)
with patch.object(processor, 'handle_list') as handle_list:
processor.handle()
handle_list.assert_called_with(
get_files.return_value, **{'remove_processed_files': False})
def test_handle_list_deletes(self, get_files):
get_files.return_value = (f for f in self.files)
processor = BaseListProcessor(remove_processed_files=True)
storage = MagicMock()
processor.storage = storage
with patch.object(processor, 'handle_list') as handle_list:
processor.handle()
handle_list.assert_called_with(
get_files.return_value, **{'remove_processed_files': True})
self.assertEqual(storage.delete.call_count, 4)
