diff options
Diffstat (limited to 'weed/mq/balancer/allocate_test.go')
| -rw-r--r-- | weed/mq/balancer/allocate_test.go | 62 |
1 files changed, 0 insertions, 62 deletions
diff --git a/weed/mq/balancer/allocate_test.go b/weed/mq/balancer/allocate_test.go deleted file mode 100644 index c714788e6..000000000 --- a/weed/mq/balancer/allocate_test.go +++ /dev/null @@ -1,62 +0,0 @@ -package balancer - -import ( - cmap "github.com/orcaman/concurrent-map/v2" - "github.com/seaweedfs/seaweedfs/weed/pb/mq_pb" - "reflect" - "testing" -) - -func Test_allocateOneBroker(t *testing.T) { - brokers := cmap.New[*BrokerStats]() - brokers.SetIfAbsent("localhost:17777", &BrokerStats{ - TopicPartitionCount: 0, - ConsumerCount: 0, - CpuUsagePercent: 0, - }) - - tests := []struct { - name string - args args - wantAssignments []*mq_pb.BrokerPartitionAssignment - }{ - { - name: "test only one broker", - args: args{ - brokers: brokers, - partitionCount: 6, - }, - wantAssignments: []*mq_pb.BrokerPartitionAssignment{ - { - LeaderBroker: "localhost:17777", - FollowerBrokers: []string{"localhost:17777"}, - Partition: &mq_pb.Partition{ - RingSize: MaxPartitionCount, - RangeStart: 0, - RangeStop: MaxPartitionCount, - }, - }, - }, - }, - } - testThem(t, tests) -} - -type args struct { - brokers cmap.ConcurrentMap[string, *BrokerStats] - partitionCount int -} - -func testThem(t *testing.T, tests []struct { - name string - args args - wantAssignments []*mq_pb.BrokerPartitionAssignment -}) { - for _, tt := range tests { - t.Run(tt.name, func(t *testing.T) { - if gotAssignments := allocateTopicPartitions(tt.args.brokers, tt.args.partitionCount); !reflect.DeepEqual(gotAssignments, tt.wantAssignments) { - t.Errorf("allocateTopicPartitions() = %v, want %v", gotAssignments, tt.wantAssignments) - } - }) - } -} |
