moved factories to tests

This commit is contained in:
Khwahish Patel
2025-02-21 16:18:43 +05:30
committed by Paul Robinson
parent ddf68ee4e0
commit df72979606
30 changed files with 59 additions and 66 deletions

View File

@ -7,12 +7,12 @@ from libp2p.host.exceptions import (
from libp2p.peer.peerinfo import (
info_from_p2p_addr,
)
from libp2p.tools.factories import (
HostFactory,
)
from libp2p.tools.utils import (
MAX_READ_LEN,
)
from tests.factories import (
HostFactory,
)
CHAT_PROTOCOL_ID = "/chat/1.0.0"
ECHO_PROTOCOL_ID = "/echo/1.0.0"

View File

@ -3,7 +3,7 @@ import pytest
from libp2p.peer.peerinfo import (
info_from_p2p_addr,
)
from libp2p.tools.factories import (
from tests.factories import (
HostFactory,
)

View File

@ -8,7 +8,7 @@ from libp2p.host.ping import (
PING_LENGTH,
PingService,
)
from libp2p.tools.factories import (
from tests.factories import (
host_pair_factory,
)

View File

@ -6,7 +6,7 @@ from libp2p.host.exceptions import (
from libp2p.peer.peerinfo import (
PeerInfo,
)
from libp2p.tools.factories import (
from tests.factories import (
HostFactory,
RoutedHostFactory,
)

View File

@ -7,7 +7,7 @@ from libp2p.identity.identify.protocol import (
ID,
_mk_identify_protobuf,
)
from libp2p.tools.factories import (
from tests.factories import (
host_pair_factory,
)

View File

@ -1,6 +1,6 @@
import pytest
from libp2p.tools.factories import (
from tests.factories import (
net_stream_pair_factory,
swarm_conn_pair_factory,
swarm_pair_factory,

View File

@ -22,12 +22,12 @@ from libp2p.tools.async_service import (
from libp2p.tools.constants import (
LISTEN_MADDR,
)
from libp2p.tools.factories import (
SwarmFactory,
)
from libp2p.tools.utils import (
connect_swarm,
)
from tests.factories import (
SwarmFactory,
)
class Event(enum.Enum):

View File

@ -10,12 +10,12 @@ from trio.testing import (
from libp2p.network.exceptions import (
SwarmException,
)
from libp2p.tools.factories import (
SwarmFactory,
)
from libp2p.tools.utils import (
connect_swarm,
)
from tests.factories import (
SwarmFactory,
)
@pytest.mark.trio

View File

@ -6,12 +6,12 @@ from trio.testing import (
from libp2p.host.exceptions import (
StreamFailure,
)
from libp2p.tools.factories import (
HostFactory,
)
from libp2p.tools.utils import (
create_echo_stream_handler,
)
from tests.factories import (
HostFactory,
)
PROTOCOL_ECHO = "/echo/1.0.0"
PROTOCOL_POTATO = "/potato/1.0.0"

View File

@ -6,9 +6,6 @@ import trio
from libp2p.peer.id import (
ID,
)
from libp2p.tools.factories import (
PubsubFactory,
)
from libp2p.tools.pubsub.floodsub_integration_test_settings import (
floodsub_protocol_pytest_params,
perform_test_from_obj,
@ -16,6 +13,9 @@ from libp2p.tools.pubsub.floodsub_integration_test_settings import (
from libp2p.tools.utils import (
connect,
)
from tests.factories import (
PubsubFactory,
)
@pytest.mark.trio

View File

@ -6,10 +6,6 @@ import trio
from libp2p.pubsub.gossipsub import (
PROTOCOL_ID,
)
from libp2p.tools.factories import (
IDFactory,
PubsubFactory,
)
from libp2p.tools.pubsub.utils import (
dense_connect,
one_to_all_connect,
@ -17,6 +13,10 @@ from libp2p.tools.pubsub.utils import (
from libp2p.tools.utils import (
connect,
)
from tests.factories import (
IDFactory,
PubsubFactory,
)
@pytest.mark.trio

View File

@ -5,13 +5,13 @@ import pytest
from libp2p.tools.constants import (
FLOODSUB_PROTOCOL_ID,
)
from libp2p.tools.factories import (
PubsubFactory,
)
from libp2p.tools.pubsub.floodsub_integration_test_settings import (
floodsub_protocol_pytest_params,
perform_test_from_obj,
)
from tests.factories import (
PubsubFactory,
)
@pytest.mark.parametrize("test_case_obj", floodsub_protocol_pytest_params)

View File

@ -21,11 +21,6 @@ from libp2p.pubsub.pubsub import (
from libp2p.tools.constants import (
MAX_READ_LEN,
)
from libp2p.tools.factories import (
IDFactory,
PubsubFactory,
net_stream_pair_factory,
)
from libp2p.tools.pubsub.utils import (
make_pubsub_msg,
)
@ -35,6 +30,11 @@ from libp2p.tools.utils import (
from libp2p.utils import (
encode_varint_prefixed,
)
from tests.factories import (
IDFactory,
PubsubFactory,
net_stream_pair_factory,
)
TESTING_TOPIC = "TEST_SUBSCRIBE"
TESTING_DATA = b"data"

View File

@ -4,7 +4,7 @@ from libp2p.security.noise.io import (
MAX_NOISE_MESSAGE_LEN,
NoisePacketReadWriter,
)
from libp2p.tools.factories import (
from tests.factories import (
raw_conn_factory,
)

View File

@ -3,7 +3,7 @@ import pytest
from libp2p.security.noise.messages import (
NoiseHandshakePayload,
)
from libp2p.tools.factories import (
from tests.factories import (
noise_conn_factory,
noise_handshake_payload_factory,
)

View File

@ -14,7 +14,7 @@ from libp2p.security.secio.transport import (
from libp2p.tools.constants import (
MAX_READ_LEN,
)
from libp2p.tools.factories import (
from tests.factories import (
raw_conn_factory,
)

View File

@ -12,7 +12,7 @@ from libp2p.security.secio.transport import ID as SECIO_PROTOCOL_ID
from libp2p.security.secure_session import (
SecureSession,
)
from libp2p.tools.factories import (
from tests.factories import (
host_pair_factory,
)

View File

@ -1,6 +1,6 @@
import pytest
from libp2p.tools.factories import (
from tests.factories import (
mplex_conn_pair_factory,
mplex_stream_pair_factory,
)

View File

@ -10,13 +10,13 @@ from libp2p.network.stream.exceptions import (
from libp2p.tools.constants import (
MAX_READ_LEN,
)
from libp2p.tools.factories import (
HostFactory,
)
from libp2p.tools.utils import (
connect,
create_echo_stream_handler,
)
from tests.factories import (
HostFactory,
)
PROTOCOL_ID_0 = TProtocol("/echo/0")
PROTOCOL_ID_1 = TProtocol("/echo/1")