aws-durable-execution-sdk-python-testing 1.1.0__tar.gz → 1.1.1__tar.gz

{aws_durable_execution_sdk_python_testing-1.1.0 → aws_durable_execution_sdk_python_testing-1.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aws-durable-execution-sdk-python-testing
-Version: 1.1.0
+Version: 1.1.1
 Summary: AWS Durable Execution Testing SDK for Python
 Project-URL: Documentation, https://github.com/aws/aws-durable-execution-sdk-python-testing#readme
 Project-URL: Issues, https://github.com/aws/aws-durable-execution-sdk-python-testing/issues

{aws_durable_execution_sdk_python_testing-1.1.0 → aws_durable_execution_sdk_python_testing-1.1.1}/aws_durable_execution_sdk_python_testing/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2025-present Amazon.com, Inc. or its affiliates.
 #
 # SPDX-License-Identifier: Apache-2.0
-__version__ = "1.1.0"
+__version__ = "1.1.1"

{aws_durable_execution_sdk_python_testing-1.1.0 → aws_durable_execution_sdk_python_testing-1.1.1}/aws_durable_execution_sdk_python_testing/execution.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-import json
 from dataclasses import replace
 from datetime import UTC, datetime
 from enum import Enum
@@ -33,7 +32,6 @@ from aws_durable_execution_sdk_python_testing.model import (
 )
 from aws_durable_execution_sdk_python_testing.token import (
     CheckpointToken,
-    CallbackToken,
 )
 
 
@@ -96,12 +94,12 @@ class Execution:
             durable_execution_arn=str(uuid4()), start_input=input, operations=[]
         )
 
