Add UTs that got deleted in refactoring

Signed-off-by: Patrick Weizhi Xu <[email protected]>
milvus-io · Jun 21, 2024 · e4eda14 · e4eda14
1 parent 3049d07
commit e4eda14
Show file tree

Hide file tree

Showing 2 changed files with 364 additions and 16 deletions.
diff --git a/internal/datacoord/task_index.go b/internal/datacoord/task_index.go
@@ -117,12 +117,14 @@ func (it *indexBuildTask) AssignTask(ctx context.Context, client types.IndexNode
 		if partitionKeyField == nil || err != nil {
 			log.Ctx(ctx).Warn("index builder get partition key field failed", zap.Int64("buildID", it.buildID), zap.Error(err))
 		} else {
-			optionalFields = append(optionalFields, &indexpb.OptionalFieldInfo{
-				FieldID:   partitionKeyField.FieldID,
-				FieldName: partitionKeyField.Name,
-				FieldType: int32(partitionKeyField.DataType),
-				DataIds:   getBinLogIDs(segment, partitionKeyField.FieldID),
-			})
+			if typeutil.IsFieldDataTypeSupportMaterializedView(partitionKeyField) {
+				optionalFields = append(optionalFields, &indexpb.OptionalFieldInfo{
+					FieldID:   partitionKeyField.FieldID,
+					FieldName: partitionKeyField.Name,
+					FieldType: int32(partitionKeyField.DataType),
+					DataIds:   getBinLogIDs(segment, partitionKeyField.FieldID),
+				})
+			}
 		}
 	}