fix tests

This commit is contained in:
chrislu
2024-08-12 09:30:42 -07:00
parent 83fe2bfc36
commit ff66155302

View File

@@ -97,9 +97,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
{
LeaderBroker: "",
Partition: &mq_pb.Partition{},
FollowerBrokers: []string{
"localhost:2",
},
FollowerBroker: "localhost:2",
},
},
},
@@ -114,9 +112,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
{
LeaderBroker: "localhost:1",
Partition: &mq_pb.Partition{},
FollowerBrokers: []string{
"",
},
FollowerBroker: "",
},
},
},
@@ -131,9 +127,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
{
LeaderBroker: "localhost:1",
Partition: &mq_pb.Partition{},
FollowerBrokers: []string{
"localhost:200",
},
FollowerBroker: "localhost:200",
},
},
},
@@ -148,43 +142,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
{
LeaderBroker: "localhost:100",
Partition: &mq_pb.Partition{},
FollowerBrokers: []string{
"localhost:200",
},
},
},
},
hasChanges: true,
},
{
name: "test missing two followers",
args: args{
activeBrokers: activeBrokers,
followerCount: 3,
assignments: []*mq_pb.BrokerPartitionAssignment{
{
LeaderBroker: "localhost:1",
Partition: &mq_pb.Partition{},
FollowerBrokers: []string{
"localhost:2",
},
},
},
},
hasChanges: true,
},
{
name: "test missing some followers",
args: args{
activeBrokers: activeBrokers,
followerCount: 10,
assignments: []*mq_pb.BrokerPartitionAssignment{
{
LeaderBroker: "localhost:1",
Partition: &mq_pb.Partition{},
FollowerBrokers: []string{
"localhost:2",
},
FollowerBroker: "localhost:200",
},
},
},
@@ -199,9 +157,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
{
LeaderBroker: "localhost:1",
Partition: &mq_pb.Partition{},
FollowerBrokers: []string{
"localhost:2",
},
FollowerBroker: "localhost:2",
},
},
},
@@ -230,9 +186,7 @@ func TestEnsureAssignmentsToActiveBrokersX(t *testing.T) {
{
LeaderBroker: "localhost:1",
Partition: &mq_pb.Partition{},
FollowerBrokers: []string{
"localhost:2",
},
FollowerBroker: "localhost:2",
},
},
},