mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 15:28:34 +01:00
Ruslan Kuprieiev
6559b9d222
* dvc: use protected mode by default Fixes #3261 Fixes #2041 * unprotect: adjust help message https://github.com/iterative/dvc.org/pull/1058/files/21aab371a487acf6f6e6201b29cd832e7c55ed23#r393324668
61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
from dvc.cache import NamedCache
|
|
from dvc.path_info import PathInfo
|
|
from dvc.remote.local import RemoteLOCAL
|
|
|
|
|
|
def test_status_download_optimization(mocker):
|
|
"""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
|
|
"""
|
|
remote = RemoteLOCAL(None, {})
|
|
|
|
infos = NamedCache()
|
|
infos.add("local", "acbd18db4cc2f85cedef654fccc4a4d8", "foo")
|
|
infos.add("local", "37b51d194a7513e45b56f6524f2d51f2", "bar")
|
|
|
|
local_exists = list(infos["local"])
|
|
mocker.patch.object(remote, "cache_exists", return_value=local_exists)
|
|
|
|
other_remote = mocker.Mock()
|
|
other_remote.url = "other_remote"
|
|
other_remote.cache_exists.return_value = []
|
|
|
|
remote.status(infos, other_remote, download=True)
|
|
|
|
assert other_remote.cache_exists.call_count == 0
|
|
|
|
|
|
@pytest.mark.parametrize("link_name", ["hardlink", "symlink"])
|
|
def test_is_protected(tmp_dir, link_name):
|
|
remote = RemoteLOCAL(None, {})
|
|
link_method = getattr(remote, link_name)
|
|
|
|
(tmp_dir / "foo").write_text("foo")
|
|
|
|
foo = PathInfo(tmp_dir / "foo")
|
|
link = PathInfo(tmp_dir / "link")
|
|
|
|
link_method(foo, link)
|
|
|
|
assert not remote.is_protected(foo)
|
|
assert not remote.is_protected(link)
|
|
|
|
remote.protect(foo)
|
|
|
|
assert remote.is_protected(foo)
|
|
assert remote.is_protected(link)
|
|
|
|
remote.unprotect(link)
|
|
|
|
assert not remote.is_protected(link)
|
|
if link_name == "symlink" and os.name == "nt":
|
|
# NOTE: Windows symlink perms don't propagate to the target
|
|
assert remote.is_protected(foo)
|
|
else:
|
|
assert not remote.is_protected(foo)
|