Felis/cmd/felis/breakglass_test.go

package main

import (
	"context"
	"encoding/json"
	"errors"
	"strings"
	"testing"

	"felis.lolicon.best/internal/api"

	tea "github.com/charmbracelet/bubbletea"
	"golang.org/x/crypto/bcrypt"
)

// fakeOwnerStore records what break-glass provisioning writes and answers the
// identity lookups, so the core logic (authentication, provisioning, accountability
// audit) is exercised without a database or a terminal.
type fakeOwnerStore struct {
	upserts  []upsertCall
	settings map[string][]byte
	audits   []api.AuditEntry
	users    map[string]*api.StaffUser // keyed by username
	admins   bool                      // AdminExists answer

	upsertErr error
	setErr    error
	auditErr  error
	userErr   error // non-not-found error from UserByUsername
	adminErr  error
}

type upsertCall struct {
	id, username, email, passwordHash string
	mustChange                        bool
}

func (f *fakeOwnerStore) AdminExists(_ context.Context) (bool, error) {
	if f.adminErr != nil {
		return false, f.adminErr
	}
	return f.admins, nil
}

func (f *fakeOwnerStore) UserByUsername(_ context.Context, username string) (*api.StaffUser, error) {
	if f.userErr != nil {
		return nil, f.userErr
	}
	if u, ok := f.users[username]; ok {
		return u, nil
	}
	return nil, api.ErrNotFound
}

func (f *fakeOwnerStore) UpsertOwner(_ context.Context, id, username, email, passwordHash string, mustChange bool) error {
	if f.upsertErr != nil {
		return f.upsertErr
	}
	f.upserts = append(f.upserts, upsertCall{id, username, email, passwordHash, mustChange})
	return nil
}

func (f *fakeOwnerStore) SetSetting(_ context.Context, key string, value []byte) error {
	if f.setErr != nil {
		return f.setErr
	}
	if f.settings == nil {
		f.settings = map[string][]byte{}
	}
	f.settings[key] = value
	return nil
}

func (f *fakeOwnerStore) Audit(_ context.Context, e api.AuditEntry) error {
	if f.auditErr != nil {
		return f.auditErr
	}
	f.audits = append(f.audits, e)
	return nil
}

// mkAdmin builds an authenticatable admin row (role=admin, real bcrypt hash) for the
// fake. MinCost keeps the hash fast — these tests are about wiring, not bcrypt.
func mkAdmin(t *testing.T, username, password string) *api.StaffUser {
	t.Helper()
	h, err := bcrypt.GenerateFromPassword([]byte(password), bcrypt.MinCost)
	if err != nil {
		t.Fatalf("hash: %v", err)
	}
	return &api.StaffUser{ID: "usr-admin", Username: username, Role: "admin", PasswordHash: string(h)}
}

func TestValidateOwnerPassword(t *testing.T) {
	cases := []struct {
		name string
		pw   string
		ok   bool
	}{
		{"too short", "1234567", false},
		{"minimum", "12345678", true},
		{"comfortable", "Mid-Range-1", true},
		{"at the bcrypt limit", strings.Repeat("a", 72), true},
		{"past the bcrypt limit", strings.Repeat("a", 73), false},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			err := validateOwnerPassword(tc.pw)
			if tc.ok && err != nil {
				t.Errorf("validateOwnerPassword(%d bytes) = %v, want nil", len(tc.pw), err)
			}
			if !tc.ok && err == nil {
				t.Errorf("validateOwnerPassword(%d bytes) = nil, want error", len(tc.pw))
			}
		})
	}
}

