Skip to content

Commit a3195fe

Browse files
committed
fix of test cases
1 parent 4ba6fe4 commit a3195fe

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

sdk/src/test/java/software/amazon/lambda/durable/DurableContextTest.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@
1111
import software.amazon.awssdk.services.lambda.model.*;
1212
import software.amazon.lambda.durable.execution.ExecutionManager;
1313
import software.amazon.lambda.durable.execution.SuspendExecutionException;
14+
import software.amazon.lambda.durable.execution.ThreadContext;
15+
import software.amazon.lambda.durable.execution.ThreadType;
1416
import software.amazon.lambda.durable.retry.RetryStrategies;
1517

1618
class DurableContextTest {
@@ -36,8 +38,11 @@ private DurableContext createTestContext(List<Operation> initialOperations) {
3638
"test-token",
3739
initialExecutionState,
3840
DurableConfig.builder().withDurableExecutionClient(client).build());
39-
return DurableContext.createRootContext(
41+
var root = DurableContext.createRootContext(
4042
executionManager, DurableConfig.builder().build(), null);
43+
executionManager.registerActiveThread(null);
44+
executionManager.setCurrentThreadContext(new ThreadContext(null, ThreadType.CONTEXT));
45+
return root;
4146
}
4247

4348
@Test

0 commit comments

Comments
 (0)