Add timeout parameter support to Pipedream and AsyncPipedream constructors

Co-authored-by: jverce <[email protected]>

Author: Copilot

src/pipedream/pipedream.py

@@ -28,6 +28,7 @@ def __init__(
         ),
         environment: PipedreamEnvironment = PipedreamEnvironment.PROD,
         workflow_domain: Optional[str] = None,
+        timeout: Optional[float] = None,
     ):
         if not project_id:
             raise ValueError("Project ID is required")
@@ -38,6 +39,7 @@ def __init__(
                 project_environment=project_environment,
                 project_id=project_id,
                 token=(lambda: access_token),
+                timeout=timeout,
             )
         else:
             super().__init__(
@@ -46,6 +48,7 @@ def __init__(
                 project_id=project_id,
                 client_id=client_id,
                 client_secret=client_secret,
+                timeout=timeout,
             )
 
         if not workflow_domain:
@@ -79,6 +82,7 @@ def __init__(
         ),
         environment: PipedreamEnvironment = PipedreamEnvironment.PROD,
         workflow_domain: Optional[str] = None,
+        timeout: Optional[float] = None,
     ):
         if not project_id:
             raise ValueError("Project ID is required")
@@ -89,6 +93,7 @@ def __init__(
                 project_environment=project_environment,
                 project_id=project_id,
                 token=(lambda: access_token),
+                timeout=timeout,
             )
         else:
             super().__init__(
@@ -97,6 +102,7 @@ def __init__(
                 project_id=project_id,
                 client_id=client_id,
                 client_secret=client_secret,
+                timeout=timeout,
             )
 
         if not workflow_domain:

tests/custom/test_client.py

@@ -1,7 +1,69 @@
 import pytest
 
+from pipedream import AsyncPipedream, Pipedream
+
 
 # Get started with writing tests with pytest at https://docs.pytest.org
 @pytest.mark.skip(reason="Unimplemented")
 def test_client() -> None:
     assert True
+
+
+def test_pipedream_accepts_timeout_with_access_token() -> None:
+    """Test that Pipedream accepts the timeout parameter with access_token."""
+    client = Pipedream(
+        project_id="test-project-id",
+        access_token="test-token",
+        timeout=5.0,
+    )
+    assert client._client_wrapper._timeout == 5.0
+
+
+def test_pipedream_accepts_timeout_with_client_credentials() -> None:
+    """Test that Pipedream accepts the timeout parameter with client credentials."""
+    client = Pipedream(
+        project_id="test-project-id",
+        client_id="test-client-id",
+        client_secret="test-client-secret",
+        timeout=10.0,
+    )
+    assert client._client_wrapper._timeout == 10.0
+
+
+def test_async_pipedream_accepts_timeout_with_access_token() -> None:
+    """Test that AsyncPipedream accepts the timeout parameter with access_token."""
+    client = AsyncPipedream(
+        project_id="test-project-id",
+        access_token="test-token",
+        timeout=5.0,
+    )
+    assert client._client_wrapper._timeout == 5.0
+
+
+def test_async_pipedream_accepts_timeout_with_client_credentials() -> None:
+    """Test that AsyncPipedream accepts the timeout parameter with client credentials."""
+    client = AsyncPipedream(
+        project_id="test-project-id",
+        client_id="test-client-id",
+        client_secret="test-client-secret",
+        timeout=15.0,
+    )
+    assert client._client_wrapper._timeout == 15.0
+
+
+def test_pipedream_default_timeout() -> None:
+    """Test that Pipedream uses the default timeout of 60 seconds when not specified."""
+    client = Pipedream(
+        project_id="test-project-id",
+        access_token="test-token",
+    )
+    assert client._client_wrapper._timeout == 60
+
+
+def test_async_pipedream_default_timeout() -> None:
+    """Test that AsyncPipedream uses the default timeout of 60 seconds when not specified."""
+    client = AsyncPipedream(
+        project_id="test-project-id",
+        access_token="test-token",
+    )
+    assert client._client_wrapper._timeout == 60