status-go/protocol/messenger_waku_wrapper_test.go

package protocol

import (
	"context"

	"go.uber.org/zap"

	gethbridge "github.com/status-im/status-go/eth-node/bridge/geth"
	"github.com/status-im/status-go/eth-node/types"
	"github.com/status-im/status-go/wakuv1"
)

type testWakuWrapper struct {
	*gethbridge.GethWakuWrapper

	publicWakuAPIWrapper *testPublicWakuAPIWrapper
}

func newTestWaku(w *wakuv1.Waku) types.Waku {
	wrapper := gethbridge.NewGethWakuWrapper(w)
	return &testWakuWrapper{
		GethWakuWrapper:      wrapper.(*gethbridge.GethWakuWrapper),
		publicWakuAPIWrapper: newTestPublicWakuAPI(wakuv1.NewPublicWakuAPI(w)).(*testPublicWakuAPIWrapper),
	}
}

func (tw *testWakuWrapper) PublicWakuAPI() types.PublicWakuAPI {
	return tw.publicWakuAPIWrapper
}

func (tw *testWakuWrapper) SubscribePostEvents() chan *PostMessageSubscription {
	subscription := make(chan *PostMessageSubscription, 100)
	tw.publicWakuAPIWrapper.postSubscriptions = append(tw.publicWakuAPIWrapper.postSubscriptions, subscription)
	return subscription
}

type PostMessageSubscription struct {
	id  []byte
	msg *types.NewMessage
}

type testPublicWakuAPIWrapper struct {
	*gethbridge.GethPublicWakuAPIWrapper

	postSubscriptions []chan *PostMessageSubscription
}

func newTestPublicWakuAPI(api *wakuv1.PublicWakuAPI) types.PublicWakuAPI {
	wrapper := gethbridge.NewGethPublicWakuAPIWrapper(api)
	return &testPublicWakuAPIWrapper{
		GethPublicWakuAPIWrapper: wrapper.(*gethbridge.GethPublicWakuAPIWrapper),
	}
}

func (tp *testPublicWakuAPIWrapper) Post(ctx context.Context, req types.NewMessage) ([]byte, error) {
	id, err := tp.GethPublicWakuAPIWrapper.Post(ctx, req)
	if err != nil {
		return nil, err
	}
	for _, s := range tp.postSubscriptions {
		select {
		case s <- &PostMessageSubscription{id: id, msg: &req}:
		default:
			// subscription channel full
		}
	}
	return id, err
}

func newTestWakuWrapper(config *wakuv1.Config, logger *zap.Logger) (*testWakuWrapper, error) {
	if config == nil {
		config = &wakuv1.DefaultConfig
	}
	w := wakuv1.New(config, logger)
	return newTestWaku(w).(*testWakuWrapper), w.Start()
}