|
2 | 2 | // SPDX-License-Identifier: Apache-2.0 |
3 | 3 | package software.amazon.lambda.durable; |
4 | 4 |
|
5 | | -import static org.junit.jupiter.api.Assertions.assertEquals; |
6 | | -import static org.junit.jupiter.api.Assertions.assertThrows; |
7 | | - |
8 | 5 | import java.time.Duration; |
9 | 6 | import java.util.concurrent.atomic.AtomicInteger; |
| 7 | + |
| 8 | +import static org.junit.jupiter.api.Assertions.assertEquals; |
| 9 | +import static org.junit.jupiter.api.Assertions.assertThrows; |
10 | 10 | import org.junit.jupiter.api.Test; |
| 11 | + |
11 | 12 | import software.amazon.awssdk.services.lambda.model.OperationStatus; |
12 | 13 | import software.amazon.awssdk.services.lambda.model.OperationType; |
13 | 14 | import software.amazon.lambda.durable.model.ExecutionStatus; |
@@ -110,12 +111,12 @@ void testFullWaitOperation() { |
110 | 111 |
|
111 | 112 | assertEquals(ExecutionStatus.SUCCEEDED, result.getStatus()); |
112 | 113 | assertEquals(3, result.getSucceededOperations().size()); |
113 | | - assertEquals("Step 1 done", result.getSucceededOperations().get(0).getStepResult(String.class)); |
| 114 | + assertEquals("Step 1 done", result.getOperation("step1").getStepResult(String.class)); |
114 | 115 | assertEquals(OperationType.WAIT, result.getSucceededOperations().get(1).getType()); |
115 | 116 | assertEquals( |
116 | 117 | OperationStatus.SUCCEEDED, |
117 | 118 | result.getSucceededOperations().get(1).getStatus()); |
118 | | - assertEquals("Step 2 done", result.getSucceededOperations().get(2).getStepResult(String.class)); |
| 119 | + assertEquals("Step 2 done", result.getOperation("step2").getStepResult(String.class)); |
119 | 120 | assertEquals("Step 1 done + Step 2 done", result.getResult(TestOutput.class).result); |
120 | 121 | } |
121 | 122 |
|
|
0 commit comments