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
|
// Copyright (C) 2026 Danilo M. <danix@danix.xyz>
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License version 2 as
// published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
package notify
import (
"bytes"
"image/png"
"testing"
"github.com/godbus/dbus/v5"
)
// rawVariant builds the (iiibiiay) struct the bus delivers for image-data.
func rawVariant(w, h, stride int, alpha bool, ch int, data []byte) dbus.Variant {
return dbus.MakeVariant([]interface{}{
int32(w), int32(h), int32(stride), alpha, int32(8), int32(ch), data,
})
}
func TestImageDataFromHints(t *testing.T) {
// 2x1 RGBA: red, green.
rgba := []byte{255, 0, 0, 255, 0, 255, 0, 255}
cases := []struct {
name string
hints map[string]dbus.Variant
wantW int
want bool
}{
{"image-data wins", map[string]dbus.Variant{
"image-data": rawVariant(2, 1, 8, true, 4, rgba),
"image-path": dbus.MakeVariant("/tmp/x.png"),
}, 2, true},
{"icon_data fallback", map[string]dbus.Variant{
"icon_data": rawVariant(2, 1, 8, true, 4, rgba),
}, 2, true},
{"image_data alias", map[string]dbus.Variant{
"image_data": rawVariant(2, 1, 8, true, 4, rgba),
}, 2, true},
{"absent", map[string]dbus.Variant{}, 0, false},
}
for _, c := range cases {
t.Run(c.name, func(t *testing.T) {
got, ok := ImageDataFromHints(c.hints)
if ok != c.want {
t.Fatalf("ok = %v, want %v", ok, c.want)
}
if ok && got.Width != c.wantW {
t.Fatalf("width = %d, want %d", got.Width, c.wantW)
}
})
}
}
func TestImagePathFromHints(t *testing.T) {
got, ok := ImagePathFromHints(map[string]dbus.Variant{"image-path": dbus.MakeVariant("/tmp/shot.png")})
if !ok || got != "/tmp/shot.png" {
t.Fatalf("got %q ok=%v", got, ok)
}
if _, ok := ImagePathFromHints(map[string]dbus.Variant{}); ok {
t.Fatal("empty hints must not report a path")
}
}
func TestRawImagePNG(t *testing.T) {
// 2x1 RGBA on a rowstride wider than the data, to prove stride is honoured.
r := &RawImage{Width: 2, Height: 1, RowStride: 12, HasAlpha: true, BitsPerSample: 8, Channels: 4,
Data: []byte{255, 0, 0, 255, 0, 255, 0, 255, 9, 9, 9, 9}}
data, err := r.PNG()
if err != nil {
t.Fatalf("PNG: %v", err)
}
img, err := png.Decode(bytes.NewReader(data))
if err != nil {
t.Fatalf("decode: %v", err)
}
if img.Bounds().Dx() != 2 || img.Bounds().Dy() != 1 {
t.Fatalf("bounds = %v", img.Bounds())
}
r0, g0, b0, a0 := img.At(0, 0).RGBA()
if r0>>8 != 255 || g0>>8 != 0 || b0>>8 != 0 || a0>>8 != 255 {
t.Fatalf("pixel 0 = %d %d %d %d", r0>>8, g0>>8, b0>>8, a0>>8)
}
}
func TestRawImagePNGRejectsBadData(t *testing.T) {
if _, err := (&RawImage{Width: 0, Height: 1, Channels: 4, BitsPerSample: 8}).PNG(); err == nil {
t.Fatal("zero width must error")
}
if _, err := (&RawImage{Width: 1, Height: 1, Channels: 2, BitsPerSample: 8}).PNG(); err == nil {
t.Fatal("channels 2 must error")
}
}
|
