Draft of actor call with RemainingTime timeout

Author: Pijukatel

tests/integration/conftest.py

@@ -30,7 +30,7 @@
 _TOKEN_ENV_VAR = 'APIFY_TEST_USER_API_TOKEN'
 _API_URL_ENV_VAR = 'APIFY_INTEGRATION_TESTS_API_URL'
 _SDK_ROOT_PATH = Path(__file__).parent.parent.parent.resolve()
-
+_DEFAULT_TEST_TIMEOUT = 600
 
 @pytest.fixture
 def prepare_test_env(monkeypatch: pytest.MonkeyPatch, tmp_path: Path) -> Callable[[], None]:
@@ -315,7 +315,7 @@ async def _make_actor(
             name=actor_name,
             default_run_build='latest',
             default_run_memory_mbytes=256,
-            default_run_timeout_secs=600,
+            default_run_timeout_secs=_DEFAULT_TEST_TIMEOUT,
             versions=[
                 {
                     'versionNumber': '0.0',
@@ -331,7 +331,7 @@ async def _make_actor(
         print(f'Building Actor {actor_name}...')
         build_result = await actor_client.build(version_number='0.0')
         build_client = client.build(build_result['id'])
-        build_client_result = await build_client.wait_for_finish(wait_secs=600)
+        build_client_result = await build_client.wait_for_finish(wait_secs=_DEFAULT_TEST_TIMEOUT)
 
         assert build_client_result is not None
         assert build_client_result['status'] == ActorJobStatus.SUCCEEDED
@@ -408,7 +408,7 @@ async def _run_actor(
 
         client = ApifyClientAsync(token=apify_token, api_url=os.getenv(_API_URL_ENV_VAR))
         run_client = client.run(call_result['id'])
-        run_result = await run_client.wait_for_finish(wait_secs=600)
+        run_result = await run_client.wait_for_finish(wait_secs=_DEFAULT_TEST_TIMEOUT)
 
         return ActorRun.model_validate(run_result)
 

tests/integration/test_actor_call.py

@@ -0,0 +1,42 @@
+import asyncio
+
+import pytest
+
+from apify import Actor
+from tests.integration.conftest import MakeActorFunction, RunActorFunction
+
+
+async def test_actor_start_remaining_timeout(
+    make_actor: MakeActorFunction,
+    run_actor: RunActorFunction,
+) -> None:
+    async def main() -> None:
+        async with Actor:
+            actor_input = (await Actor.get_input()) or {}
+            if actor_input.get("called_from_another_actor",False) is True:
+                # Do nothing and only wait for
+                await asyncio.sleep(1000)
+                return
+
+
+            self_run_client = Actor.apify_client.run(Actor.configuration.actor_run_id)
+            self_run_data_1 = await self_run_client.get()
+            run_time_1 = self_run_data_1.get('runTimeSecs', 0)
+            self_timeout = self_run_data_1.get('options', {}).get('timeoutSecs', 0)
+
+            # Start another run of this actor with timeout set to the time remaining in this actor
+            other_run_data = await Actor.start(actor_id=Actor.configuration.actor_id,
+                                               run_input={"called_from_another_actor": True},
+                                               timeout="RemainingTime")
+
+            self_run_data_2 = await self_run_client.get()
+            run_time_2 = self_run_data_2.get('runTimeSecs', 0)
+            other_actor_timeout = other_run_data.get('options', {}).get('timeoutSecs', 0)
+
+            assert other_actor_timeout > self_timeout - run_time_1
+            assert other_actor_timeout < self_timeout - run_time_2
+
+    actor = await make_actor(label='remaining timeout', main_func=main)
+    run_result = await run_actor(actor)
+
+    assert run_result.status == 'SUCCEEDED'