18
18
package kafka .admin
19
19
20
20
import java .util .Collections
21
- import kafka .controller .ReplicaAssignment
22
21
import kafka .server .{BaseRequestTest , BrokerServer }
23
22
import kafka .utils .TestUtils
24
23
import kafka .utils .TestUtils ._
@@ -43,25 +42,26 @@ class AddPartitionsTest extends BaseRequestTest {
43
42
val partitionId = 0
44
43
45
44
val topic1 = " new-topic1"
46
- val topic1Assignment = Map (0 -> ReplicaAssignment ( Seq (0 ,1 ), List (), List () ))
45
+ val topic1Assignment = Map (0 -> Seq (0 ,1 ))
47
46
val topic2 = " new-topic2"
48
- val topic2Assignment = Map (0 -> ReplicaAssignment ( Seq (1 ,2 ), List (), List () ))
47
+ val topic2Assignment = Map (0 -> Seq (1 ,2 ))
49
48
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 ))
51
50
val topic4 = " new-topic4"
52
- val topic4Assignment = Map (0 -> ReplicaAssignment ( Seq (0 ,3 ), List (), List () ))
51
+ val topic4Assignment = Map (0 -> Seq (0 ,3 ))
53
52
val topic5 = " new-topic5"
54
- val topic5Assignment = Map (1 -> ReplicaAssignment ( Seq (0 ,1 ), List (), List () ))
53
+ val topic5Assignment = Map (1 -> Seq (0 ,1 ))
55
54
var admin : Admin = _
56
55
56
+
57
57
@ BeforeEach
58
58
override def setUp (testInfo : TestInfo ): Unit = {
59
59
super .setUp(testInfo)
60
60
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)
65
65
admin = createAdminClient()
66
66
}
67
67
0 commit comments