28allday
2dd81700c0
LocalSend-compatible file-transfer TUI for headless Arch/Omarchy servers. - Pure-stdlib implementation of the LocalSend v2 protocol (discovery, HTTPS with matching cert fingerprint, send/receive, PIN). - Bubble Tea TUI: Devices, Transfers, Manage (received-file housekeeping) and Settings, theme-aware on Omarchy. - Dual-mode install.sh: curl-pipe download or build-from-clone. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
112 lines
3.2 KiB
Go
112 lines
3.2 KiB
Go
package tui
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
|
|
"omarchy-send/internal/config"
|
|
)
|
|
|
|
// manageModel returns a model whose receive dir is a temp dir seeded with the
|
|
// given filenames, already switched to the Manage tab.
|
|
func manageModel(t *testing.T, names ...string) (Model, string) {
|
|
t.Helper()
|
|
dir := t.TempDir()
|
|
for _, n := range names {
|
|
if err := os.WriteFile(filepath.Join(dir, n), []byte("x"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
cfg := config.Config{Alias: "omarchy", Port: 53317, ReceiveDir: dir, Protocol: "https"}
|
|
m := New(cfg, nil)
|
|
nm, _ := m.Update(tea.WindowSizeMsg{Width: 100, Height: 30})
|
|
m = nm.(Model)
|
|
nm, _ = m.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("3")}) // enter Manage
|
|
return nm.(Model), dir
|
|
}
|
|
|
|
func key(m Model, s string) Model {
|
|
var msg tea.KeyMsg
|
|
switch s {
|
|
case " ":
|
|
msg = tea.KeyMsg{Type: tea.KeySpace}
|
|
case "enter":
|
|
msg = tea.KeyMsg{Type: tea.KeyEnter}
|
|
default:
|
|
msg = tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune(s)}
|
|
}
|
|
nm, _ := m.Update(msg)
|
|
return nm.(Model)
|
|
}
|
|
|
|
func TestManageListsReceivedFilesSkippingPartials(t *testing.T) {
|
|
m, _ := manageModel(t, "photo.dng", "clip.mp4", "incoming.iso.part")
|
|
got := len(m.fileList.Items())
|
|
if got != 2 {
|
|
t.Fatalf("expected 2 listed files (.part skipped), got %d", got)
|
|
}
|
|
out := m.View()
|
|
for _, want := range []string{"Manage", "photo.dng", "clip.mp4"} {
|
|
if !strings.Contains(out, want) {
|
|
t.Errorf("manage view missing %q", want)
|
|
}
|
|
}
|
|
if strings.Contains(out, ".part") {
|
|
t.Error("in-progress .part file should not be shown")
|
|
}
|
|
}
|
|
|
|
func TestManageDeleteSingleViaConfirm(t *testing.T) {
|
|
m, dir := manageModel(t, "keep.txt", "drop.txt")
|
|
// Cursor starts on the newest (drop.txt was written last). Marking it and
|
|
// confirming should remove exactly that file.
|
|
m = key(m, " ") // mark file under cursor
|
|
if len(m.marked) != 1 {
|
|
t.Fatalf("expected 1 marked, got %d", len(m.marked))
|
|
}
|
|
m = key(m, "d") // request delete -> confirm card
|
|
if !m.confirmDel {
|
|
t.Fatal("expected confirm card to be showing")
|
|
}
|
|
m = key(m, "y") // confirm
|
|
if m.confirmDel {
|
|
t.Error("confirm card should be dismissed after delete")
|
|
}
|
|
if len(m.fileList.Items()) != 1 {
|
|
t.Fatalf("expected 1 file left, got %d", len(m.fileList.Items()))
|
|
}
|
|
// One file gone, one remains on disk.
|
|
remaining, _ := os.ReadDir(dir)
|
|
if len(remaining) != 1 {
|
|
t.Fatalf("expected 1 file on disk, got %d", len(remaining))
|
|
}
|
|
}
|
|
|
|
func TestManageDeleteAllAndCancel(t *testing.T) {
|
|
m, dir := manageModel(t, "a.txt", "b.txt", "c.txt")
|
|
m = key(m, "a") // mark all
|
|
if len(m.marked) != 3 {
|
|
t.Fatalf("expected 3 marked, got %d", len(m.marked))
|
|
}
|
|
m = key(m, "d")
|
|
m = key(m, "n") // cancel
|
|
if m.confirmDel {
|
|
t.Error("cancel should dismiss the confirm card")
|
|
}
|
|
if files, _ := os.ReadDir(dir); len(files) != 3 {
|
|
t.Fatalf("cancel must not delete anything; have %d files", len(files))
|
|
}
|
|
// Now actually delete all.
|
|
m = key(m, "d")
|
|
m = key(m, "y")
|
|
if files, _ := os.ReadDir(dir); len(files) != 0 {
|
|
t.Fatalf("expected all files deleted, %d remain", len(files))
|
|
}
|
|
if len(m.fileList.Items()) != 0 {
|
|
t.Errorf("list should be empty after deleting all")
|
|
}
|
|
}
|