diff --git a/openmessaging/src/test/java/io/openmessaging/rocketmq/consumer/PullConsumerImplTest.java b/openmessaging/src/test/java/io/openmessaging/rocketmq/consumer/PullConsumerImplTest.java index 277a5c65f27072162787c1096e4356eff08558d5..843ddb788053397a9811286d3bacf8c10a9ab354 100644 --- a/openmessaging/src/test/java/io/openmessaging/rocketmq/consumer/PullConsumerImplTest.java +++ b/openmessaging/src/test/java/io/openmessaging/rocketmq/consumer/PullConsumerImplTest.java @@ -46,13 +46,13 @@ public class PullConsumerImplTest { @Mock private DefaultMQPullConsumer rocketmqPullConsumer; - private LocalMessageCache localMessageCache = - spy(new LocalMessageCache(rocketmqPullConsumer, new ClientConfig())); + private LocalMessageCache localMessageCache = null; @Before public void init() throws NoSuchFieldException, IllegalAccessException { final MessagingAccessPoint messagingAccessPoint = MessagingAccessPointFactory .getMessagingAccessPoint("openmessaging:rocketmq://IP1:9876,IP2:9876/namespace"); + consumer = messagingAccessPoint.createPullConsumer(queueName, OMS.newKeyValue().put(NonStandardKeys.CONSUMER_GROUP, "TestGroup")); @@ -60,6 +60,10 @@ public class PullConsumerImplTest { field.setAccessible(true); field.set(consumer, rocketmqPullConsumer); //Replace + ClientConfig clientConfig = new ClientConfig(); + clientConfig.setOmsOperationTimeout(200); + localMessageCache = spy(new LocalMessageCache(rocketmqPullConsumer, clientConfig)); + field = PullConsumerImpl.class.getDeclaredField("localMessageCache"); field.setAccessible(true); field.set(consumer, localMessageCache);