def test_request_tracker_reserve_request_id_generated():
tracker = RequestTracker()
node_id = NodeIDFactory()
with tracker.reserve_request_id(node_id) as request_id:
assert tracker.is_request_id_active(node_id, request_id)
assert not tracker.is_request_id_active(node_id, request_id)
def __init__(self, network: NetworkAPI) -> None:
self.logger = get_extended_debug_logger("ddht.AlexandriaClient")
self.network = network
self.request_tracker = RequestTracker()
self.subscription_manager = SubscriptionManager()
network.add_talk_protocol(self)
self._active_request_ids = set()
def test_request_tracker_reserve_request_id_provided():
tracker = RequestTracker()
node_id = NodeIDFactory()
request_id = b"\x01\x02\x03\04"
assert not tracker.is_request_id_active(node_id, request_id)
with tracker.reserve_request_id(node_id, request_id) as actual_request_id:
assert actual_request_id == request_id
assert tracker.is_request_id_active(node_id, request_id)
assert not tracker.is_request_id_active(node_id, request_id)
def __init__(
self,
local_private_key: keys.PrivateKey,
listen_on: Endpoint,
enr_db: QueryableENRDatabaseAPI,
session_cache_size: int,
events: EventsAPI = None,
message_type_registry: MessageTypeRegistry = v51_registry,
) -> None:
self.local_private_key = local_private_key
self.listen_on = listen_on
self._listening = trio.Event()
self.enr_manager = ENRManager(
private_key=local_private_key,
enr_db=enr_db,
)
self.enr_db = enr_db
self._registry = message_type_registry
self.request_tracker = RequestTracker()
# Datagrams
(
self._outbound_datagram_send_channel,
self._outbound_datagram_receive_channel,
) = trio.open_memory_channel[OutboundDatagram](256)
(
self._inbound_datagram_send_channel,
self._inbound_datagram_receive_channel,
) = trio.open_memory_channel[InboundDatagram](256)
# EnvelopePair
(
self._outbound_envelope_send_channel,
self._outbound_envelope_receive_channel,
) = trio.open_memory_channel[OutboundEnvelope](256)
(
self._inbound_envelope_send_channel,
self._inbound_envelope_receive_channel,
) = trio.open_memory_channel[InboundEnvelope](256)
# Messages
(
self._outbound_message_send_channel,
self._outbound_message_receive_channel,
) = trio.open_memory_channel[AnyOutboundMessage](256)
(
self._inbound_message_send_channel,
self._inbound_message_receive_channel,
) = trio.open_memory_channel[AnyInboundMessage](256)
if events is None:
events = Events()
self.events = events
self.pool = Pool(
local_private_key=self.local_private_key,
local_node_id=self.enr_manager.enr.node_id,
enr_db=self.enr_db,
outbound_envelope_send_channel=self.
_outbound_envelope_send_channel,
inbound_message_send_channel=self._inbound_message_send_channel,
session_cache_size=session_cache_size,
message_type_registry=self._registry,
events=self.events,
)
self.dispatcher = Dispatcher(
self._inbound_envelope_receive_channel,
self._inbound_message_receive_channel,
self.pool,
self.enr_db,
self.events,
)
self.envelope_decoder = EnvelopeEncoder(
self._outbound_envelope_receive_channel,
self._outbound_datagram_send_channel,
)
self.envelope_encoder = EnvelopeDecoder(
self._inbound_datagram_receive_channel,
self._inbound_envelope_send_channel,
self.enr_manager.enr.node_id,
)
self._ready = trio.Event()