test_branch

Spitfire1900 · Dec 4, 2024 · 856c4ad · 856c4ad
1 parent c4ac13b
commit 856c4ad
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_xontrib.py b/tests/test_xontrib.py
@@ -48,6 +48,15 @@ def prompts(load_xontrib: abc.Callable[[str], None],
     yield prompts
 
 
+def test_branch(git_repo):
+    with cd(git_repo):
+        repo = Repo(git_repo).init()
+        repo.index.commit('initial commit')
+        repo.create_head('test_branch')
+        repo.git.checkout('test_branch')
+        assert PromptFormatter()('{pygitstatus.branch}') == '{CYAN}test_branch'
+
+
 def test_clean(git_repo):
     with cd(git_repo):
         assert PromptFormatter()('{pygitstatus.clean}') == '{BOLD_GREEN}✓{RESET}'