[ROCKETMQ-263]Sets a small timeout value for unit tests in OpenMessag…

…ing module
andy4ja · Aug 11, 2017 · 38bbb14 · 38bbb14
1 parent aa1c757
commit 38bbb14
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/openmessaging/src/test/java/io/openmessaging/rocketmq/consumer/PullConsumerImplTest.java b/openmessaging/src/test/java/io/openmessaging/rocketmq/consumer/PullConsumerImplTest.java
@@ -46,20 +46,24 @@ 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"));
 
         Field field = PullConsumerImpl.class.getDeclaredField("rocketmqPullConsumer");
         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);