feat: Adding stream state to the Network Stream

docs/conf.py

@@ -13,11 +13,12 @@
 # If extensions (or modules to document with autodoc) are in another directory,
 # add these directories to sys.path here. If the directory is relative to the
 # documentation root, use os.path.abspath to make it absolute, like shown here.
-# sys.path.insert(0, os.path.abspath('.'))
 
 import doctest
 import os
 import sys
+
+sys.path.insert(0, os.path.abspath('..'))
 from unittest.mock import MagicMock
 
 try:

examples/websocket/test_websocket_transport.py

@@ -3,15 +3,16 @@
 Simple test script to verify WebSocket transport functionality.
 """
 
-import asyncio
 import logging
 from pathlib import Path
 import sys
 
 # Add the libp2p directory to the path so we can import it
 sys.path.insert(0, str(Path(__file__).parent))
 
+import pytest
 import multiaddr
+import trio
 
 from libp2p.transport import create_transport, create_transport_for_multiaddr
 from libp2p.transport.upgrader import TransportUpgrader
@@ -21,6 +22,7 @@
 logger = logging.getLogger(__name__)
 
 
+@pytest.mark.trio
 async def test_websocket_transport():
     """Test basic WebSocket transport functionality."""
     print("🧪 Testing WebSocket Transport Functionality")
@@ -79,6 +81,7 @@ async def test_handler(conn):
     return True
 
 
+@pytest.mark.trio
 async def test_transport_registry():
     """Test the transport registry functionality."""
     print("\n🔧 Testing Transport Registry")
@@ -135,7 +138,7 @@ async def main():
 
 if __name__ == "__main__":
     try:
-        asyncio.run(main())
+        trio.run(main)
     except KeyboardInterrupt:
         print("\n👋 Test interrupted by user")
     except Exception as e: