test/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2024-11-30 02:48:45 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

[test] Fix compaction case for insert during compact (#20628)

Browse Source 
Signed-off-by: ThreadDao <yufen.zong@zilliz.com>

Signed-off-by: ThreadDao <yufen.zong@zilliz.com>
This commit is contained in:
ThreadDao 2022-11-16 16:53:08 +08:00 committed by GitHub
parent 31872f436c
commit a8b6906084
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 26 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
tests/python_client/testcases/test_compaction.py
View File

@ -1183,7 +1183,7 @@ class TestCompactionOperation(TestcaseBase):
for plan in c_plans.plans:
assert len(plan.sources) == 1
@pytest.mark.tags(CaseLabel.L2)
@pytest.mark.tags(CaseLabel.L1)
def test_compact_during_insert(self):
"""
target: test compact during insert and flush
@ -1211,10 +1211,30 @@ class TestCompactionOperation(TestcaseBase):
t.join()
# waitting for new segment index and compact
cost = 240
index_cost = 240
start = time()
while True:
sleep(20)
sleep(10)
collection_w.load()
# new segment compacted
seg_info = self.utility_wrap.get_query_segment_info(collection_w.name)[0]
if len(seg_info) == 2:
break
end = time()
collection_w.release()
if end - start > index_cost:
raise MilvusException(1, f"Waiting more than {index_cost}s for the new segment indexed")
# compact new segment
collection_w.compact()
collection_w.wait_for_compaction_completed()
collection_w.get_compaction_plans()
# waitting for new segment index and compact
compact_cost = 120
start = time()
while True:
sleep(10)
collection_w.load()
# new segment compacted
seg_info = self.utility_wrap.get_query_segment_info(collection_w.name)[0]
@ -1222,11 +1242,11 @@ class TestCompactionOperation(TestcaseBase):
break
end = time()
collection_w.release()
if end - start > cost:
raise MilvusException(1, f"Waitting new segment index and compact more than {cost}s")
if end - start > compact_cost:
raise MilvusException(1, f"Waiting more than {compact_cost}s for the new target segment to load")
# search
search_res, _ = collection_w.search(cf.gen_vectors(1, ct.default_dim),
search_res, _ = collection_w.search([df[ct.default_float_vec_field_name][0]],
ct.default_float_vec_field_name,
ct.default_search_params, ct.default_limit)
assert len(search_res[0]) == ct.default_limit