requiring user to pass context

This commit is contained in:
Gabriel mermelstein 2024-12-13 17:25:40 +01:00
parent fe86a6d1f1
commit 0b5ad1d78a
No known key found for this signature in database
GPG Key ID: 82B8134785FEAE0D
2 changed files with 4 additions and 4 deletions

View File

@ -425,9 +425,7 @@ func (w *Waku) DialPeer(address multiaddr.Multiaddr) error {
}
// TODO: change pubsub topic to shard notation everywhere
func (w *Waku) RelayPublish(message *pb.WakuMessage, pubsubTopic string) (pb.MessageHash, error) {
ctx, cancel := context.WithTimeout(w.ctx, requestTimeout)
defer cancel()
func (w *Waku) RelayPublish(ctx context.Context, message *pb.WakuMessage, pubsubTopic string) (pb.MessageHash, error) {
return w.node.RelayPublish(ctx, message, pubsubTopic)
}

View File

@ -457,7 +457,9 @@ func TestRelay(t *testing.T) {
}
// send message
pubsubTopic := FormatWakuRelayTopic(senderNodeWakuConfig.ClusterID, senderNodeWakuConfig.Shards[0])
senderNode.RelayPublish(message, pubsubTopic)
ctx, cancel := context.WithTimeout(context.Background(), requestTimeout)
defer cancel()
senderNode.RelayPublish(ctx, message, pubsubTopic)
wg := sync.WaitGroup{}
wg.Add(1)