From dede99cd9c1302613f3d2a48f8199ca22a3936a5 Mon Sep 17 00:00:00 2001 From: yaythomas Date: Wed, 24 Sep 2025 01:23:24 -0700 Subject: [PATCH] chore: rename to aws-durable-execution-sdk-python MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Renamed the external module references from aws_durable_functions_sdk_python to aws_durable_execution_sdk_python and updated the testing module name from aws_durable_functions_sdk_python_testing to aws_durable_execution_sdk_python_testing throughout the entire codebase. Also update Lambda Service API to include new durable_execution_arn arg. 1. Updated all import statements across 52 files including: - All Python source files in src/aws_durable_execution_sdk_python_testing/ - All test files in tests/ - Documentation files (README.md, CONTRIBUTING.md) 2. Fixed API compatibility issues that arose from the module rename: - Updated InMemoryServiceClient.checkpoint() method to include the new durable_execution_arn parameter - Updated InMemoryServiceClient.get_execution_state() method to include the new durable_execution_arn parameter - Updated corresponding test cases to use the new method signatures - Added appropriate # noqa: ARG002 comments for unused parameters in the in-memory implementation 3. Maintained code quality standards: - All 406 tests pass ✅ - Type checking passes ✅ - Code formatting passes ✅ - Test coverage remains above 99% (99.15%) ✅ --- .gitignore | 3 ++ CONTRIBUTING.md | 2 +- README.md | 4 +-- pyproject.toml | 30 +++++++++---------- .../__about__.py | 0 .../__init__.py | 0 .../checkpoint/__init__.py | 0 .../checkpoint/processor.py | 18 +++++------ .../checkpoint/processors/__init__.py | 0 .../checkpoint/processors/base.py | 4 +-- .../checkpoint/processors/callback.py | 6 ++-- .../checkpoint/processors/context.py | 6 ++-- .../checkpoint/processors/execution.py | 6 ++-- .../checkpoint/processors/step.py | 8 ++--- .../checkpoint/processors/wait.py | 6 ++-- .../checkpoint/transformer.py | 16 +++++----- .../checkpoint/validators/__init__.py | 0 .../checkpoint/validators/checkpoint.py | 20 ++++++------- .../validators/operations/__init__.py | 0 .../validators/operations/callback.py | 4 +-- .../validators/operations/context.py | 4 +-- .../validators/operations/execution.py | 4 +-- .../validators/operations/invoke.py | 4 +-- .../checkpoint/validators/operations/step.py | 4 +-- .../checkpoint/validators/operations/wait.py | 4 +-- .../checkpoint/validators/transitions.py | 16 +++++----- .../client.py | 13 ++++++-- .../exceptions.py | 0 .../execution.py | 10 +++---- .../executor.py | 18 +++++------ .../invoker.py | 10 +++---- .../model.py | 0 .../observer.py | 2 +- .../py.typed | 0 .../runner.py | 24 +++++++-------- .../scheduler.py | 0 .../store.py | 2 +- .../token.py | 0 tests/checkpoint/processor_test.py | 22 +++++++------- tests/checkpoint/processors/base_test.py | 4 +-- tests/checkpoint/processors/callback_test.py | 6 ++-- tests/checkpoint/processors/context_test.py | 6 ++-- .../processors/execution_processor_test.py | 6 ++-- tests/checkpoint/processors/step_test.py | 8 ++--- tests/checkpoint/processors/wait_test.py | 6 ++-- tests/checkpoint/transformer_test.py | 8 ++--- .../checkpoint/validators/checkpoint_test.py | 10 +++---- .../validators/operations/callback_test.py | 6 ++-- .../validators/operations/context_test.py | 6 ++-- .../validators/operations/execution_test.py | 6 ++-- .../validators/operations/invoke_test.py | 6 ++-- .../validators/operations/step_test.py | 6 ++-- .../validators/operations/wait_test.py | 6 ++-- .../checkpoint/validators/transitions_test.py | 6 ++-- tests/client_test.py | 19 ++++++++---- ..._executions_python_testing_library_test.py | 6 ++-- tests/e2e/basic_success_path_test.py | 8 ++--- tests/execution_test.py | 16 +++++----- tests/executor_test.py | 14 ++++----- tests/invoker_test.py | 12 ++++---- tests/model_test.py | 2 +- tests/observer_test.py | 4 +-- tests/runner_test.py | 28 ++++++++--------- tests/scheduler_test.py | 6 ++-- tests/store_test.py | 6 ++-- tests/token_test.py | 2 +- 66 files changed, 254 insertions(+), 235 deletions(-) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/__about__.py (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/__init__.py (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/__init__.py (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/processor.py (84%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/processors/__init__.py (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/processors/base.py (97%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/processors/callback.py (87%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/processors/context.py (90%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/processors/execution.py (89%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/processors/step.py (94%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/processors/wait.py (93%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/transformer.py (86%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/__init__.py (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/checkpoint.py (90%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/operations/__init__.py (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/operations/callback.py (92%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/operations/context.py (94%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/operations/execution.py (90%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/operations/invoke.py (92%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/operations/step.py (96%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/operations/wait.py (92%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/checkpoint/validators/transitions.py (78%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/client.py (72%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/exceptions.py (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/execution.py (96%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/executor.py (96%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/invoker.py (94%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/model.py (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/observer.py (97%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/py.typed (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/runner.py (95%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/scheduler.py (100%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/store.py (95%) rename src/{aws_durable_functions_sdk_python_testing => aws_durable_execution_sdk_python_testing}/token.py (100%) diff --git a/.gitignore b/.gitignore index 1d3b2d9..479f4d9 100644 --- a/.gitignore +++ b/.gitignore @@ -24,3 +24,6 @@ __pycache__/ .attach_* dist/ + +.vscode/ +.kiro/ \ No newline at end of file diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index 3a0db55..708b856 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -95,7 +95,7 @@ This will drop you into the Python debugger on the failed test. ### Writing tests Place test files in the `tests/` directory, using file names that end with `_test`. -Mimic the package structure in the src/aws_durable_functions_sdk_python directory. +Mimic the package structure in the src/aws_durable_execution_sdk_python directory. Name your module so that src/mypackage/mymodule.py has a dedicated unit test file tests/mypackage/mymodule_test.py diff --git a/README.md b/README.md index f35cc4e..e627a21 100644 --- a/README.md +++ b/README.md @@ -81,8 +81,8 @@ def function_under_test(event: Any, context: DurableContext) -> list[str]: ### Your test code ```python -from aws_durable_functions_sdk_python.execution import InvocationStatus -from aws_durable_functions_sdk_python_testing.runner import ( +from aws_durable_execution_sdk_python.execution import InvocationStatus +from aws_durable_execution_sdk_python_testing.runner import ( ContextOperation, DurableFunctionTestResult, DurableFunctionTestRunner, diff --git a/pyproject.toml b/pyproject.toml index 004202b..b03073d 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -3,9 +3,9 @@ requires = ["hatchling"] build-backend = "hatchling.build" [project] -name = "aws-durable-functions-sdk-python-testing" +name = "aws-durable-execution-sdk-python-testing" dynamic = ["version"] -description = 'This the Python SDK for AWS Lambda Durable Functions.' +description = 'This the Python SDK for AWS Lambda Durable Execution.' readme = "README.md" requires-python = ">=3.13" license = "Apache-2.0" @@ -22,25 +22,25 @@ classifiers = [ ] dependencies = [ "boto3>=1.40.30", - "aws_durable_functions_sdk_python @ git+ssh://git@github.com/aws/aws-durable-functions-sdk-python.git" + "aws_durable_execution_sdk_python @ git+ssh://git@github.com/aws/aws-durable-execution-sdk-python.git" ] [project.urls] -Documentation = "https://github.com/aws/aws-durable-functions-sdk-python-testing#readme" -Issues = "https://github.com/aws/aws-durable-functions-sdk-python-testing/issues" -Source = "https://github.com/aws/aws-durable-functions-sdk-python-testing" +Documentation = "https://github.com/aws/aws-durable-execution-sdk-python-testing#readme" +Issues = "https://github.com/aws/aws-durable-execution-sdk-python-testing/issues" +Source = "https://github.com/aws/aws-durable-execution-sdk-python-testing" [tool.hatch.build.targets.sdist] -packages = ["src/aws_durable_functions_sdk_python_testing"] +packages = ["src/aws_durable_execution_sdk_python_testing"] [tool.hatch.build.targets.wheel] -packages = ["src/aws_durable_functions_sdk_python_testing"] +packages = ["src/aws_durable_execution_sdk_python_testing"] [tool.hatch.metadata] allow-direct-references = true [tool.hatch.version] -path = "src/aws_durable_functions_sdk_python_testing/__about__.py" +path = "src/aws_durable_execution_sdk_python_testing/__about__.py" # [tool.hatch.envs.default] # dependencies=["pytest"] @@ -56,7 +56,7 @@ dependencies = [ ] [tool.hatch.envs.test.scripts] -cov="pytest --cov-report=term-missing --cov-config=pyproject.toml --cov=src/aws_durable_functions_sdk_python_testing --cov=tests --cov-fail-under=99" +cov="pytest --cov-report=term-missing --cov-config=pyproject.toml --cov=src/aws_durable_execution_sdk_python_testing --cov=tests --cov-fail-under=99" [tool.hatch.envs.types] extra-dependencies = [ @@ -64,19 +64,19 @@ extra-dependencies = [ "pytest" ] [tool.hatch.envs.types.scripts] -check = "mypy --install-types --non-interactive {args:src/aws_durable_functions_sdk_python_testing tests}" +check = "mypy --install-types --non-interactive {args:src/aws_durable_execution_sdk_python_testing tests}" [tool.coverage.run] -source_pkgs = ["aws_durable_functions_sdk_python_testing", "tests"] +source_pkgs = ["aws_durable_execution_sdk_python_testing", "tests"] branch = true parallel = true omit = [ - "src/aws_durable_functions_sdk_python_testing/__about__.py", + "src/aws_durable_execution_sdk_python_testing/__about__.py", ] [tool.coverage.paths] -aws_durable_functions_sdk_python_testing = ["src/aws_durable_functions_sdk_python_testing", "*/aws-durable-functions-sdk-python-testing/src/aws_durable_functions_sdk_python_testing"] -tests = ["tests", "*/aws-durable-functions-sdk-python-testing/tests"] +aws_durable_execution_sdk_python_testing = ["src/aws_durable_execution_sdk_python_testing", "*/aws-durable-execution-sdk-python-testing/src/aws_durable_execution_sdk_python_testing"] +tests = ["tests", "*/aws-durable-execution-sdk-python-testing/tests"] [tool.coverage.report] exclude_lines = [ diff --git a/src/aws_durable_functions_sdk_python_testing/__about__.py b/src/aws_durable_execution_sdk_python_testing/__about__.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/__about__.py rename to src/aws_durable_execution_sdk_python_testing/__about__.py diff --git a/src/aws_durable_functions_sdk_python_testing/__init__.py b/src/aws_durable_execution_sdk_python_testing/__init__.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/__init__.py rename to src/aws_durable_execution_sdk_python_testing/__init__.py diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/__init__.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/__init__.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/__init__.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/__init__.py diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/processor.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/processor.py similarity index 84% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/processor.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/processor.py index 733c6a7..f6681ee 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/processor.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/processor.py @@ -4,27 +4,27 @@ from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( CheckpointOutput, CheckpointUpdatedExecutionState, OperationUpdate, StateOutput, ) -from aws_durable_functions_sdk_python_testing.checkpoint.transformer import ( +from aws_durable_execution_sdk_python_testing.checkpoint.transformer import ( OperationTransformer, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.checkpoint import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.checkpoint import ( CheckpointValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError -from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier -from aws_durable_functions_sdk_python_testing.token import CheckpointToken +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier +from aws_durable_execution_sdk_python_testing.token import CheckpointToken if TYPE_CHECKING: - from aws_durable_functions_sdk_python_testing.execution import Execution - from aws_durable_functions_sdk_python_testing.scheduler import Scheduler - from aws_durable_functions_sdk_python_testing.store import ExecutionStore + from aws_durable_execution_sdk_python_testing.execution import Execution + from aws_durable_execution_sdk_python_testing.scheduler import Scheduler + from aws_durable_execution_sdk_python_testing.store import ExecutionStore class CheckpointProcessor: diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/__init__.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/__init__.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/processors/__init__.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/processors/__init__.py diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/base.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/base.py similarity index 97% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/processors/base.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/processors/base.py index 3ed5695..1444b91 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/base.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/base.py @@ -6,7 +6,7 @@ from datetime import timedelta from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( CallbackDetails, ContextDetails, ExecutionDetails, @@ -20,7 +20,7 @@ ) if TYPE_CHECKING: - from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier + from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class OperationProcessor: diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/callback.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/callback.py similarity index 87% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/processors/callback.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/processors/callback.py index 77c80e4..d7c949e 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/callback.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/callback.py @@ -4,19 +4,19 @@ from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.base import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.base import ( OperationProcessor, ) if TYPE_CHECKING: - from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier + from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class CallbackProcessor(OperationProcessor): diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/context.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/context.py similarity index 90% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/processors/context.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/processors/context.py index 9915121..d5c2f20 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/context.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/context.py @@ -4,19 +4,19 @@ from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.base import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.base import ( OperationProcessor, ) if TYPE_CHECKING: - from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier + from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class ContextProcessor(OperationProcessor): diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/execution.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/execution.py similarity index 89% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/processors/execution.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/processors/execution.py index 233f233..20195be 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/execution.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/execution.py @@ -4,19 +4,19 @@ from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, Operation, OperationAction, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.base import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.base import ( OperationProcessor, ) if TYPE_CHECKING: - from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier + from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class ExecutionProcessor(OperationProcessor): diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/step.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/step.py similarity index 94% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/processors/step.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/processors/step.py index e549a7e..eb57f69 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/step.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/step.py @@ -5,7 +5,7 @@ from datetime import UTC, datetime, timedelta from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, @@ -13,13 +13,13 @@ StepDetails, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.base import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.base import ( OperationProcessor, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError if TYPE_CHECKING: - from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier + from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class StepProcessor(OperationProcessor): diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/wait.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/wait.py similarity index 93% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/processors/wait.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/processors/wait.py index 5f7ab37..2075f96 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/processors/wait.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/processors/wait.py @@ -5,7 +5,7 @@ from datetime import UTC, datetime, timedelta from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, @@ -13,12 +13,12 @@ WaitDetails, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.base import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.base import ( OperationProcessor, ) if TYPE_CHECKING: - from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier + from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class WaitProcessor(OperationProcessor): diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/transformer.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/transformer.py similarity index 86% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/transformer.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/transformer.py index f53b951..9448fd5 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/transformer.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/transformer.py @@ -4,33 +4,33 @@ from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationType, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.callback import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.callback import ( CallbackProcessor, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.context import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.context import ( ContextProcessor, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.execution import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.execution import ( ExecutionProcessor, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.step import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.step import ( StepProcessor, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.wait import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.wait import ( WaitProcessor, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError if TYPE_CHECKING: from collections.abc import MutableMapping - from aws_durable_functions_sdk_python_testing.checkpoint.processors.base import ( + from aws_durable_execution_sdk_python_testing.checkpoint.processors.base import ( OperationProcessor, ) diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/__init__.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/__init__.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/__init__.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/__init__.py diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/checkpoint.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/checkpoint.py similarity index 90% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/checkpoint.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/checkpoint.py index 1aff793..7f22d60 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/checkpoint.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/checkpoint.py @@ -5,38 +5,38 @@ import json from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( OperationType, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.callback import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.callback import ( CallbackOperationValidator, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.context import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.context import ( ContextOperationValidator, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.execution import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.execution import ( ExecutionOperationValidator, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.invoke import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.invoke import ( InvokeOperationValidator, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.step import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.step import ( StepOperationValidator, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.wait import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.wait import ( WaitOperationValidator, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.transitions import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.transitions import ( ValidActionsByOperationTypeValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError if TYPE_CHECKING: from collections.abc import MutableMapping - from aws_durable_functions_sdk_python_testing.execution import Execution + from aws_durable_execution_sdk_python_testing.execution import Execution MAX_ERROR_PAYLOAD_SIZE_BYTES = 32768 diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/__init__.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/__init__.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/__init__.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/__init__.py diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/callback.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/callback.py similarity index 92% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/callback.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/callback.py index 5900ce7..4f935f2 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/callback.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/callback.py @@ -2,14 +2,14 @@ from __future__ import annotations -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError VALID_ACTIONS_FOR_CALLBACK = frozenset( [ diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/context.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/context.py similarity index 94% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/context.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/context.py index ffd6311..c81a29f 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/context.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/context.py @@ -2,14 +2,14 @@ from __future__ import annotations -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError VALID_ACTIONS_FOR_CONTEXT = frozenset( [ diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/execution.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/execution.py similarity index 90% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/execution.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/execution.py index 805a1ae..f52b4f7 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/execution.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/execution.py @@ -2,12 +2,12 @@ from __future__ import annotations -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( OperationAction, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError VALID_ACTIONS_FOR_EXECUTION = frozenset( [ diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/invoke.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/invoke.py similarity index 92% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/invoke.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/invoke.py index 2ce4c87..ed9f8f9 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/invoke.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/invoke.py @@ -2,14 +2,14 @@ from __future__ import annotations -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError VALID_ACTIONS_FOR_INVOKE = frozenset( [ diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/step.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/step.py similarity index 96% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/step.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/step.py index 03aee8d..896a1fe 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/step.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/step.py @@ -2,14 +2,14 @@ from __future__ import annotations -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError VALID_ACTIONS_FOR_STEP = frozenset( [ diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/wait.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/wait.py similarity index 92% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/wait.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/wait.py index 893e2ff..171efc8 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/operations/wait.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/operations/wait.py @@ -2,14 +2,14 @@ from __future__ import annotations -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError VALID_ACTIONS_FOR_WAIT = frozenset( [ diff --git a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/transitions.py b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/transitions.py similarity index 78% rename from src/aws_durable_functions_sdk_python_testing/checkpoint/validators/transitions.py rename to src/aws_durable_execution_sdk_python_testing/checkpoint/validators/transitions.py index 7ca724c..0c916a5 100644 --- a/src/aws_durable_functions_sdk_python_testing/checkpoint/validators/transitions.py +++ b/src/aws_durable_execution_sdk_python_testing/checkpoint/validators/transitions.py @@ -4,30 +4,30 @@ from typing import ClassVar -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( OperationAction, OperationType, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.callback import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.callback import ( VALID_ACTIONS_FOR_CALLBACK, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.context import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.context import ( VALID_ACTIONS_FOR_CONTEXT, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.execution import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.execution import ( VALID_ACTIONS_FOR_EXECUTION, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.invoke import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.invoke import ( VALID_ACTIONS_FOR_INVOKE, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.step import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.step import ( VALID_ACTIONS_FOR_STEP, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.wait import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.wait import ( VALID_ACTIONS_FOR_WAIT, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError class ValidActionsByOperationTypeValidator: diff --git a/src/aws_durable_functions_sdk_python_testing/client.py b/src/aws_durable_execution_sdk_python_testing/client.py similarity index 72% rename from src/aws_durable_functions_sdk_python_testing/client.py rename to src/aws_durable_execution_sdk_python_testing/client.py index c42a257..a68f0cc 100644 --- a/src/aws_durable_functions_sdk_python_testing/client.py +++ b/src/aws_durable_execution_sdk_python_testing/client.py @@ -2,14 +2,14 @@ import datetime -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( CheckpointOutput, DurableServiceClient, OperationUpdate, StateOutput, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processor import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processor import ( CheckpointProcessor, ) @@ -22,17 +22,24 @@ def __init__(self, checkpoint_processor: CheckpointProcessor): def checkpoint( self, + durable_execution_arn: str, # noqa: ARG002 checkpoint_token: str, updates: list[OperationUpdate], client_token: str | None, ) -> CheckpointOutput: + # durable_execution_arn is not used in in-memory testing return self._checkpoint_processor.process_checkpoint( checkpoint_token, updates, client_token ) def get_execution_state( - self, checkpoint_token: str, next_marker: str, max_items: int = 1000 + self, + durable_execution_arn: str, # noqa: ARG002 + checkpoint_token: str, + next_marker: str, + max_items: int = 1000, ) -> StateOutput: + # durable_execution_arn is not used in in-memory testing return self._checkpoint_processor.get_execution_state( checkpoint_token, next_marker, max_items ) diff --git a/src/aws_durable_functions_sdk_python_testing/exceptions.py b/src/aws_durable_execution_sdk_python_testing/exceptions.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/exceptions.py rename to src/aws_durable_execution_sdk_python_testing/exceptions.py diff --git a/src/aws_durable_functions_sdk_python_testing/execution.py b/src/aws_durable_execution_sdk_python_testing/execution.py similarity index 96% rename from src/aws_durable_functions_sdk_python_testing/execution.py rename to src/aws_durable_execution_sdk_python_testing/execution.py index 71c1ab1..359a211 100644 --- a/src/aws_durable_functions_sdk_python_testing/execution.py +++ b/src/aws_durable_execution_sdk_python_testing/execution.py @@ -6,11 +6,11 @@ from typing import TYPE_CHECKING from uuid import uuid4 -from aws_durable_functions_sdk_python.execution import ( +from aws_durable_execution_sdk_python.execution import ( DurableExecutionInvocationOutput, InvocationStatus, ) -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, ExecutionDetails, Operation, @@ -19,14 +19,14 @@ OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.exceptions import ( +from aws_durable_execution_sdk_python_testing.exceptions import ( IllegalStateError, InvalidParameterError, ) -from aws_durable_functions_sdk_python_testing.token import CheckpointToken +from aws_durable_execution_sdk_python_testing.token import CheckpointToken if TYPE_CHECKING: - from aws_durable_functions_sdk_python_testing.model import ( + from aws_durable_execution_sdk_python_testing.model import ( StartDurableExecutionInput, ) diff --git a/src/aws_durable_functions_sdk_python_testing/executor.py b/src/aws_durable_execution_sdk_python_testing/executor.py similarity index 96% rename from src/aws_durable_functions_sdk_python_testing/executor.py rename to src/aws_durable_execution_sdk_python_testing/executor.py index d7f0020..b68c4e6 100644 --- a/src/aws_durable_functions_sdk_python_testing/executor.py +++ b/src/aws_durable_execution_sdk_python_testing/executor.py @@ -5,31 +5,31 @@ import logging from typing import TYPE_CHECKING -from aws_durable_functions_sdk_python.execution import ( +from aws_durable_execution_sdk_python.execution import ( DurableExecutionInvocationInput, DurableExecutionInvocationOutput, InvocationStatus, ) -from aws_durable_functions_sdk_python.lambda_service import ErrorObject +from aws_durable_execution_sdk_python.lambda_service import ErrorObject -from aws_durable_functions_sdk_python_testing.exceptions import ( +from aws_durable_execution_sdk_python_testing.exceptions import ( IllegalStateError, InvalidParameterError, ResourceNotFoundError, ) -from aws_durable_functions_sdk_python_testing.execution import Execution -from aws_durable_functions_sdk_python_testing.model import ( +from aws_durable_execution_sdk_python_testing.execution import Execution +from aws_durable_execution_sdk_python_testing.model import ( StartDurableExecutionInput, StartDurableExecutionOutput, ) -from aws_durable_functions_sdk_python_testing.observer import ExecutionObserver +from aws_durable_execution_sdk_python_testing.observer import ExecutionObserver if TYPE_CHECKING: from collections.abc import Awaitable, Callable - from aws_durable_functions_sdk_python_testing.invoker import Invoker - from aws_durable_functions_sdk_python_testing.scheduler import Event, Scheduler - from aws_durable_functions_sdk_python_testing.store import ExecutionStore + from aws_durable_execution_sdk_python_testing.invoker import Invoker + from aws_durable_execution_sdk_python_testing.scheduler import Event, Scheduler + from aws_durable_execution_sdk_python_testing.store import ExecutionStore logger = logging.getLogger(__name__) diff --git a/src/aws_durable_functions_sdk_python_testing/invoker.py b/src/aws_durable_execution_sdk_python_testing/invoker.py similarity index 94% rename from src/aws_durable_functions_sdk_python_testing/invoker.py rename to src/aws_durable_execution_sdk_python_testing/invoker.py index 90bb59f..9c597fe 100644 --- a/src/aws_durable_functions_sdk_python_testing/invoker.py +++ b/src/aws_durable_execution_sdk_python_testing/invoker.py @@ -5,23 +5,23 @@ from typing import TYPE_CHECKING, Any, Protocol import boto3 # type: ignore -from aws_durable_functions_sdk_python.execution import ( +from aws_durable_execution_sdk_python.execution import ( DurableExecutionInvocationInput, DurableExecutionInvocationInputWithClient, DurableExecutionInvocationOutput, InitialExecutionState, ) -from aws_durable_functions_sdk_python.lambda_context import LambdaContext +from aws_durable_execution_sdk_python.lambda_context import LambdaContext -from aws_durable_functions_sdk_python_testing.exceptions import ( +from aws_durable_execution_sdk_python_testing.exceptions import ( DurableFunctionsTestError, ) if TYPE_CHECKING: from collections.abc import Callable - from aws_durable_functions_sdk_python_testing.client import InMemoryServiceClient - from aws_durable_functions_sdk_python_testing.execution import Execution + from aws_durable_execution_sdk_python_testing.client import InMemoryServiceClient + from aws_durable_execution_sdk_python_testing.execution import Execution def create_test_lambda_context() -> LambdaContext: diff --git a/src/aws_durable_functions_sdk_python_testing/model.py b/src/aws_durable_execution_sdk_python_testing/model.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/model.py rename to src/aws_durable_execution_sdk_python_testing/model.py diff --git a/src/aws_durable_functions_sdk_python_testing/observer.py b/src/aws_durable_execution_sdk_python_testing/observer.py similarity index 97% rename from src/aws_durable_functions_sdk_python_testing/observer.py rename to src/aws_durable_execution_sdk_python_testing/observer.py index ddf7b50..e8c6dbc 100644 --- a/src/aws_durable_functions_sdk_python_testing/observer.py +++ b/src/aws_durable_execution_sdk_python_testing/observer.py @@ -4,7 +4,7 @@ from abc import ABC, abstractmethod from collections.abc import Callable -from aws_durable_functions_sdk_python.lambda_service import ErrorObject +from aws_durable_execution_sdk_python.lambda_service import ErrorObject class ExecutionObserver(ABC): diff --git a/src/aws_durable_functions_sdk_python_testing/py.typed b/src/aws_durable_execution_sdk_python_testing/py.typed similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/py.typed rename to src/aws_durable_execution_sdk_python_testing/py.typed diff --git a/src/aws_durable_functions_sdk_python_testing/runner.py b/src/aws_durable_execution_sdk_python_testing/runner.py similarity index 95% rename from src/aws_durable_functions_sdk_python_testing/runner.py rename to src/aws_durable_execution_sdk_python_testing/runner.py index 2c111ff..0648848 100644 --- a/src/aws_durable_functions_sdk_python_testing/runner.py +++ b/src/aws_durable_execution_sdk_python_testing/runner.py @@ -3,37 +3,37 @@ from dataclasses import dataclass, field from typing import TYPE_CHECKING, Protocol, TypeVar, cast -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, OperationStatus, OperationSubType, OperationType, ) -from aws_durable_functions_sdk_python.lambda_service import Operation as SvcOperation +from aws_durable_execution_sdk_python.lambda_service import Operation as SvcOperation -from aws_durable_functions_sdk_python_testing.checkpoint.processor import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processor import ( CheckpointProcessor, ) -from aws_durable_functions_sdk_python_testing.client import InMemoryServiceClient -from aws_durable_functions_sdk_python_testing.exceptions import ( +from aws_durable_execution_sdk_python_testing.client import InMemoryServiceClient +from aws_durable_execution_sdk_python_testing.exceptions import ( DurableFunctionsTestError, ) -from aws_durable_functions_sdk_python_testing.executor import Executor -from aws_durable_functions_sdk_python_testing.invoker import InProcessInvoker -from aws_durable_functions_sdk_python_testing.model import ( +from aws_durable_execution_sdk_python_testing.executor import Executor +from aws_durable_execution_sdk_python_testing.invoker import InProcessInvoker +from aws_durable_execution_sdk_python_testing.model import ( StartDurableExecutionInput, StartDurableExecutionOutput, ) -from aws_durable_functions_sdk_python_testing.scheduler import Scheduler -from aws_durable_functions_sdk_python_testing.store import InMemoryExecutionStore +from aws_durable_execution_sdk_python_testing.scheduler import Scheduler +from aws_durable_execution_sdk_python_testing.store import InMemoryExecutionStore if TYPE_CHECKING: import datetime from collections.abc import Callable, MutableMapping - from aws_durable_functions_sdk_python.execution import InvocationStatus + from aws_durable_execution_sdk_python.execution import InvocationStatus - from aws_durable_functions_sdk_python_testing.execution import Execution + from aws_durable_execution_sdk_python_testing.execution import Execution @dataclass(frozen=True) diff --git a/src/aws_durable_functions_sdk_python_testing/scheduler.py b/src/aws_durable_execution_sdk_python_testing/scheduler.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/scheduler.py rename to src/aws_durable_execution_sdk_python_testing/scheduler.py diff --git a/src/aws_durable_functions_sdk_python_testing/store.py b/src/aws_durable_execution_sdk_python_testing/store.py similarity index 95% rename from src/aws_durable_functions_sdk_python_testing/store.py rename to src/aws_durable_execution_sdk_python_testing/store.py index 41daa4c..20733ad 100644 --- a/src/aws_durable_functions_sdk_python_testing/store.py +++ b/src/aws_durable_execution_sdk_python_testing/store.py @@ -5,7 +5,7 @@ from typing import TYPE_CHECKING, Protocol if TYPE_CHECKING: - from aws_durable_functions_sdk_python_testing.execution import Execution + from aws_durable_execution_sdk_python_testing.execution import Execution class ExecutionStore(Protocol): diff --git a/src/aws_durable_functions_sdk_python_testing/token.py b/src/aws_durable_execution_sdk_python_testing/token.py similarity index 100% rename from src/aws_durable_functions_sdk_python_testing/token.py rename to src/aws_durable_execution_sdk_python_testing/token.py diff --git a/tests/checkpoint/processor_test.py b/tests/checkpoint/processor_test.py index 89436c6..ce5e0d6 100644 --- a/tests/checkpoint/processor_test.py +++ b/tests/checkpoint/processor_test.py @@ -3,7 +3,7 @@ from unittest.mock import Mock, patch import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( CheckpointOutput, CheckpointUpdatedExecutionState, OperationAction, @@ -12,14 +12,14 @@ StateOutput, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processor import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processor import ( CheckpointProcessor, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError -from aws_durable_functions_sdk_python_testing.execution import Execution -from aws_durable_functions_sdk_python_testing.scheduler import Scheduler -from aws_durable_functions_sdk_python_testing.store import ExecutionStore -from aws_durable_functions_sdk_python_testing.token import CheckpointToken +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.execution import Execution +from aws_durable_execution_sdk_python_testing.scheduler import Scheduler +from aws_durable_execution_sdk_python_testing.store import ExecutionStore +from aws_durable_execution_sdk_python_testing.token import CheckpointToken def test_init(): @@ -49,7 +49,7 @@ def test_add_execution_observer(): @patch( - "aws_durable_functions_sdk_python_testing.checkpoint.processor.CheckpointValidator" + "aws_durable_execution_sdk_python_testing.checkpoint.processor.CheckpointValidator" ) def test_process_checkpoint_success(mock_validator): """Test successful checkpoint processing.""" @@ -107,7 +107,7 @@ def test_process_checkpoint_success(mock_validator): @patch( - "aws_durable_functions_sdk_python_testing.checkpoint.processor.CheckpointValidator" + "aws_durable_execution_sdk_python_testing.checkpoint.processor.CheckpointValidator" ) def test_process_checkpoint_invalid_token_complete_execution(mock_validator): """Test checkpoint processing with complete execution.""" @@ -136,7 +136,7 @@ def test_process_checkpoint_invalid_token_complete_execution(mock_validator): @patch( - "aws_durable_functions_sdk_python_testing.checkpoint.processor.CheckpointValidator" + "aws_durable_execution_sdk_python_testing.checkpoint.processor.CheckpointValidator" ) def test_process_checkpoint_invalid_token_sequence(mock_validator): """Test checkpoint processing with invalid token sequence.""" @@ -165,7 +165,7 @@ def test_process_checkpoint_invalid_token_sequence(mock_validator): @patch( - "aws_durable_functions_sdk_python_testing.checkpoint.processor.CheckpointValidator" + "aws_durable_execution_sdk_python_testing.checkpoint.processor.CheckpointValidator" ) def test_process_checkpoint_updates_execution_state(mock_validator): """Test that checkpoint processing updates execution state correctly.""" diff --git a/tests/checkpoint/processors/base_test.py b/tests/checkpoint/processors/base_test.py index 3a34889..fa3ac0a 100644 --- a/tests/checkpoint/processors/base_test.py +++ b/tests/checkpoint/processors/base_test.py @@ -5,7 +5,7 @@ from unittest.mock import Mock import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( CallbackDetails, ContextDetails, ErrorObject, @@ -22,7 +22,7 @@ WaitOptions, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.base import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.base import ( OperationProcessor, ) diff --git a/tests/checkpoint/processors/callback_test.py b/tests/checkpoint/processors/callback_test.py index 144f870..95d2961 100644 --- a/tests/checkpoint/processors/callback_test.py +++ b/tests/checkpoint/processors/callback_test.py @@ -3,7 +3,7 @@ from unittest.mock import Mock import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, @@ -11,10 +11,10 @@ OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.callback import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.callback import ( CallbackProcessor, ) -from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier +from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class MockNotifier(ExecutionNotifier): diff --git a/tests/checkpoint/processors/context_test.py b/tests/checkpoint/processors/context_test.py index e47f1f6..68e370d 100644 --- a/tests/checkpoint/processors/context_test.py +++ b/tests/checkpoint/processors/context_test.py @@ -4,7 +4,7 @@ from unittest.mock import Mock import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, Operation, OperationAction, @@ -13,10 +13,10 @@ OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.context import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.context import ( ContextProcessor, ) -from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier +from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class MockNotifier(ExecutionNotifier): diff --git a/tests/checkpoint/processors/execution_processor_test.py b/tests/checkpoint/processors/execution_processor_test.py index 91bff8a..37c91ea 100644 --- a/tests/checkpoint/processors/execution_processor_test.py +++ b/tests/checkpoint/processors/execution_processor_test.py @@ -2,17 +2,17 @@ from unittest.mock import Mock -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, OperationAction, OperationType, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.execution import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.execution import ( ExecutionProcessor, ) -from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier +from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class MockNotifier(ExecutionNotifier): diff --git a/tests/checkpoint/processors/step_test.py b/tests/checkpoint/processors/step_test.py index 8151ab5..46583ec 100644 --- a/tests/checkpoint/processors/step_test.py +++ b/tests/checkpoint/processors/step_test.py @@ -4,7 +4,7 @@ from unittest.mock import Mock import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, Operation, OperationAction, @@ -15,11 +15,11 @@ StepOptions, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.step import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.step import ( StepProcessor, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError -from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class MockNotifier(ExecutionNotifier): diff --git a/tests/checkpoint/processors/wait_test.py b/tests/checkpoint/processors/wait_test.py index 91f07ac..547ac94 100644 --- a/tests/checkpoint/processors/wait_test.py +++ b/tests/checkpoint/processors/wait_test.py @@ -4,7 +4,7 @@ from unittest.mock import Mock import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, @@ -13,10 +13,10 @@ WaitOptions, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.wait import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.wait import ( WaitProcessor, ) -from aws_durable_functions_sdk_python_testing.observer import ExecutionNotifier +from aws_durable_execution_sdk_python_testing.observer import ExecutionNotifier class MockNotifier(ExecutionNotifier): diff --git a/tests/checkpoint/transformer_test.py b/tests/checkpoint/transformer_test.py index 2ee9777..bda74b3 100644 --- a/tests/checkpoint/transformer_test.py +++ b/tests/checkpoint/transformer_test.py @@ -3,19 +3,19 @@ from unittest.mock import Mock import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( OperationAction, OperationType, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.processors.base import ( +from aws_durable_execution_sdk_python_testing.checkpoint.processors.base import ( OperationProcessor, ) -from aws_durable_functions_sdk_python_testing.checkpoint.transformer import ( +from aws_durable_execution_sdk_python_testing.checkpoint.transformer import ( OperationTransformer, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError class MockProcessor(OperationProcessor): diff --git a/tests/checkpoint/validators/checkpoint_test.py b/tests/checkpoint/validators/checkpoint_test.py index 4fafdf8..a0f1b1a 100644 --- a/tests/checkpoint/validators/checkpoint_test.py +++ b/tests/checkpoint/validators/checkpoint_test.py @@ -3,7 +3,7 @@ import json import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, Operation, OperationAction, @@ -12,13 +12,13 @@ OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.checkpoint import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.checkpoint import ( MAX_ERROR_PAYLOAD_SIZE_BYTES, CheckpointValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError -from aws_durable_functions_sdk_python_testing.execution import Execution -from aws_durable_functions_sdk_python_testing.model import StartDurableExecutionInput +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.execution import Execution +from aws_durable_execution_sdk_python_testing.model import StartDurableExecutionInput def _create_test_execution() -> Execution: diff --git a/tests/checkpoint/validators/operations/callback_test.py b/tests/checkpoint/validators/operations/callback_test.py index c2c7680..564d196 100644 --- a/tests/checkpoint/validators/operations/callback_test.py +++ b/tests/checkpoint/validators/operations/callback_test.py @@ -1,7 +1,7 @@ """Unit tests for callback operation validator.""" import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, @@ -9,10 +9,10 @@ OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.callback import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.callback import ( CallbackOperationValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError def test_validate_start_action_with_no_current_state(): diff --git a/tests/checkpoint/validators/operations/context_test.py b/tests/checkpoint/validators/operations/context_test.py index 51eb1d2..51229fb 100644 --- a/tests/checkpoint/validators/operations/context_test.py +++ b/tests/checkpoint/validators/operations/context_test.py @@ -1,7 +1,7 @@ """Tests for context operation validator.""" import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, Operation, OperationAction, @@ -10,11 +10,11 @@ OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.context import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.context import ( VALID_ACTIONS_FOR_CONTEXT, ContextOperationValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError def test_valid_actions_for_context(): diff --git a/tests/checkpoint/validators/operations/execution_test.py b/tests/checkpoint/validators/operations/execution_test.py index be23a69..0051143 100644 --- a/tests/checkpoint/validators/operations/execution_test.py +++ b/tests/checkpoint/validators/operations/execution_test.py @@ -1,17 +1,17 @@ """Unit tests for execution operation validator.""" import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, OperationAction, OperationType, OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.execution import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.execution import ( ExecutionOperationValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError def test_validate_succeed_action(): diff --git a/tests/checkpoint/validators/operations/invoke_test.py b/tests/checkpoint/validators/operations/invoke_test.py index 9d70f63..e7f1917 100644 --- a/tests/checkpoint/validators/operations/invoke_test.py +++ b/tests/checkpoint/validators/operations/invoke_test.py @@ -1,7 +1,7 @@ """Unit tests for invoke operation validator.""" import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, @@ -9,10 +9,10 @@ OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.invoke import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.invoke import ( InvokeOperationValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError def test_validate_start_action_with_no_current_state(): diff --git a/tests/checkpoint/validators/operations/step_test.py b/tests/checkpoint/validators/operations/step_test.py index b80f681..9d70d50 100644 --- a/tests/checkpoint/validators/operations/step_test.py +++ b/tests/checkpoint/validators/operations/step_test.py @@ -1,7 +1,7 @@ """Unit tests for step operation validator.""" import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, Operation, OperationAction, @@ -11,10 +11,10 @@ StepOptions, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.step import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.step import ( StepOperationValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError def test_validate_with_no_current_state(): diff --git a/tests/checkpoint/validators/operations/wait_test.py b/tests/checkpoint/validators/operations/wait_test.py index 4e9a7aa..5503a01 100644 --- a/tests/checkpoint/validators/operations/wait_test.py +++ b/tests/checkpoint/validators/operations/wait_test.py @@ -1,7 +1,7 @@ """Unit tests for wait operation validator.""" import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( Operation, OperationAction, OperationStatus, @@ -9,10 +9,10 @@ OperationUpdate, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.operations.wait import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.operations.wait import ( WaitOperationValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError def test_validate_start_action_with_no_current_state(): diff --git a/tests/checkpoint/validators/transitions_test.py b/tests/checkpoint/validators/transitions_test.py index ee87894..b8534b3 100644 --- a/tests/checkpoint/validators/transitions_test.py +++ b/tests/checkpoint/validators/transitions_test.py @@ -1,15 +1,15 @@ """Unit tests for transitions validator.""" import pytest -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( OperationAction, OperationType, ) -from aws_durable_functions_sdk_python_testing.checkpoint.validators.transitions import ( +from aws_durable_execution_sdk_python_testing.checkpoint.validators.transitions import ( ValidActionsByOperationTypeValidator, ) -from aws_durable_functions_sdk_python_testing.exceptions import InvalidParameterError +from aws_durable_execution_sdk_python_testing.exceptions import InvalidParameterError def test_validate_step_valid_actions(): diff --git a/tests/client_test.py b/tests/client_test.py index 3d713a8..13d44fa 100644 --- a/tests/client_test.py +++ b/tests/client_test.py @@ -3,7 +3,7 @@ import datetime from unittest.mock import Mock -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.lambda_service import ( CheckpointOutput, OperationAction, OperationType, @@ -11,7 +11,7 @@ StateOutput, ) -from aws_durable_functions_sdk_python_testing.client import InMemoryServiceClient +from aws_durable_execution_sdk_python_testing.client import InMemoryServiceClient def test_init(): @@ -41,7 +41,12 @@ def test_checkpoint(): ) ] - result = client.checkpoint("token", updates, "client-token") + result = client.checkpoint( + "arn:aws:lambda:us-east-1:123456789012:function:test", + "token", + updates, + "client-token", + ) assert result == expected_output processor.process_checkpoint.assert_called_once_with( @@ -57,7 +62,9 @@ def test_get_execution_state(): client = InMemoryServiceClient(processor) - result = client.get_execution_state("token", "marker", 500) + result = client.get_execution_state( + "arn:aws:lambda:us-east-1:123456789012:function:test", "token", "marker", 500 + ) assert result == expected_output processor.get_execution_state.assert_called_once_with("token", "marker", 500) @@ -71,7 +78,9 @@ def test_get_execution_state_default_max_items(): client = InMemoryServiceClient(processor) - result = client.get_execution_state("token", "marker") + result = client.get_execution_state( + "arn:aws:lambda:us-east-1:123456789012:function:test", "token", "marker" + ) assert result == expected_output processor.get_execution_state.assert_called_once_with("token", "marker", 1000) diff --git a/tests/durable_executions_python_testing_library_test.py b/tests/durable_executions_python_testing_library_test.py index 1f5c44f..940fd6f 100644 --- a/tests/durable_executions_python_testing_library_test.py +++ b/tests/durable_executions_python_testing_library_test.py @@ -1,6 +1,6 @@ """Tests for DurableExecutionsPythonTestingLibrary module.""" -def test_aws_durable_functions_sdk_python_testing_importable(): - """Test aws_durable_functions_sdk_python_testing is importable.""" - import aws_durable_functions_sdk_python_testing # noqa: F401 +def test_aws_durable_execution_sdk_python_testing_importable(): + """Test aws_durable_execution_sdk_python_testing is importable.""" + import aws_durable_execution_sdk_python_testing # noqa: F401 diff --git a/tests/e2e/basic_success_path_test.py b/tests/e2e/basic_success_path_test.py index 5272f59..faee614 100644 --- a/tests/e2e/basic_success_path_test.py +++ b/tests/e2e/basic_success_path_test.py @@ -2,15 +2,15 @@ from typing import Any -from aws_durable_functions_sdk_python.context import ( +from aws_durable_execution_sdk_python.context import ( DurableContext, durable_step, durable_with_child_context, ) -from aws_durable_functions_sdk_python.execution import InvocationStatus, durable_handler -from aws_durable_functions_sdk_python.types import StepContext +from aws_durable_execution_sdk_python.execution import InvocationStatus, durable_handler +from aws_durable_execution_sdk_python.types import StepContext -from aws_durable_functions_sdk_python_testing.runner import ( +from aws_durable_execution_sdk_python_testing.runner import ( ContextOperation, DurableFunctionTestResult, DurableFunctionTestRunner, diff --git a/tests/execution_test.py b/tests/execution_test.py index cf48066..c61c391 100644 --- a/tests/execution_test.py +++ b/tests/execution_test.py @@ -4,8 +4,8 @@ from unittest.mock import patch import pytest -from aws_durable_functions_sdk_python.execution import InvocationStatus -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.execution import InvocationStatus +from aws_durable_execution_sdk_python.lambda_service import ( ErrorObject, Operation, OperationStatus, @@ -13,9 +13,9 @@ StepDetails, ) -from aws_durable_functions_sdk_python_testing.exceptions import IllegalStateError -from aws_durable_functions_sdk_python_testing.execution import Execution -from aws_durable_functions_sdk_python_testing.model import StartDurableExecutionInput +from aws_durable_execution_sdk_python_testing.exceptions import IllegalStateError +from aws_durable_execution_sdk_python_testing.execution import Execution +from aws_durable_execution_sdk_python_testing.model import StartDurableExecutionInput def test_execution_init(): @@ -43,7 +43,7 @@ def test_execution_init(): assert execution.consecutive_failed_invocation_attempts == 0 -@patch("aws_durable_functions_sdk_python_testing.execution.uuid4") +@patch("aws_durable_execution_sdk_python_testing.execution.uuid4") def test_execution_new(mock_uuid4): """Test Execution.new static method.""" mock_uuid = "test-uuid-123" @@ -65,7 +65,7 @@ def test_execution_new(mock_uuid4): assert execution.operations == [] -@patch("aws_durable_functions_sdk_python_testing.execution.datetime") +@patch("aws_durable_execution_sdk_python_testing.execution.datetime") def test_execution_start(mock_datetime): """Test Execution.start method.""" mock_now = datetime(2023, 1, 1, 12, 0, 0, tzinfo=UTC) @@ -452,7 +452,7 @@ def test_find_operation_not_exists(): execution._find_operation("non-existent-id") # noqa: SLF001 -@patch("aws_durable_functions_sdk_python_testing.execution.datetime") +@patch("aws_durable_execution_sdk_python_testing.execution.datetime") def test_complete_wait_success(mock_datetime): """Test complete_wait method successful completion.""" mock_now = datetime(2023, 1, 1, 12, 0, 0, tzinfo=UTC) diff --git a/tests/executor_test.py b/tests/executor_test.py index 97e838a..f6ae4b7 100644 --- a/tests/executor_test.py +++ b/tests/executor_test.py @@ -4,20 +4,20 @@ from unittest.mock import Mock, patch import pytest -from aws_durable_functions_sdk_python.execution import ( +from aws_durable_execution_sdk_python.execution import ( DurableExecutionInvocationOutput, InvocationStatus, ) -from aws_durable_functions_sdk_python.lambda_service import ErrorObject +from aws_durable_execution_sdk_python.lambda_service import ErrorObject -from aws_durable_functions_sdk_python_testing.exceptions import ( +from aws_durable_execution_sdk_python_testing.exceptions import ( IllegalStateError, InvalidParameterError, ResourceNotFoundError, ) -from aws_durable_functions_sdk_python_testing.execution import Execution -from aws_durable_functions_sdk_python_testing.executor import Executor -from aws_durable_functions_sdk_python_testing.model import StartDurableExecutionInput +from aws_durable_execution_sdk_python_testing.execution import Execution +from aws_durable_execution_sdk_python_testing.executor import Executor +from aws_durable_execution_sdk_python_testing.model import StartDurableExecutionInput @pytest.fixture @@ -71,7 +71,7 @@ def test_init(mock_store, mock_scheduler, mock_invoker): assert executor._completion_events == {} # noqa: SLF001 -@patch("aws_durable_functions_sdk_python_testing.executor.Execution") +@patch("aws_durable_execution_sdk_python_testing.executor.Execution") def test_start_execution( mock_execution_class, executor, start_input, mock_store, mock_scheduler ): diff --git a/tests/invoker_test.py b/tests/invoker_test.py index a9d4517..fca8707 100644 --- a/tests/invoker_test.py +++ b/tests/invoker_test.py @@ -4,22 +4,22 @@ from unittest.mock import Mock, patch import pytest -from aws_durable_functions_sdk_python.execution import ( +from aws_durable_execution_sdk_python.execution import ( DurableExecutionInvocationInput, DurableExecutionInvocationInputWithClient, DurableExecutionInvocationOutput, InitialExecutionState, InvocationStatus, ) -from aws_durable_functions_sdk_python.lambda_context import LambdaContext +from aws_durable_execution_sdk_python.lambda_context import LambdaContext -from aws_durable_functions_sdk_python_testing.execution import Execution -from aws_durable_functions_sdk_python_testing.invoker import ( +from aws_durable_execution_sdk_python_testing.execution import Execution +from aws_durable_execution_sdk_python_testing.invoker import ( InProcessInvoker, LambdaInvoker, create_test_lambda_context, ) -from aws_durable_functions_sdk_python_testing.model import StartDurableExecutionInput +from aws_durable_execution_sdk_python_testing.model import StartDurableExecutionInput def test_create_test_lambda_context(): @@ -112,7 +112,7 @@ def test_lambda_invoker_init(): def test_lambda_invoker_create(): """Test creating LambdaInvoker with boto3 client.""" - with patch("aws_durable_functions_sdk_python_testing.invoker.boto3") as mock_boto3: + with patch("aws_durable_execution_sdk_python_testing.invoker.boto3") as mock_boto3: mock_client = Mock() mock_boto3.client.return_value = mock_client diff --git a/tests/model_test.py b/tests/model_test.py index 7255c6a..1740fdc 100644 --- a/tests/model_test.py +++ b/tests/model_test.py @@ -2,7 +2,7 @@ import pytest -from aws_durable_functions_sdk_python_testing.model import ( +from aws_durable_execution_sdk_python_testing.model import ( StartDurableExecutionInput, StartDurableExecutionOutput, ) diff --git a/tests/observer_test.py b/tests/observer_test.py index 33d5feb..ce6c372 100644 --- a/tests/observer_test.py +++ b/tests/observer_test.py @@ -4,9 +4,9 @@ from unittest.mock import Mock import pytest -from aws_durable_functions_sdk_python.lambda_service import ErrorObject +from aws_durable_execution_sdk_python.lambda_service import ErrorObject -from aws_durable_functions_sdk_python_testing.observer import ( +from aws_durable_execution_sdk_python_testing.observer import ( ExecutionNotifier, ExecutionObserver, ) diff --git a/tests/runner_test.py b/tests/runner_test.py index dea43ed..9fdcef4 100644 --- a/tests/runner_test.py +++ b/tests/runner_test.py @@ -4,8 +4,8 @@ from unittest.mock import Mock, patch import pytest -from aws_durable_functions_sdk_python.execution import InvocationStatus -from aws_durable_functions_sdk_python.lambda_service import ( +from aws_durable_execution_sdk_python.execution import InvocationStatus +from aws_durable_execution_sdk_python.lambda_service import ( CallbackDetails, ContextDetails, ExecutionDetails, @@ -15,17 +15,17 @@ StepDetails, WaitDetails, ) -from aws_durable_functions_sdk_python.lambda_service import Operation as SvcOperation +from aws_durable_execution_sdk_python.lambda_service import Operation as SvcOperation -from aws_durable_functions_sdk_python_testing.exceptions import ( +from aws_durable_execution_sdk_python_testing.exceptions import ( DurableFunctionsTestError, ) -from aws_durable_functions_sdk_python_testing.execution import Execution -from aws_durable_functions_sdk_python_testing.model import ( +from aws_durable_execution_sdk_python_testing.execution import Execution +from aws_durable_execution_sdk_python_testing.model import ( StartDurableExecutionInput, StartDurableExecutionOutput, ) -from aws_durable_functions_sdk_python_testing.runner import ( +from aws_durable_execution_sdk_python_testing.runner import ( OPERATION_FACTORIES, CallbackOperation, ContextOperation, @@ -657,12 +657,12 @@ def test_durable_function_test_result_get_execution(): assert found_exec.name == "test-execution" -@patch("aws_durable_functions_sdk_python_testing.runner.Scheduler") -@patch("aws_durable_functions_sdk_python_testing.runner.InMemoryExecutionStore") -@patch("aws_durable_functions_sdk_python_testing.runner.CheckpointProcessor") -@patch("aws_durable_functions_sdk_python_testing.runner.InMemoryServiceClient") -@patch("aws_durable_functions_sdk_python_testing.runner.InProcessInvoker") -@patch("aws_durable_functions_sdk_python_testing.runner.Executor") +@patch("aws_durable_execution_sdk_python_testing.runner.Scheduler") +@patch("aws_durable_execution_sdk_python_testing.runner.InMemoryExecutionStore") +@patch("aws_durable_execution_sdk_python_testing.runner.CheckpointProcessor") +@patch("aws_durable_execution_sdk_python_testing.runner.InMemoryServiceClient") +@patch("aws_durable_execution_sdk_python_testing.runner.InProcessInvoker") +@patch("aws_durable_execution_sdk_python_testing.runner.Executor") def test_durable_function_test_runner_init( mock_executor, mock_invoker, mock_client, mock_processor, mock_store, mock_scheduler ): @@ -700,7 +700,7 @@ def test_durable_function_test_runner_context_manager(): mock_close.assert_called_once() -@patch("aws_durable_functions_sdk_python_testing.runner.Scheduler") +@patch("aws_durable_execution_sdk_python_testing.runner.Scheduler") def test_durable_function_test_runner_close(mock_scheduler): """Test DurableFunctionTestRunner close method.""" handler = Mock() diff --git a/tests/scheduler_test.py b/tests/scheduler_test.py index d3f3b9d..65db932 100644 --- a/tests/scheduler_test.py +++ b/tests/scheduler_test.py @@ -7,7 +7,7 @@ import pytest -from aws_durable_functions_sdk_python_testing.scheduler import Event, Scheduler +from aws_durable_execution_sdk_python_testing.scheduler import Event, Scheduler def wait_for_condition(condition_func, timeout_iterations=100): @@ -213,7 +213,7 @@ def failing_func() -> None: raise ValueError(msg) with patch( - "aws_durable_functions_sdk_python_testing.scheduler.logger" + "aws_durable_execution_sdk_python_testing.scheduler.logger" ) as mock_logger: future = scheduler.call_later(failing_func, delay=0.01) wait_for_condition(lambda: future.done()) @@ -560,7 +560,7 @@ def failing_func() -> None: # Test that user function exceptions are propagated through the Future with patch( - "aws_durable_functions_sdk_python_testing.scheduler.logger" + "aws_durable_execution_sdk_python_testing.scheduler.logger" ) as mock_logger: future = scheduler.call_later(failing_func, delay=0.01) wait_for_condition(lambda: future.done()) diff --git a/tests/store_test.py b/tests/store_test.py index d9d4897..7099c4b 100644 --- a/tests/store_test.py +++ b/tests/store_test.py @@ -2,9 +2,9 @@ import pytest -from aws_durable_functions_sdk_python_testing.execution import Execution -from aws_durable_functions_sdk_python_testing.model import StartDurableExecutionInput -from aws_durable_functions_sdk_python_testing.store import InMemoryExecutionStore +from aws_durable_execution_sdk_python_testing.execution import Execution +from aws_durable_execution_sdk_python_testing.model import StartDurableExecutionInput +from aws_durable_execution_sdk_python_testing.store import InMemoryExecutionStore def test_in_memory_execution_store_save_and_load(): diff --git a/tests/token_test.py b/tests/token_test.py index 66ad713..714d8c9 100644 --- a/tests/token_test.py +++ b/tests/token_test.py @@ -5,7 +5,7 @@ import pytest -from aws_durable_functions_sdk_python_testing.token import ( +from aws_durable_execution_sdk_python_testing.token import ( CallbackToken, CheckpointToken, )