313 lines
11 KiB
Go
313 lines
11 KiB
Go
package cli
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
type projectCapture struct {
|
|
method string
|
|
path string
|
|
}
|
|
|
|
func projectServer(t *testing.T, status int, respBody string) (*httptest.Server, *projectCapture) {
|
|
t.Helper()
|
|
capture := &projectCapture{}
|
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
capture.method = r.Method
|
|
capture.path = r.URL.Path
|
|
if !strings.HasPrefix(r.URL.Path, "/api/v1/projects") {
|
|
http.NotFound(w, r)
|
|
return
|
|
}
|
|
w.Header().Set("Content-Type", "application/json")
|
|
w.WriteHeader(status)
|
|
_, _ = io.WriteString(w, respBody)
|
|
}))
|
|
t.Cleanup(srv.Close)
|
|
return srv, capture
|
|
}
|
|
|
|
func TestProjectList_Success(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, capture := projectServer(t, http.StatusOK, `{"projects":[{"id":"zeta","name":"Zeta","sessionPrefix":"zeta"},{"id":"alpha","name":"Alpha","sessionPrefix":"alpha","resolveError":"config missing"}]}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "ls")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
|
|
}
|
|
if capture.method != http.MethodGet || capture.path != "/api/v1/projects" {
|
|
t.Fatalf("request = %s %s, want GET /api/v1/projects", capture.method, capture.path)
|
|
}
|
|
if !strings.Contains(out, "ID") || !strings.Contains(out, "SESSION PREFIX") {
|
|
t.Fatalf("output missing table header:\n%s", out)
|
|
}
|
|
if strings.Index(out, "alpha") > strings.Index(out, "zeta") {
|
|
t.Fatalf("projects should be sorted by id in output:\n%s", out)
|
|
}
|
|
if !strings.Contains(out, "degraded: config missing") {
|
|
t.Fatalf("output missing degraded status:\n%s", out)
|
|
}
|
|
}
|
|
|
|
func TestProjectList_JSON(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, _ := projectServer(t, http.StatusOK, `{"projects":[{"id":"demo","name":"Demo","sessionPrefix":"demo"}]}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "ls", "--json")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
|
|
}
|
|
var got projectListResult
|
|
if err := json.Unmarshal([]byte(out), &got); err != nil {
|
|
t.Fatalf("decode json output: %v\nout=%s", err, out)
|
|
}
|
|
if len(got.Projects) != 1 || got.Projects[0].ID != "demo" {
|
|
t.Fatalf("projects = %#v, want demo", got.Projects)
|
|
}
|
|
}
|
|
|
|
func TestProjectList_Empty(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, _ := projectServer(t, http.StatusOK, `{"projects":[]}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "ls")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
|
|
}
|
|
if !strings.Contains(out, "No projects registered") || !strings.Contains(out, "ao project add --path") {
|
|
t.Fatalf("empty output missing hint:\n%s", out)
|
|
}
|
|
}
|
|
|
|
func TestProjectGet_Success(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, capture := projectServer(t, http.StatusOK, `{"status":"ok","project":{"id":"demo","name":"Demo","path":"/repo/demo","repo":"git@example.com:demo.git","defaultBranch":"main","agent":"codex"}}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "get", "demo")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
|
|
}
|
|
if capture.method != http.MethodGet || capture.path != "/api/v1/projects/demo" {
|
|
t.Fatalf("request = %s %s, want GET /api/v1/projects/demo", capture.method, capture.path)
|
|
}
|
|
for _, want := range []string{"Project demo (ok)", "name: Demo", "path: /repo/demo", "default branch: main", "agent: codex"} {
|
|
if !strings.Contains(out, want) {
|
|
t.Fatalf("output missing %q:\n%s", want, out)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestProjectGet_JSON(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, capture := projectServer(t, http.StatusOK, `{"status":"degraded","project":{"id":"demo","name":"Demo","path":"/repo/demo","resolveError":"config missing"}}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "get", "demo", "--json")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
|
|
}
|
|
if capture.method != http.MethodGet || capture.path != "/api/v1/projects/demo" {
|
|
t.Fatalf("request = %s %s, want GET /api/v1/projects/demo", capture.method, capture.path)
|
|
}
|
|
var got projectGetResult
|
|
if err := json.Unmarshal([]byte(out), &got); err != nil {
|
|
t.Fatalf("decode json output: %v\nout=%s", err, out)
|
|
}
|
|
if got.Status != "degraded" || got.Project.ID != "demo" || got.Project.ResolveError != "config missing" {
|
|
t.Fatalf("get json = %#v, want degraded demo with resolve error", got)
|
|
}
|
|
}
|
|
|
|
func TestProjectGet_MissingArg(t *testing.T) {
|
|
setConfigEnv(t)
|
|
_, _, err := executeCLI(t, Deps{}, "project", "get")
|
|
if err == nil {
|
|
t.Fatal("expected missing arg error")
|
|
}
|
|
if got := ExitCode(err); got != 2 {
|
|
t.Fatalf("exit code = %d, want 2", got)
|
|
}
|
|
}
|
|
|
|
func TestProjectGet_NotFound(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, _ := projectServer(t, http.StatusNotFound, `{"error":"not_found","code":"PROJECT_NOT_FOUND","message":"Unknown project"}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
_, errOut, err := executeCLI(t, Deps{
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "get", "missing")
|
|
if err == nil {
|
|
t.Fatal("expected not found error")
|
|
}
|
|
if got := ExitCode(err); got != 1 {
|
|
t.Fatalf("exit code = %d, want 1", got)
|
|
}
|
|
if !strings.Contains(err.Error(), "PROJECT_NOT_FOUND") && !strings.Contains(errOut, "PROJECT_NOT_FOUND") {
|
|
t.Fatalf("error did not surface not found envelope: %v\nstderr=%s", err, errOut)
|
|
}
|
|
}
|
|
|
|
func TestProjectRemove_RequiresID(t *testing.T) {
|
|
setConfigEnv(t)
|
|
_, _, err := executeCLI(t, Deps{}, "project", "rm")
|
|
if err == nil {
|
|
t.Fatal("expected missing id error")
|
|
}
|
|
if got := ExitCode(err); got != 2 {
|
|
t.Fatalf("exit code = %d, want 2", got)
|
|
}
|
|
}
|
|
|
|
func TestProjectRemove_NotFound(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, _ := projectServer(t, http.StatusNotFound, `{"error":"not_found","code":"PROJECT_NOT_FOUND","message":"Unknown project"}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
_, errOut, err := executeCLI(t, Deps{
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "rm", "missing", "--yes")
|
|
if err == nil {
|
|
t.Fatal("expected not found error")
|
|
}
|
|
if got := ExitCode(err); got != 1 {
|
|
t.Fatalf("exit code = %d, want 1", got)
|
|
}
|
|
if !strings.Contains(err.Error(), "PROJECT_NOT_FOUND") && !strings.Contains(errOut, "PROJECT_NOT_FOUND") {
|
|
t.Fatalf("error did not surface not found envelope: %v\nstderr=%s", err, errOut)
|
|
}
|
|
}
|
|
|
|
func TestProjectRemove_AbortsWhenConfirmationDoesNotMatch(t *testing.T) {
|
|
setConfigEnv(t)
|
|
out, _, err := executeCLI(t, Deps{
|
|
In: strings.NewReader("nope\n"),
|
|
}, "project", "rm", "demo")
|
|
if err != nil {
|
|
t.Fatalf("unexpected abort error: %v", err)
|
|
}
|
|
if !strings.Contains(out, "Type the project id to confirm") || !strings.Contains(out, "aborted") {
|
|
t.Fatalf("output missing prompt/abort:\n%s", out)
|
|
}
|
|
}
|
|
|
|
func TestProjectRemove_DeletesAfterConfirmation(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, capture := projectServer(t, http.StatusOK, `{"ok":true,"id":"demo"}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
In: strings.NewReader("demo\n"),
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "rm", "demo")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
|
|
}
|
|
if capture.method != http.MethodDelete || capture.path != "/api/v1/projects/demo" {
|
|
t.Fatalf("request = %s %s, want DELETE /api/v1/projects/demo", capture.method, capture.path)
|
|
}
|
|
if !strings.Contains(out, "removed project demo") {
|
|
t.Fatalf("output missing removal message:\n%s", out)
|
|
}
|
|
}
|
|
|
|
func TestProjectRemove_JSONDocumentedEnvelope(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, capture := projectServer(t, http.StatusOK, `{"ok":true,"id":"demo"}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
In: strings.NewReader("wrong\n"),
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "rm", "demo", "--yes", "--json")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
|
|
}
|
|
if capture.method != http.MethodDelete || capture.path != "/api/v1/projects/demo" {
|
|
t.Fatalf("request = %s %s, want DELETE /api/v1/projects/demo", capture.method, capture.path)
|
|
}
|
|
var got projectRemoveResult
|
|
if err := json.Unmarshal([]byte(out), &got); err != nil {
|
|
t.Fatalf("decode json output: %v\nout=%s", err, out)
|
|
}
|
|
if !got.OK || got.ID != "demo" || got.ProjectID != "" {
|
|
t.Fatalf("remove json = %#v, want documented ok/id envelope", got)
|
|
}
|
|
}
|
|
|
|
func TestProjectRemove_JSONBackendEnvelope(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
removedStorageDir := false
|
|
srv, _ := projectServer(t, http.StatusOK, `{"projectId":"demo","removedStorageDir":false}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "rm", "demo", "--yes", "--json")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
|
|
}
|
|
var got projectRemoveResult
|
|
if err := json.Unmarshal([]byte(out), &got); err != nil {
|
|
t.Fatalf("decode json output: %v\nout=%s", err, out)
|
|
}
|
|
if got.ProjectID != "demo" || got.RemovedStorageDir == nil || *got.RemovedStorageDir != removedStorageDir {
|
|
t.Fatalf("remove json = %#v, want backend projectId/removedStorageDir envelope", got)
|
|
}
|
|
}
|
|
|
|
func TestProjectRemove_EmptySuccessFallsBackToRequestedID(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, _ := projectServer(t, http.StatusNoContent, ``)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "rm", "demo", "--yes")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error for empty 2xx body: %v\nstderr=%s", err, errOut)
|
|
}
|
|
if !strings.Contains(out, "removed project demo") {
|
|
t.Fatalf("output missing fallback removal id:\n%s", out)
|
|
}
|
|
}
|
|
|
|
func TestProjectRemove_YesSkipsConfirmationAndSupportsBackendRemoveEnvelope(t *testing.T) {
|
|
cfg := setConfigEnv(t)
|
|
srv, capture := projectServer(t, http.StatusOK, `{"projectId":"demo","removedStorageDir":false}`)
|
|
writeRunFileFor(t, cfg, srv)
|
|
|
|
out, errOut, err := executeCLI(t, Deps{
|
|
In: strings.NewReader("wrong\n"),
|
|
ProcessAlive: func(int) bool { return true },
|
|
}, "project", "rm", "demo", "--yes")
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v\nstderr=%s", err, errOut)
|
|
}
|
|
if capture.method != http.MethodDelete || capture.path != "/api/v1/projects/demo" {
|
|
t.Fatalf("request = %s %s, want DELETE /api/v1/projects/demo", capture.method, capture.path)
|
|
}
|
|
if strings.Contains(out, "Type the project id") || !strings.Contains(out, "removed project demo") {
|
|
t.Fatalf("--yes output should skip prompt and print removal:\n%s", out)
|
|
}
|
|
}
|