201 lines
10 KiB
Go
201 lines
10 KiB
Go
package tui
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
"testing"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
|
|
"proxyctl/internal/app"
|
|
"proxyctl/internal/profile"
|
|
"proxyctl/internal/target"
|
|
)
|
|
|
|
type targetModule struct {
|
|
snapshot app.Snapshot
|
|
result app.Result
|
|
intents []app.Intent
|
|
}
|
|
|
|
func (m *targetModule) Snapshot() app.Snapshot { return m.snapshot }
|
|
func (m *targetModule) Dispatch(intent app.Intent) app.Result {
|
|
m.intents = append(m.intents, intent)
|
|
m.snapshot = m.result.Snapshot
|
|
return m.result
|
|
}
|
|
|
|
func TestTargetPanelShowsOnlyAvailableTargetsAndAppliesAsynchronously(t *testing.T) {
|
|
module := &targetModule{snapshot: app.Snapshot{ActiveProfile: "office", Targets: []app.TargetSnapshot{
|
|
{ID: "shell", Description: target.Description{Name: "Shell", Summary: "Exports for Bash/Zsh"}, Available: true, State: target.External, Evidence: "existing proxy"},
|
|
{ID: "apt", Description: target.Description{Name: "APT"}, Available: false},
|
|
}}}
|
|
module.result = app.Result{Snapshot: app.Snapshot{ActiveProfile: "office", Targets: []app.TargetSnapshot{{ID: "shell", Description: target.Description{Name: "Shell", Summary: "Exports for Bash/Zsh"}, Available: true, State: target.Enabled, Evidence: "matches"}}}, Notice: "Shell Target is now Enabled"}
|
|
model := NewModel(module)
|
|
view := model.View()
|
|
if !strings.Contains(view, "Shell") || strings.Contains(view, "APT") || !strings.Contains(view, "Exports for Bash/Zsh") {
|
|
t.Fatalf("Target panel:\n%s", view)
|
|
}
|
|
|
|
updated, command := model.Update(tea.KeyMsg{Type: tea.KeySpace})
|
|
model = updated.(Model)
|
|
if command == nil || !strings.Contains(model.View(), "Applying") || len(module.intents) != 0 {
|
|
t.Fatalf("operation did not expose applying state before effect:\n%s", model.View())
|
|
}
|
|
message := command()
|
|
updated, _ = model.Update(message)
|
|
model = updated.(Model)
|
|
if len(module.intents) != 1 || module.intents[0].Kind != app.SetTargetDesired || !module.intents[0].Enabled || !strings.Contains(model.View(), "Enabled") {
|
|
t.Fatalf("completed toggle = %#v\n%s", module.intents, model.View())
|
|
}
|
|
}
|
|
|
|
func TestTargetRestartConfirmationCancelsWithoutConfirmedMutation(t *testing.T) {
|
|
initial := app.Snapshot{ActiveProfile: "office", Targets: []app.TargetSnapshot{{ID: "docker-daemon", Description: target.Description{Name: "Docker daemon"}, Available: true, State: target.Disabled}}}
|
|
module := &targetModule{snapshot: initial, result: app.Result{Snapshot: initial, Confirmation: "Docker will restart immediately and may disrupt running work."}}
|
|
model := NewModel(module)
|
|
updated, command := model.Update(tea.KeyMsg{Type: tea.KeySpace})
|
|
model = updated.(Model)
|
|
updated, _ = model.Update(command())
|
|
model = updated.(Model)
|
|
if !strings.Contains(model.View(), "restart immediately") || len(module.intents) != 1 || module.intents[0].Confirm {
|
|
t.Fatalf("warning = intents %#v\n%s", module.intents, model.View())
|
|
}
|
|
updated, command = model.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{'n'}})
|
|
model = updated.(Model)
|
|
if command != nil || len(module.intents) != 1 || strings.Contains(model.View(), "restart immediately") {
|
|
t.Fatalf("cancel mutated = intents %#v\n%s", module.intents, model.View())
|
|
}
|
|
|
|
updated, command = model.Update(tea.KeyMsg{Type: tea.KeySpace})
|
|
model = updated.(Model)
|
|
updated, _ = model.Update(command())
|
|
model = updated.(Model)
|
|
module.result = app.Result{
|
|
Snapshot: app.Snapshot{ActiveProfile: "office", Targets: []app.TargetSnapshot{{ID: "docker-daemon", Description: target.Description{Name: "Docker daemon"}, Available: true, State: target.Enabled}}},
|
|
Notice: "Docker daemon Target is now Enabled",
|
|
}
|
|
updated, command = model.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{'y'}})
|
|
model = updated.(Model)
|
|
if command == nil {
|
|
t.Fatal("confirmation did not start operation")
|
|
}
|
|
_ = command()
|
|
if len(module.intents) != 3 || !module.intents[2].Confirm {
|
|
t.Fatalf("confirmed intents = %#v", module.intents)
|
|
}
|
|
}
|
|
|
|
func TestHelpListsEveryTargetAndMainListsOnlyAvailableTargets(t *testing.T) {
|
|
module := &targetModule{snapshot: app.Snapshot{Targets: []app.TargetSnapshot{
|
|
{ID: "shell", Description: target.Description{Name: "Shell", Summary: "summary", Details: "affects future Shell commands; does not affect daemons; owns a generated fragment; user permission; reload side effect"}, Available: true, State: target.Disabled, Evidence: "integration installed"},
|
|
{ID: "apt", Description: target.Description{Name: "APT", Summary: "persistent package proxy", Details: "owns an overlay and requires system write permission"}, Available: false, AvailabilityEvidence: "Debian APT tool apt-config was not detected"},
|
|
}}}
|
|
model := NewModel(module)
|
|
view := model.View()
|
|
if strings.Contains(view, "APT") || !strings.Contains(view, "affects future Shell commands") || !strings.Contains(view, "integration installed") {
|
|
t.Fatalf("main Target panel:\n%s", view)
|
|
}
|
|
model = updateKey(model, "?")
|
|
view = model.View()
|
|
if !strings.Contains(view, "Shell") || !strings.Contains(view, "APT") || !strings.Contains(view, "Not detected: Debian APT tool apt-config was not detected") || !strings.Contains(view, "owns an overlay") {
|
|
t.Fatalf("Target help:\n%s", view)
|
|
}
|
|
model = updateKey(model, "esc")
|
|
if !strings.Contains(model.View(), "Target panel") {
|
|
t.Fatalf("Escape did not return from help:\n%s", model.View())
|
|
}
|
|
}
|
|
|
|
func TestRefreshKeyDispatchesDetectionRefresh(t *testing.T) {
|
|
module := &targetModule{snapshot: app.Snapshot{}, result: app.Result{Snapshot: app.Snapshot{Targets: []app.TargetSnapshot{{ID: "apt", Description: target.Description{Name: "APT"}, Available: true, State: target.External}}}}}
|
|
model := NewModel(module)
|
|
model = updateKey(model, "r")
|
|
if len(module.intents) != 1 || module.intents[0].Kind != app.RefreshTargets || !strings.Contains(model.View(), "External") {
|
|
t.Fatalf("refresh = intents %#v\n%s", module.intents, model.View())
|
|
}
|
|
}
|
|
|
|
type progressiveTargetModule struct {
|
|
targetModule
|
|
updates chan app.Result
|
|
}
|
|
|
|
func (m *progressiveTargetModule) DispatchProgress(intent app.Intent) <-chan app.Result {
|
|
m.intents = append(m.intents, intent)
|
|
return m.updates
|
|
}
|
|
|
|
func TestActiveProfileTransitionRendersPerTargetProgress(t *testing.T) {
|
|
initial := app.Snapshot{ActiveProfile: "old", Profiles: []profile.Profile{{Name: "new"}}, Targets: []app.TargetSnapshot{{ID: "shell", Description: target.Description{Name: "Shell"}, Available: true, State: target.Enabled}}}
|
|
updates := make(chan app.Result, 3)
|
|
updates <- app.Result{Snapshot: app.Snapshot{ActiveProfile: "new", Targets: initial.Targets}, Progress: []app.TargetProgress{{TargetID: "shell", Applying: true}}}
|
|
updates <- app.Result{Snapshot: app.Snapshot{ActiveProfile: "new", Targets: []app.TargetSnapshot{{ID: "shell", Description: target.Description{Name: "Shell"}, Available: true, State: target.Enabled}}}, Progress: []app.TargetProgress{{TargetID: "shell", State: target.Enabled}}, Notice: "Shell Target update completed"}
|
|
updates <- app.Result{Snapshot: app.Snapshot{ActiveProfile: "new", Targets: []app.TargetSnapshot{{ID: "shell", Description: target.Description{Name: "Shell"}, Available: true, State: target.Enabled}}}, Complete: true, Notice: "Activated Profile new"}
|
|
close(updates)
|
|
module := &progressiveTargetModule{targetModule: targetModule{snapshot: initial}, updates: updates}
|
|
model := updateKey(NewModel(module), "p")
|
|
updated, command := model.Update(tea.KeyMsg{Type: tea.KeyEnter})
|
|
model = updated.(Model)
|
|
if command == nil {
|
|
t.Fatal("activation did not start progressive operation")
|
|
}
|
|
updated, command = model.Update(command())
|
|
model = updated.(Model)
|
|
if command == nil || !strings.Contains(model.View(), "Applying") || !strings.Contains(model.View(), "Active Profile: new") {
|
|
t.Fatalf("applying progress:\n%s", model.View())
|
|
}
|
|
updated, command = model.Update(command())
|
|
model = updated.(Model)
|
|
if command == nil || !strings.Contains(model.View(), "Enabled") || !strings.Contains(model.View(), "update completed") {
|
|
t.Fatalf("Target completion:\n%s", model.View())
|
|
}
|
|
updated, _ = model.Update(command())
|
|
model = updated.(Model)
|
|
if !strings.Contains(model.View(), "Activated Profile new") {
|
|
t.Fatalf("final transition:\n%s", model.View())
|
|
}
|
|
}
|
|
|
|
func TestTargetPanelRendersEveryTargetState(t *testing.T) {
|
|
states := []target.State{target.Enabled, target.Disabled, target.External, target.Drifted}
|
|
var targets []app.TargetSnapshot
|
|
for _, state := range states {
|
|
targets = append(targets, app.TargetSnapshot{ID: string(state), Description: target.Description{Name: string(state)}, Available: true, State: state})
|
|
}
|
|
view := NewModel(&targetModule{snapshot: app.Snapshot{Targets: targets}}).View()
|
|
for _, state := range states {
|
|
if !strings.Contains(view, string(state)) {
|
|
t.Fatalf("missing state %s:\n%s", state, view)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestDriftedTargetWithoutActiveProfileTogglesTowardDisabled(t *testing.T) {
|
|
initial := app.Snapshot{Targets: []app.TargetSnapshot{{ID: "shell", Description: target.Description{Name: "Shell"}, Available: true, State: target.Drifted}}}
|
|
module := &targetModule{snapshot: initial, result: app.Result{Snapshot: app.Snapshot{Targets: []app.TargetSnapshot{{ID: "shell", Description: target.Description{Name: "Shell"}, Available: true, State: target.Disabled}}}}}
|
|
model := NewModel(module)
|
|
_, command := model.Update(tea.KeyMsg{Type: tea.KeySpace})
|
|
if command == nil {
|
|
t.Fatal("Drifted Target did not start disable")
|
|
}
|
|
_ = command()
|
|
if len(module.intents) != 1 || module.intents[0].Enabled {
|
|
t.Fatalf("Drifted Target intent = %#v", module.intents)
|
|
}
|
|
}
|
|
|
|
func TestTargetFailurePreservesTruthfulStateAndRendersRedactedError(t *testing.T) {
|
|
initial := app.Snapshot{ActiveProfile: "office", Targets: []app.TargetSnapshot{{ID: "shell", Description: target.Description{Name: "Shell"}, Available: true, State: target.Enabled}}}
|
|
module := &targetModule{snapshot: initial, result: app.Result{Snapshot: initial, Err: errors.New("permission denied for alice:REDACTED@proxy")}}
|
|
model := NewModel(module)
|
|
updated, command := model.Update(tea.KeyMsg{Type: tea.KeySpace})
|
|
model = updated.(Model)
|
|
updated, _ = model.Update(command())
|
|
model = updated.(Model)
|
|
view := model.View()
|
|
if !strings.Contains(view, "Enabled") || !strings.Contains(view, "\x1b[31mError:") || !strings.Contains(view, "REDACTED") {
|
|
t.Fatalf("failed operation rendering:\n%s", view)
|
|
}
|
|
}
|