Update ServerRequestTests.java

BranchMetrics · Oct 14, 2023 · c81e744 · c81e744
1 parent 7f50ed2
commit c81e744
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/Branch-SDK/src/androidTest/java/io/branch/referral/ServerRequestTests.java b/Branch-SDK/src/androidTest/java/io/branch/referral/ServerRequestTests.java
@@ -34,7 +34,7 @@ public void tearDown() throws InterruptedException {
 
     @Test
     public void testTimedOutInitSessionCallbackInvoked() {
-        setTimeouts(1, PrefHelper.CONNECT_TIMEOUT);
+        setTimeouts(1, 1);
         initSessionResumeActivity(null, new Runnable() {
             @Override
             public void run() {
@@ -49,7 +49,7 @@ public void testTimedOutLastAttributedTouchDataCallbackInvoked() {
         initSessionResumeActivity(null, new Runnable() {
             @Override
             public void run() {
-                setTimeouts(1, PrefHelper.CONNECT_TIMEOUT);
+                setTimeouts(1, 1);
 
                 final CountDownLatch lock1 = new CountDownLatch(1);
                 Branch.getInstance().getLastAttributedTouchData(new ServerRequestGetLATD.BranchLastAttributedTouchDataListener() {
@@ -76,7 +76,7 @@ public void testTimedOutGenerateShortUrlCallbackInvoked() {
         initSessionResumeActivity(null, new Runnable() {
             @Override
             public void run() {
-                setTimeouts(1, PrefHelper.CONNECT_TIMEOUT);
+                setTimeouts(1, 1);
 
                 final CountDownLatch lock3 = new CountDownLatch(1);
                 BranchUniversalObject buo = new BranchUniversalObject()
@@ -100,9 +100,13 @@ public void run() {
                     @Override
                     public void onLinkCreate(String url, BranchError error) {
                         BranchLogger.v("error is " + error);
-                        Assert.assertTrue((BranchError.ERR_BRANCH_REQ_TIMED_OUT == error.getErrorCode())
-                                || BranchError.ERR_BRANCH_NO_CONNECTIVITY == error.getErrorCode());
-                        lock3.countDown();
+                        if (error == null) {
+                            Assert.fail("Error was null");
+                        }
+                        else {
+                            Assert.assertTrue((BranchError.ERR_BRANCH_REQ_TIMED_OUT == error.getErrorCode()) || BranchError.ERR_BRANCH_NO_CONNECTIVITY == error.getErrorCode());
+                            lock3.countDown();
+                        }
                     }
                 });
                 try {