Skip to content

Commit 023f9c2

Browse files
authored
KAFKA-18529: ConsumerRebootstrapTest should run for async consumer (#18554)
Reviewers: Kirk True <[email protected]>, Chia-Ping Tsai <[email protected]>, Lianet Magrans <[email protected]>
1 parent 6235a73 commit 023f9c2

File tree

2 files changed

+11
-11
lines changed

2 files changed

+11
-11
lines changed

core/src/test/scala/integration/kafka/api/ConsumerRebootstrapTest.scala

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
package kafka.api
1818

1919
import kafka.api.ConsumerRebootstrapTest._
20-
import kafka.server.QuorumTestHarness.getTestQuorumAndGroupProtocolParametersClassicGroupProtocolOnly
20+
import kafka.server.QuorumTestHarness.getTestQuorumAndGroupProtocolParametersAll
2121
import kafka.utils.{TestInfoUtils, TestUtils}
2222
import org.apache.kafka.clients.CommonClientConfigs
2323
import org.apache.kafka.clients.producer.{KafkaProducer, ProducerRecord}
@@ -135,12 +135,12 @@ object ConsumerRebootstrapTest {
135135

136136
final val RebootstrapTestName = s"${TestInfoUtils.TestWithParameterizedQuorumAndGroupProtocolNames}.useRebootstrapTriggerMs={2}"
137137
def rebootstrapTestParams: stream.Stream[Arguments] = {
138-
assertEquals(1, getTestQuorumAndGroupProtocolParametersClassicGroupProtocolOnly.count())
139-
val args = getTestQuorumAndGroupProtocolParametersClassicGroupProtocolOnly
140-
.findFirst().get.get
141-
stream.Stream.of(
142-
Arguments.of((args :+ true):_*),
143-
Arguments.of((args :+ false):_*)
144-
)
138+
getTestQuorumAndGroupProtocolParametersAll
139+
.flatMap { baseArgs =>
140+
stream.Stream.of(
141+
Arguments.of((baseArgs.get :+ true):_*),
142+
Arguments.of((baseArgs.get :+ false):_*)
143+
)
144+
}
145145
}
146146
}

core/src/test/scala/integration/kafka/api/PlaintextAdminIntegrationTest.scala

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -207,10 +207,10 @@ class PlaintextAdminIntegrationTest extends BaseAdminIntegrationTest {
207207
Admin.create(config)
208208
}
209209

210-
@ParameterizedTest(name = TestInfoUtils.TestWithParameterizedQuorumAndGroupProtocolNames)
211-
@MethodSource(Array("getTestQuorumAndGroupProtocolParametersAll"))
212210
@Timeout(10)
213-
def testDescribeUserScramCredentialsTimeout(quorum: String, groupProtocol: String): Unit = {
211+
@ParameterizedTest
212+
@ValueSource(strings = Array("kraft"))
213+
def testDescribeUserScramCredentialsTimeout(quorum: String): Unit = {
214214
client = createInvalidAdminClient()
215215
try {
216216
// test describeUserScramCredentials(List<String> users, DescribeUserScramCredentialsOptions options)

0 commit comments

Comments
 (0)