104 lines
4 KiB
Java
104 lines
4 KiB
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());
|
|
}
|
|
}
|