62 lines
2.8 KiB
Go
62 lines
2.8 KiB
Go
package orbit
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestTaskSourceValidationAndPrecision(t *testing.T) {
|
|
const task = "11111111111111111111111111111111"
|
|
const parent = "22222222222222222222222222222222"
|
|
const occurrence = "33333333333333333333333333333333"
|
|
at := time.Date(2026, 9, 17, 0, 0, 0, 123456000, time.UTC)
|
|
until := at.Add(time.Minute)
|
|
var reply atomic.Value
|
|
var calls atomic.Int32
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
calls.Add(1)
|
|
if r.Method != "GET" || r.URL.Path != "/v1/tasks/"+task+"/source" || r.Header.Get("Authorization") != "Bearer example-token" || r.Header.Get("Idempotency-Key") != "" {
|
|
t.Error("invalid source request")
|
|
}
|
|
json.NewEncoder(w).Encode(map[string]any{"request_id": task, "data": reply.Load()})
|
|
}))
|
|
defer server.Close()
|
|
client, _ := NewClient(server.URL, Options{Token: "example-token"})
|
|
p := parent
|
|
valid := []TaskSource{{TaskID: task, Kind: "DIRECT"}, {TaskID: task, Kind: "MANUAL_RETRY", RetryOf: &p}, {TaskID: task, Kind: "SCHEDULE", Schedule: &TaskScheduleSource{ScheduleID: parent, ScheduleVersion: 9007199254740993, OccurrenceID: occurrence, ScheduledAt: at, Decision: "FIRED"}}, {TaskID: task, Kind: "SCHEDULE", Schedule: &TaskScheduleSource{ScheduleID: parent, ScheduleVersion: 1, OccurrenceID: occurrence, ScheduledAt: at, Decision: "COALESCED", CoveredUntil: &until}}}
|
|
for _, v := range valid {
|
|
reply.Store(v)
|
|
out, err := client.GetTaskSource(context.Background(), task)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if v.Schedule != nil && out.Schedule.ScheduleVersion != v.Schedule.ScheduleVersion {
|
|
t.Fatal("version precision lost")
|
|
}
|
|
}
|
|
self := task
|
|
invalid := []TaskSource{{TaskID: parent, Kind: "DIRECT"}, {TaskID: task, Kind: "unknown"}, {TaskID: task, Kind: "DIRECT", RetryOf: &p}, {TaskID: task, Kind: "MANUAL_RETRY"}, {TaskID: task, Kind: "MANUAL_RETRY", RetryOf: &self}, {TaskID: task, Kind: "SCHEDULE"}}
|
|
for _, change := range []func(*TaskScheduleSource){func(v *TaskScheduleSource) { v.ScheduleVersion = 0 }, func(v *TaskScheduleSource) { v.Decision = "SKIPPED" }, func(v *TaskScheduleSource) { v.Decision = "COALESCED" }, func(v *TaskScheduleSource) { v.CoveredUntil = &until }, func(v *TaskScheduleSource) { v.ScheduledAt = time.Time{} }, func(v *TaskScheduleSource) { v.OccurrenceID = "invalid" }} {
|
|
v := valid[2]
|
|
s := *v.Schedule
|
|
change(&s)
|
|
v.Schedule = &s
|
|
invalid = append(invalid, v)
|
|
}
|
|
for _, v := range invalid {
|
|
reply.Store(v)
|
|
if _, err := client.GetTaskSource(context.Background(), task); err == nil {
|
|
t.Fatal("invalid source accepted", v)
|
|
}
|
|
}
|
|
before := calls.Load()
|
|
if _, err := client.GetTaskSource(context.Background(), "../tasks"); err == nil || calls.Load() != before {
|
|
t.Fatal("unsafe source path sent")
|
|
}
|
|
}
|