mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
import os
|
|
import pathlib
|
|
import sys
|
|
|
|
import pytest
|
|
|
|
from dvc.exceptions import GitHookAlreadyExistsError
|
|
from dvc.utils import file_md5
|
|
|
|
|
|
@pytest.mark.skipif(
|
|
sys.platform == "win32", reason="Git hooks aren't supported on Windows"
|
|
)
|
|
class TestInstall:
|
|
def _hook(self, name):
|
|
return pathlib.Path(".git") / "hooks" / name
|
|
|
|
def test_create_hooks(self, scm, dvc):
|
|
scm.install()
|
|
|
|
hooks_with_commands = [
|
|
("post-checkout", "exec dvc git-hook post-checkout"),
|
|
("pre-commit", "exec dvc git-hook pre-commit"),
|
|
("pre-push", "exec dvc git-hook pre-push"),
|
|
]
|
|
|
|
for fname, command in hooks_with_commands:
|
|
hook_path = self._hook(fname)
|
|
assert hook_path.is_file()
|
|
assert command in hook_path.read_text()
|
|
|
|
def test_fail_if_hook_exists(self, scm):
|
|
self._hook("post-checkout").write_text("hook content")
|
|
|
|
with pytest.raises(GitHookAlreadyExistsError):
|
|
scm.install()
|
|
|
|
def test_post_checkout(self, tmp_dir, scm, dvc):
|
|
scm.install()
|
|
tmp_dir.dvc_gen({"file": "file content"}, commit="add")
|
|
|
|
os.unlink("file")
|
|
scm.checkout("new_branch", create_new=True)
|
|
|
|
assert os.path.isfile("file")
|
|
|
|
def test_pre_push_hook(self, tmp_dir, scm, dvc, tmp_path_factory):
|
|
scm.install()
|
|
|
|
temp = tmp_path_factory.mktemp("external")
|
|
git_remote = temp / "project.git"
|
|
storage_path = temp / "dvc_storage"
|
|
|
|
with dvc.config.edit() as conf:
|
|
conf["remote"]["store"] = {"url": os.fspath(storage_path)}
|
|
conf["core"]["remote"] = "store"
|
|
tmp_dir.dvc_gen("file", "file_content", "commit message")
|
|
|
|
file_checksum = file_md5("file")[0]
|
|
expected_storage_path = (
|
|
storage_path / file_checksum[:2] / file_checksum[2:]
|
|
)
|
|
|
|
scm.repo.clone(os.fspath(git_remote))
|
|
scm.repo.create_remote("origin", os.fspath(git_remote))
|
|
|
|
assert not expected_storage_path.is_file()
|
|
scm.repo.git.push("origin", "master")
|
|
assert expected_storage_path.is_file()
|
|
assert expected_storage_path.read_text() == "file_content"
|