diff --git a/tests/python_client/chaos/chaos_objects/memory_stress/chaos_querynode_memory_stress.yaml b/tests/python_client/chaos/chaos_objects/memory_stress/chaos_querynode_memory_stress.yaml index 282219fbc5747849729d677217defc8defcb87b7..17be18d079fc9b94bb1fb68e00e3d7b0d23ce089 100644 --- a/tests/python_client/chaos/chaos_objects/memory_stress/chaos_querynode_memory_stress.yaml +++ b/tests/python_client/chaos/chaos_objects/memory_stress/chaos_querynode_memory_stress.yaml @@ -15,4 +15,5 @@ spec: stressors: memory: workers: 4 - size: 1000MB \ No newline at end of file + size: 512Mi + duration: 5m \ No newline at end of file diff --git a/tests/python_client/chaos/test_chaos_memory_stress.py b/tests/python_client/chaos/test_chaos_memory_stress.py index 538bd970aaca3ef4460833cb72e275535b85cd93..4a8f555eb3d4cc0d51be3c8e8f3d12cdc00d534a 100644 --- a/tests/python_client/chaos/test_chaos_memory_stress.py +++ b/tests/python_client/chaos/test_chaos_memory_stress.py @@ -57,37 +57,28 @@ class TestChaosData: 6. release and reload collection, verify search and query is available expected: after chaos deleted, load, search and query qre both available """ - c_name = cf.gen_unique_str('chaos_memory') - collection_w = construct_from_data(c_name) + c_name = 'chaos_memory_nx6DNW4q' + collection_w = ApiCollectionWrapper() + collection_w.init_collection(c_name) log.debug(collection_w.schema) + log.debug(collection_w._shards_num) # apply memory stress - apply_memory_stress(chaos_yaml) + # apply_memory_stress(chaos_yaml) # wait memory stress - sleep(constants.WAIT_PER_OP * 2) - - # query - collection_w.release() - collection_w.load() - term_expr = f'{ct.default_int64_field_name} in [0, 1, 999, 99]' - t0 = datetime.datetime.now() - query_res, _ = collection_w.query(term_expr) - tt = datetime.datetime.now() - t0 - log.info(f"assert query: {tt}") - assert len(query_res) == 4 - - sleep(constants.WAIT_PER_OP * 5) + # sleep(constants.WAIT_PER_OP * 2) # query collection_w.release() collection_w.load() term_expr = f'{ct.default_int64_field_name} in [0, 1, 999, 99]' - t0 = datetime.datetime.now() - query_res, _ = collection_w.query(term_expr) - tt = datetime.datetime.now() - t0 - log.info(f"assert query: {tt}") - assert len(query_res) == 4 + for i in range(4): + t0_query = datetime.datetime.now() + query_res, _ = collection_w.query(term_expr) + tt_query = datetime.datetime.now() - t0_query + log.info(f"{i} query cost: {tt_query}") + assert len(query_res) == 4 @pytest.mark.tags(CaseLabel.L3) @pytest.mark.parametrize('chaos_yaml', get_chaos_yamls())