clear

Author: Mantisus

tests/unit/test_client_timeouts.py

@@ -150,8 +150,6 @@ def mock_request(*_args: Any, **kwargs: Any) -> Response:
 ]
 
 
-# This test will probably need to be reworked or skipped when switching to `impit`.
-# Without the mock library, it's difficult to reproduce, maybe with monkeypatch?
 @pytest.mark.parametrize(
     ('client_type', 'method', 'expected_timeout', 'kwargs'),
     _timeout_params,
@@ -173,8 +171,6 @@ def test_specific_timeouts_for_specific_endpoints_sync(
     assert patch_request[0] == expected_timeout
 
 
-# This test will probably need to be reworked or skipped when switching to `impit`.
-# Without the mock library, it's difficult to reproduce, maybe with monkeypatch?
 @pytest.mark.parametrize(
     ('client_type', 'method', 'expected_timeout', 'kwargs'),
     _timeout_params,

tests/unit/test_logging.py

@@ -115,8 +115,9 @@ def _streaming_log_handler(_request: Request) -> Response:
     """Handler for streaming log requests."""
 
     def generate_logs() -> Iterator[bytes]:
-        for chunk in _MOCKED_ACTOR_LOGS:  # noqa: UP028
+        for chunk in _MOCKED_ACTOR_LOGS:
             yield chunk
+            time.sleep(0.01)
 
     total_size = sum(len(chunk) for chunk in _MOCKED_ACTOR_LOGS)