Signal-Android/test/unitTest/java/org/thoughtcrime/securesms/registration/PushChallengeRequestTest.java

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());
  }
}