aloknnikhil commented on a change in pull request #9996:
URL: https://github.com/apache/kafka/pull/9996#discussion_r568973059
##########
File path:
clients/src/test/java/org/apache/kafka/clients/admin/KafkaAdminClientTest.java
##########
@@ -5169,6 +5171,72 @@ public void testShouldRefreshCoordinator() {
assertTrue(ConsumerGroupOperationContext.shouldRefreshCoordinator(errorCounts));
}
+ @Test
+ public void testDecommissionBrokerSuccess() throws InterruptedException,
ExecutionException {
+ int decommissionedBrokerNode = 1;
+ try (AdminClientUnitTestEnv env = mockClientEnv()) {
+ env.kafkaClient().setNodeApiVersions(
+ NodeApiVersions.create(ApiKeys.DECOMMISSION_BROKER.id,
(short) 0, (short) 0));
+
env.kafkaClient().prepareResponse(prepareDecommissionBrokerResponse(Errors.NONE,
0));
+
+ DecommissionBrokerResult result =
env.adminClient().decommissionBroker(decommissionedBrokerNode);
+
+ // Validate response
+ assertNotNull(result.all());
+ result.all().get();
+ }
+ }
+
+ @Test
+ public void testDecommissionBrokerFailure() {
+ int decommissionedBrokerNode = 1;
+ try (AdminClientUnitTestEnv env = mockClientEnv()) {
+ env.kafkaClient().setNodeApiVersions(
+ NodeApiVersions.create(ApiKeys.DECOMMISSION_BROKER.id,
(short) 0, (short) 0));
+
env.kafkaClient().prepareResponse(prepareDecommissionBrokerResponse(Errors.BROKER_NOT_AVAILABLE,
0));
Review comment:
Makes sense. Updated
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
