From 189724dd1e55d5a9256ed405d6a3e82d10a47df7 Mon Sep 17 00:00:00 2001
From: Anthony Laibe <anthony@laibe.cc>
Date: Sat, 30 Oct 2021 18:42:24 +0200
Subject: [PATCH] test: add filter test

---
 waku/v2/protocol/filter/waku_filter.go      |   1 -
 waku/v2/protocol/filter/waku_filter_test.go | 144 ++++++++++++++++++++
 2 files changed, 144 insertions(+), 1 deletion(-)
 create mode 100644 waku/v2/protocol/filter/waku_filter_test.go

diff --git a/waku/v2/protocol/filter/waku_filter.go b/waku/v2/protocol/filter/waku_filter.go
index 3d0fbd6d..d0eff0e6 100644
--- a/waku/v2/protocol/filter/waku_filter.go
+++ b/waku/v2/protocol/filter/waku_filter.go
@@ -230,7 +230,6 @@ func NewWakuFilter(ctx context.Context, host host.Host, isFullNode bool, handler
 }
 
 func (wf *WakuFilter) FilterListener() {
-
 	// This function is invoked for each message received
 	// on the full node in context of Waku2-Filter
 	handle := func(envelope *protocol.Envelope) error { // async
diff --git a/waku/v2/protocol/filter/waku_filter_test.go b/waku/v2/protocol/filter/waku_filter_test.go
new file mode 100644
index 00000000..86279b3e
--- /dev/null
+++ b/waku/v2/protocol/filter/waku_filter_test.go
@@ -0,0 +1,144 @@
+package filter
+
+import (
+	"context"
+	"crypto/rand"
+	"sync"
+	"testing"
+	"time"
+
+	"github.com/libp2p/go-libp2p-core/host"
+	"github.com/libp2p/go-libp2p-core/peerstore"
+	"github.com/status-im/go-waku/tests"
+	v2 "github.com/status-im/go-waku/waku/v2"
+	"github.com/status-im/go-waku/waku/v2/protocol"
+	"github.com/status-im/go-waku/waku/v2/protocol/pb"
+	"github.com/status-im/go-waku/waku/v2/protocol/relay"
+	"github.com/stretchr/testify/require"
+)
+
+func makeWakuRelay(t *testing.T, topic relay.Topic, broadcaster v2.Broadcaster) (*relay.WakuRelay, *relay.Subscription, host.Host) {
+	port, err := tests.FindFreePort(t, "", 5)
+	require.NoError(t, err)
+
+	host, err := tests.MakeHost(context.Background(), port, rand.Reader)
+	require.NoError(t, err)
+
+	relay, err := relay.NewWakuRelay(context.Background(), host, broadcaster)
+	require.NoError(t, err)
+
+	sub, err := relay.Subscribe(context.Background(), &topic)
+	require.NoError(t, err)
+
+	return relay, sub, host
+}
+
+func makeWakuFilter(t *testing.T, filters Filters) (*WakuFilter, host.Host) {
+	port, err := tests.FindFreePort(t, "", 5)
+	require.NoError(t, err)
+
+	host, err := tests.MakeHost(context.Background(), port, rand.Reader)
+	require.NoError(t, err)
+
+	filterHandler := func(requestId string, msg pb.MessagePush) {
+		for _, message := range msg.Messages {
+			filters.Notify(message, requestId)
+		}
+	}
+
+	filter := NewWakuFilter(context.Background(), host, false, filterHandler)
+
+	return filter, host
+}
+
+// Node1: Filter subscribed to content topic A
+// Node2: Relay + Filter
+//
+// Node1 and Node2 are peers
+//
+// Node2 send a succesful message with topic A
+// Node1 receive the message
+//
+// Node2 send a succesful message with topic B
+// Node1 doesn't receive the message
+func TestWakuFilter(t *testing.T) {
+	var filters = make(Filters)
+	var testTopic relay.Topic = "/waku/2/go/filter/test"
+	testContentTopic := "TopicA"
+
+	node1, host1 := makeWakuFilter(t, filters)
+	defer node1.Stop()
+
+	broadcaster := v2.NewBroadcaster(10)
+	node2, sub2, host2 := makeWakuRelay(t, testTopic, broadcaster)
+	defer node2.Stop()
+	defer sub2.Unsubscribe()
+	filterHandler := func(requestId string, msg pb.MessagePush) {
+		for _, message := range msg.Messages {
+			filters.Notify(message, requestId)
+		}
+	}
+
+	node2Filter := NewWakuFilter(context.Background(), host2, true, filterHandler)
+	broadcaster.Register(node2Filter.MsgC)
+
+	host1.Peerstore().AddAddr(host2.ID(), tests.GetHostAddress(host2), peerstore.PermanentAddrTTL)
+	err := host1.Peerstore().AddProtocols(host2.ID(), string(FilterID_v20beta1))
+	require.NoError(t, err)
+
+	contentFilter := &ContentFilter{
+		Topic:         string(testTopic),
+		ContentTopics: []string{testContentTopic},
+	}
+	sub, err := node1.Subscribe(context.Background(), *contentFilter, []FilterSubscribeOption{WithAutomaticPeerSelection()}...)
+	require.NoError(t, err)
+
+	// Sleep to make sure the filter is subscribed
+	time.Sleep(2 * time.Second)
+
+	ch := make(chan *protocol.Envelope, 1024)
+	filters[sub.RequestID] = Filter{
+		PeerID:         sub.Peer,
+		Topic:          contentFilter.Topic,
+		ContentFilters: contentFilter.ContentTopics,
+		Chan:           ch,
+	}
+	var wg sync.WaitGroup
+
+	wg.Add(1)
+	go func() {
+		defer wg.Done()
+		env := <-ch
+		require.Equal(t, contentFilter.ContentTopics[0], env.Message().GetContentTopic())
+	}()
+
+	_, err = node2.Publish(context.Background(), &pb.WakuMessage{
+		Payload:      []byte{1},
+		Version:      0,
+		ContentTopic: testContentTopic,
+		Timestamp:    0,
+	}, &testTopic)
+	require.NoError(t, err)
+
+	wg.Wait()
+
+	wg.Add(1)
+	go func() {
+		select {
+		case <-ch:
+			require.Fail(t, "should not receive another message")
+		case <-time.After(3 * time.Second):
+			defer wg.Done()
+		}
+	}()
+
+	_, err = node2.Publish(context.Background(), &pb.WakuMessage{
+		Payload:      []byte{1},
+		Version:      0,
+		ContentTopic: "TopicB",
+		Timestamp:    0,
+	}, &testTopic)
+	require.NoError(t, err)
+
+	wg.Wait()
+}