1//go:build linux
2
3package tui
4
5import (
6 "bytes"
7 "os"
8 "path/filepath"
9 "testing"
10)
11
12func writeFakeLinuxClipboard(t *testing.T, name, script string) {
13 t.Helper()
14 path := filepath.Join(t.TempDir(), name)
15 if err := os.WriteFile(path, []byte("#!/bin/sh\n"+script), 0o700); err != nil {
16 t.Fatal(err)
17 }
18 t.Setenv("PATH", filepath.Dir(path))
19}
20
21func TestReadClipboardImagePNGWayland(t *testing.T) {
22 fixture := filepath.Join(t.TempDir(), "image.png")
23 want := testClipboardPNG(t)
24 if err := os.WriteFile(fixture, want, 0o600); err != nil {
25 t.Fatal(err)
26 }
27 t.Setenv("CLIPBOARD_IMAGE_FIXTURE", fixture)
28 t.Setenv("WAYLAND_DISPLAY", "wayland-0")
29 t.Setenv("DISPLAY", "")
30 writeFakeLinuxClipboard(t, "wl-paste", `
31if [ "$1" = "--list-types" ]; then
32 printf 'text/plain\nimage/png\n'
33else
34 /bin/cat "$CLIPBOARD_IMAGE_FIXTURE"
35fi
36`)
37
38 path, got, ok, err := ReadClipboardImagePNG()
39 if err != nil {
40 t.Fatal(err)
41 }
42 if path != "" || !ok {
43 t.Fatalf("path = %q, ok = %v", path, ok)
44 }
45 if !bytes.Equal(got, want) {
46 // Normalization may choose a different valid PNG representation, so
47 // compare dimensions when the encoded bytes differ.
48 if gotW, gotH := ImageDimensions(got); gotW != 2 || gotH != 2 {
49 t.Fatalf("normalized image dimensions = %dx%d", gotW, gotH)
50 }
51 }
52}
53
54func TestReadClipboardImagePNGX11(t *testing.T) {
55 fixture := filepath.Join(t.TempDir(), "image.png")
56 if err := os.WriteFile(fixture, testClipboardPNG(t), 0o600); err != nil {
57 t.Fatal(err)
58 }
59 t.Setenv("CLIPBOARD_IMAGE_FIXTURE", fixture)
60 t.Setenv("WAYLAND_DISPLAY", "")
61 t.Setenv("DISPLAY", ":1")
62 writeFakeLinuxClipboard(t, "xclip", `
63if [ "$4" = "TARGETS" ]; then
64 printf 'TARGETS\nimage/png\n'
65else
66 /bin/cat "$CLIPBOARD_IMAGE_FIXTURE"
67fi
68`)
69
70 _, got, ok, err := ReadClipboardImagePNG()
71 if err != nil {
72 t.Fatal(err)
73 }
74 if !ok || len(got) == 0 {
75 t.Fatalf("ok = %v, len(data) = %d", ok, len(got))
76 }
77}
78
79func TestReadClipboardImagePNGNoImageFallsBack(t *testing.T) {
80 t.Setenv("WAYLAND_DISPLAY", "wayland-0")
81 t.Setenv("DISPLAY", "")
82 writeFakeLinuxClipboard(t, "wl-paste", `
83if [ "$1" = "--list-types" ]; then
84 printf 'text/plain;charset=utf-8\n'
85 exit 0
86fi
87exit 9
88`)
89
90 _, data, ok, err := ReadClipboardImagePNG()
91 if err != nil {
92 t.Fatal(err)
93 }
94 if ok || data != nil {
95 t.Fatalf("ok = %v, data = %v", ok, data)
96 }
97}
98
99func TestReadClipboardImagePNGRejectsMalformedData(t *testing.T) {
100 t.Setenv("WAYLAND_DISPLAY", "wayland-0")
101 t.Setenv("DISPLAY", "")
102 writeFakeLinuxClipboard(t, "wl-paste", `
103if [ "$1" = "--list-types" ]; then
104 printf 'image/png\n'
105else
106 printf 'not a png'
107fi
108`)
109
110 if _, _, ok, err := ReadClipboardImagePNG(); err == nil || ok {
111 t.Fatalf("ok = %v, err = %v", ok, err)
112 }
113}
114
115func TestPreferredClipboardImageMIME(t *testing.T) {
116 mime, ok := preferredClipboardImageMIME("text/plain\nimage/jpeg\nimage/png\n")
117 if !ok || mime != "image/png" {
118 t.Fatalf("mime = %q, ok = %v", mime, ok)
119 }
120 if mime, ok := preferredClipboardImageMIME("text/plain\ntext/html\n"); ok {
121 t.Fatalf("mime = %q, want no image MIME", mime)
122 }
123}