add an integration test for session resumption

integrationtests/self/resumption_test.go

@@ -0,0 +1,96 @@
+package self_test
+
+import (
+	"crypto/tls"
+	"fmt"
+	"net"
+	"sync"
+
+	quic "github.com/lucas-clemente/quic-go"
+	"github.com/lucas-clemente/quic-go/internal/testdata"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+type clientSessionCache struct {
+	mutex sync.Mutex
+	cache map[string]*tls.ClientSessionState
+
+	gets chan<- string
+	puts chan<- string
+}
+
+func newClientSessionCache(gets, puts chan<- string) *clientSessionCache {
+	return &clientSessionCache{
+		cache: make(map[string]*tls.ClientSessionState),
+		gets:  gets,
+		puts:  puts,
+	}
+}
+
+var _ tls.ClientSessionCache = &clientSessionCache{}
+
+func (c *clientSessionCache) Get(sessionKey string) (*tls.ClientSessionState, bool) {
+	c.gets <- sessionKey
+	c.mutex.Lock()
+	session, ok := c.cache[sessionKey]
+	c.mutex.Unlock()
+	return session, ok
+}
+
+func (c *clientSessionCache) Put(sessionKey string, cs *tls.ClientSessionState) {
+	c.puts <- sessionKey
+	c.mutex.Lock()
+	c.cache[sessionKey] = cs
+	c.mutex.Unlock()
+}
+
+var _ = Describe("TLS session resumption", func() {
+	It("uses session resumption", func() {
+		server, err := quic.ListenAddr("localhost:0", testdata.GetTLSConfig(), nil)
+		Expect(err).ToNot(HaveOccurred())
+		defer server.Close()
+
+		done := make(chan struct{})
+		go func() {
+			defer close(done)
+			defer GinkgoRecover()
+			sess, err := server.Accept()
+			Expect(err).ToNot(HaveOccurred())
+			Expect(sess.ConnectionState().DidResume).To(BeFalse())
+
+			sess, err = server.Accept()
+			Expect(err).ToNot(HaveOccurred())
+			Expect(sess.ConnectionState().DidResume).To(BeTrue())
+		}()
+
+		gets := make(chan string, 100)
+		puts := make(chan string, 100)
+		cache := newClientSessionCache(gets, puts)
+		tlsConf := &tls.Config{
+			RootCAs:            testdata.GetRootCA(),
+			ClientSessionCache: cache,
+		}
+		sess, err := quic.DialAddr(
+			fmt.Sprintf("localhost:%d", server.Addr().(*net.UDPAddr).Port),
+			tlsConf,
+			nil,
+		)
+		Expect(err).ToNot(HaveOccurred())
+		var sessionKey string
+		Eventually(puts).Should(Receive(&sessionKey))
+		Expect(sess.ConnectionState().DidResume).To(BeFalse())
+
+		sess, err = quic.DialAddr(
+			fmt.Sprintf("localhost:%d", server.Addr().(*net.UDPAddr).Port),
+			tlsConf,
+			nil,
+		)
+		Expect(err).ToNot(HaveOccurred())
+		Expect(gets).To(Receive(Equal(sessionKey)))
+		Expect(sess.ConnectionState().DidResume).To(BeTrue())
+
+		Eventually(done).Should(BeClosed())
+	})
+})