Files
strukto-ai--mirage/python/tests/workspace/test_dify_resource.py
T
wehub-resource-sync bcbd1bdb22
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
chore: import upstream snapshot with attribution
2026-07-13 12:30:44 +08:00

67 lines
1.9 KiB
Python

import pytest
from mirage import MountMode, Workspace
from mirage.core.dify import tree
from mirage.resource.dify import DifyConfig, DifyResource
async def list_documents(config):
return [{
"id": "doc-1",
"name": "Quickstart",
"doc_metadata": [{
"name": "slug",
"value": "guides/quickstart"
}],
"enabled": True,
"indexing_status": "completed",
"archived": False,
"tokens": 5,
"data_source_type": "upload_file",
"data_source_detail_dict": {
"upload_file": {
"size": 321
}
},
"created_at": 1716282000,
}]
def resource() -> DifyResource:
return DifyResource(
DifyConfig(
api_key="secret",
base_url="https://dify.example/v1",
dataset_id="dataset-1",
))
def test_workspace_mount_registers_dify_commands_and_ops():
workspace = Workspace({"/knowledge": resource()}, mode=MountMode.READ)
mount = workspace.mount("/knowledge/")
assert "awk" in mount.commands()
assert "ls" in mount.commands()
assert "cat" in mount.commands()
assert "cut" in mount.commands()
assert "rg" in mount.commands()
assert "sed" in mount.commands()
assert "sort" in mount.commands()
assert "stat" in mount.commands()
assert "tree" in mount.commands()
assert "uniq" in mount.commands()
assert "du" in mount.commands()
assert "read" in mount.registered_ops()
assert "stat" in mount.registered_ops()
@pytest.mark.asyncio
async def test_workspace_executes_dify_ls(monkeypatch):
monkeypatch.setattr(tree, "list_all_documents", list_documents)
workspace = Workspace({"/knowledge": resource()}, mode=MountMode.READ)
result = await workspace.execute("ls /knowledge")
assert result.exit_code == 0
assert await result.stdout_str() == "guides\n"