Search in sources :

Example 21 with AccessToken

use of co.cask.cdap.security.authentication.client.AccessToken in project cdap-ingest by caskdata.

the class RestStreamClientTest method testNotAuthorizedEmptyTokenTruncate.

@Test
public void testNotAuthorizedEmptyTokenTruncate() throws IOException {
    AuthenticationClient authClient = Mockito.mock(AuthenticationClient.class);
    AccessToken accessToken = Mockito.mock(AccessToken.class);
    Mockito.when(authClient.getAccessToken()).thenReturn(accessToken);
    Mockito.when(accessToken.getValue()).thenReturn(StringUtils.EMPTY);
    Mockito.when(accessToken.getTokenType()).thenReturn("Bearer");
    createClient(authClient);
    try {
        streamClient.truncate(TestUtils.AUTH_STREAM_NAME);
        Assert.fail("Expected HttpFailureException");
    } catch (HttpFailureException e) {
        Assert.assertEquals(HttpURLConnection.HTTP_UNAUTHORIZED, e.getStatusCode());
    }
}
Also used : HttpFailureException(co.cask.common.http.exception.HttpFailureException) AccessToken(co.cask.cdap.security.authentication.client.AccessToken) AuthenticationClient(co.cask.cdap.security.authentication.client.AuthenticationClient) Test(org.junit.Test)

Example 22 with AccessToken

use of co.cask.cdap.security.authentication.client.AccessToken in project cdap-ingest by caskdata.

the class RestStreamWriterTest method testSuccessAuthorizedStringWrite.

@Test
public void testSuccessAuthorizedStringWrite() throws IOException, InterruptedException, ExecutionException {
    AuthenticationClient authClient = Mockito.mock(AuthenticationClient.class);
    AccessToken accessToken = Mockito.mock(AccessToken.class);
    Mockito.when(authClient.getAccessToken()).thenReturn(accessToken);
    Mockito.when(authClient.isAuthEnabled()).thenReturn(true);
    Mockito.when(accessToken.getValue()).thenReturn(RestTest.AUTH_TOKEN);
    Mockito.when(accessToken.getTokenType()).thenReturn("Bearer");
    createClient(authClient);
    streamWriter = streamClient.createWriter(TestUtils.AUTH_STREAM_NAME + TestUtils.WRITER_TEST_STREAM_NAME_POSTFIX);
    streamWriter.write(RestTest.EXPECTED_WRITER_CONTENT, Charsets.UTF_8).get();
}
Also used : AccessToken(co.cask.cdap.security.authentication.client.AccessToken) AuthenticationClient(co.cask.cdap.security.authentication.client.AuthenticationClient) Test(org.junit.Test)

Example 23 with AccessToken

use of co.cask.cdap.security.authentication.client.AccessToken in project cdap-ingest by caskdata.

the class RestStreamClientTest method testSuccessAuthSetTTL.

@Test
public void testSuccessAuthSetTTL() throws IOException {
    AuthenticationClient authClient = Mockito.mock(AuthenticationClient.class);
    AccessToken accessToken = Mockito.mock(AccessToken.class);
    Mockito.when(authClient.getAccessToken()).thenReturn(accessToken);
    Mockito.when(accessToken.getValue()).thenReturn(AUTH_TOKEN);
    Mockito.when(accessToken.getTokenType()).thenReturn("Bearer");
    createClient(authClient);
    streamClient.setTTL(TestUtils.SUCCESS_STREAM_NAME, STREAM_TTL);
}
Also used : AccessToken(co.cask.cdap.security.authentication.client.AccessToken) AuthenticationClient(co.cask.cdap.security.authentication.client.AuthenticationClient) Test(org.junit.Test)

Example 24 with AccessToken

use of co.cask.cdap.security.authentication.client.AccessToken in project cdap-ingest by caskdata.

the class RestStreamClientTest method testNotAuthorizedEmptyTokenCreate.

@Test
public void testNotAuthorizedEmptyTokenCreate() throws IOException {
    AuthenticationClient authClient = Mockito.mock(AuthenticationClient.class);
    AccessToken accessToken = Mockito.mock(AccessToken.class);
    Mockito.when(authClient.getAccessToken()).thenReturn(accessToken);
    Mockito.when(accessToken.getValue()).thenReturn(StringUtils.EMPTY);
    Mockito.when(accessToken.getTokenType()).thenReturn("Bearer");
    createClient(authClient);
    try {
        streamClient.create(TestUtils.AUTH_STREAM_NAME);
        Assert.fail("Expected HttpFailureException");
    } catch (HttpFailureException e) {
        Assert.assertEquals(HttpURLConnection.HTTP_UNAUTHORIZED, e.getStatusCode());
    }
}
Also used : HttpFailureException(co.cask.common.http.exception.HttpFailureException) AccessToken(co.cask.cdap.security.authentication.client.AccessToken) AuthenticationClient(co.cask.cdap.security.authentication.client.AuthenticationClient) Test(org.junit.Test)

Example 25 with AccessToken

use of co.cask.cdap.security.authentication.client.AccessToken in project cdap-ingest by caskdata.

the class RestStreamClientTest method testSuccessAuthCreate.

@Test
public void testSuccessAuthCreate() throws IOException {
    AuthenticationClient authClient = Mockito.mock(AuthenticationClient.class);
    AccessToken accessToken = Mockito.mock(AccessToken.class);
    Mockito.when(authClient.getAccessToken()).thenReturn(accessToken);
    Mockito.when(accessToken.getValue()).thenReturn(AUTH_TOKEN);
    Mockito.when(accessToken.getTokenType()).thenReturn("Bearer");
    createClient(authClient);
    streamClient.create(TestUtils.SUCCESS_STREAM_NAME);
}
Also used : AccessToken(co.cask.cdap.security.authentication.client.AccessToken) AuthenticationClient(co.cask.cdap.security.authentication.client.AuthenticationClient) Test(org.junit.Test)

Aggregations

AccessToken (co.cask.cdap.security.authentication.client.AccessToken)34 Test (org.junit.Test)29 AuthenticationClient (co.cask.cdap.security.authentication.client.AuthenticationClient)16 URL (java.net.URL)15 HttpRequest (co.cask.common.http.HttpRequest)13 HttpFailureException (co.cask.common.http.exception.HttpFailureException)10 HttpResponse (co.cask.common.http.HttpResponse)6 ConnectionConfig (co.cask.cdap.client.config.ConnectionConfig)2 IOException (java.io.IOException)2 Properties (java.util.Properties)2 ExecutionException (java.util.concurrent.ExecutionException)2 DatasetProperties (co.cask.cdap.api.dataset.DatasetProperties)1 StreamEvent (co.cask.cdap.api.flow.flowlet.StreamEvent)1 ClientConfig (co.cask.cdap.client.config.ClientConfig)1 BadRequestException (co.cask.cdap.common.BadRequestException)1 StreamNotFoundException (co.cask.cdap.common.StreamNotFoundException)1 UnauthenticatedException (co.cask.cdap.common.UnauthenticatedException)1 ExploreDriver (co.cask.cdap.explore.jdbc.ExploreDriver)1 Credential (co.cask.cdap.security.authentication.client.Credential)1 BasicAuthenticationClient (co.cask.cdap.security.authentication.client.basic.BasicAuthenticationClient)1