mirror of
https://github.com/infiniflow/ragflow.git
synced 2025-07-03 07:04:09 +00:00
b6f3f15f0b
### What problem does this PR solve? 1. Read KB list path parameter, page_number and page_size, which type isn't int 2. Add cases on create / list / delete datasets. ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) - [x] Test cases Signed-off-by: jinhai <haijin.chn@gmail.com>
64 lines
1.9 KiB
Python
64 lines
1.9 KiB
Python
from common import HOST_ADDRESS, create_dataset, list_dataset, rm_dataset
|
|
import requests
|
|
|
|
|
|
def test_dataset(get_auth):
|
|
# create dataset
|
|
res = create_dataset(get_auth, "test_create_dataset")
|
|
assert res.get("code") == 0, f"{res.get('message')}"
|
|
|
|
# list dataset
|
|
page_number = 1
|
|
dataset_list = []
|
|
while True:
|
|
res = list_dataset(get_auth, page_number)
|
|
data = res.get("data")
|
|
for item in data:
|
|
dataset_id = item.get("id")
|
|
dataset_list.append(dataset_id)
|
|
if len(dataset_list) < page_number * 150:
|
|
break
|
|
page_number += 1
|
|
|
|
print(f"found {len(dataset_list)} datasets")
|
|
# delete dataset
|
|
for dataset_id in dataset_list:
|
|
res = rm_dataset(get_auth, dataset_id)
|
|
assert res.get("code") == 0, f"{res.get('message')}"
|
|
print(f"{len(dataset_list)} datasets are deleted")
|
|
|
|
|
|
def test_dataset_1k_dataset(get_auth):
|
|
# create dataset
|
|
authorization = {"Authorization": get_auth}
|
|
url = f"{HOST_ADDRESS}/v1/kb/create"
|
|
for i in range(1000):
|
|
res = create_dataset(get_auth, f"test_create_dataset_{i}")
|
|
assert res.get("code") == 0, f"{res.get('message')}"
|
|
|
|
# list dataset
|
|
page_number = 1
|
|
dataset_list = []
|
|
while True:
|
|
res = list_dataset(get_auth, page_number)
|
|
data = res.get("data")
|
|
for item in data:
|
|
dataset_id = item.get("id")
|
|
dataset_list.append(dataset_id)
|
|
if len(dataset_list) < page_number * 150:
|
|
break
|
|
page_number += 1
|
|
|
|
print(f"found {len(dataset_list)} datasets")
|
|
# delete dataset
|
|
for dataset_id in dataset_list:
|
|
res = rm_dataset(get_auth, dataset_id)
|
|
assert res.get("code") == 0, f"{res.get('message')}"
|
|
print(f"{len(dataset_list)} datasets are deleted")
|
|
|
|
# delete dataset
|
|
# create invalid name dataset
|
|
# update dataset with different parameters
|
|
# create duplicated name dataset
|
|
#
|