test-2

Signed-off-by: lhy1024 <[email protected]>
lhy1024 · Jan 14, 2025 · 40981b4 · 40981b4
1 parent 437b369
commit 40981b4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/pkg/core/store.go b/pkg/core/store.go
@@ -666,6 +666,9 @@ func NewStoresInfo() *StoresInfo {
 func (s *StoresInfo) GetStore(storeID uint64) *StoreInfo {
 	s.RLock()
 	defer s.RUnlock()
+	if s.stores == nil {
+		return nil
+	}
 	store, ok := s.stores[storeID]
 	if !ok {
 		return nil

diff --git a/tests/testutil.go b/tests/testutil.go
@@ -212,7 +212,8 @@ func MustPutStore(re *require.Assertions, cluster *TestCluster, store *metapb.St
 		ts = time.Now().UnixNano()
 	}
 	raftCluster := grpcServer.GetRaftCluster()
-	storeInfo := raftCluster.GetStore(store.GetId())
+	id := store.GetId()
+	storeInfo := raftCluster.GetStore(id)
 	newStore := storeInfo.Clone(
 		core.SetStoreStats(&pdpb.StoreStats{
 			Capacity:  uint64(10 * units.GiB),