chore: import upstream snapshot with attribution

python/tests/core/gdrive/__init__.py

@@ -0,0 +1,13 @@
+# ========= 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. =========

python/tests/core/gdrive/test_read.py

@@ -0,0 +1,200 @@
+# ========= 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. =========
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from mirage.accessor.gdrive import GDriveAccessor
+from mirage.cache.index.config import IndexEntry
+from mirage.cache.index.ram import RAMIndexCacheStore
+from mirage.core.gdrive.read import read, read_bytes
+from mirage.core.google._client import TokenManager
+from mirage.core.google.config import GoogleConfig
+from mirage.types import PathSpec
+
+
+@pytest.fixture
+def config():
+    return GoogleConfig(
+        client_id="test-id",
+        client_secret="test-secret",
+        refresh_token="test-refresh",
+    )
+
+
+@pytest.fixture
+def token_manager(config):
+    mgr = TokenManager(config)
+    mgr._access_token = "fake-token"
+    mgr._expires_at = 9999999999
+    return mgr
+
+
+@pytest.fixture
+def accessor(config, token_manager):
+    return GDriveAccessor(config=config, token_manager=token_manager)
+
+
+@pytest.fixture
+def index():
+    store = RAMIndexCacheStore()
+    return store
+
+
+@pytest.mark.asyncio
+async def test_read_bytes(token_manager):
+    content = b"pdf content here"
+    with patch(
+            "mirage.core.gdrive.read.download_file",
+            new_callable=AsyncMock,
+            return_value=content,
+    ) as mock_download:
+        result = await read_bytes(token_manager, "file123")
+        assert result == content
+        mock_download.assert_called_once_with(token_manager, "file123")
+
+
+@pytest.mark.asyncio
+async def test_read_file(accessor, index):
+    await index.put(
+        "/Team Drive/report.pdf",
+        IndexEntry(
+            id="file123",
+            name="report",
+            resource_type="gdrive/file",
+            remote_time="2026-04-01T00:00:00.000Z",
+            vfs_name="report.pdf",
+            extra={"drive_id": "drive1"},
+        ))
+    content = b"pdf content here"
+    with patch(
+            "mirage.core.gdrive.read.download_file",
+            new_callable=AsyncMock,
+            return_value=content,
+    ):
+        result = await read(
+            accessor,
+            PathSpec(resource_path="Team Drive/report.pdf",
+                     virtual="/Team Drive/report.pdf",
+                     directory="/Team Drive/report.pdf"), index)
+        assert result == content
+
+
+@pytest.mark.asyncio
+async def test_read_shared_drive_raises_is_a_directory(accessor, index):
+    await index.put(
+        "/Team Drive",
+        IndexEntry(
+            id="drive1",
+            name="Team Drive",
+            resource_type="gdrive/shared_drive",
+            vfs_name="Team Drive",
+            extra={"drive_id": "drive1"},
+        ))
+    with patch(
+            "mirage.core.gdrive.read.download_file",
+            new_callable=AsyncMock,
+    ) as mock_download:
+        with pytest.raises(IsADirectoryError):
+            await read(
+                accessor,
+                PathSpec(resource_path="Team Drive",
+                         virtual="/Team Drive",
+                         directory="/Team Drive"),
+                index,
+            )
+    mock_download.assert_not_awaited()
+
+
+@pytest.mark.asyncio
+async def test_read_not_found(accessor, index):
+    with pytest.raises(FileNotFoundError):
+        await read(
+            accessor,
+            PathSpec(resource_path="missing/file.txt",
+                     virtual="/missing/file.txt",
+                     directory="/missing/file.txt"), index)
+
+
+@pytest.mark.asyncio
+async def test_read_auto_bootstraps_from_empty_index(accessor, index):
+
+    async def fake_list_files(_tm, folder_id, drive_id=None):
+        if folder_id == "root":
+            return [{
+                "id": "f1",
+                "name": "report.pdf",
+                "mimeType": "application/pdf",
+                "modifiedTime": "2026-04-01T00:00:00.000Z",
+                "owners": [],
+                "capabilities": {},
+            }]
+        raise AssertionError(f"unexpected folder_id={folder_id}")
+
+    with (
+            patch(
+                "mirage.core.gdrive.readdir.list_files",
+                new=fake_list_files,
+            ),
+            patch(
+                "mirage.core.gdrive.read.download_file",
+                new_callable=AsyncMock,
+                return_value=b"pdf-bytes",
+            ),
+    ):
+        result = await read(
+            accessor,
+            PathSpec(resource_path="report.pdf",
+                     virtual="/report.pdf",
+                     directory="/report.pdf"),
+            index,
+        )
+        assert result == b"pdf-bytes"
+
+
+@pytest.mark.asyncio
+async def test_read_missing_file_raises_after_recursion(accessor, index):
+
+    async def fake_list_files(_tm, folder_id, drive_id=None):
+        if folder_id == "root":
+            return [{
+                "id": "f1",
+                "name": "other.txt",
+                "mimeType": "text/plain",
+                "modifiedTime": "2026-04-01T00:00:00.000Z",
+                "owners": [],
+                "capabilities": {},
+            }]
+        raise AssertionError(f"unexpected folder_id={folder_id}")
+
+    with (
+            patch(
+                "mirage.core.gdrive.readdir.list_files",
+                new=fake_list_files,
+            ),
+            patch(
+                "mirage.core.gdrive.read.download_file",
+                new_callable=AsyncMock,
+                side_effect=AssertionError("should not call download_file"),
+            ),
+    ):
+        with pytest.raises(FileNotFoundError):
+            await read(
+                accessor,
+                PathSpec(resource_path="missing.txt",
+                         virtual="/missing.txt",
+                         directory="/missing.txt"),
+                index,
+            )

