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

package app

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

	tea "github.com/charmbracelet/bubbletea"
)

func typeInto(a *App, s string) {
	for _, r := range s {
		a.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune{r}})
	}
}

func TestCtrlNOpensNamePrompt(t *testing.T) {
	a := newApp()
	a.Update(tea.WindowSizeMsg{Width: 100, Height: 20})
	a.Update(tea.KeyMsg{Type: tea.KeyCtrlN})
	if a.mode != ModeNamePrompt {
		t.Fatalf("mode = %d, want ModeNamePrompt", a.mode)
	}
	if a.dialog.Title() != "New note" {
		t.Errorf("dialog title = %q, want 'New note'", a.dialog.Title())
	}
}

func TestNamePromptShowsTargetDir(t *testing.T) {
	dir := t.TempDir()
	a := newApp()
	a.Update(tea.WindowSizeMsg{Width: 100, Height: 20})
	a.openNamePrompt(namingNew, dir, "")
	if !strings.Contains(a.View(), filepath.Base(dir)) {
		t.Errorf("View() does not show the target dir %q:\n%s", dir, a.View())
	}
}

func TestEscCancelsBackToPreviousMode(t *testing.T) {
	dir := t.TempDir()
	p := filepath.Join(dir, "open.md")
	os.WriteFile(p, []byte("body"), 0o644)

	a := newApp()
	a.Update(tea.WindowSizeMsg{Width: 100, Height: 20})
	a.Load(p)
	a.Update(tea.KeyMsg{Type: tea.KeyCtrlN})
	a.Update(tea.KeyMsg{Type: tea.KeyEsc})
	if a.mode != ModeEditor {
		t.Errorf("mode = %d after Esc, want ModeEditor", a.mode)
	}
	if a.path != p {
		t.Errorf("path = %q after Esc, want the file we were editing (%q)", a.path, p)
	}
	entries, _ := os.ReadDir(dir)
	if len(entries) != 1 {
		t.Errorf("Esc wrote %d entries to disk, want 1 (the pre-existing file)", len(entries))
	}
}

func TestDirtyBufferConfirmsDiscardBeforePrompt(t *testing.T) {
	a := newApp()
	a.Update(tea.WindowSizeMsg{Width: 100, Height: 20})
	a.editor.SetContent([]byte("x"))
	a.editor.Dirty = true
	a.Update(tea.KeyMsg{Type: tea.KeyCtrlN})
	if a.mode == ModeNamePrompt {
		t.Fatal("first Ctrl+N on a dirty buffer opened the prompt; want a discard confirmation")
	}
	a.Update(tea.KeyMsg{Type: tea.KeyCtrlN})
	if a.mode != ModeNamePrompt {
		t.Errorf("second Ctrl+N: mode = %d, want ModeNamePrompt", a.mode)
	}
}

func TestPickerCtrlNPrefillsQuery(t *testing.T) {
	dir := t.TempDir()
	a := newApp()
	a.Update(tea.WindowSizeMsg{Width: 100, Height: 20})
	if err := a.StartPickerIn(dir); err != nil {
		t.Fatal(err)
	}
	typeInto(a, "draft")
	a.Update(tea.KeyMsg{Type: tea.KeyCtrlN})
	if a.mode != ModeNamePrompt {
		t.Fatalf("mode = %d, want ModeNamePrompt", a.mode)
	}
	if a.dialog.Value() != "draft" {
		t.Errorf("dialog prefill = %q, want draft", a.dialog.Value())
	}
}

func TestF2OpensRenamePrefilled(t *testing.T) {
	dir := t.TempDir()
	p := filepath.Join(dir, "before.md")
	os.WriteFile(p, []byte("body"), 0o644)

	a := newApp()
	a.Update(tea.WindowSizeMsg{Width: 100, Height: 20})
	a.Load(p)
	a.Update(tea.KeyMsg{Type: tea.KeyF2})
	if a.mode != ModeNamePrompt {
		t.Fatalf("mode = %d, want ModeNamePrompt", a.mode)
	}
	if a.dialog.Title() != "Rename" {
		t.Errorf("dialog title = %q, want Rename", a.dialog.Title())
	}
	if a.dialog.Value() != "before" {
		t.Errorf("dialog prefill = %q, want before", a.dialog.Value())
	}
}