AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-20 14:49:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Test: Update test case for #6081 (#6446)

Browse Source 
### What problem does this PR solve?

Update test case for #6081

### Type of change

- [x] Update test case
This commit is contained in:
liu an 2025-03-24 13:18:12 +08:00 committed by GitHub
parent 85eb3775d6
commit 0b63346a1a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
sdk/python/test/test_http_api/test_file_management_within_dataset/test_update_document.py
View File

@ -39,9 +39,7 @@ class TestAuthorization:
), ),
], ],
) )
def test_invalid_auth( def test_invalid_auth(self, get_http_api_auth, tmp_path, auth, expected_code, expected_message):
self, get_http_api_auth, tmp_path, auth, expected_code, expected_message
):
ids = create_datasets(get_http_api_auth, 1) ids = create_datasets(get_http_api_auth, 1)
document_ids = batch_upload_documents(get_http_api_auth, ids[0], 1, tmp_path) document_ids = batch_upload_documents(get_http_api_auth, ids[0], 1, tmp_path)
res = update_documnet(auth, ids[0], document_ids[0], {"name": "auth_test.txt"}) res = update_documnet(auth, ids[0], document_ids[0], {"name": "auth_test.txt"})
@ -91,14 +89,10 @@ class TestUpdatedDocument:
), ),
], ],
) )
def test_name( def test_name(self, get_http_api_auth, tmp_path, name, expected_code, expected_message):
self, get_http_api_auth, tmp_path, name, expected_code, expected_message
):
ids = create_datasets(get_http_api_auth, 1) ids = create_datasets(get_http_api_auth, 1)
document_ids = batch_upload_documents(get_http_api_auth, ids[0], 2, tmp_path) document_ids = batch_upload_documents(get_http_api_auth, ids[0], 2, tmp_path)
res = update_documnet( res = update_documnet(get_http_api_auth, ids[0], document_ids[0], {"name": name})
get_http_api_auth, ids[0], document_ids[0], {"name": name}
)
assert res["code"] == expected_code assert res["code"] == expected_code
if expected_code == 0: if expected_code == 0:
res = list_documnet(get_http_api_auth, ids[0], {"id": document_ids[0]}) res = list_documnet(get_http_api_auth, ids[0], {"id": document_ids[0]})
@ -117,13 +111,9 @@ class TestUpdatedDocument:
), ),
], ],
) )
def test_invalid_document_id( def test_invalid_document_id(self, get_http_api_auth, document_id, expected_code, expected_message):
self, get_http_api_auth, document_id, expected_code, expected_message
):
ids = create_datasets(get_http_api_auth, 1) ids = create_datasets(get_http_api_auth, 1)
res = update_documnet( res = update_documnet(get_http_api_auth, ids[0], document_id, {"name": "new_name.txt"})
get_http_api_auth, ids[0], document_id, {"name": "new_name.txt"}
)
assert res["code"] == expected_code assert res["code"] == expected_code
assert res["message"] == expected_message assert res["message"] == expected_message
@ -138,14 +128,10 @@ class TestUpdatedDocument:
), ),
], ],
) )
def test_invalid_dataset_id( def test_invalid_dataset_id(self, get_http_api_auth, tmp_path, dataset_id, expected_code, expected_message):
self, get_http_api_auth, tmp_path, dataset_id, expected_code, expected_message
):
ids = create_datasets(get_http_api_auth, 1) ids = create_datasets(get_http_api_auth, 1)
document_ids = batch_upload_documents(get_http_api_auth, ids[0], 1, tmp_path) document_ids = batch_upload_documents(get_http_api_auth, ids[0], 1, tmp_path)
res = update_documnet( res = update_documnet(get_http_api_auth, dataset_id, document_ids[0], {"name": "new_name.txt"})
get_http_api_auth, dataset_id, document_ids[0], {"name": "new_name.txt"}
)
assert res["code"] == expected_code assert res["code"] == expected_code
assert res["message"] == expected_message assert res["message"] == expected_message
@ -153,14 +139,10 @@ class TestUpdatedDocument:
"meta_fields, expected_code, expected_message", "meta_fields, expected_code, expected_message",
[({"test": "test"}, 0, ""), ("test", 102, "meta_fields must be a dictionary")], [({"test": "test"}, 0, ""), ("test", 102, "meta_fields must be a dictionary")],
) )
def test_meta_fields( def test_meta_fields(self, get_http_api_auth, tmp_path, meta_fields, expected_code, expected_message):
self, get_http_api_auth, tmp_path, meta_fields, expected_code, expected_message
):
ids = create_datasets(get_http_api_auth, 1) ids = create_datasets(get_http_api_auth, 1)
document_ids = batch_upload_documents(get_http_api_auth, ids[0], 1, tmp_path) document_ids = batch_upload_documents(get_http_api_auth, ids[0], 1, tmp_path)
res = update_documnet( res = update_documnet(get_http_api_auth, ids[0], document_ids[0], {"meta_fields": meta_fields})
get_http_api_auth, ids[0], document_ids[0], {"meta_fields": meta_fields}
)
if expected_code == 0: if expected_code == 0:
res = list_documnet(get_http_api_auth, ids[0], {"id": document_ids[0]}) res = list_documnet(get_http_api_auth, ids[0], {"id": document_ids[0]})
assert res["data"]["docs"][0]["meta_fields"] == meta_fields assert res["data"]["docs"][0]["meta_fields"] == meta_fields
@ -173,18 +155,16 @@ class TestUpdatedDocument:
("naive", 0, ""), ("naive", 0, ""),
("manual", 0, ""), ("manual", 0, ""),
("qa", 0, ""), ("qa", 0, ""),
pytest.param("table", 0, "", marks=pytest.mark.xfail(reason="issues/6081")), ("table", 0, ""),
("paper", 0, ""), ("paper", 0, ""),
("book", 0, ""), ("book", 0, ""),
("laws", 0, ""), ("laws", 0, ""),
("presentation", 0, ""), ("presentation", 0, ""),
pytest.param( ("picture", 0, ""),
"picture", 0, "", marks=pytest.mark.xfail(reason="issues/6081") ("one", 0, ""),
),
pytest.param("one", 0, "", marks=pytest.mark.xfail(reason="issues/6081")),
("knowledge_graph", 0, ""), ("knowledge_graph", 0, ""),
pytest.param("email", 0, "", marks=pytest.mark.xfail(reason="issues/6081")), ("email", 0, ""),
pytest.param("tag", 0, "", marks=pytest.mark.xfail(reason="issues/6081")), ("tag", 0, ""),
("", 102, "`chunk_method` doesn't exist"), ("", 102, "`chunk_method` doesn't exist"),
( (
"other_chunk_method", "other_chunk_method",
@ -193,14 +173,11 @@ class TestUpdatedDocument:
), ),
], ],
) )
def test_chunk_method( def test_chunk_method(self, get_http_api_auth, tmp_path, chunk_method, expected_code, expected_message):
self, get_http_api_auth, tmp_path, chunk_method, expected_code, expected_message
):
ids = create_datasets(get_http_api_auth, 1) ids = create_datasets(get_http_api_auth, 1)
document_ids = batch_upload_documents(get_http_api_auth, ids[0], 1, tmp_path) document_ids = batch_upload_documents(get_http_api_auth, ids[0], 1, tmp_path)
res = update_documnet( res = update_documnet(get_http_api_auth, ids[0], document_ids[0], {"chunk_method": chunk_method})
get_http_api_auth, ids[0], document_ids[0], {"chunk_method": chunk_method} print(res)
)
assert res["code"] == expected_code assert res["code"] == expected_code
if expected_code == 0: if expected_code == 0:
res = list_documnet(get_http_api_auth, ids[0], {"id": document_ids[0]}) res = list_documnet(get_http_api_auth, ids[0], {"id": document_ids[0]})