Skip to content

Commit ccafd89

Browse files
Fix usage of openapi generated client
1 parent 9fff4d4 commit ccafd89

File tree

2 files changed

+12
-3
lines changed

2 files changed

+12
-3
lines changed

src/main/kotlin/dev/restate/sdktesting/tests/PauseResumeTest.kt

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ package dev.restate.sdktesting.tests
1010

1111
import dev.restate.admin.api.InvocationApi
1212
import dev.restate.admin.client.ApiClient
13+
import dev.restate.admin.model.RestartAsNewInvocationDeploymentParameter
1314
import dev.restate.client.Client
1415
import dev.restate.client.kotlin.attachSuspend
1516
import dev.restate.sdk.annotation.Handler
@@ -90,7 +91,10 @@ class PauseResumeTest {
9091
// Resume the paused invocation on the specific endpoint
9192
val adminClient = ApiClient().setHost(adminURI.host).setPort(adminURI.port)
9293
val invocationApi = InvocationApi(adminClient)
93-
retryOnServiceUnavailable { invocationApi.resumeInvocation(invocationId, null) }
94+
retryOnServiceUnavailable {
95+
invocationApi.resumeInvocation(
96+
invocationId, RestartAsNewInvocationDeploymentParameter("keep"))
97+
}
9498

9599
assertThat(sendResult.attachSuspend().response()).isEqualTo("input")
96100

src/main/kotlin/dev/restate/sdktesting/tests/RestartAsNewInvocationTest.kt

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@ package dev.restate.sdktesting.tests
1010

1111
import dev.restate.admin.api.InvocationApi
1212
import dev.restate.admin.client.ApiClient
13+
import dev.restate.admin.model.RestartAsNewInvocationDeploymentParameter
1314
import dev.restate.client.Client
1415
import dev.restate.client.IngressException
1516
import dev.restate.client.kotlin.*
@@ -100,7 +101,10 @@ class RestartAsNewInvocationTest {
100101
val invocationApi = InvocationApi(adminClient)
101102
val newInvocationId =
102103
retryOnServiceUnavailable {
103-
invocationApi.restartAsNewInvocation(sendResult.invocationId(), null, null)
104+
invocationApi.restartAsNewInvocation(
105+
sendResult.invocationId(),
106+
null,
107+
RestartAsNewInvocationDeploymentParameter("latest"))
104108
}
105109
.newInvocationId
106110

@@ -158,7 +162,8 @@ class RestartAsNewInvocationTest {
158162
val invocationApi = InvocationApi(adminClient)
159163
val newInvocationId =
160164
retryOnServiceUnavailable {
161-
invocationApi.restartAsNewInvocation(sendResult.invocationId(), 1, null)
165+
invocationApi.restartAsNewInvocation(
166+
sendResult.invocationId(), 1, RestartAsNewInvocationDeploymentParameter("latest"))
162167
}
163168
.newInvocationId
164169

0 commit comments

Comments
 (0)