1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
|
package ackhandler
import (
"testing"
"github.com/quic-go/quic-go/internal/wire"
"github.com/stretchr/testify/require"
)
func TestIsFrameTypeAckEliciting(t *testing.T) {
testCases := map[wire.FrameType]bool{
wire.FrameTypePing: true,
wire.FrameTypeAck: false,
wire.FrameTypeAckECN: false,
wire.FrameTypeResetStream: true,
wire.FrameTypeStopSending: true,
wire.FrameTypeCrypto: true,
wire.FrameTypeNewToken: true,
wire.FrameType(0x08): true,
wire.FrameType(0x09): true,
wire.FrameType(0x0a): true,
wire.FrameType(0x0b): true,
wire.FrameType(0x0c): true,
wire.FrameType(0x0d): true,
wire.FrameType(0x0e): true,
wire.FrameType(0x0f): true,
wire.FrameTypeMaxData: true,
wire.FrameTypeMaxStreamData: true,
wire.FrameTypeBidiMaxStreams: true,
wire.FrameTypeUniMaxStreams: true,
wire.FrameTypeDataBlocked: true,
wire.FrameTypeStreamDataBlocked: true,
wire.FrameTypeBidiStreamBlocked: true,
wire.FrameTypeUniStreamBlocked: true,
wire.FrameTypeNewConnectionID: true,
wire.FrameTypeRetireConnectionID: true,
wire.FrameTypePathChallenge: true,
wire.FrameTypePathResponse: true,
wire.FrameTypeConnectionClose: false,
wire.FrameTypeApplicationClose: false,
wire.FrameTypeHandshakeDone: true,
wire.FrameTypeResetStreamAt: true,
wire.FrameTypeDatagramNoLength: true,
wire.FrameTypeDatagramWithLength: true,
}
for ft, expected := range testCases {
require.Equal(t, expected, IsFrameTypeAckEliciting(ft), "unexpected result for frame type 0x%x", ft)
}
}
func TestAckElicitingFrames(t *testing.T) {
testCases := map[wire.Frame]bool{
&wire.AckFrame{}: false,
&wire.ConnectionCloseFrame{}: false,
&wire.DataBlockedFrame{}: true,
&wire.PingFrame{}: true,
&wire.ResetStreamFrame{}: true,
&wire.StreamFrame{}: true,
&wire.DatagramFrame{}: true,
&wire.MaxDataFrame{}: true,
&wire.MaxStreamDataFrame{}: true,
&wire.StopSendingFrame{}: true,
}
for f, expected := range testCases {
require.Equal(t, expected, IsFrameAckEliciting(f))
require.Equal(t, expected, HasAckElicitingFrames([]Frame{{Frame: f}}))
}
}
|