mirror/seaweedfs
1
0
Fork 0
mirror of https://github.com/seaweedfs/seaweedfs.git synced 2024-11-27 20:59:42 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix test

Browse Source
This commit is contained in:
chrislu 2023-12-11 20:57:29 -08:00
parent 580940bf82
commit 61b05fad95
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
weed/mq/pub_balancer/allocate_test.go
View File

@ -3,7 +3,7 @@ package pub_balancer
import (
cmap "github.com/orcaman/concurrent-map/v2"
"github.com/seaweedfs/seaweedfs/weed/pb/mq_pb"
"reflect"
"github.com/stretchr/testify/assert"
"testing"
)
@ -53,8 +53,14 @@ func testThem(t *testing.T, tests []struct {
}) {
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)
gotAssignments := allocateTopicPartitions(tt.args.brokers, tt.args.partitionCount)
assert.Equal(t, len(tt.wantAssignments), len(gotAssignments))
for i, gotAssignment := range gotAssignments {
assert.Equal(t, tt.wantAssignments[i].LeaderBroker, gotAssignment.LeaderBroker)
assert.Equal(t, tt.wantAssignments[i].Partition.RangeStart, gotAssignment.Partition.RangeStart)
assert.Equal(t, tt.wantAssignments[i].Partition.RangeStop, gotAssignment.Partition.RangeStop)
assert.Equal(t, tt.wantAssignments[i].Partition.RingSize, gotAssignment.Partition.RingSize)
assert.Equal(t, tt.wantAssignments[i].Partition.UnixTimeNs, gotAssignment.Partition.UnixTimeNs)
}
})
}