Examples with TaskException com.khmelenko.lab.varis.task.TaskException used on opensource projects

Example 11 with TaskException

use of com.khmelenko.lab.varis.task.TaskException in project Varis-Android by dkhmelenko.

the class TestAuthPresenter method testTwoFactorAuth.

@Test
public void testTwoFactorAuth() {
    final String login = "login";
    final String password = "password";
    String auth = EncryptionUtils.generateBasicAuthorization(login, password);
    // rules for throwing a request for 2-factor auth
    Header header = new Header(GithubApiService.TWO_FACTOR_HEADER, "required");
    List<Header> headers = new ArrayList<>();
    headers.add(header);
    Response response = new Response("https://github.com", 401, "twoFactorAuth", headers, null);
    TaskError taskError = new TaskError(401, "twoFactorAuth");
    taskError.setResponse(response);
    TaskException exception = new TaskException(taskError);
    when(mGitHubRestClient.getApiService().createNewAuthorization(eq(auth), any(AuthorizationRequest.class))).thenThrow(exception);
    mAuthPresenter.login(login, password);
    verify(mTaskManager).createNewAuthorization(eq(auth), any(AuthorizationRequest.class));
    verify(mAuthView).showTwoFactorAuth();
    // rules for handling 2-factor auth continuation
    final String securityCode = "123456";
    final String gitHubToken = "gitHubToken";
    Authorization authorization = new Authorization();
    authorization.setToken(gitHubToken);
    authorization.setId(1L);
    when(mGitHubRestClient.getApiService().createNewAuthorization(eq(auth), eq(securityCode), any(AuthorizationRequest.class))).thenReturn(authorization);
    final String accessToken = "token";
    AccessTokenRequest request = new AccessTokenRequest();
    request.setGithubToken(gitHubToken);
    AccessToken token = new AccessToken();
    token.setAccessToken(accessToken);
    when(mTravisRestClient.getApiService().auth(request)).thenReturn(token);
    when(mGitHubRestClient.getApiService().deleteAuthorization(auth, String.valueOf(authorization.getId()))).thenReturn(null);
    mAuthPresenter.twoFactorAuth(securityCode);
    verify(mTaskManager).startAuth(gitHubToken);
    verify(mTaskManager).deleteAuthorization(auth, String.valueOf(authorization.getId()), securityCode);
    verify(mAuthView, times(2)).hideProgress();
    verify(mAuthView).finishView();
}