This is an automated email from the ASF dual-hosted git repository.

jbonofre pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/activemq.git


The following commit(s) were added to refs/heads/main by this push:
     new 82a237476 [AMQ-8981] Fix CustomBridgeFactoryTest
     new 81ee88d91 Merge pull request #899 from mattrpav/AMQ-8981b
82a237476 is described below

commit 82a2374764d4002e72f37457143f3207d732cf04
Author: Matt Pavlovich <[email protected]>
AuthorDate: Mon Aug 22 09:35:43 2022 -0500

    [AMQ-8981] Fix CustomBridgeFactoryTest
---
 .../apache/activemq/network/BaseNetworkTest.java   | 15 +++--
 .../network/CompositeNetworkBridgeListener.java    | 49 ++++++++++++++
 .../activemq/network/CustomBridgeFactoryTest.java  | 78 +---------------------
 .../network/CustomNetworkBridgeFactory.java        | 29 ++++++++
 .../network/localBroker-custom-factory.xml         |  2 +-
 5 files changed, 93 insertions(+), 80 deletions(-)

diff --git 
a/activemq-unit-tests/src/test/java/org/apache/activemq/network/BaseNetworkTest.java
 
b/activemq-unit-tests/src/test/java/org/apache/activemq/network/BaseNetworkTest.java
index e5cd967f6..161591420 100644
--- 
a/activemq-unit-tests/src/test/java/org/apache/activemq/network/BaseNetworkTest.java
+++ 
b/activemq-unit-tests/src/test/java/org/apache/activemq/network/BaseNetworkTest.java
@@ -53,10 +53,17 @@ public class BaseNetworkTest {
     }
 
     protected void doTearDown() throws Exception {
-        localConnection.close();
-        remoteConnection.close();
-        localBroker.stop();
-        remoteBroker.stop();
+        if(localConnection != null)
+            localConnection.close();
+
+        if(remoteConnection != null)
+            remoteConnection.close();
+
+        if(localBroker != null)
+            localBroker.stop();
+
+        if(remoteBroker != null)
+            remoteBroker.stop();
     }
 
     protected void doSetUp(boolean deleteAllMessages) throws Exception {
diff --git 
a/activemq-unit-tests/src/test/java/org/apache/activemq/network/CompositeNetworkBridgeListener.java
 
b/activemq-unit-tests/src/test/java/org/apache/activemq/network/CompositeNetworkBridgeListener.java
new file mode 100644
index 000000000..e9fdfcf38
--- /dev/null
+++ 
b/activemq-unit-tests/src/test/java/org/apache/activemq/network/CompositeNetworkBridgeListener.java
@@ -0,0 +1,49 @@
+package org.apache.activemq.network;
+
+import java.util.Arrays;
+import java.util.List;
+
+import org.apache.activemq.command.Message;
+
+public class CompositeNetworkBridgeListener implements NetworkBridgeListener {
+    private final List<NetworkBridgeListener> listeners;
+
+    public CompositeNetworkBridgeListener(NetworkBridgeListener... wrapped) {
+        this.listeners = Arrays.asList(wrapped);
+    }
+
+    @Override
+    public void bridgeFailed() {
+        for (NetworkBridgeListener listener : listeners) {
+            listener.bridgeFailed();
+        }
+    }
+
+    @Override
+    public void onStart(NetworkBridge bridge) {
+        for (NetworkBridgeListener listener : listeners) {
+            listener.onStart(bridge);
+        }
+    }
+
+    @Override
+    public void onStop(NetworkBridge bridge) {
+        for (NetworkBridgeListener listener : listeners) {
+            listener.onStop(bridge);
+        }
+    }
+
+    @Override
+    public void onOutboundMessage(NetworkBridge bridge, Message message) {
+        for (NetworkBridgeListener listener : listeners) {
+            listener.onOutboundMessage(bridge, message);
+        }
+    }
+
+    @Override
+    public void onInboundMessage(NetworkBridge bridge, Message message) {
+        for (NetworkBridgeListener listener : listeners) {
+            listener.onInboundMessage(bridge, message);
+        }
+    }
+}
diff --git 
a/activemq-unit-tests/src/test/java/org/apache/activemq/network/CustomBridgeFactoryTest.java
 
b/activemq-unit-tests/src/test/java/org/apache/activemq/network/CustomBridgeFactoryTest.java
index c14225352..79dedab3b 100644
--- 
a/activemq-unit-tests/src/test/java/org/apache/activemq/network/CustomBridgeFactoryTest.java
+++ 
b/activemq-unit-tests/src/test/java/org/apache/activemq/network/CustomBridgeFactoryTest.java
@@ -42,7 +42,6 @@ import javax.jms.TextMessage;
 /**
  * Basic test which verify if custom bridge factory receives any interactions 
when configured.
  */
-@Ignore // AMQ-8977 -- can't find class due to xbean or mockito upgrade
 public class CustomBridgeFactoryTest extends BaseNetworkTest {
 
     private ActiveMQQueue outgoing = new ActiveMQQueue("outgoing");
@@ -50,7 +49,7 @@ public class CustomBridgeFactoryTest extends BaseNetworkTest {
     /**
      * Verification of outgoing communication - from local broker (with 
customized bridge configured) to remote one.
      */
-    @Test(timeout = 60000)
+    @Test(timeout = 10000)
     public void verifyOutgoingCommunication() throws JMSException {
         CustomNetworkBridgeFactory bridgeFactory = 
getCustomNetworkBridgeFactory();
         NetworkBridgeListener listener = bridgeFactory.getListener();
@@ -68,7 +67,7 @@ public class CustomBridgeFactoryTest extends BaseNetworkTest {
     /**
      * Additional test which makes sure that custom bridge receives 
notification about broker shutdown.
      */
-    @Test(timeout = 60000)
+    @Test(timeout = 10000)
     public void verifyBrokerShutdown() {
         shutdownTest(() -> {
             try {
@@ -83,7 +82,7 @@ public class CustomBridgeFactoryTest extends BaseNetworkTest {
     /**
      * Verification of network connector shutdown.
      */
-    @Test(timeout = 60000)
+    @Test(timeout = 10000)
     public void verifyConnectorShutdown() {
         shutdownTest(() -> {
             try {
@@ -145,75 +144,4 @@ public class CustomBridgeFactoryTest extends 
BaseNetworkTest {
     protected String getLocalBrokerURI() {
         return "org/apache/activemq/network/localBroker-custom-factory.xml";
     }
-
-    // test classes
-    static class CustomNetworkBridgeFactory implements BridgeFactory {
-
-        private final NetworkBridgeListener listener;
-
-        CustomNetworkBridgeFactory() {
-            this(Mockito.mock(NetworkBridgeListener.class));
-        }
-
-        CustomNetworkBridgeFactory(NetworkBridgeListener listener) {
-            this.listener = listener;
-        }
-
-        public NetworkBridgeListener getListener() {
-            return listener;
-        }
-
-        @Override
-        public DemandForwardingBridge 
createNetworkBridge(NetworkBridgeConfiguration configuration, Transport 
localTransport, Transport remoteTransport, NetworkBridgeListener listener) {
-            DemandForwardingBridge bridge = new 
DemandForwardingBridge(configuration, localTransport, remoteTransport);
-            bridge.setNetworkBridgeListener(new 
CompositeNetworkBridgeListener(this.listener, listener));
-            return bridge;
-        }
-
-    }
-
-    static class CompositeNetworkBridgeListener implements 
NetworkBridgeListener {
-
-        private final List<NetworkBridgeListener> listeners;
-
-        public CompositeNetworkBridgeListener(NetworkBridgeListener ... 
wrapped) {
-            this.listeners = Arrays.asList(wrapped);
-        }
-
-        @Override
-        public void bridgeFailed() {
-            for (NetworkBridgeListener listener : listeners) {
-                listener.bridgeFailed();
-            }
-        }
-
-        @Override
-        public void onStart(NetworkBridge bridge) {
-            for (NetworkBridgeListener listener : listeners) {
-                listener.onStart(bridge);
-            }
-        }
-
-        @Override
-        public void onStop(NetworkBridge bridge) {
-            for (NetworkBridgeListener listener : listeners) {
-                listener.onStop(bridge);
-            }
-        }
-
-        @Override
-        public void onOutboundMessage(NetworkBridge bridge, Message message) {
-            for (NetworkBridgeListener listener : listeners) {
-                listener.onOutboundMessage(bridge, message);
-            }
-        }
-
-        @Override
-        public void onInboundMessage(NetworkBridge bridge, Message message) {
-            for (NetworkBridgeListener listener : listeners) {
-                listener.onInboundMessage(bridge, message);
-            }
-        }
-    }
-
 }
diff --git 
a/activemq-unit-tests/src/test/java/org/apache/activemq/network/CustomNetworkBridgeFactory.java
 
b/activemq-unit-tests/src/test/java/org/apache/activemq/network/CustomNetworkBridgeFactory.java
new file mode 100644
index 000000000..eecefe1aa
--- /dev/null
+++ 
b/activemq-unit-tests/src/test/java/org/apache/activemq/network/CustomNetworkBridgeFactory.java
@@ -0,0 +1,29 @@
+package org.apache.activemq.network;
+
+import org.apache.activemq.transport.Transport;
+import org.mockito.Mockito;
+
+public class CustomNetworkBridgeFactory implements BridgeFactory { 
+
+    private final NetworkBridgeListener listener;
+
+    public CustomNetworkBridgeFactory() {
+        this(Mockito.mock(NetworkBridgeListener.class));
+    }
+
+    public CustomNetworkBridgeFactory(NetworkBridgeListener listener) {
+        this.listener = listener;
+    }
+
+    public NetworkBridgeListener getListener() {
+        return listener;
+    }
+
+    @Override
+    public DemandForwardingBridge 
createNetworkBridge(NetworkBridgeConfiguration configuration, Transport 
localTransport, Transport remoteTransport, NetworkBridgeListener listener) {
+        DemandForwardingBridge bridge = new 
DemandForwardingBridge(configuration, localTransport, remoteTransport);
+        bridge.setNetworkBridgeListener(new 
CompositeNetworkBridgeListener(this.listener, listener));
+        return bridge;
+    }
+
+ }
diff --git 
a/activemq-unit-tests/src/test/resources/org/apache/activemq/network/localBroker-custom-factory.xml
 
b/activemq-unit-tests/src/test/resources/org/apache/activemq/network/localBroker-custom-factory.xml
index 9dc8c61be..d8f30fe87 100644
--- 
a/activemq-unit-tests/src/test/resources/org/apache/activemq/network/localBroker-custom-factory.xml
+++ 
b/activemq-unit-tests/src/test/resources/org/apache/activemq/network/localBroker-custom-factory.xml
@@ -27,7 +27,7 @@
     <networkConnectors>
       <networkConnector uri="static:(tcp://localhost:61617)" 
name="networkConnector">
           <bridgeFactory>
-              <bean xmlns="http://www.springframework.org/schema/beans"; 
class="org.apache.activemq.network.CustomBridgeFactoryTest.CustomNetworkBridgeFactory"
 />
+              <bean xmlns="http://www.springframework.org/schema/beans"; 
class="org.apache.activemq.network.CustomNetworkBridgeFactory" />
           </bridgeFactory>
       </networkConnector>
     </networkConnectors>

Reply via email to