test_wait_helpers_test.go RawHistoryFolder

1package daemon
2
3import (
4	"testing"
5	"time"
6
7	"github.com/bnema/vev/internal/protocol/wire"
8)
9
10const testWaitTimeout = 2 * time.Second
11
12func testFramesOfType(frames []wire.Envelope, name string) []wire.Envelope {
13	matched := make([]wire.Envelope, 0, len(frames))
14	for _, frame := range frames {
15		if envelopeMessageName(nil, frame.Payload) == name {
16			matched = append(matched, frame)
17		}
18	}
19	return matched
20}
21
22func awaitTestValue[T any](t *testing.T, ch <-chan T, failure string) T {
23	t.Helper()
24	timer := time.NewTimer(testWaitTimeout)
25	defer timer.Stop()
26	select {
27	case value := <-ch:
28		return value
29	case <-timer.C:
30		t.Fatal(failure)
31		var zero T
32		return zero
33	}
34}
35
36func awaitTestCompletion(t *testing.T, done <-chan struct{}, failure string) {
37	t.Helper()
38	awaitTestValue(t, done, failure)
39}
40
41func awaitCoordinatorOutput(
42	t *testing.T,
43	sends <-chan wire.Envelope,
44	timers <-chan *coordinatorMockTimer,
45	frameContext string,
46	timeoutFailure string,
47) wire.Envelope {
48	t.Helper()
49	deadline := time.NewTimer(testWaitTimeout)
50	defer deadline.Stop()
51	for {
52		select {
53		case frame := <-sends:
54			name := envelopeMessageName(t, frame.Payload)
55			if name == "RoutePosition" {
56				continue
57			}
58			if name != "Output" {
59				t.Fatalf("unexpected frame type %s %s", name, frameContext)
60			}
61			return frame
62		case timer := <-timers:
63			timer.ch <- time.Time{}
64		case <-deadline.C:
65			t.Fatal(timeoutFailure)
66			return wire.Envelope{}
67		}
68	}
69}