gr1ffon/quic-go
1
0
Fork 0
forked from quic-go/quic-go
Code Pull Requests Activity

wire: migrate tests away from Ginkgo (#4643)

Browse Source 
* wire: translate ACK frame tests

* wire: translate CONNECTION_CLOSE frame tests

* wire: translate DATA_BLOCKED frame tests

* wire: translate DATAGRAM frame tests

* wire: translate HANDSHAKE_DONE frame tests

* wire: translate MAX_DATA frame tests

* wire: translate MAX_STREAM_DATA frame tests

* wire: translate MAX_STREAMS frame tests

* wire: translate NEW_CONNECTION_ID frame tests

* wire: translate NEW_TOKEN frame tests

* wire: translate PATH_CHALLENGE frame tests

* wire: translate PATH_RESPONSE frame tests

* wire: translate PING frame test

* wire: translate RESET_STREAM frame tests

* wire: translate RETIRE_CONNECTION_ID frame tests

* wire: translate STOP_SENDING frame tests

* wire: translate STREAM_DATA_BLOCKED frame tests

* wire: translate STREAMS_BLOCKED frame tests

* wire: translate CRYPTO frame tests

* wire: translate STREAM frame tests

* wire: translate version negotiation tests

* wire: translate header tests

* wire: translate pool tests

* wire: translate frame logging tests

* wire: translate short header tests

* wire: translate framer parser tests

* wire: translate transport parameter tests
This commit is contained in:
Marten Seemann
2024-09-13 10:27:39 +08:00
committed by GitHub
parent 7a10ed602d
commit 672f906a40
30 changed files with 3768 additions and 4168 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
internal/wire/max_data_frame_test.go
View File

@@ -2,53 +2,38 @@ package wire
import (
"io"
"testing"
"github.com/quic-go/quic-go/internal/protocol"
"github.com/quic-go/quic-go/quicvarint"
. "github.com/onsi/ginkgo/v2"
. "github.com/onsi/gomega"
"github.com/stretchr/testify/require"
)
var _ = Describe("MAX_DATA frame", func() {
Context("when parsing", func() {
It("accepts sample frame", func() {
data := encodeVarInt(0xdecafbad123456) // byte offset
frame, l, err := parseMaxDataFrame(data, protocol.Version1)
Expect(err).ToNot(HaveOccurred())
Expect(frame.MaximumData).To(Equal(protocol.ByteCount(0xdecafbad123456)))
Expect(l).To(Equal(len(data)))
})
func TestParseMaxDataFrame(t *testing.T) {
data := encodeVarInt(0xdecafbad123456) // byte offset
frame, l, err := parseMaxDataFrame(data, protocol.Version1)
require.NoError(t, err)
require.Equal(t, protocol.ByteCount(0xdecafbad123456), frame.MaximumData)
require.Equal(t, len(data), l)
}
It("errors on EOFs", func() {
data := encodeVarInt(0xdecafbad1234567) // byte offset
_, l, err := parseMaxDataFrame(data, protocol.Version1)
Expect(err).NotTo(HaveOccurred())
Expect(l).To(Equal(len(data)))
for i := range data {
_, _, err := parseMaxDataFrame(data[:i], protocol.Version1)
Expect(err).To(MatchError(io.EOF))
}
})
})
func TestParseMaxDataErrorsOnEOFs(t *testing.T) {
data := encodeVarInt(0xdecafbad1234567) // byte offset
_, l, err := parseMaxDataFrame(data, protocol.Version1)
require.NoError(t, err)
require.Equal(t, len(data), l)
for i := range data {
_, _, err := parseMaxDataFrame(data[:i], protocol.Version1)
require.Equal(t, io.EOF, err)
}
}
Context("writing", func() {
It("has proper length", func() {
f := &MaxDataFrame{
MaximumData: 0xdeadbeef,
}
Expect(f.Length(protocol.Version1)).To(BeEquivalentTo(1 + quicvarint.Len(0xdeadbeef)))
})
It("writes a MAX_DATA frame", func() {
f := &MaxDataFrame{
MaximumData: 0xdeadbeefcafe,
}
b, err := f.Append(nil, protocol.Version1)
Expect(err).ToNot(HaveOccurred())
expected := []byte{maxDataFrameType}
expected = append(expected, encodeVarInt(0xdeadbeefcafe)...)
Expect(b).To(Equal(expected))
})
})
})
func TestWriteMaxDataFrame(t *testing.T) {
f := &MaxDataFrame{MaximumData: 0xdeadbeefcafe}
b, err := f.Append(nil, protocol.Version1)
require.NoError(t, err)
expected := []byte{maxDataFrameType}
expected = append(expected, encodeVarInt(0xdeadbeefcafe)...)
require.Equal(t, expected, b)
require.Len(t, b, int(f.Length(protocol.Version1)))
}