pubsub_test.go

package firestore_test

import (
	"os"
	"strconv"
	"testing"
	"time"

	"github.com/ThreeDotsLabs/watermill"
	"github.com/ThreeDotsLabs/watermill-firestore/pkg/firestore"
	"github.com/ThreeDotsLabs/watermill/message"
	"github.com/ThreeDotsLabs/watermill/pubsub/tests"
)

func createPubSub(t *testing.T) (message.Publisher, message.Subscriber) {
	return createPubSubWithSubscriptionName(t, "topic")
}

func createPubSubWithSubscriptionName(t *testing.T, subscriptionName string) (message.Publisher, message.Subscriber) {
	logger := watermill.NewStdLogger(true, true)

	pub, err := firestore.NewPublisher(
		firestore.PublisherConfig{
			ProjectID:             os.Getenv("FIRESTORE_PROJECT_ID"),
			PubSubRootCollection:  "test-root",
			MessagePublishTimeout: time.Second * 60,
		},
		logger,
	)
	if err != nil {
		t.Fatal(err)
	}

	sub, err := firestore.NewSubscriber(
		firestore.SubscriberConfig{
			ProjectID:                os.Getenv("FIRESTORE_PROJECT_ID"),
			PubSubRootCollection:     "test-root",
			GenerateSubscriptionName: func(topic string) string { return topic + "_" + subscriptionName },
			Timeout:                  time.Second * 30,
		},
		logger,
	)
	if err != nil {
		panic(err)
	}
	return pub, sub
}

func TestPublishSubscribe(t *testing.T) {
	tests.TestPubSub(
		t,
		tests.Features{
			ConsumerGroups:      true,
			ExactlyOnceDelivery: false,
			GuaranteedOrder:     false,
			Persistent:          true,
		},
		createPubSub,
		createPubSubWithSubscriptionName,
	)
}

func createPubSubBench(n int) (message.Publisher, message.Subscriber) {
	logger := watermill.NewStdLogger(true, false)

	pub, err := firestore.NewPublisher(
		firestore.PublisherConfig{
			ProjectID: os.Getenv("FIRESTORE_PROJECT_ID"),
		},
		logger,
	)
	if err != nil {
		panic(err)
	}

	sub, err := firestore.NewSubscriber(
		firestore.SubscriberConfig{
			GenerateSubscriptionName: func(topic string) string {
				return topic + strconv.Itoa(n)
			},
			ProjectID: os.Getenv("FIRESTORE_PROJECT_ID"),
		},
		logger,
	)
	if err != nil {
		panic(err)
	}
	return pub, sub
}

func BenchmarkPublishSubscribe(b *testing.B) {
	tests.BenchSubscriber(b, createPubSubBench)
}