refactor(rt): upstream HttpClientEngine interface changes (#608)

* refactor(rt): upstream HttpClientEngine interface changes * replace default retry middleware with AWS specific component that sets retry headers * add max attempts to retry header
smithy-lang · Feb 13, 2023 · 6ec30de · 6ec30de
1 parent 95cc8f5
commit 6ec30de
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...n-protocols/common/test/aws/smithy/kotlin/runtime/awsprotocol/json/AwsJsonProtocolTest.kt b/...n-protocols/common/test/aws/smithy/kotlin/runtime/awsprotocol/json/AwsJsonProtocolTest.kt
@@ -27,7 +27,7 @@ class AwsJsonProtocolTest {
     @Test
     fun testSetJsonProtocolHeaders() = runTest {
         val mockEngine = object : HttpClientEngineBase("test") {
-            override suspend fun roundTrip(request: HttpRequest): HttpCall {
+            override suspend fun roundTrip(context: ExecutionContext, request: HttpRequest): HttpCall {
                 val resp = HttpResponse(HttpStatusCode.OK, Headers.Empty, HttpBody.Empty)
                 val now = Instant.now()
                 return HttpCall(request, resp, now, now)
@@ -58,7 +58,7 @@ class AwsJsonProtocolTest {
     @Test
     fun testEmptyBody() = runTest {
         val mockEngine = object : HttpClientEngineBase("test") {
-            override suspend fun roundTrip(request: HttpRequest): HttpCall {
+            override suspend fun roundTrip(context: ExecutionContext, request: HttpRequest): HttpCall {
                 val resp = HttpResponse(HttpStatusCode.OK, Headers.Empty, HttpBody.Empty)
                 val now = Instant.now()
                 return HttpCall(request, resp, now, now)
@@ -86,7 +86,7 @@ class AwsJsonProtocolTest {
     @Test
     fun testDoesNotOverride() = runTest {
         val mockEngine = object : HttpClientEngineBase("test") {
-            override suspend fun roundTrip(request: HttpRequest): HttpCall {
+            override suspend fun roundTrip(context: ExecutionContext, request: HttpRequest): HttpCall {
                 val resp = HttpResponse(HttpStatusCode.OK, Headers.Empty, HttpBody.Empty)
                 val now = Instant.now()
                 return HttpCall(request, resp, now, now)