mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Ruslan Kuprieiev
c83a4ea406
* erepo: don't use _get_tree_for * erepo: refactor download/fetch logic * dvc: unify handling existing output
110 lines
3.3 KiB
Python
110 lines
3.3 KiB
Python
import os
|
|
from unittest.mock import call
|
|
|
|
import pytest
|
|
|
|
from dvc.external_repo import external_repo
|
|
from dvc.path_info import PathInfo
|
|
from tests.unit.tree.test_repo import make_subrepo
|
|
|
|
|
|
def test_hook_is_called(tmp_dir, erepo_dir, mocker):
|
|
subrepo_paths = [
|
|
"subrepo1",
|
|
"subrepo2",
|
|
os.path.join("dir", "subrepo3"),
|
|
os.path.join("dir", "subrepo4"),
|
|
"subrepo5",
|
|
os.path.join("subrepo5", "subrepo6"),
|
|
]
|
|
subrepos = [erepo_dir / path for path in subrepo_paths]
|
|
for repo in subrepos:
|
|
make_subrepo(repo, erepo_dir.scm)
|
|
|
|
for repo in subrepos + [erepo_dir]:
|
|
with repo.chdir():
|
|
repo.scm_gen("foo", "foo", commit=f"git add {repo}/foo")
|
|
repo.dvc_gen("bar", "bar", commit=f"dvc add {repo}/bar")
|
|
|
|
with external_repo(str(erepo_dir)) as repo:
|
|
spy = mocker.spy(repo, "make_repo")
|
|
|
|
list(repo.repo_tree.walk(repo.root_dir)) # drain
|
|
assert spy.call_count == len(subrepos)
|
|
|
|
paths = [os.path.join(repo.root_dir, path) for path in subrepo_paths]
|
|
spy.assert_has_calls([call(path) for path in paths], any_order=True)
|
|
|
|
|
|
@pytest.mark.parametrize("root_is_dvc", [False, True])
|
|
def test_subrepo_is_constructed_properly(
|
|
tmp_dir, scm, mocker, make_tmp_dir, root_is_dvc
|
|
):
|
|
if root_is_dvc:
|
|
make_subrepo(tmp_dir, scm)
|
|
|
|
subrepo = tmp_dir / "subrepo"
|
|
make_subrepo(subrepo, scm)
|
|
local_cache = subrepo.dvc.cache.local.cache_dir
|
|
|
|
tmp_dir.scm_gen("bar", "bar", commit="add bar")
|
|
subrepo.dvc_gen("foo", "foo", commit="add foo")
|
|
|
|
cache_dir = make_tmp_dir("temp-cache")
|
|
with external_repo(
|
|
str(tmp_dir), cache_dir=str(cache_dir), cache_types=["symlink"]
|
|
) as repo:
|
|
spy = mocker.spy(repo, "make_repo")
|
|
|
|
list(repo.repo_tree.walk(repo.root_dir)) # drain
|
|
assert spy.call_count == 1
|
|
subrepo = spy.return_value
|
|
|
|
assert repo.url == str(tmp_dir)
|
|
assert repo._cache_config["cache"]["dir"] == str(cache_dir)
|
|
assert repo.cache.local.cache_dir == str(cache_dir)
|
|
assert subrepo.cache.local.cache_dir == str(cache_dir)
|
|
|
|
assert repo._cache_config["cache"]["type"] == ["symlink"]
|
|
assert repo.cache.local.cache_types == ["symlink"]
|
|
assert subrepo.cache.local.cache_types == ["symlink"]
|
|
|
|
assert (
|
|
subrepo.config["remote"]["auto-generated-upstream"]["url"]
|
|
== local_cache
|
|
)
|
|
if root_is_dvc:
|
|
main_cache = tmp_dir.dvc.cache.local.cache_dir
|
|
assert repo.config["remote"]["auto-generated-upstream"][
|
|
"url"
|
|
] == str(main_cache)
|
|
|
|
|
|
def test_fetch_external_repo_jobs(tmp_dir, scm, mocker, dvc, local_remote):
|
|
tmp_dir.dvc_gen(
|
|
{
|
|
"dir1": {
|
|
"file1": "file1",
|
|
"file2": "file2",
|
|
"file3": "file3",
|
|
"file4": "file4",
|
|
},
|
|
},
|
|
commit="init",
|
|
)
|
|
|
|
dvc.push()
|
|
|
|
with external_repo(str(tmp_dir)) as repo:
|
|
spy = mocker.spy(repo.cloud, "pull")
|
|
|
|
dvc.cache.local.save(
|
|
PathInfo(repo.root_dir) / "dir1",
|
|
repo.repo_tree,
|
|
None,
|
|
follow_subrepos=False,
|
|
jobs=3,
|
|
)
|
|
run_jobs = tuple(spy.call_args_list[0])[1].get("jobs")
|
|
assert run_jobs == 3
|