func TestProvisionOwner(t *testing.T) {
	ctx := context.Background()

	t.Run("happy path mints a must-change admin with a verifiable hash", func(t *testing.T) {
		f := &fakeOwnerStore{}
		const pw = "valid-test-pw"
		if err := provisionOwner(ctx, f, "owner", "[email protected]", pw); err != nil {
			t.Fatalf("provisionOwner: %v", err)
		}
		if len(f.upserts) != 1 {
			t.Fatalf("want 1 upsert, got %d", len(f.upserts))
		}
		got := f.upserts[0]
		if got.username != "owner" {
			t.Errorf("username = %q, want owner", got.username)
		}
		if got.email != "[email protected]" {
			t.Errorf("email = %q, want [email protected]", got.email)
		}
		// must_change_password=true is load-bearing: it arms the API lockdown so the
		// Owner can do nothing but change the password on first login.
		if !got.mustChange {
			t.Error("mustChange = false, want true (forced first-login change)")
		}
		if !strings.HasPrefix(got.id, "usr-") {
			t.Errorf("id = %q, want usr- prefix", got.id)
		}
		// Only the hash is stored; the typed plaintext must verify against it.
		if bcrypt.CompareHashAndPassword([]byte(got.passwordHash), []byte(pw)) != nil {
			t.Error("typed password does not verify against the stored hash")
		}
		if got.passwordHash == pw {
			t.Error("stored hash equals plaintext — password was not hashed")
		}
	})

	t.Run("trims surrounding whitespace", func(t *testing.T) {
		f := &fakeOwnerStore{}
		if err := provisionOwner(ctx, f, "  owner  ", "  [email protected]  ", "valid-test-pw"); err != nil {
			t.Fatalf("provisionOwner: %v", err)
		}
		if f.upserts[0].username != "owner" || f.upserts[0].email != "[email protected]" {
			t.Errorf("got username=%q email=%q, want trimmed", f.upserts[0].username, f.upserts[0].email)
		}
	})

	t.Run("rejects an empty username before any write", func(t *testing.T) {
		f := &fakeOwnerStore{}
		if err := provisionOwner(ctx, f, "   ", "", "valid-test-pw"); err == nil {
			t.Fatal("want error for empty username")
		}
		if len(f.upserts) != 0 {
			t.Errorf("want no upsert on validation failure, got %d", len(f.upserts))
		}
	})

	t.Run("rejects a weak password before any write", func(t *testing.T) {
		f := &fakeOwnerStore{}
		if err := provisionOwner(ctx, f, "owner", "", "short"); err == nil {
			t.Fatal("want error for a sub-8-byte password")
		}
		if len(f.upserts) != 0 {
			t.Errorf("want no upsert on weak password, got %d", len(f.upserts))
		}
	})

	t.Run("propagates a store error", func(t *testing.T) {
		f := &fakeOwnerStore{upsertErr: errors.New("boom")}
		if err := provisionOwner(ctx, f, "owner", "", "valid-test-pw"); err == nil {
			t.Fatal("want error when the store fails")
		}
	})
}

func TestEnableLocalAuth(t *testing.T) {
	f := &fakeOwnerStore{}
	if err := enableLocalAuth(context.Background(), f); err != nil {
		t.Fatalf("enableLocalAuth: %v", err)
	}
	raw, ok := f.settings[api.LocalAuthEnabledKey]
	if !ok {
		t.Fatalf("setting %q not written", api.LocalAuthEnabledKey)
	}
	// Mirror how the live API parses the toggle (session.go localAuthEnabled): the
	// stored jsonb must round-trip to the bool true, or the gate fails closed and the
	// Owner can never log in.
	var enabled bool
	if err := json.Unmarshal(raw, &enabled); err != nil {
		t.Fatalf("setting value %q is not valid JSON: %v", raw, err)
	}
	if !enabled {
		t.Errorf("local_auth_enabled = false, want true")
	}
}

func TestGenerateBootstrapPassword(t *testing.T) {
	const want = 20
	pw, err := generateBootstrapPassword()
	if err != nil {
		t.Fatalf("generateBootstrapPassword: %v", err)
	}
	if len(pw) != want {
		t.Errorf("length = %d, want %d", len(pw), want)
	}
	for _, c := range pw {
		if !strings.ContainsRune(bootstrapPasswordAlphabet, c) {
			t.Errorf("password contains out-of-alphabet rune %q", c)
		}
	}
	// A generated password must satisfy the same rule provisionOwner enforces.
	if err := validateOwnerPassword(pw); err != nil {
		t.Errorf("generated password fails validateOwnerPassword: %v", err)
	}
	other, err := generateBootstrapPassword()
	if err != nil {
		t.Fatal(err)
	}
	if pw == other {
		t.Error("two calls produced the same password")
	}
}

