Fix tests that did not wait for IAM to display

* Some tests try to dismiss IAMs when they are added to queue but
before they are displayed. Added an additional wait before running
dismissCurrentMessage() to fix some flakyness in these tests.
* Delay is upto 1,000ms as a buffer, normally wait is ~10x less

Author: jkasten2

OneSignalSDK/unittest/src/test/java/com/test/onesignal/InAppMessageIntegrationTests.java

@@ -366,6 +366,17 @@ public void testTimedMessageIsDisplayedOncePerSession() throws Exception {
                 .pollInterval(new Duration(10, TimeUnit.MILLISECONDS))
                 .until(() -> OneSignalPackagePrivateHelper.getInAppMessageDisplayQueue().size() == 1);
 
+        // After IAM is added to display queue we now need to wait until it is shown
+        Awaitility.await()
+                .atMost(new Duration(1_000, TimeUnit.MILLISECONDS))
+                .pollInterval(new Duration(10, TimeUnit.MILLISECONDS))
+                .untilAsserted(new ThrowingRunnable() {
+                    @Override
+                    public void run() {
+                        assertTrue(OneSignalPackagePrivateHelper.isInAppMessageShowing());
+                    }
+                });
+
         OneSignalPackagePrivateHelper.dismissCurrentMessage();
 
         // Check that the IAM is not displayed again
@@ -1331,6 +1342,17 @@ public void testInAppMessageDisplayMultipleTimes_sessionDurationTrigger() throws
         // No schedule should happen, IAM should evaluate to true
         assertEquals(1, OneSignalPackagePrivateHelper.getInAppMessageDisplayQueue().size());
 
+        // After IAM is added to display queue we now need to wait until it is shown
+        Awaitility.await()
+            .atMost(new Duration(1_000, TimeUnit.MILLISECONDS))
+            .pollInterval(new Duration(10, TimeUnit.MILLISECONDS))
+            .untilAsserted(new ThrowingRunnable() {
+                @Override
+                public void run() throws Exception {
+                    assertTrue(OneSignalPackagePrivateHelper.isInAppMessageShowing());
+                }
+            });
+
         // Dismiss IAM will make display quantity increase and last display time to change
         OneSignalPackagePrivateHelper.dismissCurrentMessage();
         // Check IAMs was removed from queue