sshkeeper/internal/tui/confirm_test.go

package tui

import (
	"errors"
	"strings"
	"testing"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/mirivlad/sshkeeper/internal/model"
)

func TestServerDeleteRequiresExplicitConfirmation(t *testing.T) {
	server := &model.Server{ID: 1, Alias: "prod", Host: "prod.example", Port: 22, User: "root", AuthMethod: model.AuthKey}
	m := New([]*model.Server{server})
	deleted := 0
	oldDelete, oldList := DeleteServer, ListServers
	t.Cleanup(func() { DeleteServer, ListServers = oldDelete, oldList })
	DeleteServer = func(alias string) error {
		deleted++
		return nil
	}
	ListServers = func() ([]*model.Server, error) { return nil, nil }

	updated, cmd := m.updateList(tea.KeyMsg{Type: tea.KeyCtrlD})
	m = updated.(*tuiModel)
	if cmd != nil || deleted != 0 {
		t.Fatalf("delete ran before confirmation: cmd=%v deleted=%d", cmd != nil, deleted)
	}
	if m.screen != screenConfirm || m.confirm == nil {
		t.Fatalf("expected confirmation screen, got screen=%v confirm=%v", m.screen, m.confirm)
	}
	if m.confirm.focus != confirmCancel {
		t.Fatalf("default focus = %v, want Cancel", m.confirm.focus)
	}
	view := m.View()
	for _, want := range []string{"prod", "saved port forwards", "vault secrets", "> [ Cancel ]", "[ Delete ]"} {
		if !strings.Contains(view, want) {
			t.Fatalf("confirmation missing %q:\n%s", want, view)
		}
	}

	updated, cmd = m.updateConfirm(tea.KeyMsg{Type: tea.KeyEnter})
	m = updated.(*tuiModel)
	if cmd != nil || deleted != 0 || m.screen != screenList {
		t.Fatalf("Enter on default Cancel must cancel: screen=%v cmd=%v deleted=%d", m.screen, cmd != nil, deleted)
	}
}

func TestServerDeleteRunsOnceAndReturnsToList(t *testing.T) {
	server := &model.Server{ID: 1, Alias: "prod", Host: "prod.example", Port: 22, User: "root", AuthMethod: model.AuthKey}
	m := New([]*model.Server{server})
	deleted := 0
	oldDelete, oldList := DeleteServer, ListServers
	t.Cleanup(func() { DeleteServer, ListServers = oldDelete, oldList })
	DeleteServer = func(alias string) error {
		deleted++
		return nil
	}
	ListServers = func() ([]*model.Server, error) { return nil, nil }

	updated, _ := m.updateList(tea.KeyMsg{Type: tea.KeyCtrlD})
	m = updated.(*tuiModel)
	updated, _ = m.updateConfirm(tea.KeyMsg{Type: tea.KeyTab})
	m = updated.(*tuiModel)
	updated, cmd := m.updateConfirm(tea.KeyMsg{Type: tea.KeyEnter})
	m = updated.(*tuiModel)
	if cmd == nil || m.confirm == nil || !m.confirm.pending {
		t.Fatal("expected pending destructive command")
	}
	_, duplicate := m.updateConfirm(tea.KeyMsg{Type: tea.KeyEnter})
	if duplicate != nil {
		t.Fatal("repeated Enter must not start a duplicate delete")
	}
	msg := cmd()
	updated, _ = m.Update(msg)
	m = updated.(*tuiModel)
	if deleted != 1 {
		t.Fatalf("delete calls = %d, want 1", deleted)
	}
	if m.screen != screenList || m.confirm != nil || len(m.servers) != 0 {
		t.Fatalf("unexpected completion state: screen=%v confirm=%v servers=%d", m.screen, m.confirm, len(m.servers))
	}
}

