strukto-ai--mirage/python/tests/core/onedrive/test_stat.py

import pytest
from aioresponses import aioresponses

from mirage.accessor.onedrive import OneDriveAccessor, OneDriveConfig
from mirage.cache.index import RAMIndexCacheStore
from mirage.core.onedrive.readdir import readdir
from mirage.core.onedrive.stat import stat
from mirage.types import FileType, PathSpec
from mirage.utils.key_prefix import mount_key


def _accessor(**kw) -> OneDriveAccessor:
    return OneDriveAccessor(OneDriveConfig(access_token="tok", **kw))


_FILE_URL = ("https://graph.microsoft.com/v1.0/me/drive"
             "/root:/Docs/report.docx")
_DIR_URL = "https://graph.microsoft.com/v1.0/me/drive/root:/Docs"


@pytest.mark.asyncio
async def test_stat_root_is_directory():
    result = await stat(_accessor(), PathSpec.from_str_path("/"))
    assert result.type == FileType.DIRECTORY


@pytest.mark.asyncio
async def test_stat_file_carries_size_and_ctag_fingerprint():
    with aioresponses() as m:
        m.get(_FILE_URL,
              payload={
                  "id": "01ITEM",
                  "name": "report.docx",
                  "size": 1234,
                  "lastModifiedDateTime": "2026-05-01T10:00:00Z",
                  "cTag": "ctag-abc",
                  "eTag": "etag-xyz",
                  "file": {
                      "mimeType": "application/vnd.openxml"
                  },
              })
        result = await stat(_accessor(),
                            PathSpec.from_str_path("/Docs/report.docx"))
    assert result.name == "report.docx"
    assert result.size == 1234
    assert result.fingerprint == "ctag-abc"
    assert result.extra["id"] == "01ITEM"


@pytest.mark.asyncio
async def test_stat_folder_is_directory():
    with aioresponses() as m:
        m.get(_DIR_URL,
              payload={
                  "id": "01FOLDER",
                  "name": "Docs",
                  "size": 4096,
                  "lastModifiedDateTime": "2026-05-01T10:00:00Z",
                  "folder": {
                      "childCount": 2
                  },
              })
        result = await stat(_accessor(), PathSpec.from_str_path("/Docs"))
    assert result.type == FileType.DIRECTORY
    assert result.name == "Docs"
    assert result.size == 4096
    assert result.modified == "2026-05-01T10:00:00Z"


@pytest.mark.asyncio
async def test_stat_missing_raises_file_not_found():
    with aioresponses() as m:
        m.get(_FILE_URL,
              status=404,
              payload={"error": {
                  "code": "itemNotFound",
                  "message": "no"
              }})
        with pytest.raises(FileNotFoundError) as exc:
            await stat(
                _accessor(),
                PathSpec.from_str_path(
                    "/od/Docs/report.docx",
                    mount_key("/od/Docs/report.docx", "/od")))
    assert str(exc.value) == "/od/Docs/report.docx"


@pytest.mark.asyncio
async def test_stat_from_index_returns_modified():
    index = RAMIndexCacheStore()
    with aioresponses() as m:
        m.get("https://graph.microsoft.com/v1.0/me/drive/root/children",
              payload={
                  "value": [{
                      "id": "1",
                      "name": "notes.txt",
                      "size": 42,
                      "file": {},
                      "lastModifiedDateTime": "2026-06-19T09:28:00Z",
                  }]
              })
        await readdir(_accessor(), PathSpec.from_str_path("/"), index)
    result = await stat(_accessor(), PathSpec.from_str_path("/notes.txt"),
                        index)
    assert result.name == "notes.txt"
    assert result.size == 42
    assert result.modified == "2026-06-19T09:28:00Z"


@pytest.mark.asyncio
async def test_stat_from_index_folder_returns_modified():
    index = RAMIndexCacheStore()
    with aioresponses() as m:
        m.get("https://graph.microsoft.com/v1.0/me/drive/root/children",
              payload={
                  "value": [{
                      "id": "2",
                      "name": "Docs",
                      "folder": {
                          "childCount": 5
                      },
                      "size": 9000,
                      "lastModifiedDateTime": "2026-05-28T02:10:00Z",
                  }]
              })
        await readdir(_accessor(), PathSpec.from_str_path("/"), index)
    result = await stat(_accessor(), PathSpec.from_str_path("/Docs"), index)
    assert result.type == FileType.DIRECTORY
    assert result.modified == "2026-05-28T02:10:00Z"
    assert result.size == 9000