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
67 lines
1.9 KiB
Python
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"
|