Debug logging instead of print

Author: blast-hardcheese

tests/v2/test_v2_cancellation.py

@@ -25,6 +25,9 @@
 from tests.v2.fixtures.raw_ws_server import OuterPayload, WsServerFixture
 
 
+logger = logging.getLogger(__file__)
+
+
 async def test_rpc_cancel(ws_server: WsServerFixture) -> None:
     (urimeta, recv, conn) = ws_server
 
@@ -226,7 +229,7 @@ async def _upload_chunks() -> AsyncIterator[OuterPayload[dict[Any, Any]]]:
             lambda x: x,
             lambda x: x,
         ):
-            print(repr(chunk))
+            logger.debug(repr(chunk))
 
     receive_task = asyncio.create_task(receive_chunks())
     request_msg = parse_transport_msg(await recv.get())
@@ -358,7 +361,7 @@ async def receive_chunks() -> None:
             lambda x: x,
             lambda x: x,
         ):
-            print(repr(chunk))
+            logger.debug(repr(chunk))
 
     receive_task = asyncio.create_task(receive_chunks())
 

tests/v2/test_v2_session_lifecycle.py

@@ -22,6 +22,9 @@
 from tests.v2.fixtures.raw_ws_server import WsServerFixture
 
 
+logger = logging.getLogger(__file__)
+
+
 class _PermissiveRateLimiter(RateLimiter):
     def start_restoring_budget(self, user: str) -> None:
         pass
@@ -189,9 +192,9 @@ async def handle_server_messages() -> None:
         async for datagram in client.send_subscription(
             "test", "bigstream", {}, lambda x: x, lambda x: x, lambda x: x
         ):
-            print(datagram)
+            logger.debug(datagram)
     except Exception:
-        logging.exception("Interrupted")
+        logger.exception("Interrupted")
 
     await client.close()
     await connecting