func TestAuthenticateAdmin(t *testing.T) {
	ctx := context.Background()

	t.Run("verifies a matching admin credential", func(t *testing.T) {
		f := &fakeOwnerStore{users: map[string]*api.StaffUser{"root": mkAdmin(t, "root", "correct horse")}}
		matched, ok, err := authenticateAdmin(ctx, f, "root", "correct horse")
		if err != nil {
			t.Fatalf("authenticateAdmin: %v", err)
		}
		if !ok {
			t.Fatal("ok = false, want true for the correct password")
		}
		if matched != "root" {
			t.Errorf("matched = %q, want root", matched)
		}
	})

	t.Run("a wrong password is a non-match, not an error", func(t *testing.T) {
		f := &fakeOwnerStore{users: map[string]*api.StaffUser{"root": mkAdmin(t, "root", "correct horse")}}
		_, ok, err := authenticateAdmin(ctx, f, "root", "wrong")
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
		if ok {
			t.Error("ok = true, want false for a wrong password")
		}
	})

	t.Run("a non-admin role can never attribute a break-glass", func(t *testing.T) {
		player := mkAdmin(t, "alice", "correct horse")
		player.Role = "user" // a player row, even with a hash, is not staff
		f := &fakeOwnerStore{users: map[string]*api.StaffUser{"alice": player}}
		_, ok, err := authenticateAdmin(ctx, f, "alice", "correct horse")
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
		if ok {
			t.Error("ok = true, want false for a non-admin role")
		}
	})

	t.Run("a hashless admin row is a non-match", func(t *testing.T) {
		f := &fakeOwnerStore{users: map[string]*api.StaffUser{
			"ghost": {Username: "ghost", Role: "admin", PasswordHash: ""},
		}}
		_, ok, err := authenticateAdmin(ctx, f, "ghost", "anything")
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
		if ok {
			t.Error("ok = true, want false when no hash is set")
		}
	})

	t.Run("an unknown user is a non-match, not an error", func(t *testing.T) {
		f := &fakeOwnerStore{}
		_, ok, err := authenticateAdmin(ctx, f, "nobody", "pw")
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
		if ok {
			t.Error("ok = true, want false for an unknown user")
		}
	})

	t.Run("empty input is a non-match with no store call", func(t *testing.T) {
		f := &fakeOwnerStore{userErr: errors.New("must not be called")}
		if _, ok, err := authenticateAdmin(ctx, f, "", "pw"); ok || err != nil {
			t.Errorf("empty username: ok=%v err=%v, want false,nil", ok, err)
		}
		if _, ok, err := authenticateAdmin(ctx, f, "root", ""); ok || err != nil {
			t.Errorf("empty password: ok=%v err=%v, want false,nil", ok, err)
		}
	})

	t.Run("a datastore fault is surfaced", func(t *testing.T) {
		f := &fakeOwnerStore{userErr: errors.New("db down")}
		if _, _, err := authenticateAdmin(ctx, f, "root", "pw"); err == nil {
			t.Fatal("want error when the store fails")
		}
	})
}

// auditOf decodes the single recorded audit row's payload for assertions.
func auditOf(t *testing.T, f *fakeOwnerStore) (api.AuditEntry, map[string]any) {
	t.Helper()
	if len(f.audits) != 1 {
		t.Fatalf("want exactly 1 audit row, got %d", len(f.audits))
	}
	e := f.audits[0]
	var payload map[string]any
	if err := json.Unmarshal(e.Payload, &payload); err != nil {
		t.Fatalf("audit payload is not valid JSON: %v", err)
	}
	return e, payload
}

