mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-08 07:18:27 +01:00
Peter Rowlands (변기호)
0fb9870de6
* remote: s/get_checksum/get_hash/ * remote/tree: s/get_file_checksum/get_file_hash/ * remote: s/get_dir_checksum/get_dir_hash/ * remote: s/checksum_to_path/hash_to_path/ * remote: s/path_to_checksum/path_to_hash/ * remote: cleanup remaining checksum references * remote.index: s/checksum/hash/ * note: does not rename "checksum" field in the index DB schema
92 lines
2.6 KiB
Python
92 lines
2.6 KiB
Python
import errno
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from dvc.cache import NamedCache
|
|
from dvc.path_info import PathInfo
|
|
from dvc.remote.index import RemoteIndexNoop
|
|
from dvc.remote.local import LocalCache, LocalRemoteTree
|
|
|
|
|
|
def test_status_download_optimization(mocker, dvc):
|
|
"""When comparing the status to pull a remote cache,
|
|
And the desired files to fetch are already on the local cache,
|
|
Don't check the existence of the desired files on the remote cache
|
|
"""
|
|
cache = LocalCache(LocalRemoteTree(dvc, {}))
|
|
|
|
infos = NamedCache()
|
|
infos.add("local", "acbd18db4cc2f85cedef654fccc4a4d8", "foo")
|
|
infos.add("local", "37b51d194a7513e45b56f6524f2d51f2", "bar")
|
|
|
|
local_exists = list(infos["local"])
|
|
mocker.patch.object(cache, "hashes_exist", return_value=local_exists)
|
|
|
|
other_remote = mocker.Mock()
|
|
other_remote.url = "other_remote"
|
|
other_remote.hashes_exist.return_value = []
|
|
other_remote.index = RemoteIndexNoop()
|
|
|
|
cache.status(infos, other_remote, download=True)
|
|
|
|
assert other_remote.hashes_exist.call_count == 0
|
|
|
|
|
|
@pytest.mark.parametrize("link_name", ["hardlink", "symlink"])
|
|
def test_is_protected(tmp_dir, dvc, link_name):
|
|
tree = LocalRemoteTree(dvc, {})
|
|
link_method = getattr(tree, link_name)
|
|
|
|
(tmp_dir / "foo").write_text("foo")
|
|
|
|
foo = PathInfo(tmp_dir / "foo")
|
|
link = PathInfo(tmp_dir / "link")
|
|
|
|
link_method(foo, link)
|
|
|
|
assert not tree.is_protected(foo)
|
|
assert not tree.is_protected(link)
|
|
|
|
tree.protect(foo)
|
|
|
|
assert tree.is_protected(foo)
|
|
assert tree.is_protected(link)
|
|
|
|
tree.unprotect(link)
|
|
|
|
assert not tree.is_protected(link)
|
|
if os.name == "nt" and link_name == "hardlink":
|
|
# NOTE: NTFS doesn't allow deleting read-only files, which forces us to
|
|
# set write perms on the link, which propagates to the source.
|
|
assert not tree.is_protected(foo)
|
|
else:
|
|
assert tree.is_protected(foo)
|
|
|
|
|
|
@pytest.mark.parametrize("err", [errno.EPERM, errno.EACCES])
|
|
def test_protect_ignore_errors(tmp_dir, mocker, err):
|
|
tmp_dir.gen("foo", "foo")
|
|
foo = PathInfo("foo")
|
|
tree = LocalRemoteTree(None, {})
|
|
|
|
tree.protect(foo)
|
|
|
|
mock_chmod = mocker.patch(
|
|
"os.chmod", side_effect=OSError(err, "something")
|
|
)
|
|
tree.protect(foo)
|
|
assert mock_chmod.called
|
|
|
|
|
|
def test_protect_ignore_erofs(tmp_dir, mocker):
|
|
tmp_dir.gen("foo", "foo")
|
|
foo = PathInfo("foo")
|
|
tree = LocalRemoteTree(None, {})
|
|
|
|
mock_chmod = mocker.patch(
|
|
"os.chmod", side_effect=OSError(errno.EROFS, "read-only fs")
|
|
)
|
|
tree.protect(foo)
|
|
assert mock_chmod.called
|