Update tests to use new local runner create overload.

Author: phipag

examples/src/test/java/com/amazonaws/lambda/durable/examples/GenericTypesExampleTest.java

@@ -12,7 +12,7 @@ class GenericTypesExampleTest {
     @Test
     void testGenericTypesExample() {
         var handler = new GenericTypesExample();
-        var runner = LocalDurableTestRunner.create(GenericTypesExample.Input.class, handler::handleRequest)
+        var runner = LocalDurableTestRunner.create(GenericTypesExample.Input.class, handler)
                 .withSkipTime(true);
 
         var input = new GenericTypesExample.Input("user123");
@@ -48,7 +48,7 @@ void testGenericTypesExample() {
     @Test
     void testOperationTracking() {
         var handler = new GenericTypesExample();
-        var runner = LocalDurableTestRunner.create(GenericTypesExample.Input.class, handler::handleRequest)
+        var runner = LocalDurableTestRunner.create(GenericTypesExample.Input.class, handler)
                 .withSkipTime(true);
 
         var input = new GenericTypesExample.Input("user456");

examples/src/test/java/com/amazonaws/lambda/durable/examples/RetryExampleTest.java

@@ -11,11 +11,10 @@ class RetryExampleTest {
 
     @Test
     void testRetryExampleWithTimeBasedFailure() {
+        var handler = new RetryExample();
+
         // Test the retry example with time-based failure simulation
-        var runner = LocalDurableTestRunner.create(String.class, (input, ctx) -> {
-            var handler = new RetryExample();
-            return handler.handleRequest(input, ctx);
-        });
+        var runner = LocalDurableTestRunner.create(Object.class, handler);
 
         var result = runner.run("test-input");
 
@@ -32,11 +31,10 @@ void testRetryExampleWithTimeBasedFailure() {
     void testRetryExampleDemonstration() {
         // This test demonstrates the retry behavior without strict assertions
         // It's useful for observing the retry mechanism in action
+        var handler = new RetryExample();
 
-        var runner = LocalDurableTestRunner.create(String.class, (input, ctx) -> {
-            var handler = new RetryExample();
-            return handler.handleRequest(input, ctx);
-        });
+        // Test the retry example with time-based failure simulation
+        var runner = LocalDurableTestRunner.create(Object.class, handler);
 
         var result = runner.run("demo-input");
 
@@ -50,10 +48,10 @@ void testRetryExampleDemonstration() {
     @Test
     void testRetryExampleShowsRetryBehavior() {
         // Test that shows the different retry behaviors
-        var runner = LocalDurableTestRunner.create(String.class, (input, ctx) -> {
-            var handler = new RetryExample();
-            return handler.handleRequest(input, ctx);
-        });
+        var handler = new RetryExample();
+
+        // Test the retry example with time-based failure simulation
+        var runner = LocalDurableTestRunner.create(Object.class, handler);
 
         var result = runner.run("retry-behavior-test");
 

examples/src/test/java/com/amazonaws/lambda/durable/examples/SimpleStepExampleTest.java

@@ -16,7 +16,7 @@ void testSimpleStepExample() {
         var handler = new SimpleStepExample();
 
         // Create test runner
-        var runner = LocalDurableTestRunner.create(GreetingRequest.class, handler::handleRequest);
+        var runner = LocalDurableTestRunner.create(GreetingRequest.class, handler);
 
         // Run with input
         var input = new GreetingRequest("Alice");
@@ -30,7 +30,7 @@ void testSimpleStepExample() {
     @Test
     void testWithDefaultName() {
         var handler = new SimpleStepExample();
-        var runner = LocalDurableTestRunner.create(GreetingRequest.class, handler::handleRequest);
+        var runner = LocalDurableTestRunner.create(GreetingRequest.class, handler);
 
         var input = new GreetingRequest();
         var result = runner.run(input);
@@ -42,7 +42,7 @@ void testWithDefaultName() {
     @Test
     void testReplay() {
         var handler = new SimpleStepExample();
-        var runner = LocalDurableTestRunner.create(GreetingRequest.class, handler::handleRequest);
+        var runner = LocalDurableTestRunner.create(GreetingRequest.class, handler);
 
         // First execution
         var input = new GreetingRequest("Bob");

examples/src/test/java/com/amazonaws/lambda/durable/examples/WaitExampleTest.java

@@ -13,7 +13,7 @@ class WaitExampleTest {
     @Test
     void testWaitExampleStartsAndWaits() {
         var handler = new WaitExample();
-        var runner = LocalDurableTestRunner.create(GreetingRequest.class, handler::handleRequest);
+        var runner = LocalDurableTestRunner.create(GreetingRequest.class, handler);
 
         var input = new GreetingRequest("Bob");
 

sdk-integration-tests/src/test/java/com/amazonaws/lambda/durable/retry/RetryIntegrationTest.java

@@ -42,7 +42,7 @@ public String handleRequest(String input, DurableContext context) {
             }
         };
 
-        var runner = LocalDurableTestRunner.create(String.class, handler::handleRequest);
+        var runner = LocalDurableTestRunner.create(String.class, handler);
         var result = runner.run("test-input");
 
         assertEquals(ExecutionStatus.PENDING, result.getStatus());
@@ -69,7 +69,7 @@ public String handleRequest(String input, DurableContext context) {
             }
         };
 
-        var runner = LocalDurableTestRunner.create(String.class, handler::handleRequest);
+        var runner = LocalDurableTestRunner.create(String.class, handler);
         var result = runner.run("test-input");
 
         assertEquals(ExecutionStatus.FAILED, result.getStatus());
@@ -96,7 +96,7 @@ public String handleRequest(String input, DurableContext context) {
             }
         };
 
-        var runner = LocalDurableTestRunner.create(String.class, handler::handleRequest);
+        var runner = LocalDurableTestRunner.create(String.class, handler);
         var result = runner.run("test-input");
 
         assertEquals(ExecutionStatus.SUCCEEDED, result.getStatus());