mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
99 lines
2.8 KiB
Python
99 lines
2.8 KiB
Python
import os
|
|
import shutil
|
|
|
|
from mock import patch
|
|
|
|
from dvc.repo import Repo
|
|
from dvc.main import main
|
|
from dvc.compat import fspath
|
|
from tests.basic_env import TestDvc
|
|
from dvc.external_repo import clean_repos
|
|
|
|
|
|
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, erepo_dir):
|
|
with erepo_dir.branch("branch", new=True), erepo_dir.chdir():
|
|
erepo_dir.scm.repo.index.remove([".dvc"], r=True)
|
|
shutil.rmtree(".dvc")
|
|
erepo_dir.scm_gen("file", "first version")
|
|
erepo_dir.scm.add(["file"])
|
|
erepo_dir.scm.commit("first version")
|
|
|
|
dvc.imp(fspath(erepo_dir), "file", "file", rev="branch")
|
|
|
|
status = dvc.status(["file.dvc"])
|
|
|
|
assert status == {}
|
|
|
|
# Caching in external repos doesn't see upstream updates within single
|
|
# cli call, so we need to clean the caches to see the changes.
|
|
clean_repos()
|
|
|
|
with erepo_dir.branch("branch", new=False), erepo_dir.chdir():
|
|
erepo_dir.scm_gen("file", "second_version", commit="update file")
|
|
erepo_dir.scm.add(["file"])
|
|
erepo_dir.scm.commit("first version")
|
|
|
|
status, = dvc.status(["file.dvc"])["file.dvc"]
|
|
|
|
assert status == {
|
|
"changed deps": {
|
|
"file ({})".format(fspath(erepo_dir)): "update available"
|
|
}
|
|
}
|
|
|
|
|
|
def test_status_before_and_after_dvc_init(tmp_dir, dvc, erepo_dir):
|
|
with erepo_dir.chdir():
|
|
erepo_dir.scm.repo.index.remove([".dvc"], r=True)
|
|
shutil.rmtree(".dvc")
|
|
erepo_dir.scm_gen("file", "first version")
|
|
erepo_dir.scm.add(["file"])
|
|
erepo_dir.scm.commit("first version")
|
|
old_rev = erepo_dir.scm.get_rev()
|
|
|
|
dvc.imp(fspath(erepo_dir), "file", "file")
|
|
|
|
assert dvc.status(["file.dvc"]) == {}
|
|
|
|
with erepo_dir.chdir():
|
|
Repo.init()
|
|
erepo_dir.scm.repo.index.remove(["file"])
|
|
os.remove("file")
|
|
erepo_dir.dvc_gen("file", "second version")
|
|
erepo_dir.scm.add([".dvc", "file.dvc"])
|
|
erepo_dir.scm.commit("version with dvc")
|
|
new_rev = erepo_dir.scm.get_rev()
|
|
|
|
assert old_rev != new_rev
|
|
|
|
# Caching in external repos doesn't see upstream updates within single
|
|
# cli call, so we need to clean the caches to see the changes.
|
|
clean_repos()
|
|
|
|
status, = dvc.status(["file.dvc"])["file.dvc"]
|
|
|
|
assert status == {
|
|
"changed deps": {
|
|
"file ({})".format(fspath(erepo_dir)): "update available"
|
|
}
|
|
}
|