mirror of
https://github.com/aljazceru/opencode.git
synced 2025-12-22 10:14:22 +01:00
278 lines
8.3 KiB
Go
278 lines
8.3 KiB
Go
package chat
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/charmbracelet/bubbles/v2/spinner"
|
|
tea "github.com/charmbracelet/bubbletea/v2"
|
|
"github.com/sst/opencode/internal/app"
|
|
"github.com/sst/opencode/internal/completions"
|
|
"github.com/sst/opencode/internal/components/dialog"
|
|
"github.com/sst/opencode/internal/components/textarea"
|
|
"github.com/sst/opencode/internal/styles"
|
|
)
|
|
|
|
func newTestEditor() *editorComponent {
|
|
m := &editorComponent{
|
|
app: &app.App{},
|
|
textarea: textarea.New(),
|
|
spinner: spinner.New(),
|
|
}
|
|
return m
|
|
}
|
|
|
|
func TestPasteAtPathWithTrailingComma_PreservesPunctuation_NoDoubleSpace(t *testing.T) {
|
|
m := newTestEditor()
|
|
p := createTempTextFile(t, "", "pc.txt", "x")
|
|
|
|
paste := "See @" + p + ", next"
|
|
_, cmd := m.Update(tea.PasteMsg(paste))
|
|
if cmd == nil {
|
|
t.Fatalf("expected command to be returned for comma punctuation paste")
|
|
}
|
|
if _, ok := cmd().(AttachmentInsertedMsg); !ok {
|
|
t.Fatalf("expected AttachmentInsertedMsg for comma punctuation paste")
|
|
}
|
|
if len(m.textarea.GetAttachments()) != 1 {
|
|
t.Fatalf("expected 1 attachment, got %d", len(m.textarea.GetAttachments()))
|
|
}
|
|
v := m.Value()
|
|
if !strings.Contains(v, ", next") {
|
|
t.Fatalf("expected comma and following text to be preserved, got: %q", v)
|
|
}
|
|
if strings.Contains(v, ", next") {
|
|
t.Fatalf("did not expect double space after comma, got: %q", v)
|
|
}
|
|
}
|
|
|
|
func TestPasteAtPathWithTrailingQuestion_PreservesPunctuation_NoDoubleSpace(t *testing.T) {
|
|
m := newTestEditor()
|
|
p := createTempTextFile(t, "", "pq.txt", "x")
|
|
|
|
paste := "Check @" + p + "? Done"
|
|
_, cmd := m.Update(tea.PasteMsg(paste))
|
|
if cmd == nil {
|
|
t.Fatalf("expected command to be returned for question punctuation paste")
|
|
}
|
|
if _, ok := cmd().(AttachmentInsertedMsg); !ok {
|
|
t.Fatalf("expected AttachmentInsertedMsg for question punctuation paste")
|
|
}
|
|
if len(m.textarea.GetAttachments()) != 1 {
|
|
t.Fatalf("expected 1 attachment, got %d", len(m.textarea.GetAttachments()))
|
|
}
|
|
v := m.Value()
|
|
if !strings.Contains(v, "? Done") {
|
|
t.Fatalf("expected question mark and following text to be preserved, got: %q", v)
|
|
}
|
|
if strings.Contains(v, "? Done") {
|
|
t.Fatalf("did not expect double space after question mark, got: %q", v)
|
|
}
|
|
}
|
|
|
|
func TestPasteMultipleInlineAtPaths_AttachesEach(t *testing.T) {
|
|
m := newTestEditor()
|
|
dir := t.TempDir()
|
|
p1 := createTempTextFile(t, dir, "m1.txt", "one")
|
|
p2 := createTempTextFile(t, dir, "m2.txt", "two")
|
|
|
|
// Build a paste with text around, two @paths, and punctuation after the first
|
|
paste := "Please check @" + p1 + ", and also @" + p2 + " thanks"
|
|
|
|
_, cmd := m.Update(tea.PasteMsg(paste))
|
|
if cmd == nil {
|
|
t.Fatalf("expected command to be returned for multi inline paste")
|
|
}
|
|
if _, ok := cmd().(AttachmentInsertedMsg); !ok {
|
|
t.Fatalf("expected AttachmentInsertedMsg for multi inline paste")
|
|
}
|
|
|
|
atts := m.textarea.GetAttachments()
|
|
if len(atts) != 2 {
|
|
t.Fatalf("expected 2 attachments, got %d", len(atts))
|
|
}
|
|
v := m.Value()
|
|
if !strings.Contains(v, "Please check") || !strings.Contains(v, "and also") || !strings.Contains(v, "thanks") {
|
|
t.Fatalf("expected surrounding text to be preserved, got: %q", v)
|
|
}
|
|
}
|
|
|
|
func createTempTextFile(t *testing.T, dir, name, content string) string {
|
|
t.Helper()
|
|
if dir == "" {
|
|
td, err := os.MkdirTemp("", "editor-test-*")
|
|
if err != nil {
|
|
t.Fatalf("failed to make temp dir: %v", err)
|
|
}
|
|
dir = td
|
|
}
|
|
p := filepath.Join(dir, name)
|
|
if err := os.WriteFile(p, []byte(content), 0o600); err != nil {
|
|
t.Fatalf("failed to write temp file: %v", err)
|
|
}
|
|
abs, err := filepath.Abs(p)
|
|
if err != nil {
|
|
t.Fatalf("failed to get abs path: %v", err)
|
|
}
|
|
return abs
|
|
}
|
|
|
|
func createTempBinFile(t *testing.T, dir, name string, data []byte) string {
|
|
t.Helper()
|
|
if dir == "" {
|
|
td, err := os.MkdirTemp("", "editor-test-*")
|
|
if err != nil {
|
|
t.Fatalf("failed to make temp dir: %v", err)
|
|
}
|
|
dir = td
|
|
}
|
|
p := filepath.Join(dir, name)
|
|
if err := os.WriteFile(p, data, 0o600); err != nil {
|
|
t.Fatalf("failed to write temp bin file: %v", err)
|
|
}
|
|
abs, err := filepath.Abs(p)
|
|
if err != nil {
|
|
t.Fatalf("failed to get abs path: %v", err)
|
|
}
|
|
return abs
|
|
}
|
|
|
|
func TestPasteStartsWithAt_AttachesAndEmitsMsg(t *testing.T) {
|
|
m := newTestEditor()
|
|
p := createTempTextFile(t, "", "a.txt", "hello")
|
|
|
|
_, cmd := m.Update(tea.PasteMsg("@" + p))
|
|
if cmd == nil {
|
|
t.Fatalf("expected command to be returned")
|
|
}
|
|
msg := cmd()
|
|
if _, ok := msg.(AttachmentInsertedMsg); !ok {
|
|
t.Fatalf("expected AttachmentInsertedMsg, got %T", msg)
|
|
}
|
|
|
|
atts := m.textarea.GetAttachments()
|
|
if len(atts) != 1 {
|
|
t.Fatalf("expected 1 attachment, got %d", len(atts))
|
|
}
|
|
if v := m.Value(); !strings.HasSuffix(v, " ") {
|
|
t.Fatalf("expected trailing space after attachment, got value: %q", v)
|
|
}
|
|
}
|
|
|
|
func TestPasteAfterAt_ReplacesAtWithAttachment(t *testing.T) {
|
|
m := newTestEditor()
|
|
p := createTempTextFile(t, "", "b.txt", "hello")
|
|
|
|
m.textarea.SetValue("@")
|
|
// Cursor should be at the end after SetValue; paste absolute path
|
|
_, cmd := m.Update(tea.PasteMsg(p))
|
|
if cmd == nil {
|
|
t.Fatalf("expected command to be returned")
|
|
}
|
|
if _, ok := cmd().(AttachmentInsertedMsg); !ok {
|
|
t.Fatalf("expected AttachmentInsertedMsg from paste after '@'")
|
|
}
|
|
|
|
// Ensure the raw '@' rune was removed (attachment inserted in its place)
|
|
if m.textarea.LastRuneIndex('@') != -1 {
|
|
t.Fatalf("'@' rune should have been removed from the text slice")
|
|
}
|
|
if len(m.textarea.GetAttachments()) != 1 {
|
|
t.Fatalf("expected 1 attachment inserted")
|
|
}
|
|
if v := m.Value(); !strings.HasSuffix(v, " ") {
|
|
t.Fatalf("expected trailing space after attachment, got value: %q", v)
|
|
}
|
|
}
|
|
|
|
func TestPlainTextPaste_NoAttachment_NoMsg(t *testing.T) {
|
|
m := newTestEditor()
|
|
_, cmd := m.Update(tea.PasteMsg("hello"))
|
|
if cmd != nil {
|
|
t.Fatalf("expected no command for plain text paste")
|
|
}
|
|
if got := m.Value(); got != "hello" {
|
|
t.Fatalf("expected value 'hello', got %q", got)
|
|
}
|
|
if len(m.textarea.GetAttachments()) != 0 {
|
|
t.Fatalf("expected no attachments for plain text paste")
|
|
}
|
|
}
|
|
|
|
func TestPlainPathPng_AttachesImage(t *testing.T) {
|
|
m := newTestEditor()
|
|
// Minimal bytes; content isn't validated, extension determines mime
|
|
p := createTempBinFile(t, "", "img.png", []byte{0x89, 'P', 'N', 'G'})
|
|
|
|
_, cmd := m.Update(tea.PasteMsg(p))
|
|
if cmd == nil {
|
|
t.Fatalf("expected command to be returned for image path paste")
|
|
}
|
|
if _, ok := cmd().(AttachmentInsertedMsg); !ok {
|
|
t.Fatalf("expected AttachmentInsertedMsg for image path paste")
|
|
}
|
|
atts := m.textarea.GetAttachments()
|
|
if len(atts) != 1 {
|
|
t.Fatalf("expected 1 attachment, got %d", len(atts))
|
|
}
|
|
if atts[0].MediaType != "image/png" {
|
|
t.Fatalf("expected image/png mime, got %q", atts[0].MediaType)
|
|
}
|
|
if v := m.Value(); !strings.HasSuffix(v, " ") {
|
|
t.Fatalf("expected trailing space after attachment, got value: %q", v)
|
|
}
|
|
}
|
|
|
|
func TestPlainPathPdf_AttachesPDF(t *testing.T) {
|
|
m := newTestEditor()
|
|
p := createTempBinFile(t, "", "doc.pdf", []byte("%PDF-1.4"))
|
|
|
|
_, cmd := m.Update(tea.PasteMsg(p))
|
|
if cmd == nil {
|
|
t.Fatalf("expected command to be returned for pdf path paste")
|
|
}
|
|
if _, ok := cmd().(AttachmentInsertedMsg); !ok {
|
|
t.Fatalf("expected AttachmentInsertedMsg for pdf path paste")
|
|
}
|
|
atts := m.textarea.GetAttachments()
|
|
if len(atts) != 1 {
|
|
t.Fatalf("expected 1 attachment, got %d", len(atts))
|
|
}
|
|
if atts[0].MediaType != "application/pdf" {
|
|
t.Fatalf("expected application/pdf mime, got %q", atts[0].MediaType)
|
|
}
|
|
if v := m.Value(); !strings.HasSuffix(v, " ") {
|
|
t.Fatalf("expected trailing space after attachment, got value: %q", v)
|
|
}
|
|
}
|
|
|
|
func TestCompletionFiles_InsertsAttachment_EmitsMsg(t *testing.T) {
|
|
m := newTestEditor()
|
|
p := createTempTextFile(t, "", "c.txt", "hello")
|
|
m.textarea.SetValue("@")
|
|
|
|
item := completions.CompletionSuggestion{
|
|
ProviderID: "files",
|
|
Value: p,
|
|
Display: func(_ styles.Style) string { return p },
|
|
}
|
|
// Build the completion selected message as if the user selected from the dialog
|
|
msg := dialog.CompletionSelectedMsg{Item: item, SearchString: "@"}
|
|
|
|
_, cmd := m.Update(msg)
|
|
if cmd == nil {
|
|
t.Fatalf("expected command to be returned")
|
|
}
|
|
if _, ok := cmd().(AttachmentInsertedMsg); !ok {
|
|
t.Fatalf("expected AttachmentInsertedMsg from files completion selection")
|
|
}
|
|
if len(m.textarea.GetAttachments()) != 1 {
|
|
t.Fatalf("expected 1 attachment inserted from completion selection")
|
|
}
|
|
if v := m.Value(); !strings.HasSuffix(v, " ") {
|
|
t.Fatalf("expected trailing space after attachment, got value: %q", v)
|
|
}
|
|
}
|