Function TestSendReceive

pubsub/pubsub_test.go:110–137

(t *testing.T)

func (*driverSub) Close() error                       { return nil }

func TestSendReceive(t *testing.T) {
	ctx := context.Background()
	ds := NewDriverSub()
	dt := &driverTopic{
		subs: []*driverSub{ds},
	}
	topic := NewTopic(dt, nil)
	defer topic.Shutdown(ctx)
	m := &Message{LoggableID: "foo", Body: []byte("user signed up")}
	if err := topic.Send(ctx, m); err == nil {
		t.Fatalf("expected a Send with a non-empty LoggableID to fail")
	}
	m.LoggableID = ""
	if err := topic.Send(ctx, m); err != nil {
		t.Fatal(err)
	}

	sub := NewSubscription(ds, nil, nil)
	defer sub.Shutdown(ctx)
	m2, err := sub.Receive(ctx)
	if err != nil {
		t.Fatal(err)
	}
	if string(m2.Body) != string(m.Body) {
		t.Fatalf("received message has body %q, want %q", m2.Body, m.Body)
	}
	m2.Ack()
}

func TestConcurrentReceivesGetAllTheMessages(t *testing.T) {
	howManyToSend := int(1e3)