add more unit tests

Author: zhongkechen

sdk/src/main/java/com/amazonaws/lambda/durable/exception/StepInterruptedException.java

@@ -12,6 +12,9 @@ public StepInterruptedException(Operation operation) {
     }
 
     public static boolean isStepInterruptedException(ErrorObject errorObject) {
+        if (errorObject == null) {
+            return false;
+        }
         return StepInterruptedException.toErrorObject().errorType().equals(errorObject.errorType());
     }
 

sdk/src/test/java/com/amazonaws/lambda/durable/exception/CallbackExceptionTest.java

@@ -0,0 +1,44 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+package com.amazonaws.lambda.durable.exception;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+
+import org.junit.jupiter.api.Test;
+import software.amazon.awssdk.services.lambda.model.ErrorObject;
+import software.amazon.awssdk.services.lambda.model.Operation;
+
+class CallbackExceptionTest {
+    String CALLBACK_ID = "123";
+    Operation OPERATION = Operation.builder().build();
+    ErrorObject ERROR_OBJECT = ErrorObject.builder()
+            .errorType("MyErrorType")
+            .errorMessage("MyErrorMessage")
+            .build();
+
+    @Test
+    void testCallbackTimeoutException() {
+        var exception = new CallbackTimeoutException(CALLBACK_ID, OPERATION);
+        assertEquals(CALLBACK_ID, exception.getCallbackId());
+        assertEquals(OPERATION, exception.getOperation());
+        assertEquals("Callback timed out: " + CALLBACK_ID, exception.getMessage());
+    }
+
+    @Test
+    void testCallbackFailedException() {
+        var exception = new CallbackFailedException(CALLBACK_ID, OPERATION, ERROR_OBJECT);
+        assertEquals(CALLBACK_ID, exception.getCallbackId());
+        assertEquals(OPERATION, exception.getOperation());
+        assertEquals(ERROR_OBJECT, exception.getErrorObject());
+        assertEquals("MyErrorType: MyErrorMessage", exception.getMessage());
+    }
+
+    @Test
+    void testCallbackException() {
+        var exception = new CallbackException(CALLBACK_ID, OPERATION, ERROR_OBJECT, "myErrorMessage");
+        assertEquals(CALLBACK_ID, exception.getCallbackId());
+        assertEquals(OPERATION, exception.getOperation());
+        assertEquals(ERROR_OBJECT, exception.getErrorObject());
+        assertEquals("myErrorMessage", exception.getMessage());
+    }
+}

sdk/src/test/java/com/amazonaws/lambda/durable/exception/StepFailedExceptionTest.java

@@ -18,13 +18,17 @@ class StepFailedExceptionTest {
     @Test
     void testConstructorWithNullErrorObject() {
         var exception = new StepFailedException(OPERATION, null);
+        assertEquals(OPERATION, exception.getOperation());
+        assertNull(exception.getErrorObject());
         assertEquals("Step failed with null error", exception.getMessage());
     }
 
     @Test
     void testConstructorWithErrorObject() {
         var exception = new StepFailedException(OPERATION, ERROR_OBJECT);
 
+        assertEquals(OPERATION, exception.getOperation());
+        assertEquals(ERROR_OBJECT, exception.getErrorObject());
         assertEquals("Step failed with error of type MyErrorType. Message: MyErrorMessage", exception.getMessage());
     }
 }

sdk/src/test/java/com/amazonaws/lambda/durable/exception/StepInterruptedExceptionTest.java

@@ -5,16 +5,19 @@
 import static org.junit.jupiter.api.Assertions.*;
 
 import org.junit.jupiter.api.Test;
+import software.amazon.awssdk.services.lambda.model.ErrorObject;
 import software.amazon.awssdk.services.lambda.model.Operation;
 
 class StepInterruptedExceptionTest {
+    private static final String ERROR_TYPE = "com.amazonaws.lambda.durable.exception.StepInterruptedException";
     private static final Operation OPERATION =
             Operation.builder().id("op-123").name("my-step").build();
 
     @Test
     void testConstructorWithOperationIdAndStepName() {
         var exception = new StepInterruptedException(OPERATION);
 
+        assertEquals(ERROR_TYPE, exception.getErrorObject().errorType());
         assertInstanceOf(RuntimeException.class, exception);
         assertInstanceOf(DurableExecutionException.class, exception);
         assertEquals("op-123", exception.getOperation().id());
@@ -23,4 +26,16 @@ void testConstructorWithOperationIdAndStepName() {
         assertTrue(exception.getMessage().contains("Step Name: my-step"));
         assertTrue(exception.getMessage().contains("initiated but failed to reach completion due to an interruption"));
     }
+
+    @Test
+    void testIsStepInterruptedException() {
+        var errorObject = ErrorObject.builder().errorType(ERROR_TYPE).build();
+        assertTrue(StepInterruptedException.isStepInterruptedException(errorObject));
+
+        assertFalse(StepInterruptedException.isStepInterruptedException(null));
+        assertFalse(StepInterruptedException.isStepInterruptedException(
+                ErrorObject.builder().build()));
+        assertFalse(StepInterruptedException.isStepInterruptedException(
+                ErrorObject.builder().errorType("StepInterruptedException").build()));
+    }
 }