func TestPerformBreakGlass(t *testing.T) {
	ctx := context.Background()

	t.Run("bootstrap uses the typed password and never echoes it", func(t *testing.T) {
		f := &fakeOwnerStore{}
		op := breakGlassOp{
			mode:          "bootstrap",
			accountable:   "deploybot",
			osUser:        "deploybot",
			ownerUsername: "owner",
			ownerEmail:    "[email protected]",
			ownerPassword: "valid-test-pw",
		}
		out, err := performBreakGlass(ctx, f, op)
		if err != nil {
			t.Fatalf("performBreakGlass: %v", err)
		}
		// The operator typed their own password, so it must NOT be surfaced for display.
		if out.displayPassword != "" {
			t.Errorf("displayPassword = %q, want empty for a typed bootstrap password", out.displayPassword)
		}
		if out.auditErr != nil {
			t.Errorf("auditErr = %v, want nil", out.auditErr)
		}
		if len(f.upserts) != 1 || bcrypt.CompareHashAndPassword([]byte(f.upserts[0].passwordHash), []byte("valid-test-pw")) != nil {
			t.Error("owner was not provisioned with the typed password")
		}
		if _, ok := f.settings[api.LocalAuthEnabledKey]; !ok {
			t.Error("local auth was not enabled — login would 403")
		}
		e, payload := auditOf(t, f)
		if e.Actor != "deploybot" || e.Source != "break-glass" || e.Action != "break_glass.bootstrap" {
			t.Errorf("audit envelope = %+v, want actor=deploybot source=break-glass action=break_glass.bootstrap", e)
		}
		if payload["verified"] != false {
			t.Errorf("payload.verified = %v, want false for bootstrap", payload["verified"])
		}
		if _, present := payload["admin_account"]; present {
			t.Error("payload.admin_account present, want omitted when no admin was attempted")
		}
		if payload["os_user"] != "deploybot" || payload["owner"] != "owner" {
			t.Errorf("payload = %v, want os_user=deploybot owner=owner", payload)
		}
	})

	t.Run("recovery generates a one-time password and records a verified row", func(t *testing.T) {
		f := &fakeOwnerStore{}
		op := breakGlassOp{
			mode:           "recovery",
			accountable:    "root",
			osUser:         "alice",
			ownerUsername:  "owner",
			attemptedAdmin: "root",
		}
		out, err := performBreakGlass(ctx, f, op)
		if err != nil {
			t.Fatalf("performBreakGlass: %v", err)
		}
		if out.displayPassword == "" {
			t.Fatal("displayPassword empty, want a generated one-time password")
		}
		// The shown password must be the one actually stored (as a hash).
		if bcrypt.CompareHashAndPassword([]byte(f.upserts[0].passwordHash), []byte(out.displayPassword)) != nil {
			t.Error("displayed password does not match the stored hash")
		}
		e, payload := auditOf(t, f)
		if e.Actor != "root" || e.Action != "break_glass.recovery" {
			t.Errorf("audit envelope = %+v, want actor=root action=break_glass.recovery", e)
		}
		if payload["verified"] != true {
			t.Errorf("payload.verified = %v, want true for recovery", payload["verified"])
		}
		if payload["admin_account"] != "root" {
			t.Errorf("payload.admin_account = %v, want root", payload["admin_account"])
		}
	})

	t.Run("root override records an unverified row attributed to the OS user", func(t *testing.T) {
		f := &fakeOwnerStore{}
		op := breakGlassOp{
			mode:           "root_override",
			accountable:    "alice",
			osUser:         "alice",
			ownerUsername:  "owner",
			attemptedAdmin: "typo-admin",
		}
		out, err := performBreakGlass(ctx, f, op)
		if err != nil {
			t.Fatalf("performBreakGlass: %v", err)
		}
		if out.displayPassword == "" {
			t.Error("displayPassword empty, want a generated one-time password")
		}
		e, payload := auditOf(t, f)
		if e.Actor != "alice" || e.Action != "break_glass.root_override" {
			t.Errorf("audit envelope = %+v, want actor=alice action=break_glass.root_override", e)
		}
		if payload["verified"] != false {
			t.Errorf("payload.verified = %v, want false for root override", payload["verified"])
		}
		// The attempted (failed) admin is preserved so the trail shows what was tried.
		if payload["admin_account"] != "typo-admin" {
			t.Errorf("payload.admin_account = %v, want typo-admin", payload["admin_account"])
		}
	})

	t.Run("an audit failure does not fail the recovery", func(t *testing.T) {
		f := &fakeOwnerStore{auditErr: errors.New("audit sink down")}
		op := breakGlassOp{mode: "recovery", accountable: "root", osUser: "alice", ownerUsername: "owner", attemptedAdmin: "root"}
		out, err := performBreakGlass(ctx, f, op)
		if err != nil {
			t.Fatalf("performBreakGlass returned %v, want nil — break-glass must survive a dead audit sink", err)
		}
		if out.auditErr == nil {
			t.Error("auditErr = nil, want the surfaced audit failure")
		}
		// The load-bearing writes must still have happened.
		if len(f.upserts) != 1 {
			t.Error("owner was not provisioned despite a recoverable audit failure")
		}
		if _, ok := f.settings[api.LocalAuthEnabledKey]; !ok {
			t.Error("local auth was not enabled despite a recoverable audit failure")
		}
	})

	t.Run("does not enable local auth or audit if the owner write fails", func(t *testing.T) {
		f := &fakeOwnerStore{upsertErr: errors.New("boom")}
		op := breakGlassOp{mode: "bootstrap", accountable: "root", osUser: "root", ownerUsername: "owner", ownerPassword: "valid-test-pw"}
		if _, err := performBreakGlass(ctx, f, op); err == nil {
			t.Fatal("want error when the owner write fails")
		}
		if len(f.settings) != 0 {
			t.Error("local auth should not be enabled when provisioning failed")
		}
		if len(f.audits) != 0 {
			t.Error("no audit row should be written when provisioning failed")
		}
	})

	t.Run("records accountability before enabling local auth, surviving an enableLocalAuth failure", func(t *testing.T) {
		// The credential is reset by provisionOwner; if the audit were written only
		// after enableLocalAuth, a failed toggle write would leave that reset with no
		// "who did it" row. Order guarantees the accountability row lands first.
		f := &fakeOwnerStore{setErr: errors.New("settings write down")}
		op := breakGlassOp{mode: "recovery", accountable: "root", osUser: "alice", ownerUsername: "owner", attemptedAdmin: "root"}
		if _, err := performBreakGlass(ctx, f, op); err == nil {
			t.Fatal("want error when enableLocalAuth fails")
		}
		if len(f.upserts) != 1 {
			t.Error("owner should have been provisioned before the toggle write failed")
		}
		if len(f.audits) != 1 {
			t.Fatalf("accountability row count = %d, want 1 — the audit must precede enableLocalAuth", len(f.audits))
		}
		if f.audits[0].Actor != "root" || f.audits[0].Action != "break_glass.recovery" {
			t.Errorf("audit = %+v, want actor=root action=break_glass.recovery", f.audits[0])
		}
	})
}

