squash: add more assertions and tidy up tests

Author: hln33

packages/aws-durable-execution-sdk-python-testing/tests/execution_test.py

@@ -523,7 +523,7 @@ def test_complete_fail_without_error():
     execution.complete_fail(None)
 
     assert execution.is_complete is True
-    assert execution.result.status == InvocationStatus.FAILED
+    assert execution.result.status is InvocationStatus.FAILED
     assert execution.result.error is None
 
 

packages/aws-durable-execution-sdk-python-testing/tests/runner_test.py

@@ -838,6 +838,7 @@ def test_durable_function_test_runner_run(mock_store_class, mock_executor_class)
     # Verify result
     assert isinstance(result, DurableFunctionTestResult)
     assert result.status is InvocationStatus.SUCCEEDED
+    assert result.execution_status is ExecutionStatus.SUCCEEDED
 
 
 @patch("aws_durable_execution_sdk_python_testing.runner.Executor")
@@ -890,6 +891,7 @@ def test_durable_function_test_runner_run_with_custom_params(
     mock_executor.wait_until_complete.assert_called_once_with("test-arn", 1800)
 
     assert result.status is InvocationStatus.SUCCEEDED
+    assert result.execution_status is ExecutionStatus.SUCCEEDED
 
 
 @patch("aws_durable_execution_sdk_python_testing.runner.Executor")