func TestForwardDeleteReturnsToForwardListAndRetainsError(t *testing.T) {
	server := &model.Server{ID: 1, Alias: "prod"}
	fwd := &model.Forward{ID: 7, ServerID: 1, Name: "postgres", Type: model.ForwardLocal, LocalAddr: "127.0.0.1", LocalPort: 15432, RemoteAddr: "db", RemotePort: 5432, Enabled: true}
	m := New([]*model.Server{server})
	m.screen = screenForwardList
	m.forwardScreen = newForwardScreenModel(server.ID, server.Alias, 80, 24)
	m.forwardScreen.list = []*model.Forward{fwd}
	m.forwardScreen.selected = 0

	oldDelete := DeleteForward
	t.Cleanup(func() { DeleteForward = oldDelete })
	DeleteForward = func(id int64) error { return errors.New("database is read-only") }

	updated, _ := m.updateForwardList(tea.KeyMsg{Type: tea.KeyCtrlD})
	m = updated.(*tuiModel)
	if m.confirm == nil || m.confirm.parent != screenForwardList {
		t.Fatalf("expected forward-list parent, confirm=%#v", m.confirm)
	}
	updated, _ = m.updateConfirm(tea.KeyMsg{Type: tea.KeyTab})
	m = updated.(*tuiModel)
	updated, cmd := m.updateConfirm(tea.KeyMsg{Type: tea.KeyEnter})
	m = updated.(*tuiModel)
	if cmd == nil {
		t.Fatal("expected delete command")
	}
	updated, _ = m.Update(cmd())
	m = updated.(*tuiModel)
	if m.screen != screenForwardList || m.confirm != nil {
		t.Fatalf("expected return to forwards, screen=%v confirm=%v", m.screen, m.confirm)
	}
	if m.forwardScreen == nil || m.forwardScreen.err == nil || !strings.Contains(m.forwardScreen.View(), "database is read-only") {
		t.Fatalf("forward error not retained: model=%#v view=%q", m.forwardScreen, m.forwardScreen.View())
	}
}

func TestOtherDestructiveActionsOpenConfirmation(t *testing.T) {
	t.Run("tag", func(t *testing.T) {
		m := New(nil)
		m.screen = screenTags
		m.setTags([]string{"prod"})
		updated, cmd := m.updateTags(tea.KeyMsg{Type: tea.KeyCtrlD})
		m = updated.(*tuiModel)
		if cmd != nil || m.screen != screenConfirm || m.confirm == nil || m.confirm.parent != screenTags {
			t.Fatalf("tag deletion did not open confirmation: screen=%v confirm=%#v cmd=%v", m.screen, m.confirm, cmd != nil)
		}
	})

	t.Run("template", func(t *testing.T) {
		m := New(nil)
		m.screen = screenTemplates
		m.setTemplates([]*model.CommandTemplate{{Name: "uptime", Command: "uptime"}})
		updated, cmd := m.updateTemplates(tea.KeyMsg{Type: tea.KeyCtrlD})
		m = updated.(*tuiModel)
		if cmd != nil || m.screen != screenConfirm || m.confirm == nil || m.confirm.parent != screenTemplates {
			t.Fatalf("template deletion did not open confirmation: screen=%v confirm=%#v cmd=%v", m.screen, m.confirm, cmd != nil)
		}
	})

	t.Run("tunnel", func(t *testing.T) {
		m := New(nil)
		m.screen = screenTunnelManager
		m.tunnelScreen = newTunnelScreenModel(80, 24)
		m.tunnelScreen.tunnels = []*model.TunnelState{{ID: 11, Name: "prod tunnel", ServerAlias: "prod"}}
		m.tunnelScreen.rebuildList()
		updated, cmd := m.updateTunnelManager(tea.KeyMsg{Type: tea.KeyCtrlD})
		m = updated.(*tuiModel)
		if cmd != nil || m.screen != screenConfirm || m.confirm == nil || m.confirm.parent != screenTunnelManager {
			t.Fatalf("tunnel stop did not open confirmation: screen=%v confirm=%#v cmd=%v", m.screen, m.confirm, cmd != nil)
		}
	})
}