From ea7673618d125277ae0d124ad5cabb255dd1c296 Mon Sep 17 00:00:00 2001
From: Mak Muftic <mak@nodefactory.io>
Date: Wed, 29 Sep 2021 16:39:44 +0200
Subject: [PATCH] Add unit tests for lotus tracer

---
 node/modules/tracer/tracer_test.go | 110 +++++++++++++++++++++++++++++
 1 file changed, 110 insertions(+)
 create mode 100644 node/modules/tracer/tracer_test.go

diff --git a/node/modules/tracer/tracer_test.go b/node/modules/tracer/tracer_test.go
new file mode 100644
index 000000000..8ce809d3e
--- /dev/null
+++ b/node/modules/tracer/tracer_test.go
@@ -0,0 +1,110 @@
+package tracer
+
+import (
+	"testing"
+	"time"
+
+	"github.com/libp2p/go-libp2p-core/peer"
+	pubsub "github.com/libp2p/go-libp2p-pubsub"
+	pubsub_pb "github.com/libp2p/go-libp2p-pubsub/pb"
+	"github.com/stretchr/testify/require"
+)
+
+type testTracerTransport struct {
+	t           *testing.T
+	executeTest func(t *testing.T, evt TracerTransportEvent)
+}
+
+const peerIdA peer.ID = "12D3KooWAbSVMgRejb6ECg6fRTkCPGCfu8396msZVryu8ivcz44G"
+
+func NewTestTraceTransport(t *testing.T, executeTest func(t *testing.T, evt TracerTransportEvent)) TracerTransport {
+	return &testTracerTransport{
+		t:           t,
+		executeTest: executeTest,
+	}
+}
+
+func (ttt *testTracerTransport) Transport(evt TracerTransportEvent) error {
+	ttt.executeTest(ttt.t, evt)
+	return nil
+}
+
+func TestTracer_PeerScores(t *testing.T) {
+
+	testTransport := NewTestTraceTransport(t, func(t *testing.T, evt TracerTransportEvent) {
+		require.Equal(t, peerIdA.Pretty(), evt.lotusTraceEvent.PeerID)
+		require.Equal(t, "source-auth-token-test", evt.lotusTraceEvent.SourceAuth)
+		require.Equal(t, float64(32), evt.lotusTraceEvent.PeerScore.Score)
+
+		n := time.Now().UnixNano()
+		require.LessOrEqual(t, *evt.lotusTraceEvent.Timestamp, n)
+
+		require.Equal(t, peerIdA.Pretty(), evt.lotusTraceEvent.PeerScore.PeerID)
+		require.Equal(t, 1, len(evt.lotusTraceEvent.PeerScore.Topics))
+
+		topic := evt.lotusTraceEvent.PeerScore.Topics[0]
+		require.Equal(t, "topicA", topic.Topic)
+		require.Equal(t, float64(100), topic.FirstMessageDeliveries)
+	})
+
+	lt := NewLotusTracer(
+		[]TracerTransport{testTransport},
+		peerIdA,
+		"source-auth-token-test",
+	)
+
+	topics := make(map[string]*pubsub.TopicScoreSnapshot)
+	topics["topicA"] = &pubsub.TopicScoreSnapshot{
+		FirstMessageDeliveries: float64(100),
+	}
+
+	m := make(map[peer.ID]*pubsub.PeerScoreSnapshot)
+	m[peerIdA] = &pubsub.PeerScoreSnapshot{
+		Score:  float64(32),
+		Topics: topics,
+	}
+
+	lt.PeerScores(m)
+}
+
+func TestTracer_PubSubTrace(t *testing.T) {
+	now := time.Now()
+	n := int64(now.Unix())
+
+	testTransport := NewTestTraceTransport(t, func(t *testing.T, evt TracerTransportEvent) {
+		require.Equal(t, []byte(peerIdA), evt.pubsubTraceEvent.PeerID)
+		require.Equal(t, &n, evt.pubsubTraceEvent.Timestamp)
+	})
+
+	lt := NewLotusTracer(
+		[]TracerTransport{testTransport},
+		"pid",
+		"source-auth",
+	)
+
+	lt.Trace(&pubsub_pb.TraceEvent{
+		PeerID:    []byte(peerIdA),
+		Timestamp: &n,
+	})
+
+}
+
+func TestTracer_MultipleTransports(t *testing.T) {
+	testTransportA := NewTestTraceTransport(t, func(t *testing.T, evt TracerTransportEvent) {
+		require.Equal(t, []byte(peerIdA), evt.pubsubTraceEvent.PeerID)
+	})
+
+	testTransportB := NewTestTraceTransport(t, func(t *testing.T, evt TracerTransportEvent) {
+		require.Equal(t, []byte(peerIdA), evt.pubsubTraceEvent.PeerID)
+	})
+
+	executeTest := NewLotusTracer(
+		[]TracerTransport{testTransportA, testTransportB},
+		"pid",
+		"source-auth",
+	)
+
+	executeTest.Trace(&pubsub_pb.TraceEvent{
+		PeerID: []byte(peerIdA),
+	})
+}