uquic/internal/ackhandler/ack_eliciting_test.go

package ackhandler

import (
	"reflect"

	"github.com/lucas-clemente/quic-go/internal/wire"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

var _ = Describe("ack-eliciting frames", func() {
	for fl, el := range map[wire.Frame]bool{
		&wire.AckFrame{}:             false,
		&wire.DataBlockedFrame{}:     true,
		&wire.ConnectionCloseFrame{}: true,
		&wire.PingFrame{}:            true,
		&wire.ResetStreamFrame{}:     true,
		&wire.StreamFrame{}:          true,
		&wire.MaxDataFrame{}:         true,
		&wire.MaxStreamDataFrame{}:   true,
	} {
		f := fl
		e := el
		fName := reflect.ValueOf(f).Elem().Type().Name()

		It("works for "+fName, func() {
			Expect(IsFrameAckEliciting(f)).To(Equal(e))
		})

		It("stripping non-ack-elicinting frames works for "+fName, func() {
			s := []wire.Frame{f}
			if e {
				Expect(stripNonAckElicitingFrames(s)).To(Equal([]wire.Frame{f}))
			} else {
				Expect(stripNonAckElicitingFrames(s)).To(BeEmpty())
			}
		})

		It("HasAckElicitingFrames works for "+fName, func() {
			Expect(HasAckElicitingFrames([]wire.Frame{f})).To(Equal(e))
		})
	}
})