mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-14 12:16:20 +00:00
fix: mini issues
This commit is contained in:
1 parent
72652814c0
commit
e0649d25cf
7 files changed
+66
-33
No files matched your search
@@ -386,9 +386,7 @@ class TestConfigureSource:
|
||||
|
||||
def test_request_active_docs_used(self, monkeypatch):
|
||||
import application.api.answer.services.stream_processor as sp_mod
|
||||
from application.api.answer.services.stream_processor import (
|
||||
StreamProcessor,
|
||||
)
|
||||
StreamProcessor = sp_mod.StreamProcessor
|
||||
|
||||
monkeypatch.setattr(sp_mod, "can_access", lambda *a, **k: True)
|
||||
sp = StreamProcessor({"active_docs": "abc"}, {"sub": "u"})
|
||||
@@ -402,9 +400,7 @@ class TestConfigureSource:
|
||||
answer, while /api/sources/<id>/search correctly refused the same id.
|
||||
"""
|
||||
import application.api.answer.services.stream_processor as sp_mod
|
||||
from application.api.answer.services.stream_processor import (
|
||||
StreamProcessor,
|
||||
)
|
||||
StreamProcessor = sp_mod.StreamProcessor
|
||||
|
||||
monkeypatch.setattr(sp_mod, "can_access", lambda *a, **k: False)
|
||||
sp = StreamProcessor({"active_docs": "someone-elses-id"}, {"sub": "u"})
|
||||
@@ -414,9 +410,7 @@ class TestConfigureSource:
|
||||
|
||||
def test_mixed_access_keeps_only_the_readable_ids(self, monkeypatch):
|
||||
import application.api.answer.services.stream_processor as sp_mod
|
||||
from application.api.answer.services.stream_processor import (
|
||||
StreamProcessor,
|
||||
)
|
||||
StreamProcessor = sp_mod.StreamProcessor
|
||||
|
||||
monkeypatch.setattr(
|
||||
sp_mod, "can_access", lambda conn, kind, sid, user: sid == "mine"
|
||||
@@ -427,9 +421,7 @@ class TestConfigureSource:
|
||||
|
||||
def test_access_check_failure_fails_closed(self, monkeypatch):
|
||||
import application.api.answer.services.stream_processor as sp_mod
|
||||
from application.api.answer.services.stream_processor import (
|
||||
StreamProcessor,
|
||||
)
|
||||
StreamProcessor = sp_mod.StreamProcessor
|
||||
|
||||
def _boom(*a, **k):
|
||||
raise RuntimeError("db down")
|
||||
|
||||
Reference in new issue
Block a user