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

This commit is contained in:
wehub-resource-sync
2026-07-13 12:30:44 +08:00
commit bcbd1bdb22
5748 changed files with 562488 additions and 0 deletions
+13
View File
@@ -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. =========
+90
View File
@@ -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"
+171
View File
@@ -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)
+363
View File
@@ -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]
+156
View File
@@ -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)
+82
View File
@@ -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)
+70
View File
@@ -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"}')