Function TestPublishWithScheduleAt

jetstream/test/publish_test.go:2040–2097

(t *testing.T)

}

func TestPublishWithScheduleAt(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,
	})
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}
	cons, err := stream.CreateConsumer(ctx, jetstream.ConsumerConfig{
		FilterSubject: "target.>",
	})
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}

	schedTime := time.Now().Add(time.Second).UTC().Truncate(time.Second)
	ack, err := js.Publish(ctx, "schedule.at", []byte("hello"),
		jetstream.WithScheduleAt(schedTime),
		jetstream.WithScheduleTarget("target.at"),
	)
	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.ScheduleHeader); got != "@at "+schedTime.Format(time.RFC3339) {
		t.Fatalf("Expected schedule header %q; got: %q", "@at "+schedTime.Format(time.RFC3339), got)
	}
	if got := gotMsg.Header.Get(jetstream.ScheduleTargetHeader); got != "target.at" {
		t.Fatalf("Expected schedule target header %q; got: %q", "target.at", got)
	}

	msg, err := cons.Next(jetstream.FetchMaxWait(5 * time.Second))
	if err != nil {
		t.Fatalf("Unexpected error: %v", err)
	}
	if string(msg.Data()) != "hello" {
		t.Fatalf("Expected message data %q; got: %q", "hello", string(msg.Data()))
	}
}