milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

test: refine load state check in restful test (#33861) 

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
pull/33836/head
zhuwenxing 2024-06-14 16:22:10 +08:00 committed by GitHub
parent 9b69601dfb
commit a963afa0d3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/restful_client_v2/testcases/test_collection_operations.py
View File

@ -798,7 +798,13 @@ class TestGetCollectionLoadState(TestBase):
client.collection_describe(collection_name=name)
rsp = client.collection_load_state(collection_name=name)
assert rsp['code'] == 0
assert rsp['data']['loadState'] in ["LoadStateNotLoad", "LoadStateLoading"]
t0 = time.time()
while time.time() - t0 < 10:
rsp = client.collection_load_state(collection_name=name)
if rsp['data']['loadState'] != "LoadStateNotLoad":
break
time.sleep(1)
assert rsp['data']['loadState'] in ["LoadStateLoading", "LoadStateLoaded"]
# insert data
nb = 3000
data = get_data_by_payload(payload, nb)