How to Get Current Git Branch Name in Python with Mock Subprocess

import subprocess
from unittest.mock import patch, MagicMock
from git import Repo
import os


def get_current_branch(repo_path="."):
    """Get the current branch name of a git repository."""
    repo = Repo(repo_path)
    return repo.active_branch.name


if __name__ == "__main__":
    # Mock subprocess to control the git output
    mock_process = MagicMock()
    mock_process.stdout = "feature/user-auth"
    
    with patch.object(subprocess, "run", return_value=mock_process):
        with patch.object(Repo, "active_branch", new_callable=MagicMock):
            Repo.active_branch.return_value = MagicMock(name="feature/user-auth")
            
            # Alternative: mock at subprocess level
            with patch("git.cmd.Git.execute") as mock_execute:
                mock_execute.return_value = "feature/user-auth"
                branch = get_current_branch()
                print(f"Current branch: {branch}")