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)