AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-04-21 05:29:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Test: Update test cases per PR #5778 (#5880)

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

update test cases per PR https://github.com/infiniflow/ragflow/pull/5778

### Type of change

- [x] update test cases
This commit is contained in:
liu an 2025-03-11 11:07:09 +08:00 committed by GitHub
parent 8e965040ce
commit 21c55a2e0f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 19 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
sdk/python/test/test_http_api/test_dataset_mangement/test_delete_dataset.py
View File

@ -69,12 +69,16 @@ class TestDatasetDeletion:
"AttributeError(\"'str' object has no attribute 'get'\")", "AttributeError(\"'str' object has no attribute 'get'\")",
3, 3,
), ),
(lambda r: {"ids": r[:1]}, 0, "", 2),
(lambda r: {"ids": r}, 0, "", 0),
], ],
) )
def test_basic_scenarios( def test_basic_scenarios(
self, get_http_api_auth, payload, expected_code, expected_message, remaining self, get_http_api_auth, payload, expected_code, expected_message, remaining
): ):
create_datasets(get_http_api_auth, 3) ids = create_datasets(get_http_api_auth, 3)
if callable(payload):
payload = payload(ids)
res = delete_dataset(get_http_api_auth, payload) res = delete_dataset(get_http_api_auth, payload)
assert res["code"] == expected_code assert res["code"] == expected_code
if res["code"] != 0: if res["code"] != 0:
@ -83,58 +87,26 @@ class TestDatasetDeletion:
res = list_dataset(get_http_api_auth) res = list_dataset(get_http_api_auth)
assert len(res["data"]) == remaining assert len(res["data"]) == remaining
def test_delete_one(self, get_http_api_auth): @pytest.mark.parametrize(
count = 3 "payload",
ids = create_datasets(get_http_api_auth, count) [
res = delete_dataset(get_http_api_auth, {"ids": ids[:1]}) lambda r: {"ids": ["invalid_id"] + r},
assert res["code"] == 0 lambda r: {"ids": r[:1] + ["invalid_id"] + r[1:3]},
lambda r: {"ids": r + ["invalid_id"]},
res = list_dataset(get_http_api_auth) ],
assert len(res["data"]) == count - 1 )
def test_delete_partial_invalid_id(self, get_http_api_auth, payload):
def test_delete_multi(self, get_http_api_auth):
ids = create_datasets(get_http_api_auth, 3) ids = create_datasets(get_http_api_auth, 3)
res = delete_dataset(get_http_api_auth, {"ids": ids}) if callable(payload):
payload = payload(ids)
res = delete_dataset(get_http_api_auth, payload)
assert res["code"] == 0 assert res["code"] == 0
assert res["data"]["errors"][0] == "You don't own the dataset invalid_id"
assert res["data"]["success_count"] == 3
res = list_dataset(get_http_api_auth) res = list_dataset(get_http_api_auth)
assert len(res["data"]) == 0 assert len(res["data"]) == 0
@pytest.mark.xfail(reason="issue#5760")
def test_delete_partial_invalid_id_at_beginning(self, get_http_api_auth):
count = 3
ids = create_datasets(get_http_api_auth, count)
res = delete_dataset(get_http_api_auth, {"ids": ["invalid_id"] + ids})
assert res["code"] == 102
assert res["message"] == "You don't own the dataset invalid_id"
res = list_dataset(get_http_api_auth)
assert len(res["data"]) == 3
@pytest.mark.xfail(reason="issue#5760")
def test_delete_partial_invalid_id_in_middle(self, get_http_api_auth):
count = 3
ids = create_datasets(get_http_api_auth, count)
res = delete_dataset(
get_http_api_auth, {"ids": ids[:1] + ["invalid_id"] + ids[1:3]}
)
assert res["code"] == 102
assert res["message"] == "You don't own the dataset invalid_id"
res = list_dataset(get_http_api_auth)
assert len(res["data"]) == 3
@pytest.mark.xfail(reason="issue#5760")
def test_delete_partial_invalid_id_at_end(self, get_http_api_auth):
count = 3
ids = create_datasets(get_http_api_auth, count)
res = delete_dataset(get_http_api_auth, {"ids": ids + ["invalid_id"]})
assert res["code"] == 102
assert res["message"] == "You don't own the dataset invalid_id"
res = list_dataset(get_http_api_auth)
assert len(res["data"]) == 3
def test_repeated_deletion(self, get_http_api_auth): def test_repeated_deletion(self, get_http_api_auth):
ids = create_datasets(get_http_api_auth, 1) ids = create_datasets(get_http_api_auth, 1)
res = delete_dataset(get_http_api_auth, {"ids": ids}) res = delete_dataset(get_http_api_auth, {"ids": ids})