StarMirror/py-libp2p
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Explicitly import ID, Transport from secio

Browse Source
This commit is contained in:
mhchia 2019-08-31 22:38:46 +08:00
parent aa0866698f
commit b955c0fa02
No known key found for this signature in database
GPG Key ID: 389EFBEA1362589A
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/factories.py
View File

@ -10,7 +10,7 @@ from libp2p.pubsub.gossipsub import GossipSub
from libp2p.pubsub.pubsub import Pubsub from libp2p.pubsub.pubsub import Pubsub
from libp2p.security.base_transport import BaseSecureTransport from libp2p.security.base_transport import BaseSecureTransport
from libp2p.security.insecure.transport import PLAINTEXT_PROTOCOL_ID, InsecureTransport from libp2p.security.insecure.transport import PLAINTEXT_PROTOCOL_ID, InsecureTransport
from libp2p.security.secio.transport import ID, Transport import libp2p.security.secio.transport as secio
from libp2p.typing import TProtocol from libp2p.typing import TProtocol
from tests.pubsub.configs import ( from tests.pubsub.configs import (
FLOODSUB_PROTOCOL_ID, FLOODSUB_PROTOCOL_ID,
@ -22,15 +22,10 @@ from tests.pubsub.configs import (
def security_transport_factory( def security_transport_factory(
is_secure: bool, key_pair: KeyPair is_secure: bool, key_pair: KeyPair
) -> Dict[TProtocol, BaseSecureTransport]: ) -> Dict[TProtocol, BaseSecureTransport]:
protocol_id: TProtocol
security_transport: BaseSecureTransport
if not is_secure: if not is_secure:
protocol_id = PLAINTEXT_PROTOCOL_ID return {PLAINTEXT_PROTOCOL_ID: InsecureTransport(key_pair)}
security_transport = InsecureTransport(key_pair)
else: else:
protocol_id = ID return {secio.ID: secio.Transport(key_pair)}
security_transport = Transport(key_pair)
return {protocol_id: security_transport}
def swarm_factory(is_secure: bool): def swarm_factory(is_secure: bool):