fix(android): update test case

Author: kholood-ea

android/src/test/java/com/instabug/reactlibrary/RNInstabugSessionReplayModuleTest.java

@@ -4,6 +4,7 @@
 import static junit.framework.TestCase.assertTrue;
 
 import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.doAnswer;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.mockStatic;
 import static org.mockito.Mockito.spy;
@@ -34,6 +35,7 @@
 import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.Executors;
 import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.atomic.AtomicBoolean;
 
 
 public class RNInstabugSessionReplayModuleTest {
@@ -141,41 +143,44 @@ public void testSetSyncCallback() throws Exception  {
         MockedStatic mockArgument = mockStatic(Arguments.class);
         RNInstabugSessionReplayModule SRModule = spy(new RNInstabugSessionReplayModule(mock(ReactApplicationContext.class)));
 
-        CountDownLatch latch =new CountDownLatch(1);
-        SRModule.latch=latch;
+        AtomicBoolean result = new AtomicBoolean(false);
+        boolean shouldSync=true;
 
         when(Arguments.createMap()).thenReturn(new JavaOnlyMap());
 
         mockSessionReplay.when(() -> SessionReplay.setSyncCallback(any(SessionSyncListener.class)))
                 .thenAnswer(new Answer<Void>() {
                     @Override
                     public Void answer(InvocationOnMock invocation) {
-                        ((SessionSyncListener) invocation.getArguments()[0]).onSessionReadyToSync(new SessionMetadata("device","android","1.0",20));
+                        SessionSyncListener listener = (SessionSyncListener) invocation.getArguments()[0];
+                        SessionMetadata metadata = new SessionMetadata("device", "android", "1.0", 20);
+                         boolean shouldSync=listener.onSessionReadyToSync(metadata);
+                        result.set(shouldSync);
                         return null;
                     }
                 });
 
+        WritableMap params = Arguments.createMap();
+        params.putString("appVersion","1.0");
+        params.putString("OS","android");
+        params.putString("device","device");
+        params.putDouble("sessionDurationInSeconds",20);
+
         Thread thread= new Thread (() ->{
             try {
                 Thread.sleep(500);
             } catch (InterruptedException e) {
                 throw new RuntimeException(e);
             }
-            SRModule.evaluateSync(true);
+
+            SRModule.evaluateSync(shouldSync);
         });
+        
         thread.start();
 
         SRModule.setSyncCallback();
 
-        WritableMap params = Arguments.createMap();
-        params.putString("appVersion","1.0");
-        params.putString("OS","android");
-        params.putString("device","device");
-        params.putDouble("sessionDurationInSeconds",20);
-
-       assertEquals(SRModule.shouldSync,true);
-       assertTrue("Latch should be zero after evaluateSync is called", SRModule.latch.getCount() == 0);
-
+        assertEquals(shouldSync,result.get());
        verify(SRModule).sendEvent(Constants.IBG_SESSION_REPLAY_ON_SYNC_CALLBACK_INVOCATION, params);
        mockSessionReplay.verify(() -> SessionReplay.setSyncCallback(any(SessionSyncListener.class)));