python/tests/core/gdrive/test_readdir.py

@@ -0,0 +1,332 @@
+# ========= 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. =========
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from mirage.accessor.gdrive import GDriveAccessor
+from mirage.cache.index.config import IndexEntry
+from mirage.cache.index.ram import RAMIndexCacheStore
+from mirage.core.gdrive.readdir import readdir
+from mirage.core.google._client import TokenManager
+from mirage.core.google.config import GoogleConfig
+from mirage.types import PathSpec
+
+
+@pytest.fixture
+def config():
+    return GoogleConfig(
+        client_id="test-id",
+        client_secret="test-secret",
+        refresh_token="test-refresh",
+    )
+
+
+@pytest.fixture
+def token_manager(config):
+    mgr = TokenManager(config)
+    mgr._access_token = "fake-token"
+    mgr._expires_at = 9999999999
+    return mgr
+
+
+@pytest.fixture
+def accessor(config, token_manager):
+    return GDriveAccessor(config=config, token_manager=token_manager)
+
+
+@pytest.fixture
+def index():
+    return RAMIndexCacheStore()
+
+
+@pytest.mark.asyncio
+async def test_readdir_root(accessor, index):
+    files = [
+        {
+            "id": "f1",
+            "name": "readme.txt",
+            "mimeType": "text/plain",
+            "modifiedTime": "2026-04-01T00:00:00.000Z",
+            "owners": [{
+                "me": True,
+                "emailAddress": "me@gmail.com"
+            }],
+            "capabilities": {
+                "canEdit": True
+            },
+        },
+    ]
+    with patch(
+            "mirage.core.gdrive.readdir.list_files",
+            new_callable=AsyncMock,
+            return_value=files,
+    ):
+        result = await readdir(
+            accessor, PathSpec(resource_path="", virtual="/", directory="/"),
+            index)
+        assert "/readme.txt" in result
+
+
+@pytest.mark.asyncio
+async def test_readdir_cached(accessor, index):
+    entry = IndexEntry(
+        id="f1",
+        name="cached.txt",
+        resource_type="gdrive/file",
+        remote_time="2026-04-01T00:00:00.000Z",
+        vfs_name="cached.txt",
+    )
+    await index.set_dir("/", [("cached.txt", entry)])
+    result = await readdir(
+        accessor, PathSpec(resource_path="", virtual="/", directory="/"),
+        index)
+    assert any("cached.txt" in r for r in result)
+
+
+@pytest.mark.asyncio
+async def test_readdir_subfolder(accessor, index):
+    await index.put(
+        "/docs",
+        IndexEntry(
+            id="folder1",
+            name="docs",
+            resource_type="gdrive/folder",
+            remote_time="2026-04-01T00:00:00.000Z",
+            vfs_name="docs",
+        ))
+
+    files = [
+        {
+            "id": "f2",
+            "name": "notes.txt",
+            "mimeType": "text/plain",
+            "modifiedTime": "2026-04-01T00:00:00.000Z",
+            "owners": [],
+            "capabilities": {
+                "canEdit": False
+            },
+        },
+    ]
+    with patch(
+            "mirage.core.gdrive.readdir.list_files",
+            new_callable=AsyncMock,
+            return_value=files,
+    ) as mock_list:
+        result = await readdir(
+            accessor,
+            PathSpec(resource_path="docs", virtual="/docs", directory="/docs"),
+            index)
+        assert "/docs/notes.txt" in result
+        mock_list.assert_called_once_with(accessor.token_manager,
+                                          folder_id="folder1",
+                                          drive_id=None)
+
+
+@pytest.mark.asyncio
+async def test_readdir_repopulates_evicted_subfolder(accessor, index):
+    root_files = [{
+        "id": "folder1",
+        "name": "docs",
+        "mimeType": "application/vnd.google-apps.folder",
+        "modifiedTime": "2026-04-01T00:00:00.000Z",
+        "owners": [],
+        "capabilities": {},
+    }]
+    docs_files = [{
+        "id": "f2",
+        "name": "notes.txt",
+        "mimeType": "text/plain",
+        "modifiedTime": "2026-04-01T00:00:00.000Z",
+        "owners": [],
+        "capabilities": {},
+    }]
+
+    async def fake_list_files(_tm, folder_id, drive_id=None):
+        if folder_id == "root":
+            return root_files
+        if folder_id == "folder1":
+            return docs_files
+        raise AssertionError(f"unexpected folder_id={folder_id}")
+
+    with patch(
+            "mirage.core.gdrive.readdir.list_files",
+            new=fake_list_files,
+    ):
+        result = await readdir(
+            accessor,
+            PathSpec(resource_path="docs", virtual="/docs", directory="/docs"),
+            index)
+        assert "/docs/notes.txt" in result
+
+
+@pytest.mark.asyncio
+async def test_readdir_missing_subfolder_raises_after_recursion(
+        accessor, index):
+    root_files = [{
+        "id": "f1",
+        "name": "other.txt",
+        "mimeType": "text/plain",
+        "modifiedTime": "2026-04-01T00:00:00.000Z",
+        "owners": [],
+        "capabilities": {},
+    }]
+
+    async def fake_list_files(_tm, folder_id, drive_id=None):
+        if folder_id == "root":
+            return root_files
+        raise AssertionError(f"should not list folder_id={folder_id}")
+
+    with patch(
+            "mirage.core.gdrive.readdir.list_files",
+            new=fake_list_files,
+    ):
+        with pytest.raises(FileNotFoundError):
+            await readdir(
+                accessor,
+                PathSpec(resource_path="docs",
+                         virtual="/docs",
+                         directory="/docs"), index)
+
+
+@pytest.mark.asyncio
+async def test_readdir_root_includes_shared_drives(accessor, index):
+    files = [{
+        "id": "f1",
+        "name": "readme.txt",
+        "mimeType": "text/plain",
+        "modifiedTime": "2026-04-01T00:00:00.000Z",
+        "owners": [],
+        "capabilities": {},
+    }]
+    drives = [{"id": "drive1", "name": "Team Drive"}]
+    with patch("mirage.core.gdrive.readdir.list_files",
+               new_callable=AsyncMock, return_value=files), \
+         patch("mirage.core.gdrive.readdir.list_shared_drives",
+               new_callable=AsyncMock, return_value=drives):
+        result = await readdir(
+            accessor, PathSpec(resource_path="", virtual="/", directory="/"),
+            index)
+        assert "/readme.txt" in result
+        # Shared Drives appear as top-level directories.
+        assert "/Team Drive/" in result
+        # The drive id is carried on the cached entry for nested listings.
+        entry = (await index.get("/Team Drive")).entry
+        assert entry is not None
+        assert entry.extra.get("drive_id") == "drive1"
+
+
+@pytest.mark.asyncio
+async def test_readdir_root_uniquifies_duplicate_shared_drive_names(
+        accessor, index):
+    drives = [
+        {
+            "id": "drive1",
+            "name": "Team"
+        },
+        {
+            "id": "drive2",
+            "name": "Team"
+        },
+        {
+            "id": "drive3",
+            "name": "Team"
+        },
+    ]
+    with patch("mirage.core.gdrive.readdir.list_files",
+               new_callable=AsyncMock, return_value=[]), \
+         patch("mirage.core.gdrive.readdir.list_shared_drives",
+               new_callable=AsyncMock, return_value=drives):
+        result = await readdir(
+            accessor, PathSpec(resource_path="", virtual="/", directory="/"),
+            index)
+
+    assert result == [
+        "/Team/",
+        "/Team [Shared Drive]/",
+        "/Team [Shared Drive 2]/",
+    ]
+    assert (await index.get("/Team")).entry.id == "drive1"
+    assert (await index.get("/Team [Shared Drive]")).entry.id == "drive2"
+    assert (await index.get("/Team [Shared Drive 2]")).entry.id == "drive3"
+
+
+@pytest.mark.asyncio
+async def test_readdir_root_shared_drives_best_effort(accessor, index):
+    """If Shared Drive enumeration fails, My Drive listing still succeeds."""
+    files = [{
+        "id": "f1",
+        "name": "readme.txt",
+        "mimeType": "text/plain",
+        "modifiedTime": "2026-04-01T00:00:00.000Z",
+        "owners": [],
+        "capabilities": {},
+    }]
+    with patch("mirage.core.gdrive.readdir.list_files",
+               new_callable=AsyncMock, return_value=files), \
+         patch("mirage.core.gdrive.readdir.list_shared_drives",
+               new_callable=AsyncMock, side_effect=RuntimeError("no scope")):
+        result = await readdir(
+            accessor, PathSpec(resource_path="", virtual="/", directory="/"),
+            index)
+        assert "/readme.txt" in result
+
+
+@pytest.mark.asyncio
+async def test_readdir_workspace_files_get_extensions(accessor, index):
+    files = [
+        {
+            "id": "d1",
+            "name": "My Document",
+            "mimeType": "application/vnd.google-apps.document",
+            "modifiedTime": "2026-04-01T00:00:00.000Z",
+            "owners": [{
+                "me": True,
+                "emailAddress": "me@gmail.com"
+            }],
+            "capabilities": {
+                "canEdit": True
+            },
+        },
+        {
+            "id": "s1",
+            "name": "My Sheet",
+            "mimeType": "application/vnd.google-apps.spreadsheet",
+            "modifiedTime": "2026-04-01T00:00:00.000Z",
+            "owners": [],
+            "capabilities": {
+                "canEdit": False
+            },
+        },
+        {
+            "id": "p1",
+            "name": "My Slides",
+            "mimeType": "application/vnd.google-apps.presentation",
+            "modifiedTime": "2026-04-01T00:00:00.000Z",
+            "owners": [],
+            "capabilities": {},
+        },
+    ]
+    with patch(
+            "mirage.core.gdrive.readdir.list_files",
+            new_callable=AsyncMock,
+            return_value=files,
+    ):
+        result = await readdir(
+            accessor, PathSpec(resource_path="", virtual="/", directory="/"),
+            index)
+        assert "/My Document.gdoc.json" in result
+        assert "/My Sheet.gsheet.json" in result
+        assert "/My Slides.gslide.json" in result

