admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-06 17:18:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: fix partial update chaos checker (#45550)

Browse Source 
Issue: #45489 
pr: #45492

---------

Signed-off-by: Eric Hou <eric.hou@zilliz.com>
Co-authored-by: Eric Hou <eric.hou@zilliz.com>
This commit is contained in:
Feilong Hou 2025-11-13 18:51:39 +08:00 committed by GitHub
parent 88ce11a361
commit df05e7c12c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 14 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
tests/python_client/chaos/checker.py
View File

@ -1212,7 +1212,7 @@ class PartialUpdateChecker(Checker):
collection_name = cf.gen_unique_str("PartialUpdateChecker_") collection_name = cf.gen_unique_str("PartialUpdateChecker_")
super().__init__(collection_name=collection_name, shards_num=shards_num, schema=schema) super().__init__(collection_name=collection_name, shards_num=shards_num, schema=schema)
schema = self.get_schema() schema = self.get_schema()
self.data = cf.gen_row_data_by_schema(nb=4, schema=schema) self.data = cf.gen_row_data_by_schema(nb=constants.DELTA_PER_INS, schema=schema)
@trace() @trace()
def partial_update_entities(self): def partial_update_entities(self):
@ -1229,26 +1229,21 @@ class PartialUpdateChecker(Checker):
schema = self.get_schema() schema = self.get_schema()
pk_field_name = self.int64_field_name pk_field_name = self.int64_field_name
rows = len(self.data) rows = len(self.data)
half = rows // 2
num_fields = len(schema.fields)
pk_old = [d[pk_field_name] for d in self.data[:half]]
# if count is even, use partial update; if count is odd, use full insert
if count % 2 == 0:
# Generate a fresh full batch (used for inserts and as a source of values) # Generate a fresh full batch (used for inserts and as a source of values)
full_rows = cf.gen_row_data_by_schema(nb=rows, schema=schema) full_rows = cf.gen_row_data_by_schema(nb=rows, schema=schema)
self.data = full_rows
else:
num_fields = len(schema.fields)
# Choose subset fields to update: always include PK + one non-PK field if available # Choose subset fields to update: always include PK + one non-PK field if available
num = count % num_fields num = count % num_fields
desired_fields = [pk_field_name, schema.fields[num if num != 0 else 1].name] desired_fields = [pk_field_name, schema.fields[num if num != 0 else 1].name]
partial_rows = cf.gen_row_data_by_schema(nb=half, schema=schema, partial_rows = cf.gen_row_data_by_schema(nb=rows, schema=schema,
desired_field_names=desired_fields) desired_field_names=desired_fields)
self.data = partial_rows
# Override PKs for partial updates to target existing rows
for i in range(half):
partial_rows[i][pk_field_name] = pk_old[i]
# Assemble final batch: first half partial updates, second half full inserts
self.data = partial_rows + full_rows[half:]
res, result = self.partial_update_entities() res, result = self.partial_update_entities()
return res, result return res, result