*: add test

Signed-off-by: Weizhen Wang <[email protected]>

resourcemanager/util/mock_gpool.go

@@ -0,0 +1,96 @@
+// Copyright 2022 PingCAP, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package util
+
+import "time"
+
+// MockGPool is only for test
+type MockGPool struct {
+	name string
+}
+
+// NewMockGPool is only for test
+func NewMockGPool(name string) *MockGPool {
+	return &MockGPool{name: name}
+}
+
+// Release is only for test
+func (m *MockGPool) Release() {
+	panic("implement me")
+}
+
+// Tune is only for test
+func (m *MockGPool) Tune(size int) {
+	panic("implement me")
+}
+
+// LastTunerTs is only for test
+func (m *MockGPool) LastTunerTs() time.Time {
+	panic("implement me")
+}
+
+// MaxInFlight is only for test
+func (m *MockGPool) MaxInFlight() int64 {
+	panic("implement me")
+}
+
+// InFlight is only for test
+func (m *MockGPool) InFlight() int64 {
+	panic("implement me")
+}
+
+// MinRT is only for test
+func (m *MockGPool) MinRT() uint64 {
+	panic("implement me")
+}
+
+// MaxPASS is only for test
+func (m *MockGPool) MaxPASS() uint64 {
+	panic("implement me")
+}
+
+// Cap is only for test
+func (m *MockGPool) Cap() int {
+	panic("implement me")
+}
+
+// LongRTT is to represent the baseline latency by tracking a measurement of the long term, less volatile RTT.
+func (m *MockGPool) LongRTT() float64 {
+	panic("implement me")
+}
+
+func (m *MockGPool) UpdateLongRTT(f func(float64) float64) {
+	panic("implement me")
+}
+
+// ShortRTT is to represent the current system latency by tracking a measurement of the short time, and more volatile RTT.
+func (m *MockGPool) ShortRTT() uint64 {
+	panic("implement me")
+}
+
+// GetQueueSize is only for test
+func (m *MockGPool) GetQueueSize() int64 {
+	panic("implement me")
+}
+
+// Running is only for test
+func (m *MockGPool) Running() int {
+	panic("implement me")
+}
+
+// Name is only for test
+func (m *MockGPool) Name() string {
+	return m.name
+}

resourcemanager/util/shard_pool_map_test.go

@@ -0,0 +1,38 @@
+// Copyright 2022 PingCAP, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package util
+
+import (
+	"strconv"
+	"sync/atomic"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestShardPoolMap(t *testing.T) {
+	rc := 10
+	pm := NewShardPoolMap()
+	for i := 0; i < rc; i++ {
+		id := strconv.FormatInt(int64(i), 10)
+		require.NoError(t, pm.Add(id, &PoolContainer{Pool: NewMockGPool(id), Component: DDL}))
+	}
+	require.Error(t, pm.Add("1", &PoolContainer{Pool: NewMockGPool("1"), Component: DDL}))
+	var cnt atomic.Int32
+	pm.Iter(func(pool *PoolContainer) {
+		cnt.Add(1)
+	})
+	require.Equal(t, rc, int(cnt.Load()))
+}