mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-03 20:27:35 +03:00
d2e31c04ab
By doing so, we can mock private methods of the stream, so they cannot be type-asserted by users of quic-go.
32 lines
835 B
Go
32 lines
835 B
Go
package quic
|
|
|
|
import (
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Stream", func() {
|
|
var str *cryptoStream
|
|
|
|
str = newCryptoStream(func() {}, nil, protocol.VersionWhatever).(*cryptoStream)
|
|
|
|
It("sets the read offset", func() {
|
|
str.setReadOffset(0x42)
|
|
Expect(str.receiveStream.readOffset).To(Equal(protocol.ByteCount(0x42)))
|
|
Expect(str.receiveStream.frameQueue.readPosition).To(Equal(protocol.ByteCount(0x42)))
|
|
})
|
|
|
|
It("says if it has data for writing", func() {
|
|
Expect(str.hasDataForWriting()).To(BeFalse())
|
|
done := make(chan struct{})
|
|
go func() {
|
|
defer GinkgoRecover()
|
|
_, err := str.Write([]byte("foobar"))
|
|
Expect(err).ToNot(HaveOccurred())
|
|
close(done)
|
|
}()
|
|
Eventually(str.hasDataForWriting).Should(BeTrue())
|
|
})
|
|
})
|