Request a push challenge and supply to SMS and voice verification.

test/unitTest/java/org/thoughtcrime/securesms/registration/PushChallengeRequestTest.java

@@ -0,0 +1,115 @@
+package org.thoughtcrime.securesms.registration;
+
+import android.app.Application;
+import android.os.AsyncTask;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.robolectric.RobolectricTestRunner;
+import org.robolectric.annotation.Config;
+import org.whispersystems.libsignal.util.guava.Optional;
+import org.whispersystems.signalservice.api.SignalServiceAccountManager;
+
+import java.io.IOException;
+
+import static org.hamcrest.Matchers.greaterThanOrEqualTo;
+import static org.hamcrest.Matchers.lessThan;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.doAnswer;
+import static org.mockito.Mockito.doThrow;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.verifyNoMoreInteractions;
+import static org.mockito.Mockito.verifyZeroInteractions;
+
+@RunWith(RobolectricTestRunner.class)
+@Config(manifest = Config.NONE, application = Application.class)
+public final class PushChallengeRequestTest {
+
+  @Test
+  public void getPushChallengeBlocking_returns_absent_if_times_out() {
+    SignalServiceAccountManager signal = mock(SignalServiceAccountManager.class);
+
+    Optional<String> challenge = PushChallengeRequest.getPushChallengeBlocking(signal, Optional.of("token"), "+123456", 50L);
+
+    assertFalse(challenge.isPresent());
+  }
+
+  @Test
+  public void getPushChallengeBlocking_waits_for_specified_period() {
+    SignalServiceAccountManager signal = mock(SignalServiceAccountManager.class);
+
+    long startTime = System.currentTimeMillis();
+    PushChallengeRequest.getPushChallengeBlocking(signal, Optional.of("token"), "+123456", 250L);
+    long duration = System.currentTimeMillis() - startTime;
+
+    assertThat(duration, greaterThanOrEqualTo(250L));
+  }
+
+  @Test
+  public void getPushChallengeBlocking_completes_fast_if_posted_to_event_bus() throws IOException {
+    SignalServiceAccountManager signal = mock(SignalServiceAccountManager.class);
+    doAnswer(invocation -> {
+      AsyncTask.execute(() -> PushChallengeRequest.postChallengeResponse("CHALLENGE"));
+      return null;
+    }).when(signal).requestPushChallenge("token", "+123456");
+
+    long startTime = System.currentTimeMillis();
+    Optional<String> challenge = PushChallengeRequest.getPushChallengeBlocking(signal, Optional.of("token"), "+123456", 500L);
+    long duration = System.currentTimeMillis() - startTime;
+
+    assertThat(duration, lessThan(500L));
+    verify(signal).requestPushChallenge("token", "+123456");
+    verifyNoMoreInteractions(signal);
+
+    assertTrue(challenge.isPresent());
+    assertEquals("CHALLENGE", challenge.get());
+  }
+
+  @Test
+  public void getPushChallengeBlocking_returns_fast_if_no_fcm_token_supplied() {
+    SignalServiceAccountManager signal = mock(SignalServiceAccountManager.class);
+
+    long startTime = System.currentTimeMillis();
+    PushChallengeRequest.getPushChallengeBlocking(signal, Optional.absent(), "+123456", 500L);
+    long duration = System.currentTimeMillis() - startTime;
+
+    assertThat(duration, lessThan(500L));
+  }
+
+  @Test
+  public void getPushChallengeBlocking_returns_absent_if_no_fcm_token_supplied() {
+    SignalServiceAccountManager signal = mock(SignalServiceAccountManager.class);
+
+    Optional<String> challenge = PushChallengeRequest.getPushChallengeBlocking(signal, Optional.absent(), "+123456", 500L);
+
+    verifyZeroInteractions(signal);
+    assertFalse(challenge.isPresent());
+  }
+
+  @Test
+  public void getPushChallengeBlocking_returns_absent_if_any_IOException_is_thrown() throws IOException {
+    SignalServiceAccountManager signal = mock(SignalServiceAccountManager.class);
+
+    doThrow(new IOException()).when(signal).requestPushChallenge(any(), any());
+
+    Optional<String> challenge = PushChallengeRequest.getPushChallengeBlocking(signal, Optional.of("token"), "+123456", 500L);
+
+    assertFalse(challenge.isPresent());
+  }
+
+  @Test
+  public void getPushChallengeBlocking_returns_absent_if_any_RuntimeException_is_thrown() throws IOException {
+    SignalServiceAccountManager signal = mock(SignalServiceAccountManager.class);
+
+    doThrow(new RuntimeException()).when(signal).requestPushChallenge(any(), any());
+
+    Optional<String> challenge = PushChallengeRequest.getPushChallengeBlocking(signal, Optional.of("token"), "+123456", 500L);
+
+    assertFalse(challenge.isPresent());
+  }
+}