mirror of
https://github.com/pre-commit/pre-commit-hooks.git
synced 2026-04-05 11:36:54 +00:00
123 lines
3.9 KiB
Python
123 lines
3.9 KiB
Python
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from pre_commit_hooks.no_commit_to_branch import _default_branch
|
|
from pre_commit_hooks.no_commit_to_branch import is_on_branch
|
|
from pre_commit_hooks.no_commit_to_branch import main
|
|
from pre_commit_hooks.util import cmd_output
|
|
from testing.util import git_commit
|
|
|
|
|
|
def test_other_branch(temp_git_dir):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', 'anotherbranch')
|
|
assert is_on_branch({'placeholder'}) is False
|
|
|
|
|
|
def test_multi_branch(temp_git_dir):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', 'another/branch')
|
|
assert is_on_branch({'placeholder'}) is False
|
|
|
|
|
|
def test_multi_branch_fail(temp_git_dir):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', 'another/branch')
|
|
assert is_on_branch({'another/branch'}) is True
|
|
|
|
|
|
def test_exact_branch(temp_git_dir):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', 'branchname')
|
|
assert is_on_branch({'branchname'}) is True
|
|
|
|
|
|
def test_main_branch_call(temp_git_dir):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', 'other')
|
|
assert main(('--branch', 'other')) == 1
|
|
|
|
|
|
@pytest.mark.parametrize('branch_name', ('b1', 'b2'))
|
|
def test_forbid_multiple_branches(temp_git_dir, branch_name):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', branch_name)
|
|
assert main(('--branch', 'b1', '--branch', 'b2'))
|
|
|
|
|
|
def test_branch_pattern_fail(temp_git_dir):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', 'another/branch')
|
|
assert is_on_branch(set(), {'another/.*'}) is True
|
|
|
|
|
|
@pytest.mark.parametrize('branch_name', ('somebranch', 'another/branch'))
|
|
def test_branch_pattern_multiple_branches_fail(temp_git_dir, branch_name):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', branch_name)
|
|
assert main(('--branch', 'somebranch', '--pattern', 'another/.*'))
|
|
|
|
|
|
def test_main_default_call(temp_git_dir):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', 'anotherbranch')
|
|
assert main(()) == 0
|
|
|
|
|
|
def test_not_on_a_branch(temp_git_dir):
|
|
with temp_git_dir.as_cwd():
|
|
git_commit('--allow-empty', '-m1')
|
|
head = cmd_output('git', 'rev-parse', 'HEAD').strip()
|
|
cmd_output('git', 'checkout', head)
|
|
# we're not on a branch!
|
|
assert main(()) == 0
|
|
|
|
|
|
@pytest.mark.parametrize('branch_name', ('master', 'main'))
|
|
def test_default_branch_names(temp_git_dir, branch_name):
|
|
with temp_git_dir.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', branch_name)
|
|
assert main(()) == 1
|
|
|
|
|
|
def test_default_branch_falls_back_when_empty_branch(tmpdir):
|
|
with patch(
|
|
'pre_commit_hooks.no_commit_to_branch.cmd_output',
|
|
return_value='origin/',
|
|
):
|
|
assert _default_branch() == frozenset({'master', 'main'})
|
|
|
|
|
|
def test_default_branch_detects_from_origin(tmpdir):
|
|
remote = tmpdir.join('remote')
|
|
cmd_output('git', 'init', '--', str(remote))
|
|
with remote.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', 'develop')
|
|
git_commit('--allow-empty', '-m', 'init')
|
|
|
|
local = tmpdir.join('local')
|
|
cmd_output('git', 'clone', str(remote), str(local))
|
|
|
|
with local.as_cwd():
|
|
assert _default_branch() == frozenset({'develop'})
|
|
|
|
|
|
def test_main_blocks_detected_default_branch(tmpdir):
|
|
remote = tmpdir.join('remote')
|
|
cmd_output('git', 'init', '--', str(remote))
|
|
with remote.as_cwd():
|
|
cmd_output('git', 'checkout', '-b', 'develop')
|
|
git_commit('--allow-empty', '-m', 'init')
|
|
|
|
local = tmpdir.join('local')
|
|
cmd_output('git', 'clone', str(remote), str(local))
|
|
|
|
with local.as_cwd():
|
|
# On detected default branch — should be blocked
|
|
assert main(()) == 1
|
|
# On a feature branch — should pass
|
|
cmd_output('git', 'checkout', '-b', 'feature')
|
|
assert main(()) == 0
|