1818package kafka .admin
1919
2020import java .util .Collections
21- import kafka .controller .ReplicaAssignment
2221import kafka .server .{BaseRequestTest , BrokerServer }
2322import kafka .utils .TestUtils
2423import kafka .utils .TestUtils ._
@@ -43,25 +42,26 @@ class AddPartitionsTest extends BaseRequestTest {
4342 val partitionId = 0
4443
4544 val topic1 = " new-topic1"
46- val topic1Assignment = Map (0 -> ReplicaAssignment ( Seq (0 ,1 ), List (), List () ))
45+ val topic1Assignment = Map (0 -> Seq (0 ,1 ))
4746 val topic2 = " new-topic2"
48- val topic2Assignment = Map (0 -> ReplicaAssignment ( Seq (1 ,2 ), List (), List () ))
47+ val topic2Assignment = Map (0 -> Seq (1 ,2 ))
4948 val topic3 = " new-topic3"
50- val topic3Assignment = Map (0 -> ReplicaAssignment ( Seq (2 ,3 ,0 ,1 ), List (), List () ))
49+ val topic3Assignment = Map (0 -> Seq (2 ,3 ,0 ,1 ))
5150 val topic4 = " new-topic4"
52- val topic4Assignment = Map (0 -> ReplicaAssignment ( Seq (0 ,3 ), List (), List () ))
51+ val topic4Assignment = Map (0 -> Seq (0 ,3 ))
5352 val topic5 = " new-topic5"
54- val topic5Assignment = Map (1 -> ReplicaAssignment ( Seq (0 ,1 ), List (), List () ))
53+ val topic5Assignment = Map (1 -> Seq (0 ,1 ))
5554 var admin : Admin = _
5655
56+
5757 @ BeforeEach
5858 override def setUp (testInfo : TestInfo ): Unit = {
5959 super .setUp(testInfo)
6060 brokers.foreach(broker => broker.asInstanceOf [BrokerServer ].lifecycleManager.initialUnfenceFuture.get())
61- createTopicWithAssignment(topic1, partitionReplicaAssignment = topic1Assignment.map { case (k, v) => k -> v.replicas } )
62- createTopicWithAssignment(topic2, partitionReplicaAssignment = topic2Assignment.map { case (k, v) => k -> v.replicas } )
63- createTopicWithAssignment(topic3, partitionReplicaAssignment = topic3Assignment.map { case (k, v) => k -> v.replicas } )
64- createTopicWithAssignment(topic4, partitionReplicaAssignment = topic4Assignment.map { case (k, v) => k -> v.replicas } )
61+ createTopicWithAssignment(topic1, partitionReplicaAssignment = topic1Assignment)
62+ createTopicWithAssignment(topic2, partitionReplicaAssignment = topic2Assignment)
63+ createTopicWithAssignment(topic3, partitionReplicaAssignment = topic3Assignment)
64+ createTopicWithAssignment(topic4, partitionReplicaAssignment = topic4Assignment)
6565 admin = createAdminClient()
6666 }
6767
0 commit comments