strukto-ai--mirage/python/tests/commands/builtin/github/test_narrow.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 pytest

from mirage.commands.builtin.github.grep import grep
from mirage.commands.builtin.github.narrow import narrow_scope
from mirage.commands.builtin.github.rg import rg
from mirage.io.stream import materialize
from mirage.types import PathSpec
from tests.fixtures.github_mock import MOCK_BLOBS

_NGLOBALS = narrow_scope.__globals__


@pytest.fixture
def counting_read(monkeypatch):
    reads: list[str] = []

    async def _read_bytes(config, owner, repo, sha):
        reads.append(sha)
        return MOCK_BLOBS.get(sha, b"")

    monkeypatch.setattr("mirage.core.github.read.read_bytes", _read_bytes)
    return reads


def _root() -> PathSpec:
    return PathSpec(resource_path="",
                    virtual="/",
                    directory="/",
                    resolved=False)


def _subdir() -> PathSpec:
    return PathSpec(resource_path="src",
                    virtual="/src",
                    directory="/src",
                    resolved=False)


@pytest.mark.asyncio
async def test_subdir_narrows_and_fetches_fewer(mock_github_api, github_env,
                                                counting_read, monkeypatch):
    accessor, index = github_env
    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 1)
    await grep(accessor, [_subdir()], "import", r=True, index=index)
    # /src holds 7 files; code search narrows to the import-matching subset.
    assert 0 < len(counting_read) < 7


@pytest.mark.asyncio
async def test_regex_narrows_via_extracted_literal(mock_github_api, github_env,
                                                   counting_read, monkeypatch):
    accessor, index = github_env
    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 1)
    await grep(accessor, [_root()], "import.*os", r=True, index=index)
    # "import" is the required literal; only files containing it are fetched.
    assert 0 < len(counting_read) < len(MOCK_BLOBS)


@pytest.mark.asyncio
async def test_files_only_shortcircuit_reads_nothing(mock_github_api,
                                                     github_env, counting_read,
                                                     monkeypatch):
    accessor, index = github_env
    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 1)
    stdout, io = await grep(accessor, [_root()],
                            "import",
                            r=True,
                            args_l=True,
                            index=index)
    body = (await materialize(stdout)).decode()
    assert len(counting_read) == 0
    assert io.exit_code == 0
    assert "src/main.py" in body


@pytest.mark.asyncio
async def test_files_only_shortcircuit_matches_generic(mock_github_api,
                                                       github_env,
                                                       monkeypatch):
    accessor, index = github_env

    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 10_000)
    out_generic, _ = await grep(accessor, [_root()],
                                "import",
                                r=True,
                                args_l=True,
                                index=index)
    text_generic = (await materialize(out_generic)).decode()

    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 1)
    out_short, _ = await grep(accessor, [_root()],
                              "import",
                              r=True,
                              args_l=True,
                              index=index)
    text_short = (await materialize(out_short)).decode()

    assert text_short == text_generic
    assert text_short


@pytest.mark.asyncio
async def test_rg_files_only_shortcircuit_reads_nothing(
        mock_github_api, github_env, counting_read, monkeypatch):
    accessor, index = github_env
    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 1)
    stdout, io = await rg(accessor, [_root()],
                          "import",
                          args_l=True,
                          index=index)
    body = (await materialize(stdout)).decode()
    assert len(counting_read) == 0
    assert io.exit_code == 0
    assert "src/main.py" in body


@pytest.mark.asyncio
async def test_rg_files_only_shortcircuit_matches_generic(
        mock_github_api, github_env, monkeypatch):
    accessor, index = github_env

    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 10_000)
    out_generic, _ = await rg(accessor, [_root()],
                              "import",
                              args_l=True,
                              index=index)
    text_generic = (await materialize(out_generic)).decode()

    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 1)
    out_short, _ = await rg(accessor, [_root()],
                            "import",
                            args_l=True,
                            index=index)
    text_short = (await materialize(out_short)).decode()

    assert text_short == text_generic
    assert text_short


@pytest.mark.asyncio
async def test_rg_files_only_shortcircuit_respects_glob(
        mock_github_api, github_env, monkeypatch):
    accessor, index = github_env

    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 10_000)
    out_generic, _ = await rg(accessor, [_root()],
                              "import",
                              args_l=True,
                              glob="main*.py",
                              index=index)
    text_generic = (await materialize(out_generic)).decode()

    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 1)
    out_short, _ = await rg(accessor, [_root()],
                            "import",
                            args_l=True,
                            glob="main*.py",
                            index=index)
    text_short = (await materialize(out_short)).decode()

    assert text_short == text_generic
    assert "main.py" in text_short
    assert "utils.py" not in text_short


@pytest.mark.asyncio
async def test_rg_shortcircuit_no_match_exit_1(mock_github_api, github_env,
                                               monkeypatch):
    accessor, index = github_env
    monkeypatch.setitem(_NGLOBALS, "SCOPE_WARN", 1)
    stdout, io = await rg(accessor, [_root()],
                          "import",
                          args_l=True,
                          glob="*.nomatch",
                          index=index)
    body = (await materialize(stdout)).decode()
    assert io.exit_code == 1
    assert body == ""