Package software.amazon.lambda.durable.testing.local

Class LocalMemoryExecutionClient

java.lang.Object

software.amazon.lambda.durable.testing.local.LocalMemoryExecutionClient

All Implemented Interfaces:

software.amazon.lambda.durable.client.DurableExecutionClient

public class LocalMemoryExecutionClient
extends Object
implements software.amazon.lambda.durable.client.DurableExecutionClient

In-memory implementation of DurableExecutionClient for local testing. Stores operations and checkpoint state in memory, simulating the durable execution backend without AWS infrastructure.

Constructor Summary

Constructors

Constructor	Description
LocalMemoryExecutionClient()

Method Summary

Modifier and Type	Method	Description
boolean	advanceTime()	Advance all operations (simulates time passing for retries/waits).
software.amazon.awssdk.services.lambda.model.CheckpointDurableExecutionResponse	checkpoint(String arn, String token, List<software.amazon.awssdk.services.lambda.model.OperationUpdate> updates)
void	completeCallback(String callbackId, OperationResult result)	Simulate external system completing callback.
void	completeChainedInvoke(String name, OperationResult result)	Completes a chained invoke operation with the given result, simulating a child Lambda response.
List<software.amazon.awssdk.services.lambda.model.Operation>	getAllOperations()	Returns all operations currently stored.
String	getCallbackId(String operationName)	Get callback ID for a named callback operation.
software.amazon.awssdk.services.lambda.model.GetDurableExecutionStateResponse	getExecutionState(String arn, String checkpointToken, String marker)
software.amazon.awssdk.services.lambda.model.Operation	getOperationByName(String name)	Returns the operation with the given name, or null if not found.
List<software.amazon.awssdk.services.lambda.model.OperationUpdate>	getOperationUpdates()	Get all operation updates that have been sent to this client.
void	resetCheckpointToStarted(String stepName)	Simulate checkpoint failure by forcing an operation into STARTED state
void	simulateFireAndForgetCheckpointLoss(String stepName)	Simulate fire-and-forget checkpoint loss by removing the operation entirely
<O> TestResult<O>	toTestResult(software.amazon.lambda.durable.model.DurableExecutionOutput output, software.amazon.lambda.durable.TypeToken<O> resultType, software.amazon.lambda.durable.serde.SerDes serDes)	Build TestResult from current state.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

LocalMemoryExecutionClient

public LocalMemoryExecutionClient()

Method Details

checkpoint

public software.amazon.awssdk.services.lambda.model.CheckpointDurableExecutionResponse checkpoint(String arn,
 String token,
 List<software.amazon.awssdk.services.lambda.model.OperationUpdate> updates)

Specified by:

checkpoint in interface software.amazon.lambda.durable.client.DurableExecutionClient

getExecutionState

public software.amazon.awssdk.services.lambda.model.GetDurableExecutionStateResponse getExecutionState(String arn,
 String checkpointToken,
 String marker)

Specified by:

getExecutionState in interface software.amazon.lambda.durable.client.DurableExecutionClient

getOperationUpdates

public List<software.amazon.awssdk.services.lambda.model.OperationUpdate> getOperationUpdates()

Get all operation updates that have been sent to this client. Useful for testing and verification.

advanceTime

public boolean advanceTime()

Advance all operations (simulates time passing for retries/waits).

Returns:

true if any operations were advanced, false otherwise

completeChainedInvoke

public void completeChainedInvoke(String name,
 OperationResult result)

Completes a chained invoke operation with the given result, simulating a child Lambda response.

getOperationByName

public software.amazon.awssdk.services.lambda.model.Operation getOperationByName(String name)

Returns the operation with the given name, or null if not found.

getAllOperations

public List<software.amazon.awssdk.services.lambda.model.Operation> getAllOperations()

Returns all operations currently stored.

toTestResult

public <O> TestResult<O> toTestResult(software.amazon.lambda.durable.model.DurableExecutionOutput output,
 software.amazon.lambda.durable.TypeToken<O> resultType,
 software.amazon.lambda.durable.serde.SerDes serDes)

Build TestResult from current state.

resetCheckpointToStarted

public void resetCheckpointToStarted(String stepName)

Simulate checkpoint failure by forcing an operation into STARTED state

simulateFireAndForgetCheckpointLoss

public void simulateFireAndForgetCheckpointLoss(String stepName)

Simulate fire-and-forget checkpoint loss by removing the operation entirely

getCallbackId

public String getCallbackId(String operationName)

Get callback ID for a named callback operation.

completeCallback

public void completeCallback(String callbackId,
 OperationResult result)

Simulate external system completing callback.