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

84
internal/wire/stream_data_blocked_frame_test.go
View File

@@ -2,59 +2,45 @@ 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("STREAM_DATA_BLOCKED frame", func() {
Context("parsing", func() {
It("accepts sample frame", func() {
data := encodeVarInt(0xdeadbeef) // stream ID
data = append(data, encodeVarInt(0xdecafbad)...) // offset
frame, l, err := parseStreamDataBlockedFrame(data, protocol.Version1)
Expect(err).ToNot(HaveOccurred())
Expect(frame.StreamID).To(Equal(protocol.StreamID(0xdeadbeef)))
Expect(frame.MaximumStreamData).To(Equal(protocol.ByteCount(0xdecafbad)))
Expect(l).To(Equal(len(data)))
})
func TestParseStreamDataBlocked(t *testing.T) {
data := encodeVarInt(0xdeadbeef) // stream ID
data = append(data, encodeVarInt(0xdecafbad)...) // offset
frame, l, err := parseStreamDataBlockedFrame(data, protocol.Version1)
require.NoError(t, err)
require.Equal(t, protocol.StreamID(0xdeadbeef), frame.StreamID)
require.Equal(t, protocol.ByteCount(0xdecafbad), frame.MaximumStreamData)
require.Equal(t, len(data), l)
}
It("errors on EOFs", func() {
data := encodeVarInt(0xdeadbeef)
data = append(data, encodeVarInt(0xc0010ff)...)
_, l, err := parseStreamDataBlockedFrame(data, protocol.Version1)
Expect(err).NotTo(HaveOccurred())
Expect(l).To(Equal(len(data)))
for i := range data {
_, _, err := parseStreamDataBlockedFrame(data[:i], protocol.Version1)
Expect(err).To(MatchError(io.EOF))
}
})
})
func TestParseStreamDataBlockedErrorsOnEOFs(t *testing.T) {
data := encodeVarInt(0xdeadbeef)
data = append(data, encodeVarInt(0xc0010ff)...)
_, l, err := parseStreamDataBlockedFrame(data, protocol.Version1)
require.NoError(t, err)
require.Equal(t, len(data), l)
for i := range data {
_, _, err := parseStreamDataBlockedFrame(data[:i], protocol.Version1)
require.Equal(t, io.EOF, err)
}
}
Context("writing", func() {
It("has proper min length", func() {
f := &StreamDataBlockedFrame{
StreamID: 0x1337,
MaximumStreamData: 0xdeadbeef,
}
Expect(f.Length(0)).To(BeEquivalentTo(1 + quicvarint.Len(0x1337) + quicvarint.Len(0xdeadbeef)))
})
It("writes a sample frame", func() {
f := &StreamDataBlockedFrame{
StreamID: 0xdecafbad,
MaximumStreamData: 0x1337,
}
b, err := f.Append(nil, protocol.Version1)
Expect(err).ToNot(HaveOccurred())
expected := []byte{streamDataBlockedFrameType}
expected = append(expected, encodeVarInt(uint64(f.StreamID))...)
expected = append(expected, encodeVarInt(uint64(f.MaximumStreamData))...)
Expect(b).To(Equal(expected))
})
})
})
func TestWriteStreamDataBlocked(t *testing.T) {
f := &StreamDataBlockedFrame{
StreamID: 0xdecafbad,
MaximumStreamData: 0x1337,
}
b, err := f.Append(nil, protocol.Version1)
require.NoError(t, err)
expected := []byte{streamDataBlockedFrameType}
expected = append(expected, encodeVarInt(uint64(f.StreamID))...)
expected = append(expected, encodeVarInt(uint64(f.MaximumStreamData))...)
require.Equal(t, expected, b)
require.Equal(t, int(f.Length(protocol.Version1)), len(b))
}