137 lines
3.8 KiB
Go
137 lines
3.8 KiB
Go
package tray
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type fakeMenuItem struct {
|
|
clicked chan struct{}
|
|
title string
|
|
tooltip string
|
|
}
|
|
|
|
func (i *fakeMenuItem) Clicked() <-chan struct{} {
|
|
return i.clicked
|
|
}
|
|
|
|
func (i *fakeMenuItem) SetTitle(title string) {
|
|
i.title = title
|
|
}
|
|
|
|
func (i *fakeMenuItem) SetTooltip(tooltip string) {
|
|
i.tooltip = tooltip
|
|
}
|
|
|
|
type fakeBackend struct {
|
|
icon []byte
|
|
tooltip string
|
|
items map[string]*fakeMenuItem
|
|
quitCalls int
|
|
registering bool
|
|
}
|
|
|
|
func (b *fakeBackend) Register(onReady func(), _ func()) {
|
|
b.registering = true
|
|
onReady()
|
|
}
|
|
|
|
func (b *fakeBackend) SetIcon(icon []byte) {
|
|
b.icon = append([]byte(nil), icon...)
|
|
}
|
|
|
|
func (b *fakeBackend) SetTooltip(tooltip string) {
|
|
b.tooltip = tooltip
|
|
}
|
|
|
|
func (b *fakeBackend) AddMenuItem(title, _ string) MenuItem {
|
|
item := &fakeMenuItem{clicked: make(chan struct{}, 1), title: title}
|
|
b.items[title] = item
|
|
return item
|
|
}
|
|
|
|
func (b *fakeBackend) Quit() {
|
|
b.quitCalls++
|
|
}
|
|
|
|
func waitFor(t *testing.T, signal <-chan struct{}) {
|
|
t.Helper()
|
|
select {
|
|
case <-signal:
|
|
case <-time.After(time.Second):
|
|
t.Fatal("timed out waiting for tray action")
|
|
}
|
|
}
|
|
|
|
func TestControllerInitializesTrayAndRoutesMenuActions(t *testing.T) {
|
|
backend := &fakeBackend{items: make(map[string]*fakeMenuItem)}
|
|
showCalls := make(chan struct{}, 1)
|
|
quitCalls := make(chan struct{}, 1)
|
|
controller := New(backend, []byte{1, 2, 3})
|
|
|
|
controller.Start(Actions{
|
|
Show: func() { showCalls <- struct{}{} },
|
|
Quit: func() { quitCalls <- struct{}{} },
|
|
})
|
|
|
|
if !backend.registering || string(backend.icon) != string([]byte{1, 2, 3}) || backend.tooltip != "Verstak" {
|
|
t.Fatalf("tray initialization = registering:%t icon:%v tooltip:%q", backend.registering, backend.icon, backend.tooltip)
|
|
}
|
|
showItem := backend.items["Show Verstak"]
|
|
quitItem := backend.items["Quit"]
|
|
if showItem == nil || quitItem == nil {
|
|
t.Fatalf("tray menu = %#v, want Show Verstak and Quit", backend.items)
|
|
}
|
|
|
|
showItem.clicked <- struct{}{}
|
|
waitFor(t, showCalls)
|
|
quitItem.clicked <- struct{}{}
|
|
waitFor(t, quitCalls)
|
|
}
|
|
|
|
func TestControllerStopsNativeTrayBackend(t *testing.T) {
|
|
backend := &fakeBackend{items: make(map[string]*fakeMenuItem)}
|
|
controller := New(backend, []byte{1})
|
|
|
|
controller.Stop()
|
|
|
|
if backend.quitCalls != 1 {
|
|
t.Fatalf("backend quit calls = %d, want 1", backend.quitCalls)
|
|
}
|
|
}
|
|
|
|
func TestControllerUsesAndUpdatesLocalizedLabels(t *testing.T) {
|
|
backend := &fakeBackend{items: make(map[string]*fakeMenuItem)}
|
|
controller := New(backend, []byte{1})
|
|
controller.SetLabels(LabelsForPreference("ru"))
|
|
controller.Start(Actions{})
|
|
|
|
show := backend.items["Показать Верстак"]
|
|
quit := backend.items["Выйти"]
|
|
if show == nil || quit == nil {
|
|
t.Fatalf("Russian tray menu = %#v", backend.items)
|
|
}
|
|
if show.tooltip != "Показать окно Верстака" || quit.tooltip != "Завершить Верстак" {
|
|
t.Fatalf("Russian tray tooltips = show:%q quit:%q", show.tooltip, quit.tooltip)
|
|
}
|
|
|
|
controller.SetLabels(LabelsForPreference("en"))
|
|
if show.title != "Show Verstak" || quit.title != "Quit" {
|
|
t.Fatalf("English tray menu after update = show:%q quit:%q", show.title, quit.title)
|
|
}
|
|
if show.tooltip != "Show the Verstak window" || quit.tooltip != "Quit Verstak" {
|
|
t.Fatalf("English tray tooltips after update = show:%q quit:%q", show.tooltip, quit.tooltip)
|
|
}
|
|
}
|
|
|
|
func TestLabelsForSystemRussianLocale(t *testing.T) {
|
|
labels := LabelsForPreference("system", "", "ru_RU.UTF-8", "en_US.UTF-8")
|
|
if labels.ShowTitle != "Показать Верстак" || labels.QuitTitle != "Выйти" {
|
|
t.Fatalf("system Russian labels = %#v", labels)
|
|
}
|
|
labels = LabelsForPreference("system", "C.UTF-8", "", "ru_RU.UTF-8")
|
|
if labels.ShowTitle != "Show Verstak" || labels.QuitTitle != "Quit" {
|
|
t.Fatalf("higher-priority system locale must win, got %#v", labels)
|
|
}
|
|
}
|