chore: import upstream snapshot with attribution
Integ / changes (push) Has been skipped
Pre-commit / pre-commit (push) Failing after 1s
CLI exit codes / changes (push) Has been skipped
Test (Install) / changes (push) Has been skipped
Test (Python) / changes (push) Has been skipped
Test (TypeScript) / changes (push) Has been skipped
CLI exit codes / cli-gate (push) Has been cancelled
Test (Install) / test-install-gate (push) Has been cancelled
Integ / integ-gate (push) Has been cancelled
Test (Python) / test-python-gate (push) Has been cancelled
Test (TypeScript) / test-typescript-gate (push) Has been cancelled
Test (Install) / python-minimal (3.12) (push) Has been cancelled
Test (Install) / python-minimal (3.11) (push) Has been cancelled
Test (Install) / python-extra (agno, mirage.agents.agno) (push) Has been cancelled
Test (Install) / python-extra (chroma, mirage.resource.chroma) (push) Has been cancelled
Test (Install) / python-extra (pdf, mirage.core.filetype.pdf) (push) Has been cancelled
Integ / integ (push) Has been cancelled
Integ / integ-database (push) Has been cancelled
Integ / integ-database-ts (push) Has been cancelled
Integ / integ-data (push) Has been cancelled
Integ / integ-ssh (push) Has been cancelled
Integ / integ-ssh-ts (push) Has been cancelled
Test (Python) / audit (push) Has been cancelled
Test (TypeScript) / test (push) Has been cancelled
Test (TypeScript) / python-fs-shim (push) Has been cancelled
CLI exit codes / Python CLI (push) Has been cancelled
CLI exit codes / TypeScript CLI (push) Has been cancelled
CLI exit codes / Cross-language snapshot interop (push) Has been cancelled
Test (Python) / test (push) Has been cancelled
Test (Python) / import-isolation (deepagents, openai, mirage.agents.openai_agents) (push) Has been cancelled
Test (Python) / import-isolation (deepagents, pydantic-ai, mirage.agents.pydantic_ai) (push) Has been cancelled
Integ / integ-ts (push) Has been cancelled
Integ / integ-fuse (push) Has been cancelled
Test (Install) / python-extra (databricks, mirage.resource.databricks_volume) (push) Has been cancelled
Test (Install) / python-extra (deepagents, mirage.agents.langchain) (push) Has been cancelled
Test (Install) / python-extra (email, mirage.resource.email) (push) Has been cancelled
Test (Install) / python-extra (fuse, mirage.fuse.mount) (push) Has been cancelled
Test (Install) / python-extra (hdf5, mirage.core.filetype.hdf5) (push) Has been cancelled
Test (Install) / python-extra (hf, mirage.resource.hf_buckets) (push) Has been cancelled
Test (Install) / python-extra (lancedb, mirage.resource.lancedb) (push) Has been cancelled
Test (Install) / python-extra (langfuse, mirage.resource.langfuse) (push) Has been cancelled
Test (Install) / python-extra (mongodb, mirage.resource.mongodb) (push) Has been cancelled
Test (Install) / python-extra (nextcloud, mirage.resource.nextcloud) (push) Has been cancelled
Test (Install) / python-extra (openai, mirage.agents.openai_agents) (push) Has been cancelled
Test (Install) / python-extra (openhands, mirage.agents.openhands, 3.12) (push) Has been cancelled
Test (Install) / python-extra (parquet, mirage.core.filetype.parquet) (push) Has been cancelled
Test (Install) / python-extra (postgres, mirage.resource.postgres) (push) Has been cancelled
Test (Install) / python-extra (pydantic-ai, mirage.agents.pydantic_ai) (push) Has been cancelled
Test (Install) / python-extra (qdrant, mirage.resource.qdrant) (push) Has been cancelled
Test (Install) / python-extra (redis, mirage.resource.redis) (push) Has been cancelled
Test (Install) / python-extra (s3, mirage.resource.s3) (push) Has been cancelled
Test (Install) / python-extra (ssh, mirage.resource.ssh) (push) Has been cancelled
Test (Install) / ts-minimal (push) Has been cancelled
Integ / changes (push) Has been skipped
Pre-commit / pre-commit (push) Failing after 1s
CLI exit codes / changes (push) Has been skipped
Test (Install) / changes (push) Has been skipped
Test (Python) / changes (push) Has been skipped
Test (TypeScript) / changes (push) Has been skipped
CLI exit codes / cli-gate (push) Has been cancelled
Test (Install) / test-install-gate (push) Has been cancelled
Integ / integ-gate (push) Has been cancelled
Test (Python) / test-python-gate (push) Has been cancelled
Test (TypeScript) / test-typescript-gate (push) Has been cancelled
Test (Install) / python-minimal (3.12) (push) Has been cancelled
Test (Install) / python-minimal (3.11) (push) Has been cancelled
Test (Install) / python-extra (agno, mirage.agents.agno) (push) Has been cancelled
Test (Install) / python-extra (chroma, mirage.resource.chroma) (push) Has been cancelled
Test (Install) / python-extra (pdf, mirage.core.filetype.pdf) (push) Has been cancelled
Integ / integ (push) Has been cancelled
Integ / integ-database (push) Has been cancelled
Integ / integ-database-ts (push) Has been cancelled
Integ / integ-data (push) Has been cancelled
Integ / integ-ssh (push) Has been cancelled
Integ / integ-ssh-ts (push) Has been cancelled
Test (Python) / audit (push) Has been cancelled
Test (TypeScript) / test (push) Has been cancelled
Test (TypeScript) / python-fs-shim (push) Has been cancelled
CLI exit codes / Python CLI (push) Has been cancelled
CLI exit codes / TypeScript CLI (push) Has been cancelled
CLI exit codes / Cross-language snapshot interop (push) Has been cancelled
Test (Python) / test (push) Has been cancelled
Test (Python) / import-isolation (deepagents, openai, mirage.agents.openai_agents) (push) Has been cancelled
Test (Python) / import-isolation (deepagents, pydantic-ai, mirage.agents.pydantic_ai) (push) Has been cancelled
Integ / integ-ts (push) Has been cancelled
Integ / integ-fuse (push) Has been cancelled
Test (Install) / python-extra (databricks, mirage.resource.databricks_volume) (push) Has been cancelled
Test (Install) / python-extra (deepagents, mirage.agents.langchain) (push) Has been cancelled
Test (Install) / python-extra (email, mirage.resource.email) (push) Has been cancelled
Test (Install) / python-extra (fuse, mirage.fuse.mount) (push) Has been cancelled
Test (Install) / python-extra (hdf5, mirage.core.filetype.hdf5) (push) Has been cancelled
Test (Install) / python-extra (hf, mirage.resource.hf_buckets) (push) Has been cancelled
Test (Install) / python-extra (lancedb, mirage.resource.lancedb) (push) Has been cancelled
Test (Install) / python-extra (langfuse, mirage.resource.langfuse) (push) Has been cancelled
Test (Install) / python-extra (mongodb, mirage.resource.mongodb) (push) Has been cancelled
Test (Install) / python-extra (nextcloud, mirage.resource.nextcloud) (push) Has been cancelled
Test (Install) / python-extra (openai, mirage.agents.openai_agents) (push) Has been cancelled
Test (Install) / python-extra (openhands, mirage.agents.openhands, 3.12) (push) Has been cancelled
Test (Install) / python-extra (parquet, mirage.core.filetype.parquet) (push) Has been cancelled
Test (Install) / python-extra (postgres, mirage.resource.postgres) (push) Has been cancelled
Test (Install) / python-extra (pydantic-ai, mirage.agents.pydantic_ai) (push) Has been cancelled
Test (Install) / python-extra (qdrant, mirage.resource.qdrant) (push) Has been cancelled
Test (Install) / python-extra (redis, mirage.resource.redis) (push) Has been cancelled
Test (Install) / python-extra (s3, mirage.resource.s3) (push) Has been cancelled
Test (Install) / python-extra (ssh, mirage.resource.ssh) (push) Has been cancelled
Test (Install) / ts-minimal (push) Has been cancelled
This commit is contained in:
@@ -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. =========
|
||||
@@ -0,0 +1,90 @@
|
||||
# ========= 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 time
|
||||
from unittest.mock import AsyncMock, patch
|
||||
|
||||
import pytest
|
||||
|
||||
from mirage.core.google._client import TokenManager, google_headers
|
||||
from mirage.core.google.config import GoogleConfig
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def config():
|
||||
return GoogleConfig(
|
||||
client_id="test-id",
|
||||
client_secret="test-secret",
|
||||
refresh_token="test-refresh",
|
||||
)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_token_manager_refreshes_on_first_call(config):
|
||||
mgr = TokenManager(config)
|
||||
with patch(
|
||||
"mirage.core.google._client.refresh_access_token",
|
||||
new_callable=AsyncMock,
|
||||
return_value=("new-token", 3600),
|
||||
) as mock_refresh:
|
||||
token = await mgr.get_token()
|
||||
assert token == "new-token"
|
||||
mock_refresh.assert_called_once_with(config)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_token_manager_caches_token(config):
|
||||
mgr = TokenManager(config)
|
||||
with patch(
|
||||
"mirage.core.google._client.refresh_access_token",
|
||||
new_callable=AsyncMock,
|
||||
return_value=("cached-token", 3600),
|
||||
) as mock_refresh:
|
||||
t1 = await mgr.get_token()
|
||||
t2 = await mgr.get_token()
|
||||
assert t1 == t2 == "cached-token"
|
||||
assert mock_refresh.call_count == 1
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_token_manager_refreshes_when_expired(config):
|
||||
mgr = TokenManager(config)
|
||||
with patch(
|
||||
"mirage.core.google._client.refresh_access_token",
|
||||
new_callable=AsyncMock,
|
||||
return_value=("token-1", 3600),
|
||||
):
|
||||
await mgr.get_token()
|
||||
|
||||
mgr._expires_at = time.time() - 1
|
||||
|
||||
with patch(
|
||||
"mirage.core.google._client.refresh_access_token",
|
||||
new_callable=AsyncMock,
|
||||
return_value=("token-2", 3600),
|
||||
):
|
||||
token = await mgr.get_token()
|
||||
assert token == "token-2"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_google_headers(config):
|
||||
mgr = TokenManager(config)
|
||||
with patch(
|
||||
"mirage.core.google._client.refresh_access_token",
|
||||
new_callable=AsyncMock,
|
||||
return_value=("my-token", 3600),
|
||||
):
|
||||
headers = await google_headers(mgr)
|
||||
assert headers["Authorization"] == "Bearer my-token"
|
||||
@@ -0,0 +1,171 @@
|
||||
# ========= 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.accessor.gdocs import GDocsAccessor
|
||||
from mirage.cache.index import IndexEntry
|
||||
from mirage.cache.index.ram import RAMIndexCacheStore
|
||||
from mirage.core.gdocs._client import TokenManager
|
||||
from mirage.core.gdocs.read import read, read_doc
|
||||
from mirage.resource.gdocs.config import GDocsConfig
|
||||
from mirage.types import PathSpec
|
||||
from mirage.utils.key_prefix import mount_key
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def token_manager():
|
||||
config = GDocsConfig(
|
||||
client_id="test-id",
|
||||
client_secret="test-secret",
|
||||
refresh_token="test-refresh",
|
||||
)
|
||||
mgr = TokenManager(config)
|
||||
mgr._access_token = "fake-token"
|
||||
mgr._expires_at = 9999999999
|
||||
return mgr
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def accessor(token_manager):
|
||||
return GDocsAccessor(config=None, token_manager=token_manager)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def index():
|
||||
return RAMIndexCacheStore()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_read_doc(token_manager):
|
||||
doc_json = {
|
||||
"documentId": "abc123",
|
||||
"title": "Test Doc",
|
||||
"body": {
|
||||
"content": []
|
||||
},
|
||||
}
|
||||
with patch(
|
||||
"mirage.core.gdocs.read.google_get",
|
||||
new_callable=AsyncMock,
|
||||
return_value=doc_json,
|
||||
):
|
||||
result = await read_doc(token_manager, "abc123")
|
||||
parsed = json.loads(result)
|
||||
assert parsed["documentId"] == "abc123"
|
||||
assert parsed["title"] == "Test Doc"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_read_via_index(accessor, index):
|
||||
await index.set_dir("/gdocs/owned", [
|
||||
("2026-04-01_My_Doc__doc1.gdoc.json",
|
||||
IndexEntry(id="abc123",
|
||||
name="Test Doc",
|
||||
resource_type="gdocs/file",
|
||||
vfs_name="2026-04-01_My_Doc__doc1.gdoc.json")),
|
||||
])
|
||||
doc_json = {
|
||||
"documentId": "abc123",
|
||||
"title": "Test Doc",
|
||||
"body": {
|
||||
"content": []
|
||||
},
|
||||
}
|
||||
with patch(
|
||||
"mirage.core.gdocs.read.google_get",
|
||||
new_callable=AsyncMock,
|
||||
return_value=doc_json,
|
||||
):
|
||||
result = await read(
|
||||
accessor,
|
||||
PathSpec(
|
||||
resource_path=mount_key(
|
||||
"/gdocs/owned/2026-04-01_My_Doc__doc1.gdoc.json",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/owned/2026-04-01_My_Doc__doc1.gdoc.json",
|
||||
directory="/gdocs/owned/2026-04-01_My_Doc__doc1.gdoc.json"),
|
||||
index)
|
||||
parsed = json.loads(result)
|
||||
assert parsed["documentId"] == "abc123"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_read_no_index(accessor):
|
||||
with pytest.raises(FileNotFoundError):
|
||||
await read(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key(
|
||||
"/gdocs/owned/nonexistent.gdoc.json", "/gdocs"),
|
||||
virtual="/gdocs/owned/nonexistent.gdoc.json",
|
||||
directory="/gdocs/owned/nonexistent.gdoc.json"), None)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_read_auto_bootstraps_from_empty_index(accessor, index):
|
||||
files = [{
|
||||
"id": "doc1",
|
||||
"name": "Notes",
|
||||
"modifiedTime": "2026-04-01T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}],
|
||||
}]
|
||||
with (
|
||||
patch(
|
||||
"mirage.core.gdocs.readdir.list_all_files",
|
||||
new_callable=AsyncMock,
|
||||
return_value=files,
|
||||
),
|
||||
patch(
|
||||
"mirage.core.gdocs.read.read_doc",
|
||||
new_callable=AsyncMock,
|
||||
return_value=b'{"documentId":"doc1"}',
|
||||
),
|
||||
):
|
||||
path = PathSpec(
|
||||
resource_path=mount_key(
|
||||
"/gdocs/owned/2026-04-01_Notes__doc1.gdoc.json", "/gdocs"),
|
||||
virtual="/gdocs/owned/2026-04-01_Notes__doc1.gdoc.json",
|
||||
directory="/gdocs/owned/2026-04-01_Notes__doc1.gdoc.json",
|
||||
)
|
||||
result = await read(accessor, path, index)
|
||||
assert b"doc1" in result
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_read_missing_file_raises_after_recursion(accessor, index):
|
||||
with (
|
||||
patch(
|
||||
"mirage.core.gdocs.readdir.list_all_files",
|
||||
new_callable=AsyncMock,
|
||||
return_value=[],
|
||||
),
|
||||
patch(
|
||||
"mirage.core.gdocs.read.read_doc",
|
||||
new_callable=AsyncMock,
|
||||
side_effect=AssertionError("should not call read_doc"),
|
||||
),
|
||||
):
|
||||
path = PathSpec(
|
||||
resource_path=mount_key("/gdocs/owned/Missing__xyz.gdoc.json",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/owned/Missing__xyz.gdoc.json",
|
||||
directory="/gdocs/owned/Missing__xyz.gdoc.json",
|
||||
)
|
||||
with pytest.raises(FileNotFoundError):
|
||||
await read(accessor, path, index)
|
||||
@@ -0,0 +1,363 @@
|
||||
# ========= 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.gdocs import GDocsAccessor
|
||||
from mirage.cache.index.ram import RAMIndexCacheStore
|
||||
from mirage.core.gdocs.readdir import readdir
|
||||
from mirage.core.gdocs.stat import stat
|
||||
from mirage.types import PathSpec
|
||||
from mirage.utils.key_prefix import mount_key
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def accessor():
|
||||
return GDocsAccessor(config=None, token_manager=None)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def index():
|
||||
return RAMIndexCacheStore()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_root(accessor, index):
|
||||
result = await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs", "/gdocs"),
|
||||
virtual="/gdocs",
|
||||
directory="/gdocs"), index)
|
||||
assert result == ["/gdocs/owned", "/gdocs/shared"]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_owned(accessor, index):
|
||||
files = [
|
||||
{
|
||||
"id": "doc1",
|
||||
"name": "My Doc",
|
||||
"modifiedTime": "2026-04-01T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}],
|
||||
},
|
||||
]
|
||||
with patch(
|
||||
"mirage.core.gdocs.readdir.list_all_files",
|
||||
new_callable=AsyncMock,
|
||||
return_value=files,
|
||||
):
|
||||
result = await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned", "/gdocs"),
|
||||
virtual="/gdocs/owned",
|
||||
directory="/gdocs/owned"), index)
|
||||
assert len(result) == 1
|
||||
assert "doc1" in result[0]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_shared(accessor, index):
|
||||
files = [
|
||||
{
|
||||
"id": "doc2",
|
||||
"name": "Shared Doc",
|
||||
"modifiedTime": "2026-03-15T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": False
|
||||
}],
|
||||
},
|
||||
]
|
||||
with patch(
|
||||
"mirage.core.gdocs.readdir.list_all_files",
|
||||
new_callable=AsyncMock,
|
||||
return_value=files,
|
||||
):
|
||||
result = await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/shared", "/gdocs"),
|
||||
virtual="/gdocs/shared",
|
||||
directory="/gdocs/shared"), index)
|
||||
assert len(result) == 1
|
||||
assert "doc2" in result[0]
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_file_path_raises(accessor, index):
|
||||
with pytest.raises(FileNotFoundError):
|
||||
await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned/file.gdoc.json",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/owned/file.gdoc.json",
|
||||
directory="/gdocs/owned/file.gdoc.json"), index)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_invalid_path_raises(accessor, index):
|
||||
with pytest.raises(FileNotFoundError):
|
||||
await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/bogus", "/gdocs"),
|
||||
virtual="/gdocs/bogus",
|
||||
directory="/gdocs/bogus"), index)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_owned_pushes_modified_range(accessor, index):
|
||||
captured = {}
|
||||
|
||||
async def fake_list(token_manager, mime_type=None, **kwargs):
|
||||
captured.update(kwargs)
|
||||
captured["mime_type"] = mime_type
|
||||
return []
|
||||
|
||||
with patch("mirage.core.gdocs.readdir.list_all_files", new=fake_list):
|
||||
await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned/2026-05-*",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/owned/2026-05-*",
|
||||
directory="/gdocs/owned",
|
||||
pattern="2026-05-*"), index)
|
||||
|
||||
assert captured["modified_after"] == "2026-05-01T00:00:00Z"
|
||||
assert captured["modified_before"] == "2026-06-01T00:00:00Z"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_owned_filtered_does_not_cache(accessor, index):
|
||||
files = [{
|
||||
"id": "may",
|
||||
"name": "MayDoc",
|
||||
"modifiedTime": "2026-05-15T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}]
|
||||
}]
|
||||
full_files = files + [{
|
||||
"id": "jan",
|
||||
"name": "JanDoc",
|
||||
"modifiedTime": "2026-01-15T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}]
|
||||
}]
|
||||
|
||||
call_count = {"n": 0}
|
||||
|
||||
async def fake_list(token_manager,
|
||||
mime_type=None,
|
||||
modified_after=None,
|
||||
modified_before=None,
|
||||
**kwargs):
|
||||
call_count["n"] += 1
|
||||
if modified_after:
|
||||
return files
|
||||
return full_files
|
||||
|
||||
with patch("mirage.core.gdocs.readdir.list_all_files", new=fake_list):
|
||||
await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned/2026-05-*",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/owned/2026-05-*",
|
||||
directory="/gdocs/owned",
|
||||
pattern="2026-05-*"), index)
|
||||
result = await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned", "/gdocs"),
|
||||
virtual="/gdocs/owned",
|
||||
directory="/gdocs/owned"), index)
|
||||
|
||||
assert call_count["n"] == 2
|
||||
assert len(result) == 2
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_owned_filtered_bypasses_warm_cache(accessor, index):
|
||||
full_files = [
|
||||
{
|
||||
"id": "may",
|
||||
"name": "MayDoc",
|
||||
"modifiedTime": "2026-05-15T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}]
|
||||
},
|
||||
{
|
||||
"id": "jan",
|
||||
"name": "JanDoc",
|
||||
"modifiedTime": "2026-01-15T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}]
|
||||
},
|
||||
]
|
||||
may_only = [full_files[0]]
|
||||
|
||||
call_count = {"n": 0}
|
||||
|
||||
async def fake_list(token_manager,
|
||||
mime_type=None,
|
||||
modified_after=None,
|
||||
modified_before=None,
|
||||
**kwargs):
|
||||
call_count["n"] += 1
|
||||
if modified_after:
|
||||
return may_only
|
||||
return full_files
|
||||
|
||||
with patch("mirage.core.gdocs.readdir.list_all_files", new=fake_list):
|
||||
await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned", "/gdocs"),
|
||||
virtual="/gdocs/owned",
|
||||
directory="/gdocs/owned"), index)
|
||||
await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned/2026-05-*",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/owned/2026-05-*",
|
||||
directory="/gdocs/owned",
|
||||
pattern="2026-05-*"), index)
|
||||
|
||||
assert call_count["n"] == 2
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_owned_no_pattern_omits_range(accessor, index):
|
||||
captured = {}
|
||||
|
||||
async def fake_list(token_manager, mime_type=None, **kwargs):
|
||||
captured.update(kwargs)
|
||||
return []
|
||||
|
||||
with patch("mirage.core.gdocs.readdir.list_all_files", new=fake_list):
|
||||
await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned", "/gdocs"),
|
||||
virtual="/gdocs/owned",
|
||||
directory="/gdocs/owned"), index)
|
||||
|
||||
assert captured.get("modified_after") is None
|
||||
assert captured.get("modified_before") is None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_owned_non_date_pattern_omits_range(accessor, index):
|
||||
captured = {}
|
||||
|
||||
async def fake_list(token_manager, mime_type=None, **kwargs):
|
||||
captured.update(kwargs)
|
||||
return []
|
||||
|
||||
with patch("mirage.core.gdocs.readdir.list_all_files", new=fake_list):
|
||||
await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned/*foo*", "/gdocs"),
|
||||
virtual="/gdocs/owned/*foo*",
|
||||
directory="/gdocs/owned",
|
||||
pattern="*foo*"), index)
|
||||
|
||||
assert captured.get("modified_after") is None
|
||||
assert captured.get("modified_before") is None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_filtered_then_stat_succeeds(accessor, index):
|
||||
files = [{
|
||||
"id": "may1",
|
||||
"name": "MayDoc",
|
||||
"modifiedTime": "2026-05-15T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": False
|
||||
}]
|
||||
}]
|
||||
with patch(
|
||||
"mirage.core.gdocs.readdir.list_all_files",
|
||||
new_callable=AsyncMock,
|
||||
return_value=files,
|
||||
):
|
||||
listed = await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/shared/2026-05-*",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/shared/2026-05-*",
|
||||
directory="/gdocs/shared",
|
||||
pattern="2026-05-*"), index)
|
||||
assert len(listed) == 1
|
||||
matched = listed[0]
|
||||
result = await stat(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key(matched, "/gdocs"),
|
||||
virtual=matched,
|
||||
directory=matched),
|
||||
index,
|
||||
)
|
||||
assert result.extra["doc_id"] == "may1"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_readdir_owned_newest_first_across_cache(accessor, index):
|
||||
"""Relies on API mock newest-first, mirroring orderBy modifiedTime desc."""
|
||||
files = [
|
||||
{
|
||||
"id": "new",
|
||||
"name": "Latest",
|
||||
"modifiedTime": "2026-05-03T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}]
|
||||
},
|
||||
{
|
||||
"id": "mid",
|
||||
"name": "Middle",
|
||||
"modifiedTime": "2026-04-01T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}]
|
||||
},
|
||||
{
|
||||
"id": "old",
|
||||
"name": "Oldest",
|
||||
"modifiedTime": "2026-01-01T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}]
|
||||
},
|
||||
]
|
||||
with patch(
|
||||
"mirage.core.gdocs.readdir.list_all_files",
|
||||
new_callable=AsyncMock,
|
||||
return_value=files,
|
||||
) as mock_list:
|
||||
first = await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned", "/gdocs"),
|
||||
virtual="/gdocs/owned",
|
||||
directory="/gdocs/owned"), index)
|
||||
second = await readdir(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned", "/gdocs"),
|
||||
virtual="/gdocs/owned",
|
||||
directory="/gdocs/owned"), index)
|
||||
assert mock_list.call_count == 1
|
||||
assert first == second
|
||||
assert "new" in first[0]
|
||||
assert "mid" in first[1]
|
||||
assert "old" in first[-1]
|
||||
@@ -0,0 +1,156 @@
|
||||
# ========= 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.gdocs import GDocsAccessor
|
||||
from mirage.cache.index import IndexEntry
|
||||
from mirage.cache.index.ram import RAMIndexCacheStore
|
||||
from mirage.core.gdocs.stat import stat
|
||||
from mirage.types import FileType, PathSpec
|
||||
from mirage.utils.key_prefix import mount_key
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def accessor():
|
||||
return GDocsAccessor(config=None, token_manager=None)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def index():
|
||||
return RAMIndexCacheStore()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stat_root(accessor, index):
|
||||
result = await stat(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs", "/gdocs"),
|
||||
virtual="/gdocs",
|
||||
directory="/gdocs"), index)
|
||||
assert result.type == FileType.DIRECTORY
|
||||
assert result.name == "/"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stat_owned_dir(accessor, index):
|
||||
result = await stat(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned", "/gdocs"),
|
||||
virtual="/gdocs/owned",
|
||||
directory="/gdocs/owned"), index)
|
||||
assert result.type == FileType.DIRECTORY
|
||||
assert result.name == "owned"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stat_shared_dir(accessor, index):
|
||||
result = await stat(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/shared", "/gdocs"),
|
||||
virtual="/gdocs/shared",
|
||||
directory="/gdocs/shared"), index)
|
||||
assert result.type == FileType.DIRECTORY
|
||||
assert result.name == "shared"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stat_doc(accessor, index):
|
||||
await index.set_dir("/gdocs/owned", [
|
||||
("2026-04-01_My_Doc__doc1.gdoc.json",
|
||||
IndexEntry(id="doc1",
|
||||
name="My Doc",
|
||||
resource_type="gdocs/file",
|
||||
remote_time="2026-04-01T00:00:00.000Z",
|
||||
vfs_name="2026-04-01_My_Doc__doc1.gdoc.json",
|
||||
size=1000)),
|
||||
])
|
||||
result = await stat(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key(
|
||||
"/gdocs/owned/2026-04-01_My_Doc__doc1.gdoc.json", "/gdocs"),
|
||||
virtual="/gdocs/owned/2026-04-01_My_Doc__doc1.gdoc.json",
|
||||
directory="/gdocs/owned/2026-04-01_My_Doc__doc1.gdoc.json"),
|
||||
index,
|
||||
)
|
||||
assert result.name == "2026-04-01_My_Doc__doc1.gdoc.json"
|
||||
assert result.type == FileType.JSON
|
||||
assert result.modified == "2026-04-01T00:00:00.000Z"
|
||||
assert result.extra["doc_id"] == "doc1"
|
||||
assert result.extra["doc_name"] == "My Doc"
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stat_not_found(accessor, index):
|
||||
files = [{
|
||||
"id": "doc1",
|
||||
"name": "My Doc",
|
||||
"modifiedTime": "2026-04-01T00:00:00.000Z",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}],
|
||||
}]
|
||||
with patch(
|
||||
"mirage.core.gdocs.readdir.list_all_files",
|
||||
new_callable=AsyncMock,
|
||||
return_value=files,
|
||||
):
|
||||
with pytest.raises(FileNotFoundError):
|
||||
await stat(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key(
|
||||
"/gdocs/owned/nonexistent.gdoc.json", "/gdocs"),
|
||||
virtual="/gdocs/owned/nonexistent.gdoc.json",
|
||||
directory="/gdocs/owned/nonexistent.gdoc.json"),
|
||||
index)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stat_cache_miss_falls_back_via_readdir(accessor, index):
|
||||
files = [{
|
||||
"id": "doc1",
|
||||
"name": "My Doc",
|
||||
"modifiedTime": "2026-04-01T00:00:00.000Z",
|
||||
"size": "1234",
|
||||
"owners": [{
|
||||
"me": True
|
||||
}],
|
||||
}]
|
||||
target = "/gdocs/owned/2026-04-01_My_Doc__doc1.gdoc.json"
|
||||
with patch(
|
||||
"mirage.core.gdocs.readdir.list_all_files",
|
||||
new_callable=AsyncMock,
|
||||
return_value=files,
|
||||
) as mock_list:
|
||||
result = await stat(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key(target, "/gdocs"),
|
||||
virtual=target,
|
||||
directory=target), index)
|
||||
assert result.type == FileType.JSON
|
||||
assert result.extra["doc_id"] == "doc1"
|
||||
assert mock_list.call_count == 1
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_stat_cache_miss_with_index_none_raises(accessor):
|
||||
with pytest.raises(FileNotFoundError):
|
||||
await stat(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned/doc.gdoc.json",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/owned/doc.gdoc.json",
|
||||
directory="/gdocs/owned/doc.gdoc.json"), None)
|
||||
@@ -0,0 +1,82 @@
|
||||
# ========= 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.gdocs import GDocsAccessor
|
||||
from mirage.cache.index import IndexEntry
|
||||
from mirage.cache.index.ram import RAMIndexCacheStore
|
||||
from mirage.core.gdocs.unlink import unlink
|
||||
from mirage.types import PathSpec
|
||||
from mirage.utils.key_prefix import mount_key
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def accessor():
|
||||
return GDocsAccessor(config=None, token_manager=None)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def index():
|
||||
return RAMIndexCacheStore()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_unlink_calls_delete_with_doc_id(accessor, index):
|
||||
await index.set_dir("/gdocs/owned", [
|
||||
("foo.gdoc.json",
|
||||
IndexEntry(id="doc1",
|
||||
name="Foo",
|
||||
resource_type="gdocs/file",
|
||||
vfs_name="foo.gdoc.json")),
|
||||
])
|
||||
with patch("mirage.core.google.tree_ops.delete_file",
|
||||
new_callable=AsyncMock) as mock_delete:
|
||||
await unlink(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned/foo.gdoc.json",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/owned/foo.gdoc.json",
|
||||
directory="/gdocs/owned/foo.gdoc.json"), index)
|
||||
mock_delete.assert_awaited_once()
|
||||
assert mock_delete.await_args.args[1] == "doc1"
|
||||
listing = await index.list_dir("/gdocs/owned")
|
||||
assert listing.entries is None or listing.entries == []
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_unlink_virtual_dir_raises(accessor, index):
|
||||
with pytest.raises(IsADirectoryError):
|
||||
await unlink(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned", "/gdocs"),
|
||||
virtual="/gdocs/owned",
|
||||
directory="/gdocs/owned"), index)
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_unlink_missing_raises(accessor, index):
|
||||
files = []
|
||||
with patch("mirage.core.gdocs.readdir.list_all_files",
|
||||
new_callable=AsyncMock,
|
||||
return_value=files):
|
||||
with pytest.raises(FileNotFoundError):
|
||||
await unlink(
|
||||
accessor,
|
||||
PathSpec(resource_path=mount_key("/gdocs/owned/nope.gdoc.json",
|
||||
"/gdocs"),
|
||||
virtual="/gdocs/owned/nope.gdoc.json",
|
||||
directory="/gdocs/owned/nope.gdoc.json"), index)
|
||||
@@ -0,0 +1,70 @@
|
||||
# ========= 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.gdocs._client import TokenManager
|
||||
from mirage.core.gdocs.update import batch_update
|
||||
from mirage.resource.gdocs.config import GDocsConfig
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def token_manager():
|
||||
config = GDocsConfig(
|
||||
client_id="test-id",
|
||||
client_secret="test-secret",
|
||||
refresh_token="test-refresh",
|
||||
)
|
||||
mgr = TokenManager(config)
|
||||
mgr._access_token = "fake-token"
|
||||
mgr._expires_at = 9999999999
|
||||
return mgr
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_batch_update(token_manager):
|
||||
payload = json.dumps({
|
||||
"requests": [{
|
||||
"insertText": {
|
||||
"location": {
|
||||
"index": 1
|
||||
},
|
||||
"text": "Hello"
|
||||
}
|
||||
}]
|
||||
})
|
||||
api_response = {"documentId": "abc123", "replies": [{}]}
|
||||
with patch(
|
||||
"mirage.core.gdocs.update.google_post",
|
||||
new_callable=AsyncMock,
|
||||
return_value=api_response,
|
||||
) as mock_post:
|
||||
result = await batch_update(token_manager, "abc123", payload)
|
||||
assert result["documentId"] == "abc123"
|
||||
mock_post.assert_called_once()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_batch_update_invalid_json(token_manager):
|
||||
with pytest.raises(ValueError, match="requests"):
|
||||
await batch_update(token_manager, "abc123", "not json")
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_batch_update_missing_requests_key(token_manager):
|
||||
with pytest.raises(ValueError, match="requests"):
|
||||
await batch_update(token_manager, "abc123", '{"foo": "bar"}')
|
||||
Reference in New Issue
Block a user