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