add some cryptosetup tests

handshake/crypto_setup_test.go

@@ -1,9 +1,72 @@
 package handshake
 
 import (
+	"github.com/lucas-clemente/quic-go/protocol"
+
 	. "github.com/onsi/ginkgo"
-	// . "github.com/onsi/gomega"
+	. "github.com/onsi/gomega"
 )
 
-var _ = Describe("Handshake", func() {
+type mockKEX struct{}
+
+func (*mockKEX) PublicKey() []byte {
+	return []byte("publickey")
+}
+func (*mockKEX) CalculateSharedKey(otherPublic []byte) ([]byte, error) {
+	return []byte("shared key"), nil
+}
+
+type mockSigner struct {
+	gotCHLO bool
+}
+
+func (s *mockSigner) SignServerProof(chlo []byte, serverConfigData []byte) ([]byte, error) {
+	if len(chlo) > 0 {
+		s.gotCHLO = true
+	}
+	return []byte("proof"), nil
+}
+func (*mockSigner) GetCertCompressed() []byte {
+	return []byte("certcompressed")
+}
+func (*mockSigner) GetCertUncompressed() []byte {
+	return []byte("certuncompressed")
+}
+
+var _ = Describe("Crypto setup", func() {
+	var (
+		kex    *mockKEX
+		signer *mockSigner
+		scfg   *ServerConfig
+		cs     *CryptoSetup
+	)
+
+	BeforeEach(func() {
+		kex = &mockKEX{}
+		signer = &mockSigner{}
+		scfg = NewServerConfig(kex, signer)
+		v := protocol.SupportedVersions[len(protocol.SupportedVersions)-1]
+		cs = NewCryptoSetup(protocol.ConnectionID(42), v, scfg)
+	})
+
+	It("has a nonce", func() {
+		Expect(cs.nonce).To(HaveLen(32))
+		Expect(cs.nonce[10]).ToNot(BeZero())
+	})
+
+	It("generates REJ messages", func() {
+		response, err := cs.handleInchoateCHLO(sampleCHLO)
+		Expect(err).ToNot(HaveOccurred())
+		Expect(response).To(HavePrefix("REJ"))
+		Expect(response).To(ContainSubstring("certcompressed"))
+		Expect(response).To(ContainSubstring("publickey"))
+		Expect(signer.gotCHLO).To(BeTrue())
+	})
+
+	It("generates REJ messages for version 30", func() {
+		cs.version = protocol.VersionNumber(30)
+		_, err := cs.handleInchoateCHLO(sampleCHLO)
+		Expect(err).ToNot(HaveOccurred())
+		Expect(signer.gotCHLO).To(BeFalse())
+	})
 })