fix: pass MessageBroker to rqueueMessageHandler in unit tests

rqueueMessageHandler now requires a MessageBroker argument (added to wire
primary-handler-dispatch capability). Update both test classes to supply a
mocked MessageBroker stubbed with REDIS_DEFAULTS capabilities.

Assisted-By: Claude Code

Author: sonus21

rqueue-spring-boot-starter/src/test/java/com/github/sonus21/rqueue/spring/boot/tests/unit/RqueueListenerAutoConfigTest.java

@@ -20,6 +20,7 @@
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.when;
 
 import com.github.sonus21.TestBase;
 import com.github.sonus21.rqueue.config.SimpleRqueueListenerContainerFactory;
@@ -30,6 +31,8 @@
 import com.github.sonus21.rqueue.core.RqueueMessageEnqueuer;
 import com.github.sonus21.rqueue.core.RqueueMessageTemplate;
 import com.github.sonus21.rqueue.core.impl.UuidV4RqueueMessageIdGenerator;
+import com.github.sonus21.rqueue.core.spi.Capabilities;
+import com.github.sonus21.rqueue.core.spi.MessageBroker;
 import com.github.sonus21.rqueue.listener.RqueueMessageHandler;
 import com.github.sonus21.rqueue.spring.boot.RqueueListenerAutoConfig;
 import com.github.sonus21.rqueue.spring.boot.tests.SpringBootUnitTest;
@@ -58,6 +61,9 @@ class RqueueListenerAutoConfigTest extends TestBase {
   @Mock
   private RqueueMessageHandler rqueueMessageHandler;
 
+  @Mock
+  private MessageBroker messageBroker;
+
   @Mock
   private RedisConnectionFactory redisConnectionFactory;
 
@@ -77,7 +83,8 @@ public void init() throws IllegalAccessException {
   @Test
   void rqueueMessageHandlerDefaultCreation()
       throws ClassNotFoundException, InstantiationException, IllegalAccessException {
-    assertNotNull(rqueueMessageAutoConfig.rqueueMessageHandler());
+    when(messageBroker.capabilities()).thenReturn(Capabilities.REDIS_DEFAULTS);
+    assertNotNull(rqueueMessageAutoConfig.rqueueMessageHandler(messageBroker));
   }
 
   @Test
@@ -92,9 +99,10 @@ void rqueueMessageHandlerReused()
         "com.github.sonus21.rqueue.converter.DefaultMessageConverterProvider",
         true);
     FieldUtils.writeField(messageAutoConfig, "simpleRqueueListenerContainerFactory", factory, true);
+    when(messageBroker.capabilities()).thenReturn(Capabilities.REDIS_DEFAULTS);
     assertEquals(
         rqueueMessageHandler.hashCode(),
-        messageAutoConfig.rqueueMessageHandler().hashCode());
+        messageAutoConfig.rqueueMessageHandler(messageBroker).hashCode());
   }
 
   @Test

rqueue-spring/src/test/java/com/github/sonus21/rqueue/spring/tests/unit/RqueueMessageConfigTest.java

@@ -19,6 +19,7 @@
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
 import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.when;
 
 import com.github.sonus21.TestBase;
 import com.github.sonus21.rqueue.config.SimpleRqueueListenerContainerFactory;
@@ -27,6 +28,8 @@
 import com.github.sonus21.rqueue.core.DefaultRqueueMessageConverter;
 import com.github.sonus21.rqueue.core.RqueueMessageTemplate;
 import com.github.sonus21.rqueue.core.impl.UuidV4RqueueMessageIdGenerator;
+import com.github.sonus21.rqueue.core.spi.Capabilities;
+import com.github.sonus21.rqueue.core.spi.MessageBroker;
 import com.github.sonus21.rqueue.listener.RqueueMessageHandler;
 import com.github.sonus21.rqueue.spring.RqueueListenerConfig;
 import com.github.sonus21.rqueue.spring.tests.SpringUnitTest;
@@ -50,6 +53,9 @@ class RqueueMessageConfigTest extends TestBase {
   @Mock
   RqueueMessageHandler rqueueMessageHandler;
 
+  @Mock
+  private MessageBroker messageBroker;
+
   @Mock
   private SimpleRqueueListenerContainerFactory simpleRqueueListenerContainerFactory;
 
@@ -79,7 +85,8 @@ void rqueueMessageHandlerDefaultCreation() throws IllegalAccessException {
         "messageConverterProviderClass",
         "com.github.sonus21.rqueue.converter.DefaultMessageConverterProvider",
         true);
-    assertNotNull(rqueueMessageConfig.rqueueMessageHandler());
+    when(messageBroker.capabilities()).thenReturn(Capabilities.REDIS_DEFAULTS);
+    assertNotNull(rqueueMessageConfig.rqueueMessageHandler(messageBroker));
   }
 
   @Test
@@ -93,8 +100,9 @@ void rqueueMessageHandlerReused() throws IllegalAccessException {
         "com.github.sonus21.rqueue.converter.DefaultMessageConverterProvider",
         true);
     FieldUtils.writeField(messageConfig, "simpleRqueueListenerContainerFactory", factory, true);
+    when(messageBroker.capabilities()).thenReturn(Capabilities.REDIS_DEFAULTS);
     assertEquals(
-        rqueueMessageHandler.hashCode(), messageConfig.rqueueMessageHandler().hashCode());
+        rqueueMessageHandler.hashCode(), messageConfig.rqueueMessageHandler(messageBroker).hashCode());
   }
 
   @Test