From f73f4d5ff16f1b95674ad365bafbcdf43dad4e84 Mon Sep 17 00:00:00 2001 From: zhuwenxing Date: Tue, 23 May 2023 13:25:31 +0800 Subject: [PATCH] [test]Update the method for check insert number (#24289) Signed-off-by: zhuwenxing --- .../testcases/test_all_collections_after_chaos.py | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/tests/python_client/chaos/testcases/test_all_collections_after_chaos.py b/tests/python_client/chaos/testcases/test_all_collections_after_chaos.py index f94c5d3824..1bbf66d7e3 100644 --- a/tests/python_client/chaos/testcases/test_all_collections_after_chaos.py +++ b/tests/python_client/chaos/testcases/test_all_collections_after_chaos.py @@ -41,7 +41,8 @@ class TestAllCollection(TestcaseBase): log.info(f"assert create collection: {tt}, init_entities: {entities}") # insert - data = cf.gen_default_list_data() + insert_batch = 3000 + data = cf.gen_default_list_data(start=-insert_batch) t0 = time.time() _, res = collection_w.insert(data) tt = time.time() - t0 @@ -52,7 +53,7 @@ class TestAllCollection(TestcaseBase): t0 = time.time() _, check_result = collection_w.flush(timeout=180) assert check_result - assert collection_w.num_entities == len(data[0]) + entities + # assert collection_w.num_entities == len(data[0]) + entities tt = time.time() - t0 entities = collection_w.num_entities log.info(f"assert flush: {tt}, entities: {entities}") @@ -86,6 +87,13 @@ class TestAllCollection(TestcaseBase): tt = time.time() - t0 log.info(f"assert search: {tt}") assert len(res_1) == 1 + # query + term_expr = f'{ct.default_int64_field_name} in {[i for i in range(-insert_batch,0)]}' + t0 = time.time() + res, _ = collection_w.query(term_expr) + tt = time.time() - t0 + log.info(f"assert query result {len(res)}: {tt}") + assert len(res) >= insert_batch collection_w.release() # insert data