Fixing test method

Author: “Akshay

iterableapi/src/test/java/com/iterable/iterableapi/IterableInboxTest.java

@@ -15,6 +15,8 @@
 
 import java.io.IOException;
 import java.util.List;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
 
 import okhttp3.mockwebserver.MockResponse;
 import okhttp3.mockwebserver.MockWebServer;
@@ -83,28 +85,32 @@ public void testInboxMessageOrdering() throws Exception {
     }
 
     @Test
-    public void testRemoveMessage() throws Exception {
+    public void testRemoveMessageSuccessCallbackOnSuccessfulResponse() throws Exception {
+        final CountDownLatch signal = new CountDownLatch(1);
         dispatcher.enqueueResponse("/inApp/getMessages", new MockResponse().setBody(IterableTestUtils.getResourceString("inapp_payload_inbox_multiple.json")));
         final IterableInAppManager inAppManager = IterableApi.getInstance().getInAppManager();
         inAppManager.syncInApp();
         shadowOf(getMainLooper()).idle();
         List<IterableInAppMessage> inboxMessages = inAppManager.getInboxMessages();
         assertEquals(2, inboxMessages.size());
         assertEquals(1, inAppManager.getUnreadInboxMessagesCount());
+
+        final JSONObject responseData = new JSONObject("{\"key\":\"value\"}");
+        dispatcher.enqueueResponse("/events/inAppConsume", new MockResponse().setResponseCode(200).setBody(responseData.toString()));
+
         inAppManager.removeMessage(inboxMessages.get(0), new IterableHelper.SuccessHandler() {
             @Override
             public void onSuccess(@NonNull JSONObject data) {
-                assertEquals(1, inAppManager.getInboxMessages().size());
-                assertEquals("message2", inAppManager.getInboxMessages().get(0).getMessageId());
+                signal.countDown();
             }
         }, new IterableHelper.FailureHandler() {
             @Override
             public void onFailure(@NonNull String reason, @Nullable JSONObject data) {
                 assertFalse(true);
             }
         });
-        assertEquals(1, inAppManager.getInboxMessages().size());
-        assertEquals("message2", inAppManager.getInboxMessages().get(0).getMessageId());
+        shadowOf(getMainLooper()).idle();
+        assertTrue("Message remove success callback called", signal.await(1, TimeUnit.SECONDS));
     }
 
     @Test