def test_remove_downloads_exact(get_zip_path, safe_remove):
file_path = '/path/file.zip'
get_zip_path.return_value = file_path
safe_remove.return_value = True
assert mod.remove_downloads('/root', content_ids=['md5']) == 1
safe_remove.assert_called_once_with(file_path)
get_zip_path.assert_called_once_with('md5', '/root')
def test_remove_downloads_all(get_downloads, safe_remove):
file_path = '/path/file.zip'
get_downloads.return_value = [file_path]
safe_remove.return_value = True
assert mod.remove_downloads('/root', extension='zip') == 1
safe_remove.assert_called_once_with(file_path)
get_downloads.assert_called_once_with('/root', 'zip')
def test_remove_downloads_fail():
with pytest.raises(TypeError):
mod.remove_downloads('/path/dir')
