Function TestScheduleCronDelivery

jetstream/test/publish_test.go:2572–2623

(t *testing.T)

}

func TestScheduleCronDelivery(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(), 10*time.Second)
	defer cancel()
	_, err = js.CreateStream(ctx, jetstream.StreamConfig{
		Name:              "SCHED",
		Subjects:          []string{"schedule.>", "target.>"},
		AllowMsgSchedules: true,
	})
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}

	cons, err := js.CreateConsumer(ctx, "SCHED", jetstream.ConsumerConfig{
		FilterSubject: "target.cron",
	})
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}

	// Schedule every second using 6-field cron.
	_, err = js.Publish(ctx, "schedule.cron", []byte("cron payload"),
		jetstream.WithScheduleCron("* * * * * *"),
		jetstream.WithScheduleTarget("target.cron"),
	)
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}

	msg, err := cons.Next(jetstream.FetchMaxWait(5 * time.Second))
	if err != nil {
		t.Fatalf("Expected to receive scheduled message: %v", err)
	}
	if string(msg.Data()) != "cron payload" {
		t.Fatalf("Expected payload %q; got: %q", "cron payload", string(msg.Data()))
	}
	if next := msg.Headers().Get(jetstream.ScheduleNextHeader); next == "" {
		t.Fatal("Expected Nats-Schedule-Next header to be set on delivered message")
	}
}