1 /* 2 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"). 5 * You may not use this file except in compliance with the License. 6 * A copy of the License is located at 7 * 8 * http://aws.amazon.com/apache2.0 9 * 10 * or in the "license" file accompanying this file. This file is distributed 11 * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either 12 * express or implied. See the License for the specific language governing 13 * permissions and limitations under the License. 14 */ 15 16 package software.amazon.awssdk.services.codecatalyst; 17 18 import static org.assertj.core.api.Assertions.assertThat; 19 20 import org.junit.Test; 21 import software.amazon.awssdk.auth.token.credentials.StaticTokenProvider; 22 import software.amazon.awssdk.regions.Region; 23 import software.amazon.awssdk.testutils.service.AwsIntegrationTestBase; 24 import software.amazon.awssdk.testutils.service.http.MockAsyncHttpClient; 25 import software.amazon.awssdk.testutils.service.http.MockSyncHttpClient; 26 27 public class BearerCredentialTest extends AwsIntegrationTestBase { 28 @Test syncClientSendsBearerToken()29 public void syncClientSendsBearerToken() { 30 try (MockSyncHttpClient httpClient = new MockSyncHttpClient(); 31 CodeCatalystClient codeCatalyst = 32 CodeCatalystClient.builder() 33 .region(Region.US_WEST_2) 34 .tokenProvider(StaticTokenProvider.create(() -> "foo-token")) 35 .httpClient(httpClient) 36 .build()) { 37 httpClient.stubNextResponse200(); 38 codeCatalyst.listSpaces(r -> {}); 39 40 assertThat(httpClient.getLastRequest().firstMatchingHeader("Authorization")) 41 .hasValue("Bearer foo-token"); 42 } 43 } 44 45 @Test asyncClientSendsBearerToken()46 public void asyncClientSendsBearerToken() { 47 try (MockAsyncHttpClient httpClient = new MockAsyncHttpClient(); 48 CodeCatalystAsyncClient codeCatalyst = 49 CodeCatalystAsyncClient.builder() 50 .region(Region.US_WEST_2) 51 .tokenProvider(StaticTokenProvider.create(() -> "foo-token")) 52 .httpClient(httpClient) 53 .build()) { 54 httpClient.stubNextResponse200(); 55 codeCatalyst.listSpaces(r -> {}).join(); 56 57 assertThat(httpClient.getLastRequest().firstMatchingHeader("Authorization")) 58 .hasValue("Bearer foo-token"); 59 } 60 } 61 } 62