From e21397d68aadd1e8fd86537a843bf09a39cdac9f Mon Sep 17 00:00:00 2001 From: sunby <bingyi.sun@zilliz.com> Date: Fri, 25 Dec 2020 16:08:35 +0800 Subject: [PATCH] Add runtime stats unittest Signed-off-by: sunby <bingyi.sun@zilliz.com> --- internal/master/runtime_stats_test.go | 30 +++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/internal/master/runtime_stats_test.go b/internal/master/runtime_stats_test.go index 539bd2894..7e9949cec 100644 --- a/internal/master/runtime_stats_test.go +++ b/internal/master/runtime_stats_test.go @@ -41,3 +41,33 @@ func TestRuntimeStats_UpdateFieldStats(t *testing.T) { assert.EqualValues(t, 1, found) } } + +func TestRuntimeStats_GetTotalNumOfRelatedSegments(t *testing.T) { + runtimeStats := NewRuntimeStats() + runtimeStats.collStats = make(map[UniqueID]*CollRuntimeStats) + + runtimeStats.collStats[1] = &CollRuntimeStats{ + fieldIndexStats: map[UniqueID][]*FieldIndexRuntimeStats{ + 100: { + {1, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}}, 10}, + {3, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}}, 20}, + {2, []*commonpb.KeyValuePair{{Key: "k2", Value: "v2"}}, 20}, + }, + 200: { + {1, []*commonpb.KeyValuePair{}, 20}, + }, + }, + } + + runtimeStats.collStats[2] = &CollRuntimeStats{ + fieldIndexStats: map[UniqueID][]*FieldIndexRuntimeStats{ + 100: { + {1, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}}, 10}, + }, + }, + } + assert.EqualValues(t, 30, runtimeStats.GetTotalNumOfRelatedSegments(1, 100, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}})) + assert.EqualValues(t, 20, runtimeStats.GetTotalNumOfRelatedSegments(1, 100, []*commonpb.KeyValuePair{{Key: "k2", Value: "v2"}})) + assert.EqualValues(t, 20, runtimeStats.GetTotalNumOfRelatedSegments(1, 200, []*commonpb.KeyValuePair{})) + assert.EqualValues(t, 10, runtimeStats.GetTotalNumOfRelatedSegments(2, 100, []*commonpb.KeyValuePair{{Key: "k1", Value: "v1"}})) +} -- GitLab