content_feed_test.go RawHistoryFolder

1package headless
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// contentFeed is the wayland side of Run's content input in tests: batches go
12// out on ch, and the generated ContentReturn mock checks that each comes back
13// reset.
14type contentFeed struct {
15	ch       chan *ports.ContentBatch
16	ret      *portsmocks.MockContentReturn
17	returned chan *ports.ContentBatch
18}
19
20// newContentFeed makes a feed whose channel holds buffer batches.
21func newContentFeed(t *testing.T, buffer int) *contentFeed {
22	t.Helper()
23	f := &contentFeed{ch: make(chan *ports.ContentBatch, buffer), ret: portsmocks.NewMockContentReturn(t), returned: make(chan *ports.ContentBatch, 64)}
24	f.ret.EXPECT().Return(mock.Anything).Run(func(b *ports.ContentBatch) {
25		if len(b.Items) != 0 {
26			t.Errorf("batch returned with %d items, want it reset", len(b.Items))
27		}
28		f.returned <- b
29	}).Return().Maybe()
30	return f
31}
32
33// send hands the output one batch holding cs.
34func (f *contentFeed) send(cs ...ports.SurfaceContent) {
35	f.ch <- &ports.ContentBatch{Items: cs}
36}
37
38// in is the channel Run reads.
39func (f *contentFeed) in() <-chan *ports.ContentBatch { return f.ch }