mirror of
https://github.com/varun-r-mallya/py-libp2p.git
synced 2025-12-31 20:36:24 +00:00
ft. modernise py-libp2p (#618)
* fix pyproject.toml , add ruff * rm lock * make progress * add poetry lock ignore * fix type issues * fix tcp type errors * fix text example - type error - wrong args * add setuptools to dev * test ci * fix docs build * fix type issues for new_swarm & new_host * fix types in gossipsub * fix type issues in noise * wip: factories * revert factories * fix more type issues * more type fixes * fix: add null checks for noise protocol initialization and key handling * corrected argument-errors in peerId and Multiaddr in peer tests * fix: Noice - remove redundant type casts in BaseNoiseMsgReadWriter * fix: update test_notify.py to use SwarmFactory.create_batch_and_listen, fix type hints, and comment out ClosedStream assertions * Fix type checks for pubsub module Signed-off-by: sukhman <sukhmansinghsaluja@gmail.com> * Fix type checks for pubsub module-tests Signed-off-by: sukhman <sukhmansinghsaluja@gmail.com> * noise: add checks for uninitialized protocol and key states in PatternXX Signed-off-by: varun-r-mallya <varunrmallya@gmail.com> * pubsub: add None checks for optional fields in FloodSub and Pubsub Signed-off-by: varun-r-mallya <varunrmallya@gmail.com> * Fix type hints and improve testing Signed-off-by: varun-r-mallya <varunrmallya@gmail.com> * remove redundant checks Signed-off-by: varun-r-mallya <varunrmallya@gmail.com> * fix build issues * add optional to trio service * fix types * fix type errors * Fix type errors Signed-off-by: varun-r-mallya <varunrmallya@gmail.com> * fixed more-type checks in crypto and peer_data files * wip: factories * replaced union with optional * fix: type-error in interp-utils and peerinfo * replace pyright with pyrefly * add pyrefly.toml * wip: fix multiselect issues * try typecheck * base check * mcache test fixes , typecheck ci update * fix ci * will this work * minor fix * use poetry * fix wokflow * use cache,fix err * fix pyrefly.toml * fix pyrefly.toml * fix cache in ci * deploy commit * add main baseline * update to v5 * improve typecheck ci (#14) * fix typo * remove holepunching code (#16) * fix gossipsub typeerrors (#17) * fix: ensure initiator user includes remote peer id in handshake (#15) * fix ci (#19) * typefix: custom_types | core/peerinfo/test_peer_info | io/abc | pubsub/floodsub | protocol_muxer/multiselect (#18) * fix: Typefixes in PeerInfo (#21) * fix minor type issue (#22) * fix type errors in pubsub (#24) * fix: Minor typefixes in tests (#23) * Fix failing tests for type-fixed test/pubsub (#8) * move pyrefly & ruff to pyproject.toml & rm .project-template (#28) * move the async_context file to tests/core * move crypto test to crypto folder * fix: some typefixes (#25) * fix type errors * fix type issues * fix: update gRPC API usage in autonat_pb2_grpc.py (#31) * md: typecheck ci * rm comments * clean up : from review suggestions * use | None over Optional as per new python standards * drop supporto for py3.9 * newsfragments --------- Signed-off-by: sukhman <sukhmansinghsaluja@gmail.com> Signed-off-by: varun-r-mallya <varunrmallya@gmail.com> Co-authored-by: acul71 <luca.pisani@birdo.net> Co-authored-by: kaneki003 <sakshamchauhan707@gmail.com> Co-authored-by: sukhman <sukhmansinghsaluja@gmail.com> Co-authored-by: varun-r-mallya <varunrmallya@gmail.com> Co-authored-by: varunrmallya <100590632+varun-r-mallya@users.noreply.github.com> Co-authored-by: lla-dane <abhinavagarwalla6@gmail.com> Co-authored-by: Collins <ArtemisfowlX@protonmail.com> Co-authored-by: Abhinav Agarwalla <120122716+lla-dane@users.noreply.github.com> Co-authored-by: guha-rahul <52607971+guha-rahul@users.noreply.github.com> Co-authored-by: Sukhman Singh <63765293+sukhman-sukh@users.noreply.github.com> Co-authored-by: acul71 <34693171+acul71@users.noreply.github.com> Co-authored-by: pacrob <5199899+pacrob@users.noreply.github.com>
This commit is contained in:
@ -1,10 +1,7 @@
|
||||
import pytest
|
||||
from trio.testing import (
|
||||
RaisesGroup,
|
||||
)
|
||||
|
||||
from libp2p.host.exceptions import (
|
||||
StreamFailure,
|
||||
from libp2p.custom_types import (
|
||||
TProtocol,
|
||||
)
|
||||
from libp2p.tools.utils import (
|
||||
create_echo_stream_handler,
|
||||
@ -13,10 +10,10 @@ from tests.utils.factories import (
|
||||
HostFactory,
|
||||
)
|
||||
|
||||
PROTOCOL_ECHO = "/echo/1.0.0"
|
||||
PROTOCOL_POTATO = "/potato/1.0.0"
|
||||
PROTOCOL_FOO = "/foo/1.0.0"
|
||||
PROTOCOL_ROCK = "/rock/1.0.0"
|
||||
PROTOCOL_ECHO = TProtocol("/echo/1.0.0")
|
||||
PROTOCOL_POTATO = TProtocol("/potato/1.0.0")
|
||||
PROTOCOL_FOO = TProtocol("/foo/1.0.0")
|
||||
PROTOCOL_ROCK = TProtocol("/rock/1.0.0")
|
||||
|
||||
ACK_PREFIX = "ack:"
|
||||
|
||||
@ -61,19 +58,12 @@ async def test_single_protocol_succeeds(security_protocol):
|
||||
|
||||
@pytest.mark.trio
|
||||
async def test_single_protocol_fails(security_protocol):
|
||||
# using trio.testing.RaisesGroup b/c pytest.raises does not handle ExceptionGroups
|
||||
# yet: https://github.com/pytest-dev/pytest/issues/11538
|
||||
# but switch to that once they do
|
||||
|
||||
# the StreamFailure is within 2 nested ExceptionGroups, so we use strict=False
|
||||
# to unwrap down to the core Exception
|
||||
with RaisesGroup(StreamFailure, allow_unwrapped=True, flatten_subgroups=True):
|
||||
# Expect that protocol negotiation fails when no common protocols exist
|
||||
with pytest.raises(Exception):
|
||||
await perform_simple_test(
|
||||
"", [PROTOCOL_ECHO], [PROTOCOL_POTATO], security_protocol
|
||||
)
|
||||
|
||||
# Cleanup not reached on error
|
||||
|
||||
|
||||
@pytest.mark.trio
|
||||
async def test_multiple_protocol_first_is_valid_succeeds(security_protocol):
|
||||
@ -103,16 +93,16 @@ async def test_multiple_protocol_second_is_valid_succeeds(security_protocol):
|
||||
|
||||
@pytest.mark.trio
|
||||
async def test_multiple_protocol_fails(security_protocol):
|
||||
protocols_for_client = [PROTOCOL_ROCK, PROTOCOL_FOO, "/bar/1.0.0"]
|
||||
protocols_for_listener = ["/aspyn/1.0.0", "/rob/1.0.0", "/zx/1.0.0", "/alex/1.0.0"]
|
||||
protocols_for_client = [PROTOCOL_ROCK, PROTOCOL_FOO, TProtocol("/bar/1.0.0")]
|
||||
protocols_for_listener = [
|
||||
TProtocol("/aspyn/1.0.0"),
|
||||
TProtocol("/rob/1.0.0"),
|
||||
TProtocol("/zx/1.0.0"),
|
||||
TProtocol("/alex/1.0.0"),
|
||||
]
|
||||
|
||||
# using trio.testing.RaisesGroup b/c pytest.raises does not handle ExceptionGroups
|
||||
# yet: https://github.com/pytest-dev/pytest/issues/11538
|
||||
# but switch to that once they do
|
||||
|
||||
# the StreamFailure is within 2 nested ExceptionGroups, so we use strict=False
|
||||
# to unwrap down to the core Exception
|
||||
with RaisesGroup(StreamFailure, allow_unwrapped=True, flatten_subgroups=True):
|
||||
# Expect that protocol negotiation fails when no common protocols exist
|
||||
with pytest.raises(Exception):
|
||||
await perform_simple_test(
|
||||
"", protocols_for_client, protocols_for_listener, security_protocol
|
||||
)
|
||||
@ -142,8 +132,8 @@ async def test_multistream_command(security_protocol):
|
||||
for protocol in supported_protocols:
|
||||
assert protocol in response
|
||||
|
||||
assert "/does/not/exist" not in response
|
||||
assert "/foo/bar/1.2.3" not in response
|
||||
assert TProtocol("/does/not/exist") not in response
|
||||
assert TProtocol("/foo/bar/1.2.3") not in response
|
||||
|
||||
# Dialer asks for unspoorted command
|
||||
with pytest.raises(ValueError, match="Command not supported"):
|
||||
|
||||
Reference in New Issue
Block a user