Fix test

Signed-off-by: Zixuan Liu <[email protected]>
ascentstream · Dec 25, 2024 · 298982a · 298982a
1 parent 8cf7b92
commit 298982a
Showing 1 changed file with 11 additions and 12 deletions.
diff --git a/...src/test/java/org/apache/pulsar/tests/integration/cli/ReplicateSubscriptionStateTest.java b/...src/test/java/org/apache/pulsar/tests/integration/cli/ReplicateSubscriptionStateTest.java
@@ -20,8 +20,11 @@
 
 import static org.awaitility.Awaitility.await;
 import static org.testng.Assert.assertEquals;
+import static org.testng.Assert.assertThrows;
 import static org.testng.Assert.assertTrue;
+import static org.testng.Assert.expectThrows;
 import org.apache.pulsar.common.naming.TopicName;
+import org.apache.pulsar.tests.integration.docker.ContainerExecException;
 import org.apache.pulsar.tests.integration.docker.ContainerExecResult;
 import org.apache.pulsar.tests.integration.suites.PulsarCliTestSuite;
 import org.testng.annotations.AfterClass;
@@ -142,18 +145,14 @@ public void testReplicateSubscriptionStateCmd() throws Exception {
 
     @Test
     public void testReplicateSubscriptionStateCmdWithInvalidParameters() throws Exception {
-        ContainerExecResult result = pulsarCluster.runAdminCommandOnAnyBroker("namespaces",
-                "set-replicate-subscription-state", "public/default");
-        assertTrue(result.getExitCode() != 0);
-        result = pulsarCluster.runAdminCommandOnAnyBroker("namespaces",
-                "set-replicate-subscription-state", "--enabled", "public/default");
-        assertTrue(result.getExitCode() != 0);
+        assertThrows(ContainerExecException.class, () -> pulsarCluster.runAdminCommandOnAnyBroker("namespaces",
+                "set-replicate-subscription-state", "public/default"));
+        assertThrows(ContainerExecException.class, () -> pulsarCluster.runAdminCommandOnAnyBroker("namespaces",
+                "set-replicate-subscription-state", "--enabled", "public/default"));
 
-        result = pulsarCluster.runAdminCommandOnAnyBroker("topicPolicies",
-                "set-replicate-subscription-state", "public/default/test");
-        assertTrue(result.getExitCode() != 0);
-        result = pulsarCluster.runAdminCommandOnAnyBroker("topicPolicies",
-                "set-replicate-subscription-state", "--enabled", "public/default/test");
-        assertTrue(result.getExitCode() != 0);
+        assertThrows(ContainerExecException.class, () -> pulsarCluster.runAdminCommandOnAnyBroker("topicPolicies",
+                "set-replicate-subscription-state", "public/default/test"));
+        assertThrows(ContainerExecException.class, () -> pulsarCluster.runAdminCommandOnAnyBroker("topicPolicies",
+                "set-replicate-subscription-state", "--enabled", "public/default/test"));
     }
 }