nullkey
7eeb3bec5d
weknora doc wait <doc-id> [<doc-id>...] blocks until every given document reaches a terminal parse_status (completed / failed), --timeout expires, or the user interrupts (SIGINT). * --timeout DURATION (default 10m; exit 124 on timeout, matches GNU timeout(1) convention) * --interval DURATION (default 2s; exponential backoff to 15s + jitter) * Multi-id polled concurrently (max 5 parallel) * Exit code priority 1 > 124 > 0 (failed > timeout > completed) New typed errors: * operation.timeout → exit 124 * operation.failed → exit 1 * operation.cancelled → exit 1 (main raises to 130 on signal) server.session_create_failed gets a special case in ExitCode to map to exit 1 (workflow failure, not transient retry). doc view and doc download positional id namespaced to <doc-id>.
61 lines
2.0 KiB
Go
61 lines
2.0 KiB
Go
package cmdutil
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestExitCode(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
err error
|
|
want int
|
|
}{
|
|
{"nil success", nil, 0},
|
|
{"flag error", NewFlagError(errors.New("bad flag")), 2},
|
|
{"silent", SilentError, 1},
|
|
{"auth.* prefix", NewError(CodeAuthUnauthenticated, "x"), 3},
|
|
{"auth.token_expired", NewError(CodeAuthTokenExpired, "x"), 3},
|
|
{"resource.not_found", NewError(CodeResourceNotFound, "x"), 4},
|
|
{"input.* prefix", NewError(CodeInputInvalidArgument, "x"), 5},
|
|
{"input.missing_flag", NewError(CodeInputMissingFlag, "x"), 5},
|
|
{"server.rate_limited", NewError(CodeServerRateLimited, "x"), 6},
|
|
{"server.* prefix", NewError(CodeServerError, "x"), 7},
|
|
{"server.timeout", NewError(CodeServerTimeout, "x"), 7},
|
|
{"network.* prefix", NewError(CodeNetworkError, "x"), 7},
|
|
{"unknown error", errors.New("plain"), 1},
|
|
{"local.* prefix", NewError(CodeLocalConfigCorrupt, "x"), 1},
|
|
{"operation.timeout", NewError(CodeOperationTimeout, "timed out"), 124},
|
|
{"operation.failed → 1 (fall-through bucket)", NewError(CodeOperationFailed, "failed"), 1},
|
|
{"operation.cancelled → 1 (main overrides to 130 on signal-cancelled ctx)", NewError(CodeOperationCancelled, "cancelled"), 1},
|
|
{"server.session_create_failed → 1 (workflow, not transient)", NewError(CodeSessionCreateFailed, "x"), 1},
|
|
}
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
assert.Equal(t, tc.want, ExitCode(tc.err))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPrintError(t *testing.T) {
|
|
t.Run("nil is silent", func(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
PrintError(&buf, nil)
|
|
assert.Empty(t, buf.String())
|
|
})
|
|
t.Run("SilentError is silent", func(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
PrintError(&buf, SilentError)
|
|
assert.Empty(t, buf.String())
|
|
})
|
|
t.Run("typed error prints message", func(t *testing.T) {
|
|
var buf bytes.Buffer
|
|
PrintError(&buf, NewError(CodeAuthUnauthenticated, "no creds"))
|
|
assert.Contains(t, buf.String(), "auth.unauthenticated")
|
|
assert.Contains(t, buf.String(), "no creds")
|
|
})
|
|
}
|