move pubsub testing tools into tests/utils

This commit is contained in:
pacrob
2025-03-24 18:05:43 -06:00
committed by Paul Robinson
parent 24b2704d8c
commit 4514e86ee4
13 changed files with 699 additions and 56 deletions

View File

@ -1,12 +1,12 @@
import pytest
import trio
from libp2p.tools.pubsub.dummy_account_node import (
DummyAccountNode,
)
from libp2p.tools.utils import (
connect,
)
from tests.utils.pubsub.dummy_account_node import (
DummyAccountNode,
)
async def perform_test(num_nodes, adjacency_map, action_func, assertion_func):

View File

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

View File

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

View File

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

View File

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