func TestAccountableOSUser(t *testing.T) {
	t.Run("prefers SUDO_USER", func(t *testing.T) {
		t.Setenv("SUDO_USER", "alice")
		if got := accountableOSUser(); got != "alice" {
			t.Errorf("accountableOSUser() = %q, want alice", got)
		}
	})
	t.Run("falls back to root when SUDO_USER is unset", func(t *testing.T) {
		t.Setenv("SUDO_USER", "")
		if got := accountableOSUser(); got != "root" {
			t.Errorf("accountableOSUser() = %q, want root", got)
		}
	})
}

// testRoot is the sanctioned placeholder root domain for tests (never a real host).
const testRoot = "mc.example.net"

// advance feeds one message to the model and returns it re-typed as *bgModel, so the
// state-machine assertions can read the resolved fields. The returned cmd is dropped:
// these tests drive the gating transitions directly (authResultMsg / key presses)
// rather than running the off-goroutine store commands.
func advance(t *testing.T, m *bgModel, msg tea.Msg) *bgModel {
	t.Helper()
	next, _ := m.Update(msg)
	bm, ok := next.(*bgModel)
	if !ok {
		t.Fatalf("Update returned %T, want *bgModel", next)
	}
	return bm
}

// enterProvisionViaMenu drives the top-level router into the Owner provision/reset
// flow the way an operator does on the emergency path: the menu opens with option 1
// (provision) focused, so a single Enter selects it. The gating sub-tests use this to
// reach stepAuth (recovery) or stepProvision (bootstrap) through the REAL entry path
// before asserting the accountability transitions — not by reaching past the menu.
func enterProvisionViaMenu(t *testing.T, m *bgModel) *bgModel {
	t.Helper()
	if m.step != stepMenu {
		t.Fatalf("expected the model to open on stepMenu, got %v", m.step)
	}
	return advance(t, m, tea.KeyMsg{Type: tea.KeyEnter})
}

