Catacombs/ui/model_test.go

package ui

import (
	"os"
	"testing"

	tea "github.com/charmbracelet/bubbletea"
	"github.com/tolelom/catacombs/config"
	"github.com/tolelom/catacombs/game"
	"github.com/tolelom/catacombs/store"
)

func testDB(t *testing.T) *store.DB {
	db, err := store.Open("test_ui.db")
	if err != nil {
		t.Fatal(err)
	}
	return db
}

func TestTitleToLobby(t *testing.T) {
	lobby := game.NewLobby(func() *config.Config { c, _ := config.Load(""); return c }())
	db := testDB(t)
	defer func() { db.Close(); os.Remove("test_ui.db") }()

	m := NewModel(80, 24, "testfp", lobby, db)

	if m.screenType() != screenTitle {
		t.Fatalf("initial screen: got %d, want screenTitle(0)", m.screenType())
	}

	// First-time player: Enter goes to nickname screen
	result, _ := m.Update(tea.KeyMsg{Type: tea.KeyEnter})
	m2 := result.(Model)

	if m2.screenType() != screenNickname {
		t.Errorf("after Enter (first time): screen=%d, want screenNickname(%d)", m2.screenType(), screenNickname)
	}

	// Type a name
	for _, ch := range []rune("Hero") {
		result, _ = m2.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{ch}})
		m2 = result.(Model)
	}

	// Confirm nickname
	result, _ = m2.Update(tea.KeyMsg{Type: tea.KeyEnter})
	m3 := result.(Model)

	if m3.screenType() != screenLobby {
		t.Errorf("after nickname Enter: screen=%d, want screenLobby(1)", m3.screenType())
	}
	if m3.playerName() == "" {
		t.Error("playerName should be set")
	}
}

func TestLobbyCreateRoom(t *testing.T) {
	lobby := game.NewLobby(func() *config.Config { c, _ := config.Load(""); return c }())
	db := testDB(t)
	defer func() { db.Close(); os.Remove("test_ui.db") }()

	m := NewModel(80, 24, "testfp2", lobby, db)

	// Go to nickname screen (first-time player)
	result, _ := m.Update(tea.KeyMsg{Type: tea.KeyEnter})
	m2 := result.(Model)

	// Type name and confirm
	for _, ch := range []rune("Hero") {
		result, _ = m2.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{ch}})
		m2 = result.(Model)
	}
	result, _ = m2.Update(tea.KeyMsg{Type: tea.KeyEnter})
	m2 = result.(Model)

	// Press 'c' to create room
	result, _ = m2.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{'c'}})
	m3 := result.(Model)

	if m3.screenType() != screenClassSelect {
		t.Errorf("after 'c': screen=%d, want screenClassSelect(2)", m3.screenType())
	}
	if m3.roomCode() == "" {
		t.Error("roomCode should be set")
	}
}

func TestClassSelectToGame(t *testing.T) {
	lobby := game.NewLobby(func() *config.Config { c, _ := config.Load(""); return c }())
	db := testDB(t)
	defer func() { db.Close(); os.Remove("test_ui.db") }()

	m := NewModel(80, 24, "testfp3", lobby, db)

	// Title -> Nickname -> Lobby
	result, _ := m.Update(tea.KeyMsg{Type: tea.KeyEnter})
	m2 := result.(Model)
	for _, ch := range []rune("Hero") {
		result, _ = m2.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{ch}})
		m2 = result.(Model)
	}
	result, _ = m2.Update(tea.KeyMsg{Type: tea.KeyEnter})
	m2 = result.(Model)

	// Lobby -> Class Select
	result, _ = m2.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{'c'}})
	m3 := result.(Model)

	if m3.screenType() != screenClassSelect {
		t.Fatalf("should be at class select, got %d", m3.screenType())
	}

	// Press Enter to select Warrior (default cursor=0) → WaitingScreen
	result, _ = m3.Update(tea.KeyMsg{Type: tea.KeyEnter})
	m4 := result.(Model)

	if m4.screenType() != screenWaiting {
		t.Fatalf("after class select Enter: screen=%d, want screenWaiting(%d)", m4.screenType(), screenWaiting)
	}

	// Press Enter to ready up (solo room → immediately starts game)
	result, _ = m4.Update(tea.KeyMsg{Type: tea.KeyEnter})
	m5 := result.(Model)

	if m5.screenType() != screenGame {
		t.Errorf("after ready Enter: screen=%d, want screenGame(%d)", m5.screenType(), screenGame)
	}
	if m5.session() == nil {
		t.Error("session should be set")
	}
}

func TestKeyHelpers(t *testing.T) {
	enter := tea.KeyMsg{Type: tea.KeyEnter}
	if !isEnter(enter) {
		t.Error("isEnter should match KeyEnter type")
	}

	enterStr := tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{'\r'}}
	_ = enterStr // might not match, that's ok

	up := tea.KeyMsg{Type: tea.KeyUp}
	if !isUp(up) {
		t.Error("isUp should match KeyUp type")
	}

	q := tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{'q'}}
	if !isQuit(q) {
		t.Error("isQuit should match 'q'")
	}
}