mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-07 07:18:27 +01:00
Saugat PachhaiandRuslan Kuprieiev
1937527249
* dvc: use lockfile as yaml instead of json * refactor Dvcfile into Pipeline file and Single stage file * fix tests * dvc: fix outputs * add more tests for collection of outputs * add tests for data cloud/get/import/ls * tests: test for checkouts * Allow other checksums other than md5 * tests: use iterdir instead of os.listdir * cleanup errors reported by cc and ds * utils: throw DvcException instead of plain Exception * tests: use yaml.load instead of json.load * run: split assignments * Update dvc/stage/exceptions.py Co-authored-by: Ruslan Kuprieiev <[email protected]>
96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
import os
|
|
|
|
from mock import patch
|
|
|
|
from dvc.main import main
|
|
from dvc.compat import fspath
|
|
from tests.basic_env import TestDvc
|
|
|
|
|
|
class TestStatus(TestDvc):
|
|
def test_quiet(self):
|
|
self.dvc.add(self.FOO)
|
|
|
|
ret = main(["status", "--quiet"])
|
|
self.assertEqual(ret, 0)
|
|
|
|
os.remove(self.FOO)
|
|
os.rename(self.BAR, self.FOO)
|
|
|
|
ret = main(["status", "--quiet"])
|
|
self.assertEqual(ret, 1)
|
|
|
|
@patch("dvc.repo.status._cloud_status", return_value=True)
|
|
def test_implied_cloud(self, mock_status):
|
|
main(["status", "--remote", "something"])
|
|
mock_status.assert_called()
|
|
|
|
|
|
def test_status_non_dvc_repo_import(tmp_dir, dvc, git_dir):
|
|
with git_dir.branch("branch", new=True):
|
|
git_dir.scm_gen("file", "first version", commit="first version")
|
|
|
|
dvc.imp(fspath(git_dir), "file", "file", rev="branch")
|
|
|
|
assert dvc.status(["file.dvc"]) == {}
|
|
|
|
with git_dir.branch("branch", new=False):
|
|
git_dir.scm_gen("file", "second version", commit="update file")
|
|
|
|
(status,) = dvc.status(["file.dvc"])["file.dvc"]
|
|
assert status == {
|
|
"changed deps": {"file ({})".format(git_dir): "update available"}
|
|
}
|
|
|
|
|
|
def test_status_before_and_after_dvc_init(tmp_dir, dvc, git_dir):
|
|
git_dir.scm_gen("file", "first version", commit="first verison")
|
|
old_rev = git_dir.scm.get_rev()
|
|
|
|
dvc.imp(fspath(git_dir), "file", "file")
|
|
|
|
assert dvc.status(["file.dvc"]) == {}
|
|
|
|
with git_dir.chdir():
|
|
git_dir.init(dvc=True)
|
|
git_dir.scm.repo.index.remove(["file"])
|
|
os.remove("file")
|
|
git_dir.dvc_gen("file", "second version", commit="with dvc")
|
|
new_rev = git_dir.scm.get_rev()
|
|
|
|
assert old_rev != new_rev
|
|
|
|
(status,) = dvc.status(["file.dvc"])["file.dvc"]
|
|
assert status == {
|
|
"changed deps": {
|
|
"file ({})".format(fspath(git_dir)): "update available"
|
|
}
|
|
}
|
|
|
|
|
|
def test_status_on_pipeline_stages(tmp_dir, dvc, run_copy):
|
|
tmp_dir.dvc_gen("foo", "foo")
|
|
stage = run_copy("foo", "bar", name="copy-foo-bar")
|
|
|
|
stage.cmd = " ".join(stage.cmd.split())
|
|
stage.dvcfile._dump_pipeline_file(stage)
|
|
assert dvc.status() == {"pipelines.yaml:copy-foo-bar": ["changed command"]}
|
|
|
|
# delete outputs
|
|
(tmp_dir / "bar").unlink()
|
|
assert dvc.status() == {
|
|
"pipelines.yaml:copy-foo-bar": [
|
|
{"changed outs": {"bar": "deleted"}},
|
|
"changed command",
|
|
]
|
|
}
|
|
(tmp_dir / "foo").unlink()
|
|
assert dvc.status() == {
|
|
"foo.dvc": [{"changed outs": {"foo": "deleted"}}],
|
|
"pipelines.yaml:copy-foo-bar": [
|
|
{"changed deps": {"foo": "deleted"}},
|
|
{"changed outs": {"bar": "deleted"}},
|
|
"changed command",
|
|
],
|
|
}
|