Function TestPublishWithScheduleTTL

jetstream/test/publish_test.go:2312–2373

(t *testing.T)

}

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

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

	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()
	stream, err := js.CreateStream(ctx, jetstream.StreamConfig{
		Name:              "SCHED",
		Subjects:          []string{"schedule.>", "target.>"},
		AllowMsgSchedules: true,
		AllowMsgTTL:       true,
	})
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}

	t.Run("duration", func(t *testing.T) {
		ack, err := js.Publish(ctx, "schedule.ttl1", nil,
			jetstream.WithScheduleEvery(5*time.Second),
			jetstream.WithScheduleTarget("target.ttl1"),
			jetstream.WithScheduleTTL(10*time.Minute),
		)
		if err != nil {
			t.Fatalf("Unexpected error: %v", err)
		}
		gotMsg, err := stream.GetMsg(ctx, ack.Sequence)
		if err != nil {
			t.Fatalf("Unexpected error: %v", err)
		}
		if got := gotMsg.Header.Get(jetstream.ScheduleTTLHeader); got != "10m0s" {
			t.Fatalf("Expected schedule TTL header %q; got: %q", "10m0s", got)
		}
	})

	t.Run("never", func(t *testing.T) {
		ack, err := js.Publish(ctx, "schedule.ttl2", nil,
			jetstream.WithScheduleEvery(5*time.Second),
			jetstream.WithScheduleTarget("target.ttl2"),
			jetstream.WithScheduleTTLNever(),
		)
		if err != nil {
			t.Fatalf("Unexpected error: %v", err)
		}
		gotMsg, err := stream.GetMsg(ctx, ack.Sequence)
		if err != nil {
			t.Fatalf("Unexpected error: %v", err)
		}
		if got := gotMsg.Header.Get(jetstream.ScheduleTTLHeader); got != "never" {