zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-15 09:28:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Agent bot 2 permission bugs (#818) (#16909)

Browse Source
This commit is contained in:
Wang Qi
2026-07-14 20:11:16 +08:00
committed by GitHub
parent 760203be42
commit 3087eb854c
5 changed files with 37 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
test/unit_test/api/apps/restful_apis/test_attachment_download_missing_blob.py
View File

@@ -71,8 +71,11 @@ def _load_agent_api(monkeypatch, *, storage_get):
_stub(
monkeypatch,
"api.apps",
AUTH_JWT="jwt",
AUTH_API="api",
AUTH_BETA="beta",
current_user=SimpleNamespace(id="tenant-1"),
login_required=lambda func: func,
login_required=lambda func=None, **_kwargs: (lambda f: f) if func is None else func,
)
_stub(monkeypatch, "api.apps.services.canvas_replica_service", CanvasReplicaService=SimpleNamespace())
_stub(monkeypatch, "api.db", CanvasCategory=SimpleNamespace())

11
test/unit_test/api/apps/restful_apis/test_get_agent_session.py
View File

@@ -44,7 +44,16 @@ def _load_agent_api(monkeypatch, get_by_id_result, delete_calls=None):
delete_calls.append(session_id)
return True
_stub(monkeypatch, "api.apps", current_user=SimpleNamespace(id="tenant-1"), login_required=lambda func: func)
_stub(
monkeypatch,
"api.apps",
AUTH_JWT="jwt",
AUTH_API="api",
AUTH_BETA="beta",
QuartAuthUnauthorized=Exception,
current_user=SimpleNamespace(id="tenant-1", is_superuser=True),
login_required=lambda func=None, **_kwargs: (lambda f: f) if func is None else func,
)
_stub(monkeypatch, "api.apps.services.canvas_replica_service", CanvasReplicaService=SimpleNamespace())
_stub(monkeypatch, "api.db", CanvasCategory=SimpleNamespace())
_stub(monkeypatch, "api.db.db_models", Task=SimpleNamespace())

11
test/unit_test/api/apps/restful_apis/test_rerun_agent_authorization.py
View File

@@ -48,7 +48,16 @@ def _load_agent_api_for_rerun(monkeypatch, *, documents_info, accessible):
acc_fn = accessible if callable(accessible) else (lambda *_a, **_k: accessible)
_stub(monkeypatch, "api.apps", current_user=SimpleNamespace(id="user-owner"), login_required=lambda func: func)
_stub(
monkeypatch,
"api.apps",
AUTH_JWT="jwt",
AUTH_API="api",
AUTH_BETA="beta",
QuartAuthUnauthorized=Exception,
current_user=SimpleNamespace(id="user-owner", is_superuser=True),
login_required=lambda func=None, **_kwargs: (lambda f: f) if func is None else func,
)
_stub(monkeypatch, "api.apps.services.canvas_replica_service", CanvasReplicaService=SimpleNamespace())
_stub(monkeypatch, "api.db", CanvasCategory=SimpleNamespace())