-    def to_dict(self) -> dict[str, Any]:
-        """Serialize execution to dictionary."""
+    def to_json_dict(self) -> dict[str, Any]:
+        """Serialize execution to JSON-serializable dictionary"""
         return {
             "DurableExecutionArn": self.durable_execution_arn,
             "StartInput": self.start_input.to_dict(),
-            "Operations": [op.to_dict() for op in self.operations],
+            "Operations": [op.to_json_dict() for op in self.operations],
             "Updates": [update.to_dict() for update in self.updates],
             "InvocationCompletions": [
                 completion.to_dict() for completion in self.invocation_completions
@@ -115,13 +113,15 @@ class Execution:
         }
 
     @classmethod
-    def from_dict(cls, data: dict[str, Any]) -> Execution:
+    def from_json_dict(cls, data: dict[str, Any]) -> Execution:
         """Deserialize execution from dictionary."""
         # Reconstruct start_input
         start_input = StartDurableExecutionInput.from_dict(data["StartInput"])
 
         # Reconstruct operations
-        operations = [Operation.from_dict(op_data) for op_data in data["Operations"]]
+        operations = [
+            Operation.from_json_dict(op_data) for op_data in data["Operations"]
+        ]
 
         # Create execution
         execution = cls(

{aws_durable_execution_sdk_python_testing-1.1.0 → aws_durable_execution_sdk_python_testing-1.1.1}/aws_durable_execution_sdk_python_testing/invoker.py

@@ -12,12 +12,10 @@ from aws_durable_execution_sdk_python.execution import (
     DurableExecutionInvocationInputWithClient,
     DurableExecutionInvocationOutput,
     InitialExecutionState,
-    InvocationStatus,
 )
 
 from aws_durable_execution_sdk_python_testing.exceptions import (
     DurableFunctionsTestError,
-    ServiceException,
 )
 from aws_durable_execution_sdk_python_testing.model import LambdaContext
 
@@ -239,7 +237,7 @@ class LambdaInvoker(Invoker):
             response = client.invoke(
                 FunctionName=function_name,
                 InvocationType="RequestResponse",  # Synchronous invocation
-                Payload=json.dumps(input.to_dict(), default=str),
+                Payload=json.dumps(input.to_json_dict()),
             )
 
             # Check HTTP status code

{aws_durable_execution_sdk_python_testing-1.1.0 → aws_durable_execution_sdk_python_testing-1.1.1}/aws_durable_execution_sdk_python_testing/stores/filesystem.py

@@ -4,7 +4,6 @@ from __future__ import annotations
 
 import json
 import logging
-from datetime import UTC, datetime
 from pathlib import Path
 
 from aws_durable_execution_sdk_python_testing.exceptions import (
@@ -16,30 +15,6 @@ from aws_durable_execution_sdk_python_testing.stores.base import (
 )
 
 
-class DateTimeEncoder(json.JSONEncoder):
-    """Custom JSON encoder that handles datetime objects."""
-
-    def default(self, obj):
-        if isinstance(obj, datetime):
-            return obj.timestamp()
-        return super().default(obj)
-
-
-def datetime_object_hook(obj):
-    """JSON object hook to convert unix timestamps back to datetime objects."""
-    if isinstance(obj, dict):
-        for key, value in obj.items():
-            if isinstance(value, int | float) and key.endswith(
-                ("_timestamp", "_time", "Timestamp", "Time")
-            ):
-                try:  # noqa: SIM105
-                    obj[key] = datetime.fromtimestamp(value, tz=UTC)
-                except (ValueError, OSError):
-                    # Leave as number if not a valid timestamp
-                    pass
-    return obj
-
-
 class FileSystemExecutionStore(BaseExecutionStore):
     """File system-based execution store for persistence."""
 
@@ -69,10 +44,10 @@ class FileSystemExecutionStore(BaseExecutionStore):
     def save(self, execution: Execution) -> None:
         """Save execution to file system."""
         file_path = self._get_file_path(execution.durable_execution_arn)
-        data = execution.to_dict()
+        data = execution.to_json_dict()
 
         with open(file_path, "w", encoding="utf-8") as f:
-            json.dump(data, f, indent=2, cls=DateTimeEncoder)
+            json.dump(data, f, indent=2)
 
     def load(self, execution_arn: str) -> Execution:
         """Load execution from file system."""
@@ -82,9 +57,9 @@ class FileSystemExecutionStore(BaseExecutionStore):
             raise ResourceNotFoundException(msg)
 
         with open(file_path, encoding="utf-8") as f:
-            data = json.load(f, object_hook=datetime_object_hook)
+            data = json.load(f)
 
-        return Execution.from_dict(data)
+        return Execution.from_json_dict(data)
 
     def update(self, execution: Execution) -> None:
         """Update execution in file system (same as save)."""
@@ -96,8 +71,8 @@ class FileSystemExecutionStore(BaseExecutionStore):
         for file_path in self._storage_dir.glob("*.json"):
             try:
                 with open(file_path, encoding="utf-8") as f:
-                    data = json.load(f, object_hook=datetime_object_hook)
-                executions.append(Execution.from_dict(data))
+                    data = json.load(f)
+                executions.append(Execution.from_json_dict(data))
             except (json.JSONDecodeError, KeyError, OSError) as e:
                 logging.warning("Skipping corrupted file %s: %s", file_path, e)
                 continue

{aws_durable_execution_sdk_python_testing-1.1.0 → aws_durable_execution_sdk_python_testing-1.1.1}/aws_durable_execution_sdk_python_testing/stores/sqlite.py

@@ -17,10 +17,6 @@ from aws_durable_execution_sdk_python_testing.execution import Execution
 from aws_durable_execution_sdk_python_testing.stores.base import (
     ExecutionStore,
 )
-from aws_durable_execution_sdk_python_testing.stores.filesystem import (
-    DateTimeEncoder,
-    datetime_object_hook,
-)
 
 
 class SQLiteExecutionStore(ExecutionStore):
@@ -102,7 +98,7 @@ class SQLiteExecutionStore(ExecutionStore):
                         execution_op.end_timestamp.timestamp()
                         if execution_op.end_timestamp
                         else None,
-                        json.dumps(execution.to_dict(), cls=DateTimeEncoder),
+                        json.dumps(execution.to_json_dict()),
                     ),
                 )
         except sqlite3.Error as e:
@@ -125,9 +121,7 @@ class SQLiteExecutionStore(ExecutionStore):
             if not row:
                 raise ResourceNotFoundException(f"Execution {execution_arn} not found")
 
-            return Execution.from_dict(
-                json.loads(row[0], object_hook=datetime_object_hook)
-            )
+            return Execution.from_json_dict(json.loads(row[0]))
         except sqlite3.Error as e:
             raise RuntimeError(f"Failed to load execution {execution_arn}: {e}") from e
         except json.JSONDecodeError as e:
@@ -222,11 +216,7 @@ class SQLiteExecutionStore(ExecutionStore):
             executions: list[Execution] = []
             for durable_execution_arn, data in rows:
                 try:
-                    executions.append(
-                        Execution.from_dict(
-                            json.loads(data, object_hook=datetime_object_hook)
-                        )
-                    )
+                    executions.append(Execution.from_json_dict(json.loads(data)))
                 except (json.JSONDecodeError, ValueError) as e:
                     # Log corrupted data but continue with other records
                     print(