feat: implement file system persistence with save and load functionality

This commit is contained in:
Stevan Freeborn
2025-12-29 12:48:05 -06:00
parent 7b4f9d5a4f
commit 3bc730db87
2 changed files with 186 additions and 0 deletions
+62
View File
@@ -0,0 +1,62 @@
package persistence
import (
"encoding/json"
"os"
"github.com/StevanFreeborn/term-tier/internal/core"
)
type FileSystem interface {
WriteFile(name string, data []byte, perm os.FileMode) error
ReadFile(name string) ([]byte, error)
}
type osFileSystem struct{}
func (osFileSystem) WriteFile(name string, data []byte, perm os.FileMode) error {
return os.WriteFile(name, data, perm)
}
func (osFileSystem) ReadFile(name string) ([]byte, error) {
return os.ReadFile(name)
}
type fileSaver struct {
fs FileSystem
}
func NewFileSaver() core.Saver {
return &fileSaver{
fs: osFileSystem{},
}
}
func NewFileSaverWithFS(fs FileSystem) core.Saver {
return &fileSaver{
fs: fs,
}
}
func (s *fileSaver) Save(filename string, state core.SaveState) error {
data, err := json.MarshalIndent(state, "", " ")
if err != nil {
return err
}
return s.fs.WriteFile(filename, data, 0644)
}
func (s *fileSaver) Load(filename string) (core.SaveState, error) {
data, err := s.fs.ReadFile(filename)
if err != nil {
return core.SaveState{}, err
}
var state core.SaveState
if err := json.Unmarshal(data, &state); err != nil {
return core.SaveState{}, err
}
return state, nil
}
+124
View File
@@ -0,0 +1,124 @@
package persistence_test
import (
"errors"
"os"
"reflect"
"testing"
"github.com/StevanFreeborn/term-tier/internal/core"
"github.com/StevanFreeborn/term-tier/internal/persistence"
)
func TestFileSaver_Save(t *testing.T) {
t.Run("Successfully saves content to mock FS", func(t *testing.T) {
mock := newMockFS()
saver := persistence.NewFileSaverWithFS(mock)
inputState := core.SaveState{}
filename := "test_tierlist.json"
err := saver.Save(filename, inputState)
if err != nil {
t.Errorf("Expected no error, got %v", err)
}
if _, exists := mock.files[filename]; !exists {
t.Error("Expected file to be created in mock FS")
}
})
t.Run("Returns error on write failure", func(t *testing.T) {
mock := newMockFS()
mock.writeErr = errors.New("disk full")
saver := persistence.NewFileSaverWithFS(mock)
err := saver.Save("fail.json", core.SaveState{})
if err == nil {
t.Error("Expected error due to write failure, got nil")
}
})
}
func TestFileSaver_Load(t *testing.T) {
t.Run("Successfully loads existing file", func(t *testing.T) {
mock := newMockFS()
filename := "data.json"
validJSON := []byte(`{
"tiers": []
}`)
mock.files[filename] = validJSON
saver := persistence.NewFileSaverWithFS(mock)
result, err := saver.Load(filename)
if err != nil {
t.Fatalf("Expected no error, got %v", err)
}
if reflect.ValueOf(result).IsZero() && len(validJSON) == 0 {
t.Error("Loaded data does not match expected content")
}
})
t.Run("Returns error when file does not exist", func(t *testing.T) {
mock := newMockFS()
saver := persistence.NewFileSaverWithFS(mock)
_, err := saver.Load("nonexistent.json")
if err == nil {
t.Error("Expected error for missing file, got nil")
}
if !errors.Is(err, os.ErrNotExist) {
t.Errorf("Expected os.ErrNotExist, got %v", err)
}
})
t.Run("Returns error on invalid JSON", func(t *testing.T) {
mock := newMockFS()
mock.files["bad.json"] = []byte(`{ broken_json: `)
saver := persistence.NewFileSaverWithFS(mock)
_, err := saver.Load("bad.json")
if err == nil {
t.Error("Expected JSON unmarshal error, got nil")
}
})
}
type mockFS struct {
files map[string][]byte
writeErr error
readErr error
}
func newMockFS() *mockFS {
return &mockFS{
files: make(map[string][]byte),
}
}
func (m *mockFS) WriteFile(name string, data []byte, perm os.FileMode) error {
if m.writeErr != nil {
return m.writeErr
}
m.files[name] = data
return nil
}
func (m *mockFS) ReadFile(name string) ([]byte, error) {
if m.readErr != nil {
return nil, m.readErr
}
data, exists := m.files[name]
if !exists {
return nil, os.ErrNotExist
}
return data, nil
}