Yellhorn MCP

"""Unit tests for handling large content in GitHub operations."""

import asyncio
import os
import tempfile
from pathlib import Path
from unittest.mock import AsyncMock, MagicMock, call, patch

import pytest

from yellhorn_mcp.integrations.github_integration import (
    add_issue_comment,
    create_github_issue,
    update_issue_with_workplan,
)
from yellhorn_mcp.utils.git_utils import (
    add_github_issue_comment,
    create_github_subissue,
    update_github_issue,
)


class TestLargeContentHandling:
    """Test suite for handling large content in GitHub operations."""

    @pytest.mark.asyncio
    async def test_create_issue_with_large_body(self):
        """Test that create_github_issue uses --body-file for large content."""
        repo_path = Path("/test/repo")

        # Create a large body (simulate content that would cause "Argument list too long")
        large_body = "x" * 200000  # 200KB of content

        # Track the commands executed
        executed_commands = []

        async def mock_command_func(path, command):
            executed_commands.append(command)
            # Check if --body-file is used
            if "--body-file" in command:
                # Verify the file was created
                file_idx = command.index("--body-file") + 1
                tmp_file = command[file_idx]
                # In real execution, the file would exist
                # For testing, we just verify the command structure
                return "https://github.com/test/repo/issues/123"
            return ""

        # Mock ensure_label_exists to avoid actual GitHub calls
        with patch("yellhorn_mcp.integrations.github_integration.ensure_label_exists"):
            result = await create_github_issue(
                repo_path,
                "Test Issue",
                large_body,
                github_command_func=mock_command_func,
            )

        # Verify --body-file was used
        assert len(executed_commands) == 1
        assert "--body-file" in executed_commands[0]
        assert "--body" not in executed_commands[0]
        assert result["number"] == "123"
        assert result["url"] == "https://github.com/test/repo/issues/123"

    @pytest.mark.asyncio
    async def test_update_issue_with_large_workplan(self):
        """Test that update_issue_with_workplan uses --body-file for large content."""
        repo_path = Path("/test/repo")

        # Create a large workplan
        large_workplan = "# Large Workplan\n\n" + "x" * 200000

        # Track commands executed
        executed_commands = []

        async def mock_command_func(path, command):
            executed_commands.append(command)
            return ""

        await update_issue_with_workplan(
            repo_path,
            "123",
            large_workplan,
            None,
            github_command_func=mock_command_func,
        )

        # Verify --body-file was used
        assert len(executed_commands) == 1
        assert "--body-file" in executed_commands[0]
        assert "--body" not in executed_commands[0]

    @pytest.mark.asyncio
    async def test_add_comment_with_large_content(self):
        """Test that add_issue_comment uses --body-file for large content."""
        repo_path = Path("/test/repo")

        # Create a large comment
        large_comment = "This is a very large comment:\n\n" + "y" * 200000

        # Track commands executed
        executed_commands = []

        async def mock_command_func(path, command):
            executed_commands.append(command)
            return ""

        await add_issue_comment(
            repo_path,
            "456",
            large_comment,
            github_command_func=mock_command_func,
        )

        # Verify --body-file was used
        assert len(executed_commands) == 1
        assert "--body-file" in executed_commands[0]
        assert "--body" not in executed_commands[0]

    @pytest.mark.asyncio
    async def test_add_github_issue_comment_with_large_content(self):
        """Test that add_github_issue_comment in git_utils uses --body-file."""
        repo_path = Path("/test/repo")

        # Create a comment
        comment_text = "GitHub comment text"

        # Simply verify the command is constructed correctly
        with patch("yellhorn_mcp.utils.git_utils.run_github_command") as mock_run:
            mock_run.return_value = ""

            await add_github_issue_comment(repo_path, "789", comment_text)

            # Verify run_github_command was called with --body-file
            assert mock_run.called
            call_args = mock_run.call_args[0]
            assert call_args[0] == repo_path
            command = call_args[1]
            assert "issue" in command
            assert "comment" in command
            assert "789" in command
            assert "--body-file" in command
            assert "--body" not in command

    @pytest.mark.asyncio
    async def test_update_github_issue_with_large_body(self):
        """Test that update_github_issue in git_utils uses --body-file."""
        repo_path = Path("/test/repo")

        # Create a body
        body_text = "Updated issue body"

        # Mock run_github_command to verify command
        with patch("yellhorn_mcp.utils.git_utils.run_github_command") as mock_run:
            mock_run.return_value = ""

            await update_github_issue(repo_path, "999", body=body_text)

            # Verify run_github_command was called with --body-file
            assert mock_run.called
            call_args = mock_run.call_args[0]
            assert call_args[0] == repo_path
            command = call_args[1]
            assert "issue" in command
            assert "edit" in command
            assert "999" in command
            assert "--body-file" in command
            assert "--body" not in command

    @pytest.mark.asyncio
    async def test_create_github_subissue_with_large_body(self):
        """Test that create_github_subissue uses --body-file."""
        repo_path = Path("/test/repo")

        # Create a body
        body_text = "Sub-issue body"

        # Mock dependencies
        with patch("yellhorn_mcp.utils.git_utils.ensure_label_exists"):
            with patch("yellhorn_mcp.utils.git_utils.add_github_issue_comment"):
                with patch("yellhorn_mcp.utils.git_utils.run_github_command") as mock_run:
                    # Return URL on issue creation
                    mock_run.return_value = "https://github.com/test/repo/issues/555"

                    result = await create_github_subissue(
                        repo_path, "111", "Test Sub-issue", body_text
                    )

                    # Verify run_github_command was called with --body-file
                    assert mock_run.called
                    # Get the first call (issue creation)
                    first_call_args = mock_run.call_args_list[0][0]
                    assert first_call_args[0] == repo_path
                    command = first_call_args[1]
                    assert "issue" in command
                    assert "create" in command
                    assert "--body-file" in command
                    assert "--body" not in command
                    assert result == "https://github.com/test/repo/issues/555"