Skip to content

Commit 22486da

Browse files
committed
Add two proxy methods in code-gen template
1 parent cf0f7cb commit 22486da

1 file changed

Lines changed: 15 additions & 1 deletion

File tree

python/rpdk/java/templates/init/guided_aws/AbstractTestBase.java

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,10 @@
44
import java.util.function.Function;
55
import software.amazon.awssdk.awscore.AwsRequest;
66
import software.amazon.awssdk.awscore.AwsResponse;
7-
import software.amazon.awssdk.core.SdkClient;
7+
import software.amazon.awssdk.core.ResponseBytes;
8+
import software.amazon.awssdk.core.ResponseInputStream;
89
import software.amazon.awssdk.core.pagination.sync.SdkIterable;
10+
import software.amazon.awssdk.services.cloudformation.CloudFormationClient;
911
import software.amazon.cloudformation.proxy.AmazonWebServicesClientProxy;
1012
import software.amazon.cloudformation.proxy.Credentials;
1113
import software.amazon.cloudformation.proxy.LoggerProxy;
@@ -43,6 +45,18 @@ static ProxyClient<SdkClient> MOCK_PROXY(
4345
return proxy.injectCredentialsAndInvokeIterableV2(request, requestFunction);
4446
}
4547

48+
@Override
49+
public <RequestT extends AwsRequest, ResponseT extends AwsResponse> ResponseInputStream<ResponseT>
50+
injectCredentialsAndInvokeV2InputStream(RequestT requestT, Function<RequestT, ResponseInputStream<ResponseT>> function) {
51+
throw new UnsupportedOperationException();
52+
}
53+
54+
@Override
55+
public <RequestT extends AwsRequest, ResponseT extends AwsResponse> ResponseBytes<ResponseT>
56+
injectCredentialsAndInvokeV2Bytes(RequestT requestT, Function<RequestT, ResponseBytes<ResponseT>> function) {
57+
throw new UnsupportedOperationException();
58+
}
59+
4660
@Override
4761
public SdkClient client() {
4862
return sdkClient;

0 commit comments

Comments
 (0)