feat: implement load and save handlers with corresponding tests
This commit is contained in:
@@ -0,0 +1,48 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"slices"
|
||||||
|
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/core"
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/state"
|
||||||
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
|
)
|
||||||
|
|
||||||
|
type loadHandler struct {
|
||||||
|
keys []string
|
||||||
|
description string
|
||||||
|
saver core.Saver
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewLoadHandler(saver core.Saver) KeyMsgHandler {
|
||||||
|
return loadHandler{
|
||||||
|
keys: []string{"ctrl+o"},
|
||||||
|
description: "Load tiers",
|
||||||
|
saver: saver,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h loadHandler) KeysHandled() []string {
|
||||||
|
return h.keys
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h loadHandler) Description() string {
|
||||||
|
return h.description
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h loadHandler) Match(s state.State, msg tea.KeyMsg) bool {
|
||||||
|
return s.Mode() == state.NavigationMode && slices.Contains(h.keys, msg.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h loadHandler) Handle(s state.State, msg tea.KeyMsg) tea.Cmd {
|
||||||
|
return s.StartInputMode("Load tiers", func(val string, s state.State) tea.Cmd {
|
||||||
|
if data, err := h.saver.Load(val); err != nil {
|
||||||
|
s.SetStatus("Error: " + err.Error())
|
||||||
|
} else {
|
||||||
|
s.Load(data)
|
||||||
|
s.SetStatus("Loaded " + val)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
@@ -0,0 +1,141 @@
|
|||||||
|
package handlers_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"slices"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/core"
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/handlers"
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/state"
|
||||||
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNewLoadHandler(t *testing.T) {
|
||||||
|
saver := &mockSaver{}
|
||||||
|
|
||||||
|
h := handlers.NewLoadHandler(saver)
|
||||||
|
|
||||||
|
if h.Description() != "Load tiers" {
|
||||||
|
t.Errorf("Expected description 'Load tiers', got %q", h.Description())
|
||||||
|
}
|
||||||
|
|
||||||
|
keys := h.KeysHandled()
|
||||||
|
|
||||||
|
if !slices.Contains(keys, "ctrl+o") {
|
||||||
|
t.Errorf("Expected keys ['ctrl+o'], got %v", keys)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadHandler_Match(t *testing.T) {
|
||||||
|
saver := &mockSaver{}
|
||||||
|
|
||||||
|
h := handlers.NewLoadHandler(saver)
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
mode state.Mode
|
||||||
|
msg tea.KeyMsg
|
||||||
|
expected bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Matches ctrl+o in NavigationMode",
|
||||||
|
mode: state.NavigationMode,
|
||||||
|
msg: tea.KeyMsg{Type: tea.KeyCtrlO},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Ignores ctrl+o in InputMode",
|
||||||
|
mode: state.InputMode,
|
||||||
|
msg: tea.KeyMsg{Type: tea.KeyCtrlO},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Ignores other keys",
|
||||||
|
mode: state.NavigationMode,
|
||||||
|
msg: tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("a")},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
s := state.New(state.WithMode(tt.mode))
|
||||||
|
if got := h.Match(s, tt.msg); got != tt.expected {
|
||||||
|
t.Errorf("Match() = %v, want %v", got, tt.expected)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoadHandler_Handle(t *testing.T) {
|
||||||
|
t.Run("Successfully loads data", func(t *testing.T) {
|
||||||
|
mockData := core.SaveState{
|
||||||
|
Tiers: []struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Items []core.Item `json:"items"`
|
||||||
|
}{
|
||||||
|
{Name: "Tier 1", Items: []core.Item{"Loaded Item"}},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
saver := &mockSaver{dataToReturn: mockData}
|
||||||
|
h := handlers.NewLoadHandler(saver)
|
||||||
|
|
||||||
|
initialTiers := []core.Tier{{Name: "Tier 1", Items: []core.Item{}}}
|
||||||
|
|
||||||
|
s := state.New(
|
||||||
|
state.WithMode(state.NavigationMode),
|
||||||
|
state.WithTiers(initialTiers),
|
||||||
|
)
|
||||||
|
|
||||||
|
msg := tea.KeyMsg{Type: tea.KeyCtrlO}
|
||||||
|
h.Handle(s, msg)
|
||||||
|
|
||||||
|
if s.Mode() != state.InputMode {
|
||||||
|
t.Fatal("Expected state to switch to InputMode")
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, r := range "savefile.json" {
|
||||||
|
s.UpdateInput(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{r}})
|
||||||
|
}
|
||||||
|
|
||||||
|
s.SubmitInput()
|
||||||
|
|
||||||
|
if s.Mode() != state.NavigationMode {
|
||||||
|
t.Error("Expected to return to NavigationMode after loading")
|
||||||
|
}
|
||||||
|
|
||||||
|
expectedStatus := "Loaded savefile.json"
|
||||||
|
|
||||||
|
if s.StatusMsg() != expectedStatus {
|
||||||
|
t.Errorf("Expected status %q, got %q", expectedStatus, s.StatusMsg())
|
||||||
|
}
|
||||||
|
|
||||||
|
currentItems := s.Tiers()[0].Items
|
||||||
|
|
||||||
|
if len(currentItems) != 1 || currentItems[0] != "Loaded Item" {
|
||||||
|
t.Errorf("Expected loaded items ['Loaded Item'], got %v", currentItems)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("Handles load errors gracefully", func(t *testing.T) {
|
||||||
|
saver := &mockSaver{shouldError: true}
|
||||||
|
|
||||||
|
h := handlers.NewLoadHandler(saver)
|
||||||
|
s := state.New(state.WithMode(state.NavigationMode))
|
||||||
|
|
||||||
|
h.Handle(s, tea.KeyMsg{Type: tea.KeyCtrlO})
|
||||||
|
|
||||||
|
for _, r := range "missing.json" {
|
||||||
|
s.UpdateInput(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{r}})
|
||||||
|
}
|
||||||
|
|
||||||
|
s.SubmitInput()
|
||||||
|
|
||||||
|
expectedPrefix := "Error:"
|
||||||
|
|
||||||
|
if len(s.StatusMsg()) < len(expectedPrefix) || s.StatusMsg()[:len(expectedPrefix)] != expectedPrefix {
|
||||||
|
t.Errorf("Expected error status, got %q", s.StatusMsg())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
@@ -0,0 +1,47 @@
|
|||||||
|
package handlers
|
||||||
|
|
||||||
|
import (
|
||||||
|
"slices"
|
||||||
|
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/core"
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/state"
|
||||||
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
|
)
|
||||||
|
|
||||||
|
type saveHandler struct {
|
||||||
|
keys []string
|
||||||
|
description string
|
||||||
|
saver core.Saver
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewSaveHandler(saver core.Saver) KeyMsgHandler {
|
||||||
|
return saveHandler{
|
||||||
|
keys: []string{"ctrl+s"},
|
||||||
|
description: "Save tiers",
|
||||||
|
saver: saver,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h saveHandler) KeysHandled() []string {
|
||||||
|
return h.keys
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h saveHandler) Description() string {
|
||||||
|
return h.description
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h saveHandler) Match(s state.State, msg tea.KeyMsg) bool {
|
||||||
|
return s.Mode() == state.NavigationMode && slices.Contains(h.keys, msg.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h saveHandler) Handle(s state.State, msg tea.KeyMsg) tea.Cmd {
|
||||||
|
return s.StartInputMode("Save tiers", func(val string, s state.State) tea.Cmd {
|
||||||
|
if err := h.saver.Save(val, s.Save()); err != nil {
|
||||||
|
s.SetStatus("Error: " + err.Error())
|
||||||
|
} else {
|
||||||
|
s.SetStatus("Saved to " + val)
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
}
|
||||||
@@ -0,0 +1,132 @@
|
|||||||
|
package handlers_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"slices"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/core"
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/handlers"
|
||||||
|
"github.com/StevanFreeborn/term-tier/internal/state"
|
||||||
|
tea "github.com/charmbracelet/bubbletea"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNewSaveHandler(t *testing.T) {
|
||||||
|
saver := &mockSaver{}
|
||||||
|
|
||||||
|
h := handlers.NewSaveHandler(saver)
|
||||||
|
|
||||||
|
if h.Description() != "Save tiers" {
|
||||||
|
t.Errorf("Expected description 'Save tiers', got %q", h.Description())
|
||||||
|
}
|
||||||
|
|
||||||
|
keys := h.KeysHandled()
|
||||||
|
|
||||||
|
if !slices.Contains(keys, "ctrl+s") {
|
||||||
|
t.Errorf("Expected keys ['ctrl+s'], got %v", keys)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSaveHandler_Match(t *testing.T) {
|
||||||
|
saver := &mockSaver{}
|
||||||
|
h := handlers.NewSaveHandler(saver)
|
||||||
|
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
mode state.Mode
|
||||||
|
msg tea.KeyMsg
|
||||||
|
expected bool
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Matches ctrl+s in NavigationMode",
|
||||||
|
mode: state.NavigationMode,
|
||||||
|
msg: tea.KeyMsg{Type: tea.KeyCtrlS},
|
||||||
|
expected: true,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Ignores ctrl+s in InputMode",
|
||||||
|
mode: state.InputMode,
|
||||||
|
msg: tea.KeyMsg{Type: tea.KeyCtrlS},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Ignores other keys",
|
||||||
|
mode: state.NavigationMode,
|
||||||
|
msg: tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("s")},
|
||||||
|
expected: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
s := state.New(state.WithMode(tt.mode))
|
||||||
|
if got := h.Match(s, tt.msg); got != tt.expected {
|
||||||
|
t.Errorf("Match() = %v, want %v", got, tt.expected)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSaveHandler_Handle(t *testing.T) {
|
||||||
|
t.Run("Successfully saves data", func(t *testing.T) {
|
||||||
|
saver := &mockSaver{}
|
||||||
|
|
||||||
|
h := handlers.NewSaveHandler(saver)
|
||||||
|
|
||||||
|
tiers := []core.Tier{{Name: "S Tier", Items: []core.Item{"Godzilla"}}}
|
||||||
|
|
||||||
|
s := state.New(
|
||||||
|
state.WithMode(state.NavigationMode),
|
||||||
|
state.WithTiers(tiers),
|
||||||
|
)
|
||||||
|
|
||||||
|
msg := tea.KeyMsg{Type: tea.KeyCtrlS}
|
||||||
|
cmd := h.Handle(s, msg)
|
||||||
|
|
||||||
|
if s.Mode() != state.InputMode {
|
||||||
|
t.Fatal("Expected state to switch to InputMode")
|
||||||
|
}
|
||||||
|
|
||||||
|
if cmd == nil {
|
||||||
|
t.Error("Expected command from Handle (blink)")
|
||||||
|
}
|
||||||
|
|
||||||
|
filename := "my_list.json"
|
||||||
|
|
||||||
|
for _, r := range filename {
|
||||||
|
s.UpdateInput(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{r}})
|
||||||
|
}
|
||||||
|
|
||||||
|
s.SubmitInput()
|
||||||
|
|
||||||
|
if len(saver.savedData.Tiers) != 1 {
|
||||||
|
t.Error("Expected saver to receive 1 tier")
|
||||||
|
}
|
||||||
|
|
||||||
|
expectedStatusMsg := "Saved to " + filename
|
||||||
|
|
||||||
|
if s.StatusMsg() != expectedStatusMsg {
|
||||||
|
t.Errorf("Expected status message %q, got %q", expectedStatusMsg, s.StatusMsg())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
t.Run("Handles save errors", func(t *testing.T) {
|
||||||
|
saver := &mockSaver{shouldError: true}
|
||||||
|
|
||||||
|
h := handlers.NewSaveHandler(saver)
|
||||||
|
s := state.New(state.WithMode(state.NavigationMode))
|
||||||
|
|
||||||
|
h.Handle(s, tea.KeyMsg{Type: tea.KeyCtrlS})
|
||||||
|
|
||||||
|
for _, r := range "bad_perms.json" {
|
||||||
|
s.UpdateInput(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{r}})
|
||||||
|
}
|
||||||
|
|
||||||
|
s.SubmitInput()
|
||||||
|
|
||||||
|
expectedPrefix := "Error:"
|
||||||
|
|
||||||
|
if len(s.StatusMsg()) < len(expectedPrefix) || s.StatusMsg()[:len(expectedPrefix)] != expectedPrefix {
|
||||||
|
t.Errorf("Expected error status, got %q", s.StatusMsg())
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user