Skip to content

Commit 6d27177

Browse files
Fix cancellation test issue
1 parent 5118ef6 commit 6d27177

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test-services/src/main/kotlin/dev/restate/sdk/testservices/CancelTestImpl.kt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@ class CancelTestImpl {
2424
}
2525

2626
override suspend fun startTest(context: ObjectContext, operation: BlockingOperation) {
27-
val client = CancelTestBlockingServiceClient.fromContext(context, "")
27+
val client = CancelTestBlockingServiceClient.fromContext(context, context.key())
2828

2929
try {
3030
client.block(operation).await()
@@ -44,8 +44,8 @@ class CancelTestImpl {
4444

4545
class BlockingService : CancelTest.BlockingService {
4646
override suspend fun block(context: ObjectContext, operation: BlockingOperation) {
47-
val self = CancelTestBlockingServiceClient.fromContext(context, "")
48-
val client = AwakeableHolderClient.fromContext(context, "cancel")
47+
val self = CancelTestBlockingServiceClient.fromContext(context, context.key())
48+
val client = AwakeableHolderClient.fromContext(context, context.key())
4949

5050
val awakeable = context.awakeable<String>()
5151
client.hold(awakeable.id).await()

0 commit comments

Comments
 (0)