test: share ec repair helpers

test/plugin_workers/ec_repair/detection_test.go

@@ -2,7 +2,6 @@ package ec_repair_test
 
 import (
 	"context"
-	"sort"
 	"testing"
 	"time"
 
@@ -15,14 +14,6 @@ import (
 	"google.golang.org/grpc/credentials/insecure"
 )
 
-type nodeSpec struct {
-	id       string
-	address  string
-	diskType string
-	diskID   uint32
-	ecShards []*master_pb.VolumeEcShardInformationMessage
-}
-
 func TestEcRepairDetectionFindsCandidates(t *testing.T) {
 	const (
 		volumeID   = uint32(100)
@@ -85,58 +76,3 @@ func TestEcRepairDetectionFindsCandidates(t *testing.T) {
 	require.NotEmpty(t, proposals)
 	require.Equal(t, "ec_repair", proposals[0].JobType)
 }
-
-func buildTopology(nodes []nodeSpec) *master_pb.TopologyInfo {
-	dataNodes := make([]*master_pb.DataNodeInfo, 0, len(nodes))
-	for _, node := range nodes {
-		diskInfo := &master_pb.DiskInfo{
-			DiskId:         node.diskID,
-			MaxVolumeCount: 100,
-			VolumeCount:    10,
-			EcShardInfos:   node.ecShards,
-		}
-		dataNodes = append(dataNodes, &master_pb.DataNodeInfo{
-			Id:        node.id,
-			Address:   node.address,
-			DiskInfos: map[string]*master_pb.DiskInfo{node.diskType: diskInfo},
-		})
-	}
-
-	return &master_pb.TopologyInfo{
-		DataCenterInfos: []*master_pb.DataCenterInfo{
-			{
-				Id: "dc1",
-				RackInfos: []*master_pb.RackInfo{
-					{
-						Id:            "rack1",
-						DataNodeInfos: dataNodes,
-					},
-				},
-			},
-		},
-	}
-}
-
-func buildEcShardInfo(volumeID uint32, collection, diskType string, diskID uint32, shardSizes map[uint32]int64) *master_pb.VolumeEcShardInformationMessage {
-	shardIDs := make([]int, 0, len(shardSizes))
-	for shardID := range shardSizes {
-		shardIDs = append(shardIDs, int(shardID))
-	}
-	sort.Ints(shardIDs)
-
-	var bits uint32
-	sizes := make([]int64, 0, len(shardIDs))
-	for _, shardID := range shardIDs {
-		bits |= (1 << shardID)
-		sizes = append(sizes, shardSizes[uint32(shardID)])
-	}
-
-	return &master_pb.VolumeEcShardInformationMessage{
-		Id:          volumeID,
-		Collection:  collection,
-		EcIndexBits: bits,
-		DiskType:    diskType,
-		DiskId:      diskID,
-		ShardSizes:  sizes,
-	}
-}