[AMQ-8354] Fix unit tests.

amazon-mq · Jun 27, 2024 · 2dc8fc2 · 2dc8fc2
1 parent 9aa294e
commit 2dc8fc2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/activemq-broker/src/test/java/org/apache/activemq/replica/ReplicaPluginInstallationTest.java b/activemq-broker/src/test/java/org/apache/activemq/replica/ReplicaPluginInstallationTest.java
@@ -67,7 +67,9 @@ public void testInstallPluginWithDefaultRole() throws Exception {
         assertThat(installedBroker).isInstanceOf(ReplicaAuthorizationBroker.class);
         Broker nextBroker = ((BrokerFilter) installedBroker).getNext();
         assertThat(nextBroker).isInstanceOf(ReplicaRoleManagementBroker.class);
-        assertThat(((BrokerFilter) nextBroker).getNext()).isEqualTo(broker);
+        Broker next = ((BrokerFilter) nextBroker).getNext();
+        assertThat(next).isInstanceOf(ReplicaJmxBroker.class);
+        assertThat(((BrokerFilter) next).getNext()).isEqualTo(broker);
         assertThat(ReplicaRole.source).isEqualTo(pluginUnderTest.getRole());
     }
 
@@ -80,6 +82,8 @@ public void testInstallPluginWithReplicaRole() throws Exception {
         Broker nextBroker = ((BrokerFilter) installedBroker).getNext();
         assertThat(nextBroker).isInstanceOf(ReplicaRoleManagementBroker.class);
 
-        assertThat(((BrokerFilter) nextBroker).getNext()).isEqualTo(broker);
+        Broker next = ((BrokerFilter) nextBroker).getNext();
+        assertThat(next).isInstanceOf(ReplicaJmxBroker.class);
+        assertThat(((BrokerFilter) next).getNext()).isEqualTo(broker);
     }
 }