presented_sink_test.go RawHistoryFolder

1package drm
2
3import (
4	"testing"
5
6	portsmocks "github.com/bnema/neferwl/internal/mocks/ports"
7	"github.com/bnema/neferwl/internal/ports"
8	"github.com/stretchr/testify/mock"
9)
10
11// presentedSink is a generated PresentationSink mock that records every push
12// on the returned channel. A full channel drops the report: Push never blocks,
13// as the port requires.
14func presentedSink(t *testing.T, size int) (ports.PresentationSink, chan ports.OutputPresented) {
15	t.Helper()
16	ch := make(chan ports.OutputPresented, size)
17	sink := portsmocks.NewMockPresentationSink(t)
18	sink.EXPECT().Push(mock.Anything).RunAndReturn(func(r ports.OutputPresented) {
19		select {
20		case ch <- r:
21		default:
22		}
23	}).Maybe()
24	return sink, ch
25}
26
27// discardSink is a PresentationSink for tests that ignore the reports.
28func discardSink(t *testing.T) ports.PresentationSink {
29	t.Helper()
30	sink, _ := presentedSink(t, 1)
31	return sink
32}