@@ -140,7 +140,7 @@ Assert that files in tests/ end in `_test.py`.
 #### `no-commit-to-branch`
 Protect specific branches from direct checkins.
 - Use `args: [--branch, staging, --branch, master]` to set the branch.
-`master` is the default if no branch argument is set.
+Both `master` and `main` are protected by default if no branch argument is set.
 - `-b` / `--branch` may be specified multiple times to protect multiple
 branches.
 - `-p` / `--pattern` can be used to protect branches that match a supplied regex

@@ -38,7 +38,7 @@ def main(argv: Optional[Sequence[str]] = None) -> int:
 )
 args = parser.parse_args(argv)
 
-protected = frozenset(args.branch or ('master',))
+protected = frozenset(args.branch or ('master', 'main'))
 patterns = frozenset(args.pattern or ())
 return int(is_on_branch(protected, patterns))
 

@@ -67,3 +67,10 @@ def test_not_on_a_branch(temp_git_dir):
 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