ui_clock_test.go RawHistoryFolder

1package client
2
3import (
4	"sync"
5	"testing"
6	"time"
7
8	"github.com/bnema/vev/internal/ports"
9	"github.com/bnema/vev/internal/protocol"
10)
11
12type attachPaletteClock struct {
13	timers          chan *attachPaletteTimer
14	handshakeTimers chan *attachPaletteTimer
15}
16
17type attachPaletteTimer struct {
18	ch       chan time.Time
19	stopped  chan struct{}
20	stopOnce sync.Once
21	duration time.Duration
22}
23
24func newAttachPaletteClock() *attachPaletteClock {
25	return &attachPaletteClock{
26		timers:          make(chan *attachPaletteTimer, 16),
27		handshakeTimers: make(chan *attachPaletteTimer, 1),
28	}
29}
30
31func (*attachPaletteClock) Now() time.Time { return time.Time{} }
32func (c *attachPaletteClock) NewTimer(delay time.Duration) ports.Timer {
33	timer := &attachPaletteTimer{ch: make(chan time.Time, 1), stopped: make(chan struct{}), duration: delay}
34	if delay == protocol.HandshakeTimeout {
35		select {
36		case c.handshakeTimers <- timer:
37		default:
38			panic("unexpected extra handshake timer")
39		}
40	} else {
41		c.timers <- timer
42	}
43	return timer
44}
45func (t *attachPaletteTimer) C() <-chan time.Time    { return t.ch }
46func (*attachPaletteTimer) Reset(time.Duration) bool { return false }
47func (t *attachPaletteTimer) Stop() bool {
48	t.stopOnce.Do(func() { close(t.stopped) })
49	return true
50}
51func (t *attachPaletteTimer) fire() { t.ch <- time.Time{} }
52
53func requireTimer(t *testing.T, timers <-chan *attachPaletteTimer, message string) *attachPaletteTimer {
54	t.Helper()
55	select {
56	case timer := <-timers:
57		return timer
58	case <-time.After(time.Second):
59		t.Fatal(message)
60		return nil
61	}
62}
63
64func requireSignal(t *testing.T, signal <-chan struct{}, message string) {
65	t.Helper()
66	select {
67	case <-signal:
68	case <-time.After(time.Second):
69		t.Fatal(message)
70	}
71}