package h2quic

import (
	"bytes"
	"net/http"

	"github.com/lucas-clemente/quic-go/protocol"
	"github.com/lucas-clemente/quic-go/utils"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

type mockStream struct {
	bytes.Buffer
}

func (mockStream) Close() error { return nil }

type mockSession struct {
	stream *mockStream
}

func (s *mockSession) NewStream(id protocol.StreamID) (utils.Stream, error) {
	Expect(id).To(Equal(protocol.StreamID(5)))
	return s.stream, nil
}

var _ = Describe("Response Writer", func() {
	var (
		w            *responseWriter
		headerStream *mockStream
		dataStream   *mockStream
		s            *mockSession
	)

	BeforeEach(func() {
		headerStream = &mockStream{}
		dataStream = &mockStream{}
		s = &mockSession{stream: dataStream}
		w = newResponseWriter(headerStream, 5, s)
	})

	It("writes status", func() {
		w.WriteHeader(http.StatusTeapot)
		Expect(headerStream.Bytes()).To(Equal([]byte{
			0x0, 0x0, 0x5, 0x1, 0x4, 0x0, 0x0, 0x0, 0x5, 'H', 0x3, '4', '1', '8',
		}))
	})

	It("writes headers", func() {
		w.Header().Add("content-length", "42")
		w.WriteHeader(http.StatusTeapot)
		Expect(headerStream.Bytes()).To(Equal([]byte{
			0x0, 0x0, 0x14, 0x1, 0x4, 0x0, 0x0, 0x0, 0x5, 0x48, 0x3, 0x34, 0x31, 0x38,
			0x40, 0x8a, 0xbc, 0x7a, 0x92, 0x5a, 0x92, 0xb6, 0x72, 0xd5, 0x32, 0x67,
			0x2, 0x34, 0x32,
		}))
	})

	It("writes data", func() {
		n, err := w.Write([]byte("foobar"))
		Expect(n).To(Equal(6))
		Expect(err).ToNot(HaveOccurred())
		// Should have written 200 on the header stream
		Expect(headerStream.Bytes()).To(Equal([]byte{
			0x0, 0x0, 0x1, 0x1, 0x4, 0x0, 0x0, 0x0, 0x5, 0x88,
		}))
		// And foobar on the data stream
		Expect(dataStream.Bytes()).To(Equal([]byte{
			0x66, 0x6f, 0x6f, 0x62, 0x61, 0x72,
		}))
	})

	It("writes data after WriteHeader is called", func() {
		w.WriteHeader(http.StatusTeapot)
		n, err := w.Write([]byte("foobar"))
		Expect(n).To(Equal(6))
		Expect(err).ToNot(HaveOccurred())
		// Should have written 418 on the header stream
		Expect(headerStream.Bytes()).To(Equal([]byte{
			0x0, 0x0, 0x5, 0x1, 0x4, 0x0, 0x0, 0x0, 0x5, 'H', 0x3, '4', '1', '8',
		}))
		// And foobar on the data stream
		Expect(dataStream.Bytes()).To(Equal([]byte{
			0x66, 0x6f, 0x6f, 0x62, 0x61, 0x72,
		}))
	})
})