mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-03 20:27:35 +03:00
a5d3eaea61
This callback also takes the stream ID, which will later be used to create a queue of streams that have data for writing available.
39 lines
1 KiB
Go
39 lines
1 KiB
Go
package quic
|
|
|
|
import (
|
|
"github.com/lucas-clemente/quic-go/internal/protocol"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Crypto Stream", func() {
|
|
var (
|
|
str *cryptoStream
|
|
mockSender *MockStreamSender
|
|
)
|
|
|
|
BeforeEach(func() {
|
|
mockSender = NewMockStreamSender(mockCtrl)
|
|
str = newCryptoStream(mockSender, 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() {
|
|
mockSender.EXPECT().onHasStreamData(str.version.CryptoStreamID())
|
|
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())
|
|
})
|
|
})
|