zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-07-29 20:19:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Go: add context, part11 (#17426)

Browse Source 
Signed-off-by: Jin Hai <haijin.chn@gmail.com>
This commit is contained in:
Jin Hai
2026-07-27 17:48:44 +08:00
committed by GitHub
parent 30b40caff8
commit 19d861b797
28 changed files with 111 additions and 89 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/testcases/test_web_api/test_chunk_app/test_chunk_routes_unit.py
View File

@@ -598,7 +598,7 @@ def test_restful_chunk_guard_branches_unit(monkeypatch):
monkeypatch.setattr(module.KnowledgebaseService, "accessible", lambda **_kwargs: True)
monkeypatch.setattr(module.DocumentService, "query", lambda **_kwargs: [])
res = _run(_route_core(module.list_chunks)("tenant-1", "kb-1", "doc-1"))
assert res["message"] == "You don't own the document doc-1.", res
assert res["message"] == "you don't own the document doc-1", res
monkeypatch.setattr(module.DocumentService, "query", lambda **_kwargs: [_DummyDoc()])
module.request = SimpleNamespace(args={"id": "chunk-1"}, headers={})

4
test/testcases/test_web_api/test_chunk_app/test_create_chunk.py
View File

@@ -145,7 +145,7 @@ class TestAddChunk:
@pytest.mark.parametrize(
"document_id, expected_code, expected_message",
[
("invalid_document_id", 102, "You don't own the document invalid_document_id."),
("invalid_document_id", 102, "you don't own the document invalid_document_id"),
],
)
def test_invalid_document_id(self, WebApiAuth, add_document, document_id, expected_code, expected_message):
@@ -177,7 +177,7 @@ class TestAddChunk:
delete_document(WebApiAuth, dataset_id, {"ids": [document_id]})
res = add_chunk(WebApiAuth, dataset_id, document_id, {"content": "chunk test"})
assert res["code"] == 102, res
assert res["message"] == f"You don't own the document {document_id}.", res
assert res["message"] == f"you don't own the document {document_id}", res
@pytest.mark.skip(reason="issues/6411")
@pytest.mark.p3

4
test/testcases/test_web_api/test_chunk_app/test_update_chunk.py
View File

@@ -156,7 +156,7 @@ class TestUpdateChunk:
@pytest.mark.parametrize(
"document_id, expected_code, expected_message",
[
("invalid_doc_id", 102, "You don't own the document invalid_doc_id."),
("invalid_doc_id", 102, "you don't own the document invalid_doc_id"),
],
)
def test_invalid_document_id_for_update(self, WebApiAuth, add_chunks, document_id, expected_code, expected_message):
@@ -217,4 +217,4 @@ class TestUpdateChunk:
delete_document(WebApiAuth, dataset_id, {"ids": [document_id]})
res = update_chunk(WebApiAuth, dataset_id, document_id, chunk_ids[0], {"content": "test content"})
assert res["code"] == 102, res
assert res["message"] in [f"You don't own the document {document_id}.", f"Can't find this chunk {chunk_ids[0]}"]
assert res["message"] in [f"you don't own the document {document_id}", f"Can't find this chunk {chunk_ids[0]}"]

2
test/testcases/test_web_api/test_document_app/test_list_documents.py
View File

@@ -197,7 +197,7 @@ class TestDocumentsList:
# Use a non-existent document ID
res = list_documents(WebApiAuth, {"kb_id": kb_id, "id": "non_existent_doc_id"})
assert res["code"] == 102
assert "You don't own the document" in res["message"]
assert "you don't own the document" in res["message"]
@pytest.mark.p3
def test_create_time_filter(self, WebApiAuth, add_documents):