Function TestPushConsumerConsume_WithQueue

jetstream/test/push_test.go:509–582

(t *testing.T)

}

func TestPushConsumerConsume_WithQueue(t *testing.T) {
	srv := RunBasicJetStreamServer()
	defer shutdownJSServerAndRemoveStorage(t, srv)
	nc, err := nats.Connect(srv.ClientURL())
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}

	js, err := jetstream.New(nc)
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}
	defer nc.Close()

	ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
	defer cancel()
	s, err := js.CreateStream(ctx, jetstream.StreamConfig{Name: "foo", Subjects: []string{"FOO.*"}})
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}
	c1, err := s.CreatePushConsumer(ctx, jetstream.ConsumerConfig{
		DeliverSubject: nats.NewInbox(),
		DeliverGroup:   "workers",
		AckPolicy:      jetstream.AckExplicitPolicy,
	})
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}

	c2, err := s.PushConsumer(ctx, c1.CachedInfo().Name)
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}

	msgs := make([]jetstream.Msg, 0)
	lock := sync.Mutex{}
	wg := &sync.WaitGroup{}
	l1, err := c1.Consume(func(msg jetstream.Msg) {
		msg.Ack()
		lock.Lock()
		msgs = append(msgs, msg)
		lock.Unlock()
		wg.Done()
	})
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}
	defer l1.Stop()
	l2, err := c2.Consume(func(msg jetstream.Msg) {
		msg.Ack()
		lock.Lock()
		msgs = append(msgs, msg)
		lock.Unlock()
		wg.Done()
	})
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}