add stream.StreamID()

h2quic/response_writer_test.go

@@ -4,16 +4,20 @@ import (
 	"bytes"
 	"net/http"
 
+	"github.com/lucas-clemente/quic-go/protocol"
 	. "github.com/onsi/ginkgo"
 	. "github.com/onsi/gomega"
 )
 
 type mockStream struct {
+	id protocol.StreamID
 	bytes.Buffer
 }
 
 func (mockStream) Close() error { return nil }
 
+func (s mockStream) StreamID() protocol.StreamID { return s.id }
+
 var _ = Describe("Response Writer", func() {
 	var (
 		w            *responseWriter

handshake/crypto_setup_test.go

@@ -92,6 +92,8 @@ func (s *mockStream) Close() error {
 	panic("not implemented")
 }
 
+func (s mockStream) StreamID() protocol.StreamID { panic("not implemented") }
+
 var _ = Describe("Crypto setup", func() {
 	var (
 		kex         *mockKEX

stream.go

@@ -213,3 +213,7 @@ func (s *stream) RegisterError(err error) {
 func (s *stream) finishedReading() bool {
 	return atomic.LoadInt32(&s.eof) != 0
 }
+
+func (s *stream) StreamID() protocol.StreamID {
+	return s.streamID
+}

stream_test.go

@@ -33,6 +33,10 @@ var _ = Describe("Stream", func() {
 		str, _ = newStream(handler, cpm, 1337)
 	})
 
+	It("gets stream id", func() {
+		Expect(str.StreamID()).To(Equal(protocol.StreamID(1337)))
+	})
+
 	Context("reading", func() {
 		It("reads a single StreamFrame", func() {
 			frame := frames.StreamFrame{

utils/utils.go

@@ -4,6 +4,8 @@ import (
 	"bytes"
 	"crypto/rand"
 	"io"
+
+	"github.com/lucas-clemente/quic-go/protocol"
 )
 
 // ReadStream is the read part of a QUIC stream
@@ -18,6 +20,7 @@ type Stream interface {
 	io.ByteReader
 	io.Writer
 	io.Closer
+	StreamID() protocol.StreamID
 }
 
 // ReadUintN reads N bytes