forge-tools/tests/test_context.py

import unittest.mock

import pytest

from frg.context import GitContext, get_git_context, parse_remote_url


@pytest.fixture(name="mock_context")
def fixture_mock_context():
    return GitContext(
        repo_name="test-repo",
        repo_author="test-user",
        current_branch="main",
        host="https://git.my.server",
    )


@pytest.fixture(autouse=True)
def mock_subprocess(mock_context):
    def mock_run_impl(*args, **kwargs):
        call_args = args[0][1:]
        return_val = unittest.mock.Mock()

        if call_args == ["branch", "--show-current"]:
            return_val.stdout = mock_context.current_branch.encode("utf8")
        if call_args == ["config", "--get", "remote.origin.url"]:
            return_val.stdout = f"{mock_context.host}/{mock_context.repo_author}/{mock_context.repo_name}.git".encode(
                "utf8"
            )

        return_val.stderr = b""
        return_val.returncode = 0

        return return_val

    with unittest.mock.patch("subprocess.run") as mock_run:
        mock_run.side_effect = mock_run_impl
        yield


def test_git_context_pulls_current_branch(mock_context):
    assert get_git_context().current_branch == mock_context.current_branch


def test_git_context_pulls_host(mock_context):
    assert get_git_context().host == mock_context.host


def test_git_context_pulls_repo_owner(mock_context):
    assert get_git_context().repo_author == mock_context.repo_author


def test_git_context_pulls_repo_name(mock_context):
    assert get_git_context().repo_name == mock_context.repo_name


def test_git_context_remaps_host_if_aliase_provided(mock_context):
    aliases = {mock_context.host: "other.host"}

    assert get_git_context(domain_aliases=aliases).host == "other.host"


@pytest.mark.parametrize(
    "url, parts",
    [
        ["git@myhost.com:user/repo.git", ("https://myhost.com", "user", "repo")],
        ["https://myhost.com/user/repo.git", ("https://myhost.com", "user", "repo")],
    ],
    ids=["ssh", "http"],
)
def test_parse_remote_url_parses_url(url, parts):
    actual_parts = parse_remote_url(url)

    assert parts == actual_parts