strukto-ai--mirage/python/tests/core/github/test_tree.py

# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ========= Copyright 2026 @ Strukto.AI All Rights Reserved. =========

import logging
from unittest.mock import AsyncMock, patch

import pytest

from mirage.core.github.config import GitHubConfig
from mirage.core.github.tree import fetch_tree
from mirage.core.github.tree_entry import TreeEntry


@pytest.fixture
def config():
    return GitHubConfig(token="ghp_test")


@pytest.mark.asyncio
@patch("mirage.core.github.tree.github_get", new_callable=AsyncMock)
async def test_fetch_tree_parses_entries(mock_get, config):
    mock_get.return_value = {
        "truncated":
        False,
        "tree": [
            {
                "path": "src",
                "type": "tree",
                "sha": "aaa",
                "size": None
            },
            {
                "path": "src/main.py",
                "type": "blob",
                "sha": "bbb",
                "size": 120
            },
        ],
    }
    tree, truncated = await fetch_tree(config, "acme", "proj", "main")
    assert "src" in tree
    assert "src/main.py" in tree
    assert tree["src"] == TreeEntry(path="src",
                                    type="tree",
                                    sha="aaa",
                                    size=None)
    assert tree["src/main.py"] == TreeEntry(path="src/main.py",
                                            type="blob",
                                            sha="bbb",
                                            size=120)


@pytest.mark.asyncio
@patch("mirage.core.github.tree.github_get", new_callable=AsyncMock)
async def test_fetch_tree_truncation_warning(mock_get, config, caplog):
    mock_get.return_value = {"truncated": True, "tree": []}
    with caplog.at_level(logging.WARNING):
        await fetch_tree(config, "acme", "proj", "main")
    assert "truncated" in caplog.text


@pytest.mark.asyncio
@patch("mirage.core.github.tree.github_get", new_callable=AsyncMock)
async def test_fetch_tree_passes_params(mock_get, config):
    mock_get.return_value = {"tree": []}
    await fetch_tree(config, "acme", "proj", "v1")
    mock_get.assert_awaited_once_with(
        config.token,
        "/repos/{owner}/{repo}/git/trees/{ref}",
        owner="acme",
        repo="proj",
        ref="v1",
        params={"recursive": "1"},
    )