// TestBGModelGating drives the break-glass state machine headlessly to lock in the
// accountability gate: a credential never advances to provisioning without either a
// verified admin (recovery) or a deliberate, explicit OVERRIDE (root override), and
// the resolved actor matches the path taken. This is the "which SysAdmin" guarantee.
func TestBGModelGating(t *testing.T) {
	ctx := context.Background()

	t.Run("recovery starts at auth; a non-matching credential offers override, never provision", func(t *testing.T) {
		m := newBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "", "", "alice", true)
		m = enterProvisionViaMenu(t, m)
		if m.step != stepAuth || m.mode != "" {
			t.Fatalf("initial step/mode = %v/%q, want stepAuth and an unresolved mode", m.step, m.mode)
		}
		m = advance(t, m, authResultMsg{ok: false})
		if m.step != stepOverride {
			t.Errorf("after a non-matching credential step = %v, want stepOverride", m.step)
		}
		if m.mode == "recovery" {
			t.Error("mode must NOT become recovery on a failed credential — that would forge attribution")
		}
	})

	t.Run("recovery with a verified admin enters provision attributed to that admin", func(t *testing.T) {
		m := newBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "", "", "alice", true)
		m = enterProvisionViaMenu(t, m)
		m = advance(t, m, authResultMsg{matched: "bob", ok: true})
		if m.step != stepProvision {
			t.Fatalf("step = %v, want stepProvision", m.step)
		}
		if m.mode != "recovery" || m.accountable != "bob" {
			t.Errorf("mode/accountable = %q/%q, want recovery/bob (the verified admin, not the OS user)", m.mode, m.accountable)
		}
		// Recovery generates the one-time password, so no password fields are shown.
		if len(m.inputs) != 2 {
			t.Errorf("recovery provision inputs = %d, want 2 (owner, email — no typed password)", len(m.inputs))
		}
	})

	t.Run("an auth lookup error surfaces an error screen, not a silent override", func(t *testing.T) {
		m := newBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "", "", "alice", true)
		m = enterProvisionViaMenu(t, m)
		m = advance(t, m, authResultMsg{err: errors.New("db unreachable")})
		if m.step != stepError || m.err == nil {
			t.Errorf("step/err = %v/%v, want stepError with a non-nil err", m.step, m.err)
		}
	})

	t.Run("the root override requires the exact OVERRIDE token", func(t *testing.T) {
		m := newBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "", "", "alice", true)
		m = enterProvisionViaMenu(t, m)
		m = advance(t, m, authResultMsg{ok: false}) // → stepOverride
		m.inputs[0].SetValue("override")            // wrong case must not pass
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEnter})
		if m.step != stepOverride || m.formErr == "" {
			t.Errorf("wrong token: step/formErr = %v/%q, want stay on stepOverride with an error", m.step, m.formErr)
		}
		if m.mode == "root_override" {
			t.Error("mode must not flip to root_override without the exact token")
		}
		m.inputs[0].SetValue(breakGlassOverrideToken)
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEnter})
		if m.step != stepProvision || m.mode != "root_override" || m.accountable != "alice" {
			t.Errorf("after OVERRIDE: step/mode/accountable = %v/%q/%q, want stepProvision/root_override/alice (the OS user)", m.step, m.mode, m.accountable)
		}
	})

	t.Run("empty admin credentials do not start a verification", func(t *testing.T) {
		m := newBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "", "", "alice", true)
		m = enterProvisionViaMenu(t, m)
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEnter}) // both inputs blank
		if m.step != stepAuth || m.formErr == "" {
			t.Errorf("blank submit: step/formErr = %v/%q, want stay on stepAuth with an error", m.step, m.formErr)
		}
	})

	t.Run("bootstrap starts at provision as the OS user and requires a valid, matching password", func(t *testing.T) {
		f := &fakeOwnerStore{}
		m := newBGModel(ctx, f, testRoot, "", "", "deploybot", false)
		m = enterProvisionViaMenu(t, m)
		if m.step != stepProvision || m.mode != "bootstrap" || m.accountable != "deploybot" {
			t.Fatalf("initial step/mode/accountable = %v/%q/%q, want stepProvision/bootstrap/deploybot", m.step, m.mode, m.accountable)
		}
		if len(m.inputs) != 4 {
			t.Fatalf("bootstrap inputs = %d, want 4 (owner, email, password, confirm)", len(m.inputs))
		}
		// Too-short password is blocked, with no writes.
		m.inputs[2].SetValue("short")
		m.inputs[3].SetValue("short")
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEnter})
		if m.step != stepProvision || m.formErr == "" {
			t.Errorf("weak password: step/formErr = %v/%q, want stay on stepProvision with an error", m.step, m.formErr)
		}
		// A mismatched confirmation is blocked.
		m.inputs[2].SetValue("valid-test-pw")
		m.inputs[3].SetValue("valid-test-XX")
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEnter})
		if m.step != stepProvision || m.formErr == "" {
			t.Errorf("mismatch: step/formErr = %v/%q, want stay on stepProvision with an error", m.step, m.formErr)
		}
		if len(f.upserts) != 0 {
			t.Error("no owner should be provisioned while the form is invalid")
		}
		// Valid + matching advances to the working state (the write is dispatched).
		m.inputs[3].SetValue("valid-test-pw")
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEnter})
		if m.step != stepWorking || m.ownerUsername != "owner" {
			t.Errorf("valid submit: step/owner = %v/%q, want stepWorking/owner", m.step, m.ownerUsername)
		}
	})
}

