66
77import pytest
88
9+ from aws_durable_execution_sdk_python .__about__ import __version__
910from aws_durable_execution_sdk_python .exceptions import (
1011 CallableRuntimeError ,
1112 CheckpointError ,
@@ -1940,6 +1941,7 @@ def test_lambda_client_initialize_client_default(
19401941 config = call_args [1 ]["config" ]
19411942 assert config .connect_timeout == 5
19421943 assert config .read_timeout == 50
1944+ assert config .user_agent_extra == f"@aws/durable-execution-sdk-python/{ __version__ } "
19431945 assert isinstance (client , LambdaClient )
19441946
19451947
@@ -1963,6 +1965,7 @@ def test_lambda_client_initialize_client_with_endpoint(
19631965 config = call_args [1 ]["config" ]
19641966 assert config .connect_timeout == 5
19651967 assert config .read_timeout == 50
1968+ assert config .user_agent_extra == f"@aws/durable-execution-sdk-python/{ __version__ } "
19661969 assert isinstance (client , LambdaClient )
19671970
19681971
@@ -2038,6 +2041,8 @@ def test_lambda_client_initialize_client_no_endpoint(
20382041 call_args = mock_boto_client .call_args
20392042 assert call_args [0 ] == ("lambda" ,)
20402043 assert "config" in call_args [1 ]
2044+ config = call_args [1 ]["config" ]
2045+ assert config .user_agent_extra == f"@aws/durable-execution-sdk-python/{ __version__ } "
20412046 assert isinstance (client , LambdaClient )
20422047
20432048
0 commit comments