mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Ruslan Kuprieiev
58295ca0ab
We used to have it named `cache` before, but these days it is clear that you can have three odbs: just odb (e.g. in-memory), cache (typically local) or remote (typically cloud). Fixing naming here before proceeding with introducing `out.odb` and `out.cache`. In preparation for migration to https://github.com/iterative/dvc-data/pull/304
108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
import os
|
|
|
|
import pytest
|
|
|
|
from dvc.cli import main
|
|
from dvc.fs import system
|
|
from dvc.stage.exceptions import StageFileDoesNotExistError, StageFileIsNotDvcFileError
|
|
from dvc.utils.fs import remove
|
|
from dvc_objects.errors import ObjectDBError
|
|
from tests.utils import get_gitignore_content
|
|
|
|
|
|
@pytest.mark.parametrize("remove_outs", [True, False])
|
|
def test_remove(tmp_dir, scm, dvc, run_copy, remove_outs):
|
|
(stage1,) = tmp_dir.dvc_gen("foo", "foo")
|
|
stage2 = run_copy("foo", "bar", single_stage=True)
|
|
stage3 = run_copy("bar", "foobar", name="copy-bar-foobar")
|
|
|
|
assert "/foo" in get_gitignore_content()
|
|
assert "/bar" in get_gitignore_content()
|
|
assert "/foobar" in get_gitignore_content()
|
|
|
|
for stage in [stage1, stage2, stage3]:
|
|
dvc.remove(stage.addressing, outs=remove_outs)
|
|
out_exists = (out.exists for out in stage.outs)
|
|
assert stage not in dvc.index.stages
|
|
if remove_outs:
|
|
assert not any(out_exists)
|
|
else:
|
|
assert all(out_exists)
|
|
|
|
assert not (tmp_dir / ".gitignore").exists()
|
|
|
|
|
|
def test_remove_file_target(tmp_dir, dvc):
|
|
tmp_dir.dvc_gen("foo", "foo")
|
|
|
|
with pytest.raises(
|
|
StageFileIsNotDvcFileError,
|
|
match="'foo' is not a .dvc file. Do you mean 'foo.dvc'?",
|
|
):
|
|
dvc.remove("foo")
|
|
|
|
dvc.remove("foo.dvc")
|
|
|
|
|
|
def test_remove_non_existent_file(tmp_dir, dvc):
|
|
with pytest.raises(StageFileDoesNotExistError):
|
|
dvc.remove("non_existent_dvc_file.dvc")
|
|
with pytest.raises(StageFileDoesNotExistError):
|
|
dvc.remove("non_existent_stage_name")
|
|
|
|
|
|
def test_remove_broken_symlink(tmp_dir, dvc):
|
|
tmp_dir.gen("foo", "foo")
|
|
dvc.cache.local.cache_types = ["symlink"]
|
|
|
|
(stage,) = dvc.add("foo")
|
|
remove(dvc.cache.local.path)
|
|
assert system.is_symlink("foo")
|
|
|
|
with pytest.raises(ObjectDBError):
|
|
dvc.remove(stage.addressing)
|
|
assert os.path.lexists("foo")
|
|
assert (tmp_dir / stage.relpath).exists()
|
|
|
|
dvc.remove(stage.addressing, outs=True)
|
|
assert not os.path.lexists("foo")
|
|
assert not (tmp_dir / stage.relpath).exists()
|
|
|
|
|
|
def test_cmd_remove(tmp_dir, dvc):
|
|
assert main(["remove", "non-existing-dvc-file"]) == 1
|
|
|
|
(stage,) = tmp_dir.dvc_gen("foo", "foo")
|
|
assert main(["remove", stage.addressing]) == 0
|
|
assert not (tmp_dir / stage.relpath).exists()
|
|
assert (tmp_dir / "foo").exists()
|
|
|
|
(stage,) = tmp_dir.dvc_gen("foo", "foo")
|
|
assert main(["remove", stage.addressing, "--outs"]) == 0
|
|
assert not (tmp_dir / stage.relpath).exists()
|
|
assert not (tmp_dir / "foo").exists()
|
|
|
|
|
|
def test_cmd_remove_gitignore_single_stage(tmp_dir, scm, dvc, run_copy):
|
|
stage = dvc.run(name="my", cmd='echo "hello" > out', deps=[], outs=["out"])
|
|
|
|
assert (tmp_dir / ".gitignore").exists()
|
|
|
|
assert main(["remove", stage.addressing]) == 0
|
|
assert not (tmp_dir / stage.relpath).exists()
|
|
assert not (stage.dvcfile._lockfile).exists()
|
|
assert not (tmp_dir / ".gitignore").exists()
|
|
|
|
|
|
def test_cmd_remove_gitignore_multistage(tmp_dir, scm, dvc, run_copy):
|
|
(stage,) = tmp_dir.dvc_gen("foo", "foo")
|
|
stage1 = run_copy("foo", "foo1", single_stage=True)
|
|
stage2 = run_copy("foo1", "foo2", name="copy-foo1-foo2")
|
|
|
|
assert (tmp_dir / ".gitignore").exists()
|
|
|
|
assert main(["remove", stage2.addressing]) == 0
|
|
assert main(["remove", stage1.addressing]) == 0
|
|
assert main(["remove", stage.addressing]) == 0
|
|
assert not (tmp_dir / ".gitignore").exists()
|