update timeout and handling of timeout errors

Author: zhongkechen

src/aws_durable_execution_sdk_python_testing/cli.py

@@ -360,7 +360,7 @@ def invoke_command(self, args: argparse.Namespace) -> int:
             )
 
             try:
-                with urlopen(req, timeout=30) as response:  # noqa: S310
+                with urlopen(req, timeout=10) as response:  # noqa: S310
                     result = json.loads(response.read().decode("utf-8"))
                     print(json.dumps(result, indent=2))  # noqa: T201
                     return 0
@@ -381,9 +381,6 @@ def invoke_command(self, args: argparse.Namespace) -> int:
                 "Error: Could not connect to the local runner server. Is it running?"
             )
             return 1
-        except TimeoutError:
-            logger.exception("Request timeout")
-            return 1
         except Exception:
             logger.exception("Unexpected error in invoke command")
             return 1

tests/cli_test.py

@@ -638,7 +638,7 @@ def test_invoke_command_makes_http_request_to_start_execution_endpoint() -> None
             req = call_args[0][0]
             assert req.full_url.endswith("/start-durable-execution")
             assert req.get_header("Content-type") == "application/json"
-            assert call_args[1]["timeout"] == 30
+            assert call_args[1]["timeout"] == 10
 
             # Verify payload structure
             payload = json.loads(req.data.decode("utf-8"))
@@ -697,24 +697,6 @@ def test_invoke_command_handles_connection_error() -> None:
         assert exit_code == 1
 
 
-def test_invoke_command_handles_timeout_error() -> None:
-    """Test that invoke command handles timeout errors gracefully."""
-    app = CliApp()
-
-    with patch("aws_durable_execution_sdk_python_testing.cli.urlopen") as mock_urlopen:
-        mock_urlopen.side_effect = TimeoutError("Request timed out")
-
-        exit_code = app.invoke_command(
-            argparse.Namespace(
-                function_name="test-function",
-                input="{}",
-                durable_execution_name=None,
-            )
-        )
-
-        assert exit_code == 1
-
-
 def test_invoke_command_handles_http_error_response() -> None:
     """Test that invoke command handles HTTP error responses."""
     app = CliApp()