tests: add test for clone(shallow_branch=...)

iterative · Mar 16, 2022 · 9891e36 · 9891e36
1 parent 9f36ffc
commit 9891e36
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_scmrepo.py b/tests/test_scmrepo.py
@@ -18,3 +18,13 @@ def test_clone(tmp_dir: TmpDir, matcher: Type[Matcher]):
 
     progress.assert_called_with(matcher.instance_of(GitProgressEvent))
     assert (tmp_dir / "dir").exists()
+
+
+def test_clone_shallow(tmp_dir: TmpDir):
+    url = "https://github.com/iterative/dvcyaml-schema"
+    shallow_branch = "master"
+
+    repo = Git.clone(url, "dir", shallow_branch=shallow_branch)
+    shallow_file = tmp_dir / "dir" / ".git" / "shallow"
+    assert shallow_file.exists()
+    assert repo.get_rev() in shallow_file.read_text().splitlines()