90 lines
3.3 KiB
Go
90 lines
3.3 KiB
Go
package orbit
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"net/http"
|
||
"net/http/httptest"
|
||
"strings"
|
||
"sync/atomic"
|
||
"testing"
|
||
)
|
||
|
||
func TestWorkflowHistoryIdentityBoundsAndResponseValidation(t *testing.T) {
|
||
const id = "11111111111111111111111111111111"
|
||
const child = "22222222222222222222222222222222"
|
||
ctx := context.Background()
|
||
var calls atomic.Int32
|
||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||
calls.Add(1)
|
||
if r.Method != "GET" || r.Header.Get("Idempotency-Key") != "" || r.Header.Get("Authorization") != "Bearer test-token" {
|
||
t.Error("invalid read request")
|
||
}
|
||
if r.URL.Query().Get("cursor") != "opaque+cursor/=" || r.URL.Query().Get("limit") != "1" {
|
||
t.Error("query changed")
|
||
}
|
||
// 合法响应信封中的错误父资源，SDK不得将其作为请求资源的数据返回。
|
||
json.NewEncoder(w).Encode(map[string]any{"request_id": id, "data": map[string]any{"instance_id": child, "workflow_id": child, "step_id": child, "revision": "1", "items": []any{}}})
|
||
}))
|
||
defer server.Close()
|
||
c, err := NewClient(server.URL, Options{Token: "test-token"})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
for _, bad := range []string{"../other", strings.Repeat("A", 32), "a?x=b"} {
|
||
if _, err = c.ListWorkflowActivations(ctx, bad, "work", HistoryPageRequest{}); err == nil {
|
||
t.Fatal("bad workflow accepted")
|
||
}
|
||
if _, err = c.GetWorkflowActivation(ctx, id, "work", bad); err == nil {
|
||
t.Fatal("bad activation accepted")
|
||
}
|
||
if _, err = c.ListWorkflowPollRounds(ctx, id, "work", bad, HistoryPageRequest{}); err == nil {
|
||
t.Fatal("bad round scope accepted")
|
||
}
|
||
if _, err = c.ListWorkflowSignals(ctx, bad, HistoryPageRequest{}); err == nil {
|
||
t.Fatal("bad signal parent accepted")
|
||
}
|
||
if _, err = c.ListWorkflowSignalOperations(ctx, id, bad, HistoryPageRequest{}); err == nil {
|
||
t.Fatal("bad signal accepted")
|
||
}
|
||
if _, err = c.GetWorkflowSignalOperation(ctx, id, child, bad); err == nil {
|
||
t.Fatal("bad operation accepted")
|
||
}
|
||
}
|
||
for _, key := range []string{"$root", "../other", "", strings.Repeat("x", 65)} {
|
||
if _, err = c.ListWorkflowActivations(ctx, id, key, HistoryPageRequest{}); err == nil {
|
||
t.Fatal("bad key accepted")
|
||
}
|
||
}
|
||
for _, in := range []HistoryPageRequest{{Limit: -1}, {Limit: 101}, {Cursor: strings.Repeat("x", 1025)}} {
|
||
if _, err = c.ListWorkflowActivations(ctx, id, "work", in); err == nil {
|
||
t.Fatal("bad activation page")
|
||
}
|
||
if _, err = c.ListWorkflowPollRounds(ctx, id, "work", child, in); err == nil {
|
||
t.Fatal("bad round page")
|
||
}
|
||
if _, err = c.ListWorkflowSignals(ctx, id, in); err == nil {
|
||
t.Fatal("bad signal page")
|
||
}
|
||
if _, err = c.ListWorkflowSignalOperations(ctx, id, child, in); err == nil {
|
||
t.Fatal("bad operation page")
|
||
}
|
||
}
|
||
if calls.Load() != 0 {
|
||
t.Fatal("invalid arguments reached transport")
|
||
}
|
||
page := HistoryPageRequest{Limit: 1, Cursor: "opaque+cursor/="}
|
||
if _, err = c.ListWorkflowActivations(ctx, id, "work", page); err == nil {
|
||
t.Fatal("foreign activation page accepted")
|
||
}
|
||
if _, err = c.ListWorkflowPollRounds(ctx, id, "work", child, page); err == nil {
|
||
t.Fatal("foreign round page accepted")
|
||
}
|
||
if _, err = c.ListWorkflowSignals(ctx, id, page); err == nil {
|
||
t.Fatal("foreign signal page accepted")
|
||
}
|
||
if calls.Load() != 3 {
|
||
t.Fatal("unexpected automatic read retry")
|
||
}
|
||
}
|