feat: add scoped task source tracing and SDK examples
This commit is contained in:
61
sdk/go/task_source_test.go
Normal file
61
sdk/go/task_source_test.go
Normal file
@@ -0,0 +1,61 @@
|
||||
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")
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user