replay_test.go RawHistoryFolder

1package quic
2
3import (
4	"context"
5	"sync"
6	"testing"
7	"time"
8
9	"github.com/bnema/vev/internal/protocol/wire"
10	"github.com/bnema/vev/internal/testutil/replaytest"
11	"github.com/stretchr/testify/require"
12)
13
14// TestTransportReplay runs the canonical typed transcript over one QUIC
15// stream: transport differences must not alter the application
16// conversation (P7.1).
17func TestTransportReplay(t *testing.T) {
18	replaytest.Run(t, func(t *testing.T, envelopes []wire.Envelope) []wire.Envelope {
19		cert, fingerprint, err := GenerateEphemeralCert()
20		require.NoError(t, err)
21		listener, err := ListenConfig("127.0.0.1:0", cert, Config{}, 4)
22		require.NoError(t, err)
23		defer func() { _ = listener.Close() }()
24
25		accepted := make(chan wire.Transport, 1)
26		go func() {
27			transport, err := listener.Accept()
28			if err != nil {
29				return
30			}
31			accepted <- transport
32		}()
33		dialer := DialConfig(listener.Addr(), "vev-bootstrap", fingerprint, Config{}, 10*time.Second)
34		sender, err := dialer.Dial(context.Background())
35		require.NoError(t, err)
36		defer func() { _ = sender.Close() }()
37
38		var receiver wire.Transport
39		select {
40		case receiver = <-accepted:
41		case <-time.After(10 * time.Second):
42			t.Fatal("QUIC listener did not accept")
43		}
44		defer func() { _ = receiver.Close() }()
45
46		var wg sync.WaitGroup
47		wg.Go(func() {
48			for _, frame := range envelopes {
49				if err := sender.Send(frame); err != nil {
50					t.Errorf("Send: %v", err)
51					return
52				}
53			}
54		})
55		got := make([]wire.Envelope, 0, len(envelopes))
56		for range envelopes {
57			frame, err := receiver.Recv()
58			if err != nil {
59				t.Errorf("Recv: %v", err)
60				_ = receiver.Close()
61				break
62			}
63			got = append(got, frame)
64		}
65		wg.Wait()
66		return got
67	})
68}