dvc/tests/func/test_used_objs.py

import json
import os

import pytest

from dvc.exceptions import NoOutputOrStageError
from dvc.stage.exceptions import StageFileDoesNotExistError

gitfs_xfail = pytest.mark.xfail(
    raises=(NoOutputOrStageError, StageFileDoesNotExistError),
    reason="gitfs works on absolute paths only",
)


@pytest.mark.parametrize(
    "stage_wdir, cwd, target",
    [
        (os.curdir, os.curdir, "foo"),
        (os.curdir, os.curdir, "train"),
        (os.curdir, os.curdir, "dvc.yaml:train"),
        (os.curdir, "sub", os.path.join(os.pardir, "foo")),
        pytest.param(
            os.curdir,
            "sub",
            os.path.join(os.pardir, "dvc.yaml:train"),
            marks=gitfs_xfail,
        ),
        ("sub", os.curdir, os.path.join("sub", "foo")),
        ("sub", os.curdir, os.path.join("sub", "dvc.yaml:train")),
        ("sub", "sub", "foo"),
        pytest.param("sub", "sub", "train", marks=gitfs_xfail),
        pytest.param("sub", "sub", "dvc.yaml:train", marks=gitfs_xfail),
        ("sub", "dir", os.path.join(os.pardir, "sub", "foo")),
        pytest.param(
            "sub",
            "dir",
            os.path.join(os.pardir, "sub", "dvc.yaml:train"),
            marks=gitfs_xfail,
        ),
    ],
)
def test_from_gitfs_when_pwd_not_in_root(
    tmp_dir, scm, dvc, stage_wdir, cwd, target
):
    path = tmp_dir.joinpath(stage_wdir).resolve()
    path.mkdir(parents=True, exist_ok=True)
    wdir = tmp_dir.joinpath(cwd).resolve()
    wdir.mkdir(parents=True, exist_ok=True)

    (path / "dvc.yaml").write_text(
        json.dumps(
            {"stages": {"train": {"cmd": "echo foo > foo", "outs": ["foo"]}}}
        )
    )
    path.gen({"foo": "foo"})
    dvc.commit(None, force=True)
    tmp_dir.scm_add(
        [path / file for file in ("dvc.yaml", "dvc.lock", ".gitignore")],
        commit="add files",
    )

    with wdir.chdir():
        assert dvc.used_objs([target], revs=[scm.get_rev()])