diff --git a/libp2p/pubsub/pubsub.py b/libp2p/pubsub/pubsub.py index af5c41d1..ecfa544d 100644 --- a/libp2p/pubsub/pubsub.py +++ b/libp2p/pubsub/pubsub.py @@ -361,8 +361,7 @@ class Pubsub(Service, IPubsub): if origin_id in self.peer_topics[sub_message.topicid]: self.peer_topics[sub_message.topicid].discard(origin_id) - # FIXME(mhchia): Change the function name? - async def handle_talk(self, publish_message: rpc_pb2.Message) -> None: + def notify_subscriptions(self, publish_message: rpc_pb2.Message) -> None: """ Put incoming message from a peer onto my blocking queue. @@ -576,7 +575,7 @@ class Pubsub(Service, IPubsub): return self._mark_msg_seen(msg) - await self.handle_talk(msg) + self.notify_subscriptions(msg) await self.router.publish(msg_forwarder, msg) def _next_seqno(self) -> bytes: diff --git a/tests/pubsub/test_pubsub.py b/tests/pubsub/test_pubsub.py index 6a22008c..d6c29310 100644 --- a/tests/pubsub/test_pubsub.py +++ b/tests/pubsub/test_pubsub.py @@ -378,14 +378,14 @@ async def test_handle_talk(): data=b"1234", seqno=b"\x00" * 8, ) - await pubsubs_fsub[0].handle_talk(msg_0) + pubsubs_fsub[0].notify_subscriptions(msg_0) msg_1 = make_pubsub_msg( origin_id=pubsubs_fsub[0].my_id, topic_ids=["NOT_SUBSCRIBED"], data=b"1234", seqno=b"\x11" * 8, ) - await pubsubs_fsub[0].handle_talk(msg_1) + pubsubs_fsub[0].notify_subscriptions(msg_1) assert ( len(pubsubs_fsub[0].topic_ids) == 1 and sub == pubsubs_fsub[0].subscribed_topics_receive[TESTING_TOPIC]