mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 23:10:42 +01:00
skshetry
9a1bcd7f1d
The code are wrapped due to us using line-length=79 before, but now some of the code can easily fit in 88 chars. The formatter did not change this automatically when we changed line-length, because of magic trailing commas that gets added when they turn into multiple lines. Black/ruff wrapping them in parens and splitting them into multiple lines was making it difficult to read.
109 lines
3.2 KiB
Python
109 lines
3.2 KiB
Python
from datetime import datetime
|
|
|
|
from scmrepo.exceptions import SCMError
|
|
|
|
from dvc.repo.experiments import ExpRefInfo
|
|
from dvc.scm import iter_revs
|
|
|
|
|
|
def test_iter_revs(tmp_dir, scm, mocker):
|
|
"""
|
|
new other
|
|
│ │
|
|
old (tag) ────┘
|
|
│
|
|
root
|
|
"""
|
|
old = scm.active_branch()
|
|
tmp_dir.scm_gen("foo", "init", commit="init")
|
|
rev_root = scm.get_rev()
|
|
tmp_dir.scm_gen("foo", "old", commit="old")
|
|
rev_old = scm.get_rev()
|
|
scm.checkout("new", create_new=True)
|
|
tmp_dir.scm_gen("foo", "new", commit="new")
|
|
rev_new = scm.get_rev()
|
|
scm.checkout(old)
|
|
scm.tag("tag")
|
|
scm.checkout("other", create_new=True)
|
|
tmp_dir.scm_gen("foo", "other", commit="new")
|
|
rev_other = scm.get_rev()
|
|
|
|
ref = ExpRefInfo(rev_root, "exp1")
|
|
scm.set_ref(str(ref), rev_new)
|
|
ref = ExpRefInfo(rev_root, "exp2")
|
|
scm.set_ref(str(ref), rev_old)
|
|
|
|
gen = iter_revs(scm, [rev_root, "new"], 1)
|
|
assert gen == {rev_root: [rev_root], rev_new: ["new"]}
|
|
gen = iter_revs(scm, ["new"], 2)
|
|
assert gen == {rev_new: ["new"], rev_old: [rev_old]}
|
|
gen = iter_revs(scm, ["other"], -1)
|
|
assert gen == {
|
|
rev_other: ["other"],
|
|
rev_old: [rev_old],
|
|
rev_root: [rev_root],
|
|
}
|
|
gen = iter_revs(scm, ["tag"])
|
|
assert gen == {rev_old: ["tag"]}
|
|
gen = iter_revs(scm, all_branches=True)
|
|
assert gen == {rev_old: [old], rev_new: ["new"], rev_other: ["other"]}
|
|
gen = iter_revs(scm, all_tags=True)
|
|
assert gen == {rev_old: ["tag"]}
|
|
gen = iter_revs(scm, all_commits=True)
|
|
assert gen == {
|
|
rev_old: [rev_old],
|
|
rev_new: [rev_new],
|
|
rev_other: [rev_other],
|
|
rev_root: [rev_root],
|
|
}
|
|
gen = iter_revs(scm, all_experiments=True)
|
|
assert gen == {
|
|
rev_new: [rev_new],
|
|
rev_old: [rev_old],
|
|
}
|
|
|
|
def _resolve_commit(rev):
|
|
from scmrepo.git.objects import GitCommit
|
|
|
|
if rev == rev_root:
|
|
return GitCommit(
|
|
"dummy",
|
|
commit_time=datetime(2022, 6, 28).timestamp(),
|
|
commit_time_offset=0,
|
|
message="dummy",
|
|
parents=["dummy"],
|
|
committer_name="dummy",
|
|
committer_email="dummy",
|
|
author_name="dummy",
|
|
author_email="dummy",
|
|
author_time=datetime(2022, 6, 28).timestamp(),
|
|
author_time_offset=0,
|
|
)
|
|
if rev == rev_old:
|
|
raise SCMError
|
|
return GitCommit(
|
|
"dummy",
|
|
commit_time=datetime(2022, 6, 30).timestamp(),
|
|
commit_time_offset=0,
|
|
message="dummy",
|
|
parents=["dummy"],
|
|
committer_name="dummy",
|
|
committer_email="dummy",
|
|
author_name="dummy",
|
|
author_email="dummy",
|
|
author_time=datetime(2022, 6, 28).timestamp(),
|
|
author_time_offset=0,
|
|
)
|
|
|
|
mocker.patch(
|
|
"scmrepo.git.Git.resolve_commit",
|
|
mocker.MagicMock(side_effect=_resolve_commit),
|
|
)
|
|
|
|
gen = iter_revs(scm, commit_date="2022-06-29")
|
|
assert gen == {
|
|
rev_new: [rev_new],
|
|
rev_old: [rev_old],
|
|
rev_other: [rev_other],
|
|
}
|