inline new execution token

Author: bchampp

src/aws_durable_execution_sdk_python_testing/executor.py

@@ -500,15 +500,13 @@ def checkpoint_execution(
             msg: str = f"Invalid checkpoint token: {checkpoint_token}"
             raise InvalidParameterValueException(msg)
 
-        # Process operation updates using the checkpoint processor
         if updates:
             checkpoint_output = self._checkpoint_processor.process_checkpoint(
                 checkpoint_token=checkpoint_token,
                 updates=updates,
                 client_token=client_token,
             )
 
-            # Convert SDK CheckpointUpdatedExecutionState to testing library version
             new_execution_state = None
             if checkpoint_output.new_execution_state:
                 new_execution_state = CheckpointUpdatedExecutionState(
@@ -521,11 +519,8 @@ def checkpoint_execution(
                 new_execution_state=new_execution_state,
             )
 
-        # Generate new checkpoint token for case with no updates
-        new_checkpoint_token = execution.get_new_checkpoint_token()
-
         return CheckpointDurableExecutionResponse(
-            checkpoint_token=new_checkpoint_token,
+            checkpoint_token=execution.get_new_checkpoint_token(),
             new_execution_state=None,
         )