// TestBGModelEdgeRouting locks in the setup-only Cloudflare edge flow WITHOUT
// touching the operator's real Cloudflare account: setup option 2 reaches the edge
// intro as an independent peer of Owner creation; breakGlass has no edge option;
// hostnames are collected in the setup form; and invalid inputs are refused before
// any cfsetup.Setup side effect. The real cloudflared/cert.pem detection and the
// integration Setup (which shells out / calls the live API) are deliberately NOT exercised.
func TestBGModelEdgeRouting(t *testing.T) {
	ctx := context.Background()

	t.Run("setup menu option 2 enters the edge intro as a peer of provisioning, leaving the Owner credential untouched", func(t *testing.T) {
		f := &fakeOwnerStore{admins: true}
		m := newSetupBGModel(ctx, f, testRoot, "op.console."+testRoot, "console."+testRoot, "alice", true)
		if m.step != stepMenu {
			t.Fatalf("initial step = %v, want stepMenu", m.step)
		}
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("2")})
		if m.step != stepEdgeIntro {
			t.Fatalf("after selecting option 2 step = %v, want stepEdgeIntro", m.step)
		}
		// Reaching the edge must NOT have provisioned or reset an Owner.
		if len(f.upserts) != 0 {
			t.Error("the edge flow must not write any Owner record")
		}
	})

	t.Run("esc from the edge intro returns to the setup router with the edge option highlighted", func(t *testing.T) {
		m := newSetupBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "op.console."+testRoot, "console."+testRoot, "alice", true)
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("2")})
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEsc})
		if m.step != stepMenu || m.focus != 1 {
			t.Errorf("after esc step/focus = %v/%d, want stepMenu with the edge option (1) focused", m.step, m.focus)
		}
	})

	t.Run("breakGlass has no edge option 2", func(t *testing.T) {
		m := newBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "op.console."+testRoot, "console."+testRoot, "alice", true)
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("2")})
		if m.step != stepMenu {
			t.Fatalf("breakGlass option 2 advanced to %v, want to stay on stepMenu", m.step)
		}
	})

	t.Run("submitEdge refuses empty credentials before any Cloudflare side effect", func(t *testing.T) {
		m := newSetupBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "op.console."+testRoot, "console."+testRoot, "alice", true)
		// Install the edge inputs directly: reaching them via the menu requires a real
		// cloudflared login (edgeReady()), which this unit test must not depend on.
		m.enterEdgeInput()
		if m.step != stepEdgeInput || len(m.inputs) != 7 {
			t.Fatalf("enterEdgeInput: step/inputs = %v/%d, want stepEdgeInput with 7 inputs", m.step, len(m.inputs))
		}
		// All inputs blank: submit (via the real key path) must report an error and stay
		// put — NOT reach stepEdgeWorking, which is what launches cfsetup.Setup against
		// the live cloudflared binary / Cloudflare API.
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEnter})
		if m.step != stepEdgeInput || m.formErr == "" {
			t.Errorf("blank edge submit: step/formErr = %v/%q, want stay on stepEdgeInput with an error", m.step, m.formErr)
		}
		if m.step == stepEdgeWorking {
			t.Error("empty credentials must never reach stepEdgeWorking — that would invoke the integration runner")
		}
	})

	t.Run("submitEdge rejects a bare @ identity that would scope Access to an empty domain", func(t *testing.T) {
		m := newSetupBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "op.console."+testRoot, "console."+testRoot, "alice", true)
		m.enterEdgeInput()
		// Token + account present, but identity is a bare "@" (empty domain). This passes
		// the non-empty check yet must be refused before cfsetup.Setup, because an empty
		// EmailDomain admits no one — a silent lock-out the operator should fix.
		m.inputs[0].SetValue("token-value")
		m.inputs[1].SetValue("1234567890abcdef1234567890abcdef")
		m.inputs[2].SetValue("@")
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEnter})
		if m.step != stepEdgeInput || m.formErr == "" {
			t.Errorf("bare @ submit: step/formErr = %v/%q, want stay on stepEdgeInput with an error", m.step, m.formErr)
		}
		if m.step == stepEdgeWorking {
			t.Error("a bare @ identity must never reach stepEdgeWorking — that would invoke the integration runner")
		}
	})

	t.Run("submitEdge requires an admin hostname and rejects URLs", func(t *testing.T) {
		m := newSetupBGModel(ctx, &fakeOwnerStore{admins: true}, testRoot, "", "", "alice", true)
		m.enterEdgeInput()
		m.inputs[0].SetValue("token-value")
		m.inputs[1].SetValue("1234567890abcdef1234567890abcdef")
		m.inputs[2].SetValue("[email protected]")
		m.inputs[3].SetValue("https://console." + testRoot)
		m.inputs[4].SetValue("")
		m = advance(t, m, tea.KeyMsg{Type: tea.KeyEnter})
		if m.step != stepEdgeInput || m.formErr == "" {
			t.Errorf("bad hostnames: step/formErr = %v/%q, want stay on stepEdgeInput with an error", m.step, m.formErr)
		}
		if m.step == stepEdgeWorking {
			t.Error("invalid hostnames must never reach stepEdgeWorking")
		}
	})
}