110 lines
2.6 KiB
Go
110 lines
2.6 KiB
Go
package orch
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
inboxcli "ai-workflow-skill/internal/cli/inbox"
|
|
)
|
|
|
|
func runOrchCommand(t *testing.T, args ...string) string {
|
|
t.Helper()
|
|
|
|
stdout, stderr, exitCode := executeOrchCommand(args...)
|
|
if exitCode != 0 {
|
|
t.Fatalf("execute orch command %v: exit=%d\nstderr:\n%s\nstdout:\n%s", args, exitCode, stderr, stdout)
|
|
}
|
|
|
|
return stdout
|
|
}
|
|
|
|
func executeOrchCommand(args ...string) (string, string, int) {
|
|
var stdout bytes.Buffer
|
|
var stderr bytes.Buffer
|
|
exitCode := Execute(args, &stdout, &stderr)
|
|
return stdout.String(), stderr.String(), exitCode
|
|
}
|
|
|
|
func runInboxCommand(t *testing.T, args ...string) string {
|
|
t.Helper()
|
|
|
|
stdout, stderr, exitCode := executeInboxCommand(args...)
|
|
if exitCode != 0 {
|
|
t.Fatalf("execute inbox command %v: exit=%d\nstderr:\n%s\nstdout:\n%s", args, exitCode, stderr, stdout)
|
|
}
|
|
|
|
return stdout
|
|
}
|
|
|
|
func executeInboxCommand(args ...string) (string, string, int) {
|
|
var stdout bytes.Buffer
|
|
var stderr bytes.Buffer
|
|
exitCode := inboxcli.Execute(args, &stdout, &stderr)
|
|
return stdout.String(), stderr.String(), exitCode
|
|
}
|
|
|
|
func mustDecodeJSON(t *testing.T, raw string, target any) {
|
|
t.Helper()
|
|
|
|
if err := json.Unmarshal([]byte(raw), target); err != nil {
|
|
t.Fatalf("decode json %q: %v", raw, err)
|
|
}
|
|
}
|
|
|
|
func nestedString(t *testing.T, value map[string]any, keys ...string) string {
|
|
t.Helper()
|
|
|
|
current := nestedValue(t, value, keys...)
|
|
str, ok := current.(string)
|
|
if !ok {
|
|
t.Fatalf("expected string at %v, got %#v", keys, current)
|
|
}
|
|
return str
|
|
}
|
|
|
|
func nestedValue(t *testing.T, value map[string]any, keys ...string) any {
|
|
t.Helper()
|
|
|
|
var current any = value
|
|
for _, key := range keys {
|
|
obj, ok := current.(map[string]any)
|
|
if !ok {
|
|
t.Fatalf("expected object at %q in %v, got %#v", key, keys, current)
|
|
}
|
|
current, ok = obj[key]
|
|
if !ok {
|
|
t.Fatalf("missing key %q in %v", key, keys)
|
|
}
|
|
}
|
|
return current
|
|
}
|
|
|
|
func nestedArray(t *testing.T, value map[string]any, keys ...string) []any {
|
|
t.Helper()
|
|
|
|
current := nestedValue(t, value, keys...)
|
|
items, ok := current.([]any)
|
|
if !ok {
|
|
t.Fatalf("expected array at %v, got %#v", keys, current)
|
|
}
|
|
return items
|
|
}
|
|
|
|
func assertErrorJSON(t *testing.T, raw string, expectedCode string) {
|
|
t.Helper()
|
|
|
|
var payload map[string]any
|
|
mustDecodeJSON(t, raw, &payload)
|
|
if ok, _ := payload["ok"].(bool); ok {
|
|
t.Fatalf("expected ok=false error payload, got %#v", payload)
|
|
}
|
|
errorValue, ok := payload["error"].(map[string]any)
|
|
if !ok {
|
|
t.Fatalf("expected error object, got %#v", payload["error"])
|
|
}
|
|
if code, _ := errorValue["code"].(string); code != expectedCode {
|
|
t.Fatalf("expected error code %q, got %#v", expectedCode, errorValue["code"])
|
|
}
|
|
}
|