strukto-ai--mirage/python/tests/core/slack/test_search.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 json
from unittest.mock import AsyncMock, patch

import pytest

from mirage.core.slack.formatters import format_grep_results
from mirage.core.slack.scope import SlackScope
from mirage.core.slack.search import search_messages
from mirage.resource.slack.config import SlackConfig


@pytest.mark.asyncio
async def test_search_messages_defaults_include_page_1():
    cfg = SlackConfig(token="xoxp-test")
    with patch(
            "mirage.core.slack.search.slack_get",
            new=AsyncMock(return_value={"ok": True}),
    ) as fake_get:
        await search_messages(cfg, "hello")
    params = fake_get.call_args.kwargs["params"]
    assert params["query"] == "hello"
    assert params["count"] == 20
    assert params["page"] == 1
    assert params["sort"] == "timestamp"


@pytest.mark.asyncio
async def test_search_messages_forwards_explicit_count_and_page():
    cfg = SlackConfig(token="xoxp-test")
    with patch(
            "mirage.core.slack.search.slack_get",
            new=AsyncMock(return_value={"ok": True}),
    ) as fake_get:
        await search_messages(cfg, "hello", count=50, page=3)
    params = fake_get.call_args.kwargs["params"]
    assert params["count"] == 50
    assert params["page"] == 3


def test_format_grep_results_path_uses_chat_jsonl():
    raw_payload = {
        "messages": {
            "matches": [
                {
                    "channel": {
                        "id": "C001",
                        "name": "general"
                    },
                    "user": "U1",
                    "ts": "1712707200.0",
                    "text": "hello",
                },
            ],
        },
    }
    raw = json.dumps(raw_payload).encode()
    scope = SlackScope(
        use_native=True,
        container="channels",
        channel_name="general",
        channel_id="C001",
        target="messages",
    )
    lines = format_grep_results(raw, scope, "/slack")
    assert len(lines) == 1
    line = lines[0]
    assert line.startswith(
        "/slack/channels/general__C001/2024-04-10/chat.jsonl:"), line