Handle format and mypy

Author: Pijukatel

tests/integration/actor_source_base/test_server.py

@@ -13,6 +13,7 @@
 import asyncio
 import logging
 from collections.abc import Awaitable, Callable, Coroutine
+from socket import socket
 from typing import Any
 
 from uvicorn import Config
@@ -72,8 +73,10 @@ def url(self) -> URL:
         protocol = 'https' if self.config.is_ssl else 'http'
         return URL(f'{protocol}://{self.config.host}:{self.config.port}/')
 
-    async def serve(self) -> None:
+    async def serve(self, sockets: list[socket] | None = None) -> None:
         """Run the server."""
+        if sockets:
+            raise RuntimeError('Simple TestServer does not support custom sockets')
         self.restart_requested = asyncio.Event()
 
         loop = asyncio.get_event_loop()

tests/integration/test_crawlers_with_storages.py

@@ -1,4 +1,9 @@
-from tests.integration.conftest import MakeActorFunction, RunActorFunction
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .conftest import MakeActorFunction, RunActorFunction
 
 
 async def test_actor_on_platform_max_crawl_depth(
@@ -79,8 +84,7 @@ async def test_actor_on_platform_max_request_retries(
 
     async def main() -> None:
         """The crawler entry point."""
-        from crawlee._types import BasicCrawlingContext
-        from crawlee.crawlers import ParselCrawler, ParselCrawlingContext
+        from crawlee.crawlers import BasicCrawlingContext, ParselCrawler, ParselCrawlingContext
 
         from apify import Actor
 
@@ -99,7 +103,8 @@ async def default_handler(_: ParselCrawlingContext) -> None:
                 raise RuntimeError('Some error')
 
             await crawler.run(['http://localhost:8080/'])
-            assert failed_counter == max_retries, f'{failed_counter=}' # TODO max_retries + 1
+            # https://github.com/apify/crawlee-python/issues/1326 , should be max_retries + 1
+            assert failed_counter == max_retries, f'{failed_counter=}'
 
     actor = await make_actor(label='crawler-max-retries', main_func=main)
     run_result = await run_actor(actor)