Debug logging instead of print

Author: blast-hardcheese

tests/v2/test_v2_cancellation.py

@@ -24,6 +24,8 @@
 from replit_river.v2.session import STREAM_CANCEL_BIT, STREAM_CLOSED_BIT
 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 +228,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 +360,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

@@ -21,6 +21,8 @@
 from replit_river.v2.session import STREAM_CLOSED_BIT, Session
 from tests.v2.fixtures.raw_ws_server import WsServerFixture
 
+logger = logging.getLogger(__file__)
+
 
 class _PermissiveRateLimiter(RateLimiter):
     def start_restoring_budget(self, user: str) -> None:
@@ -189,9 +191,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