renamed file into pipe_test

pkg/pipe/pipe_test.go

@@ -0,0 +1,58 @@
+package pipe
+
+import (
+	"context"
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/nbd-wtf/go-nostr"
+)
+
+var (
+	ctx = context.Background()
+
+	odell string = "04c915daefee38317fa734444acee390a8269fe5810b2241e5e6dd343dfbecc9"
+	calle string = "50d94fc2d8580c682b071a542f8b1e31a200b0508bab95a33bef0855df281d63"
+	pip   string = "f683e87035f7ad4f44e0b98cfbd9537e16455a92cd38cefc4cb31db7557f5ef2"
+)
+
+// Manually change pip's follow list and see if the events gets printed. Works only with `go test`
+func TestFirehose(t *testing.T) {
+	ctx, cancel := context.WithTimeout(ctx, time.Second*20)
+	defer cancel()
+
+	checker := mockChecker{pubkey: pip}
+	config := NewFirehoseConfig()
+	Firehose(ctx, config, checker, print)
+}
+
+func TestFetch(t *testing.T) {
+	pubkeys := []string{odell, calle, pip}
+	config := NewFetcherConfig()
+
+	events, err := fetch(ctx, config, pubkeys)
+	if err != nil {
+		t.Fatalf("expected error nil, got %v", err)
+	}
+
+	expected := len(pubkeys) * 2
+	if len(events) != expected {
+		t.Fatalf("expected %d events, got %d", expected, len(events))
+	}
+}
+
+type mockChecker struct {
+	pubkey string
+}
+
+func (c mockChecker) Exists(ctx context.Context, pubkey string) (bool, error) {
+	return pubkey == c.pubkey, nil
+}
+
+func print(e *nostr.Event) error {
+	fmt.Printf("\nevent ID: %v", e.ID)
+	fmt.Printf("\nevent pubkey: %v", e.PubKey)
+	fmt.Printf("\nevent kind: %d\n", e.Kind)
+	return nil
+}