notebloom/app_test.go at master · ChloeVPin/notebloom · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
package main

import (
	"os"
	"path/filepath"
	"strings"
	"testing"
)

func newTestApp(t *testing.T) *App {
	t.Helper()

	dir := t.TempDir()
	app := &App{notesDir: filepath.Join(dir, "notes")}
	if err := app.ensureNotesDir(); err != nil {
		t.Fatalf("ensureNotesDir() error = %v", err)
	}
	return app
}

func TestSaveNoteAndLoadNotes(t *testing.T) {
	app := newTestApp(t)

	note := Note{
		ID:        "note-1",
		Title:     "Hello",
		Content:   "One two three",
		Pinned:    true,
		CreatedAt: 1000,
		UpdatedAt: 2000,
	}

	if err := app.SaveNote(note); err != nil {
		t.Fatalf("SaveNote() error = %v", err)
	}

	loaded, err := app.LoadNotes()
	if err != nil {
		t.Fatalf("LoadNotes() error = %v", err)
	}
	if len(loaded) != 1 {
		t.Fatalf("LoadNotes() length = %d, want 1", len(loaded))
	}

	got := loaded[0]
	if got.ID != note.ID || got.Title != note.Title || got.Content != note.Content || !got.Pinned {
		t.Fatalf("LoadNotes() returned %+v, want %+v", got, note)
	}
}

func TestSaveNoteOverwritesExistingContent(t *testing.T) {
	app := newTestApp(t)

	initial := Note{
		ID:        "note-1",
		Title:     "Old",
		Content:   "first version",
		CreatedAt: 1000,
		UpdatedAt: 2000,
	}
	updated := Note{
		ID:        "note-1",
		Title:     "New",
		Content:   "second version",
		CreatedAt: 1000,
		UpdatedAt: 3000,
	}

	if err := app.SaveNote(initial); err != nil {
		t.Fatalf("initial SaveNote() error = %v", err)
	}
	if err := app.SaveNote(updated); err != nil {
		t.Fatalf("updated SaveNote() error = %v", err)
	}

	data, err := os.ReadFile(app.getNoteFilePath(updated.ID))
	if err != nil {
		t.Fatalf("ReadFile() error = %v", err)
	}
	if !strings.Contains(string(data), `"title": "New"`) {
		t.Fatalf("saved file does not contain updated title: %s", string(data))
	}
	if !strings.Contains(string(data), `"content": "second version"`) {
		t.Fatalf("saved file does not contain updated content: %s", string(data))
	}
}

func TestLoadNotesSkipsInvalidFiles(t *testing.T) {
	app := newTestApp(t)

	valid := Note{
		ID:        "valid-note",
		Title:     "Valid",
		Content:   "This should load",
		CreatedAt: 111,
		UpdatedAt: 222,
	}
	if err := app.SaveNote(valid); err != nil {
		t.Fatalf("SaveNote() error = %v", err)
	}

	if err := os.WriteFile(filepath.Join(app.notesDir, "broken.json"), []byte("{not json"), 0644); err != nil {
		t.Fatalf("WriteFile() error = %v", err)
	}
	if err := os.WriteFile(filepath.Join(app.notesDir, "ignore.txt"), []byte("skip"), 0644); err != nil {
		t.Fatalf("WriteFile() error = %v", err)
	}

	loaded, err := app.LoadNotes()
	if err != nil {
		t.Fatalf("LoadNotes() error = %v", err)
	}
	if len(loaded) != 1 {
		t.Fatalf("LoadNotes() length = %d, want 1", len(loaded))
	}
	if loaded[0].ID != valid.ID {
		t.Fatalf("LoadNotes() returned %+v, want note %q", loaded[0], valid.ID)
	}
}