From 7e65940188df25e00c71c89b7843f72a4345669c Mon Sep 17 00:00:00 2001 From: Lucas Clemente Date: Sun, 17 Apr 2016 00:37:07 +0200 Subject: [PATCH] add more cryptosetup tests --- handshake/crypto_setup_test.go | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/handshake/crypto_setup_test.go b/handshake/crypto_setup_test.go index 6d0e68937..2cd69bdee 100644 --- a/handshake/crypto_setup_test.go +++ b/handshake/crypto_setup_test.go @@ -1,6 +1,8 @@ package handshake import ( + "bytes" + "github.com/lucas-clemente/quic-go/protocol" . "github.com/onsi/ginkgo" @@ -75,10 +77,27 @@ var _ = Describe("Crypto setup", func() { TagPUBS: []byte("pubs-c"), }) Expect(err).ToNot(HaveOccurred()) + Expect(response).To(HavePrefix("SHLO")) Expect(response).To(ContainSubstring("pubs-s")) // TODO: Should be new pubs Expect(response).To(ContainSubstring(string(cs.nonce))) Expect(response).To(ContainSubstring(string(protocol.SupportedVersionsAsTags))) Expect(cs.secureAEAD).ToNot(BeNil()) Expect(cs.forwardSecureAEAD).ToNot(BeNil()) }) + + It("recognizes SCID", func() { + var data bytes.Buffer + WriteHandshakeMessage(&data, TagCHLO, map[Tag][]byte{TagSCID: scfg.ID}) + response, err := cs.HandleCryptoMessage(data.Bytes()) + Expect(err).ToNot(HaveOccurred()) + Expect(response).To(HavePrefix("SHLO")) + }) + + It("recognizes missing SCID", func() { + var data bytes.Buffer + WriteHandshakeMessage(&data, TagCHLO, map[Tag][]byte{}) + response, err := cs.HandleCryptoMessage(data.Bytes()) + Expect(err).ToNot(HaveOccurred()) + Expect(response).To(HavePrefix("REJ")) + }) })