python/tests/core/gdrive/test_stat.py

@@ -0,0 +1,183 @@
+# ========= 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. =========
+
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from mirage.accessor.gdrive import GDriveAccessor
+from mirage.cache.index.config import IndexEntry
+from mirage.cache.index.ram import RAMIndexCacheStore
+from mirage.core.gdrive.stat import stat
+from mirage.core.google._client import TokenManager
+from mirage.core.google.config import GoogleConfig
+from mirage.types import FileType, PathSpec
+
+
+@pytest.fixture
+def config():
+    return GoogleConfig(
+        client_id="test-id",
+        client_secret="test-secret",
+        refresh_token="test-refresh",
+    )
+
+
+@pytest.fixture
+def token_manager(config):
+    mgr = TokenManager(config)
+    mgr._access_token = "fake-token"
+    mgr._expires_at = 9999999999
+    return mgr
+
+
+@pytest.fixture
+def accessor(config, token_manager):
+    return GDriveAccessor(config=config, token_manager=token_manager)
+
+
+@pytest.fixture
+def index():
+    store = RAMIndexCacheStore()
+    return store
+
+
+async def _populate_index(index):
+    await index.put(
+        "/report.pdf",
+        IndexEntry(
+            id="f1",
+            name="report",
+            resource_type="gdrive/file",
+            remote_time="2026-04-01T00:00:00.000Z",
+            vfs_name="report.pdf",
+            size=1024,
+        ))
+    await index.put(
+        "/docs",
+        IndexEntry(
+            id="folder1",
+            name="docs",
+            resource_type="gdrive/folder",
+            remote_time="2026-04-01T00:00:00.000Z",
+            vfs_name="docs",
+        ))
+    return index
+
+
+@pytest.mark.asyncio
+async def test_stat_root(accessor, index):
+    idx = await _populate_index(index)
+    result = await stat(accessor,
+                        PathSpec(resource_path="", virtual="/", directory="/"),
+                        idx)
+    assert result.type == FileType.DIRECTORY
+    assert result.name == "/"
+
+
+@pytest.mark.asyncio
+async def test_stat_file(accessor, index):
+    idx = await _populate_index(index)
+    result = await stat(
+        accessor,
+        PathSpec(resource_path="report.pdf",
+                 virtual="/report.pdf",
+                 directory="/report.pdf"), idx)
+    assert result.name == "report.pdf"
+    assert result.size == 1024
+    assert result.extra["file_id"] == "f1"
+    assert result.extra["resource_type"] == "gdrive/file"
+
+
+@pytest.mark.asyncio
+async def test_stat_folder(accessor, index):
+    idx = await _populate_index(index)
+    result = await stat(
+        accessor,
+        PathSpec(resource_path="docs", virtual="/docs", directory="/docs"),
+        idx)
+    assert result.type == FileType.DIRECTORY
+    assert result.extra["file_id"] == "folder1"
+
+
+@pytest.mark.asyncio
+async def test_stat_shared_drive_is_directory(accessor, index):
+    await index.put(
+        "/Team Drive",
+        IndexEntry(
+            id="drive1",
+            name="Team Drive",
+            resource_type="gdrive/shared_drive",
+            vfs_name="Team Drive",
+            extra={"drive_id": "drive1"},
+        ))
+    result = await stat(
+        accessor,
+        PathSpec(resource_path="Team Drive",
+                 virtual="/Team Drive",
+                 directory="/Team Drive"),
+        index,
+    )
+    assert result.type == FileType.DIRECTORY
+    assert result.extra["file_id"] == "drive1"
+
+
+@pytest.mark.asyncio
+async def test_stat_not_found(accessor, index):
+    idx = await _populate_index(index)
+    with patch(
+            "mirage.core.gdrive.readdir.list_files",
+            new_callable=AsyncMock,
+            return_value=[],
+    ):
+        with pytest.raises(FileNotFoundError):
+            await stat(
+                accessor,
+                PathSpec(resource_path="nonexistent.txt",
+                         virtual="/nonexistent.txt",
+                         directory="/nonexistent.txt"), idx)
+
+
+@pytest.mark.asyncio
+async def test_stat_cache_miss_falls_back_via_readdir(accessor, index):
+    files = [{
+        "id": "f99",
+        "name": "fresh.pdf",
+        "mimeType": "application/pdf",
+        "modifiedTime": "2026-04-15T00:00:00.000Z",
+        "size": "2048",
+    }]
+    with patch(
+            "mirage.core.gdrive.readdir.list_files",
+            new_callable=AsyncMock,
+            return_value=files,
+    ) as mock_list:
+        result = await stat(
+            accessor,
+            PathSpec(resource_path="fresh.pdf",
+                     virtual="/fresh.pdf",
+                     directory="/fresh.pdf"), index)
+    assert result.name == "fresh.pdf"
+    assert result.extra["file_id"] == "f99"
+    assert mock_list.call_count == 1
+
+
+@pytest.mark.asyncio
+async def test_stat_index_none_raises(accessor):
+    with pytest.raises(FileNotFoundError):
+        await stat(
+            accessor,
+            PathSpec(resource_path="x.pdf",
+                     virtual="/x.pdf",
+                     directory="/x.pdf"), None)

