mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
28 lines
898 B
Python
28 lines
898 B
Python
import pytest
|
|
|
|
from libp2p.custom_types import (
|
|
TMuxerOptions,
|
|
TSecurityOptions,
|
|
)
|
|
from libp2p.transport.upgrader import (
|
|
TransportUpgrader,
|
|
)
|
|
|
|
|
|
@pytest.mark.trio
|
|
async def test_transport_upgrader_security_and_muxer_initialization():
|
|
"""Test TransportUpgrader initializes security and muxer multistreams correctly."""
|
|
secure_transports: TSecurityOptions = {}
|
|
muxer_transports: TMuxerOptions = {}
|
|
negotiate_timeout = 15
|
|
|
|
upgrader = TransportUpgrader(
|
|
secure_transports, muxer_transports, negotiate_timeout=negotiate_timeout
|
|
)
|
|
|
|
# Verify security multistream initialization
|
|
assert upgrader.security_multistream.transports == secure_transports
|
|
# Verify muxer multistream initialization and timeout
|
|
assert upgrader.muxer_multistream.transports == muxer_transports
|
|
assert upgrader.muxer_multistream.negotiate_timeout == negotiate_timeout
|