varun-r-mallya/py-libp2p
1
0
Fork 0
mirror of https://github.com/varun-r-mallya/py-libp2p.git synced 2025-12-31 20:36:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

run lint and fix errors, except mypy

Browse Source
This commit is contained in:
pacrob
2024-02-19 15:56:20 -07:00
parent 42605c0288
commit 94483714a3
171 changed files with 4809 additions and 2290 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
tests/peer/test_addrbook.py
View File

@ -1,6 +1,9 @@
import pytest
from libp2p.peer.peerstore import PeerStore, PeerStoreError
from libp2p.peer.peerstore import (
PeerStore,
PeerStoreError,
)
# Testing methods from IAddrBook base class.
@ -51,8 +54,8 @@ def test_peers_with_addrs():
store.add_addrs("peer2", ["/foo"], 10)
store.add_addrs("peer3", ["/bar"], 10)
assert set(store.peers_with_addrs()) == set(["peer2", "peer3"])
assert set(store.peers_with_addrs()) == {"peer2", "peer3"}
store.clear_addrs("peer2")
assert set(store.peers_with_addrs()) == set(["peer3"])
assert set(store.peers_with_addrs()) == {"peer3"}

8
tests/peer/test_interop.py
View File

@ -3,8 +3,12 @@ import base64
import Crypto.PublicKey.RSA as RSA
from libp2p.crypto.pb import crypto_pb2 as pb
from libp2p.crypto.rsa import RSAPrivateKey
from libp2p.peer.id import ID
from libp2p.crypto.rsa import (
RSAPrivateKey,
)
from libp2p.peer.id import (
ID,
)
# ``PRIVATE_KEY_PROTOBUF_SERIALIZATION`` is a protobuf holding an RSA private key.
PRIVATE_KEY_PROTOBUF_SERIALIZATION = """

8
tests/peer/test_peerid.py
View File

@ -3,9 +3,13 @@ import random
import base58
import multihash
from libp2p.crypto.rsa import create_new_key_pair
from libp2p.crypto.rsa import (
create_new_key_pair,
)
import libp2p.peer.id as PeerID
from libp2p.peer.id import ID
from libp2p.peer.id import (
ID,
)
ALPHABETS = "123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz"

10
tests/peer/test_peerinfo.py
View File

@ -3,8 +3,14 @@ import random
import multiaddr
import pytest
from libp2p.peer.id import ID
from libp2p.peer.peerinfo import InvalidAddrError, PeerInfo, info_from_p2p_addr
from libp2p.peer.id import (
ID,
)
from libp2p.peer.peerinfo import (
InvalidAddrError,
PeerInfo,
info_from_p2p_addr,
)
ALPHABETS = "123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz"
VALID_MULTI_ADDR_STR = "/ip4/127.0.0.1/tcp/8000/p2p/3YgLAeMKSAPcGqZkAt8mREqhQXmJT8SN8VCMN4T6ih4GNX9wvK8mWJnWZ1qA2mLdCQ" # noqa: E501

5
tests/peer/test_peermetadata.py
View File

@ -1,6 +1,9 @@
import pytest
from libp2p.peer.peerstore import PeerStore, PeerStoreError
from libp2p.peer.peerstore import (
PeerStore,
PeerStoreError,
)
# Testing methods from IPeerMetadata base class.

17
tests/peer/test_peerstore.py
View File

@ -1,6 +1,9 @@
import pytest
from libp2p.peer.peerstore import PeerStore, PeerStoreError
from libp2p.peer.peerstore import (
PeerStore,
PeerStoreError,
)
# Testing methods from IPeerStore base class.
@ -25,8 +28,8 @@ def test_add_get_protocols_basic():
store.add_protocols("peer1", ["p1", "p2"])
store.add_protocols("peer2", ["p3"])
assert set(store.get_protocols("peer1")) == set(["p1", "p2"])
assert set(store.get_protocols("peer2")) == set(["p3"])
assert set(store.get_protocols("peer1")) == {"p1", "p2"}
assert set(store.get_protocols("peer2")) == {"p3"}
def test_add_get_protocols_extend():
@ -34,7 +37,7 @@ def test_add_get_protocols_extend():
store.add_protocols("peer1", ["p1", "p2"])
store.add_protocols("peer1", ["p3"])
assert set(store.get_protocols("peer1")) == set(["p1", "p2", "p3"])
assert set(store.get_protocols("peer1")) == {"p1", "p2", "p3"}
def test_set_protocols():
@ -45,8 +48,8 @@ def test_set_protocols():
store.set_protocols("peer1", ["p4"])
store.set_protocols("peer2", [])
assert set(store.get_protocols("peer1")) == set(["p4"])
assert set(store.get_protocols("peer2")) == set([])
assert set(store.get_protocols("peer1")) == {"p4"}
assert set(store.get_protocols("peer2")) == set()
# Test with methods from other Peer interfaces.
@ -56,4 +59,4 @@ def test_peers():
store.put("peer2", "key", "val")
store.add_addr("peer3", "/foo", 10)
assert set(store.peer_ids()) == set(["peer1", "peer2", "peer3"])
assert set(store.peer_ids()) == {"peer1", "peer2", "peer3"}