python/tests/core/gdrive/test_stream.py

@@ -0,0 +1,188 @@
+# ========= 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. =========
+
+from unittest.mock import patch
+
+import pytest
+
+from mirage.accessor.gdrive import GDriveAccessor
+from mirage.cache.index.config import IndexEntry
+from mirage.cache.index.ram import RAMIndexCacheStore
+from mirage.core.gdrive.stream import stream
+from mirage.core.google._client import TokenManager
+from mirage.core.google.config import GoogleConfig
+from mirage.types import PathSpec
+
+
+@pytest.fixture
+def config():
+    return GoogleConfig(
+        client_id="test-id",
+        client_secret="test-secret",
+        refresh_token="test-refresh",
+    )
+
+
+@pytest.fixture
+def token_manager(config):
+    mgr = TokenManager(config)
+    mgr._access_token = "fake-token"
+    mgr._expires_at = 9999999999
+    return mgr
+
+
+@pytest.fixture
+def accessor(config, token_manager):
+    return GDriveAccessor(config=config, token_manager=token_manager)
+
+
+@pytest.fixture
+def index():
+    return RAMIndexCacheStore()
+
+
+async def _fake_download_stream(*args, **kwargs):
+    yield b"chunk1"
+    yield b"chunk2"
+
+
+async def _collect(source):
+    data = b""
+    async for chunk in source:
+        data += chunk
+    return data
+
+
+@pytest.mark.asyncio
+async def test_stream_file(accessor, index):
+    await index.put(
+        "/Team Drive/report.pdf",
+        IndexEntry(
+            id="file123",
+            name="report",
+            resource_type="gdrive/file",
+            remote_time="2026-04-01T00:00:00.000Z",
+            vfs_name="report.pdf",
+            extra={"drive_id": "drive1"},
+        ))
+    with patch(
+            "mirage.core.gdrive.stream.download_file_stream",
+            side_effect=_fake_download_stream,
+    ):
+        data = await _collect(
+            stream(
+                accessor,
+                PathSpec(resource_path="Team Drive/report.pdf",
+                         virtual="/Team Drive/report.pdf",
+                         directory="/Team Drive/report.pdf"), index))
+        assert data == b"chunk1chunk2"
+
+
+@pytest.mark.asyncio
+async def test_stream_not_found(accessor, index):
+
+    async def fake_list_files(_tm, folder_id, drive_id=None):
+        return []
+
+    with patch("mirage.core.gdrive.readdir.list_files", new=fake_list_files):
+        with pytest.raises(FileNotFoundError):
+            await _collect(
+                stream(
+                    accessor,
+                    PathSpec(resource_path="missing/file.txt",
+                             virtual="/missing/file.txt",
+                             directory="/missing/file.txt"), index))
+
+
+@pytest.mark.asyncio
+async def test_stream_auto_bootstraps_from_empty_index(accessor, index):
+
+    async def fake_list_files(_tm, folder_id, drive_id=None):
+        if folder_id == "root":
+            return [{
+                "id": "f1",
+                "name": "report.pdf",
+                "mimeType": "application/pdf",
+                "modifiedTime": "2026-04-01T00:00:00.000Z",
+                "owners": [],
+                "capabilities": {},
+            }]
+        raise AssertionError(f"unexpected folder_id={folder_id}")
+
+    with (
+            patch(
+                "mirage.core.gdrive.readdir.list_files",
+                new=fake_list_files,
+            ),
+            patch(
+                "mirage.core.gdrive.stream.download_file_stream",
+                side_effect=_fake_download_stream,
+            ),
+    ):
+        data = await _collect(
+            stream(
+                accessor,
+                PathSpec(resource_path="report.pdf",
+                         virtual="/report.pdf",
+                         directory="/report.pdf"),
+                index,
+            ))
+        assert data == b"chunk1chunk2"
+
+
+@pytest.mark.asyncio
+async def test_stream_folder_raises(accessor, index):
+    await index.put(
+        "/data",
+        IndexEntry(
+            id="folder1",
+            name="data",
+            resource_type="gdrive/folder",
+            remote_time="2026-04-01T00:00:00.000Z",
+            vfs_name="data",
+        ))
+    with pytest.raises(IsADirectoryError):
+        await _collect(
+            stream(
+                accessor,
+                PathSpec(resource_path="data",
+                         virtual="/data",
+                         directory="/data"), index))
+
+
+@pytest.mark.asyncio
+async def test_stream_shared_drive_raises(accessor, index):
+    await index.put(
+        "/Team Drive",
+        IndexEntry(
+            id="drive1",
+            name="Team Drive",
+            resource_type="gdrive/shared_drive",
+            vfs_name="Team Drive",
+            extra={"drive_id": "drive1"},
+        ))
+    with patch(
+            "mirage.core.gdrive.stream.download_file_stream",
+            side_effect=_fake_download_stream,
+    ) as mock_download:
+        with pytest.raises(IsADirectoryError):
+            await _collect(
+                stream(
+                    accessor,
+                    PathSpec(resource_path="Team Drive",
+                             virtual="/Team Drive",
+                             directory="/Team Drive"),
+                    index,
+                ))
+    mock_download.assert_not_called()