use the non-blocking of mint to cycle through the handshake

internal/handshake/mint_utils_test.go

@@ -0,0 +1,44 @@
+package handshake
+
+import (
+	"bytes"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("Fake Conn", func() {
+	var (
+		c      *fakeConn
+		stream *bytes.Buffer
+	)
+
+	BeforeEach(func() {
+		stream = &bytes.Buffer{}
+		c = &fakeConn{stream: stream}
+	})
+
+	Context("Reading", func() {
+		It("doesn't return any new data after one Read call", func() {
+			stream.Write([]byte("foobar"))
+			b := make([]byte, 3)
+			_, err := c.Read(b)
+			Expect(err).ToNot(HaveOccurred())
+			Expect(b).To(Equal([]byte("foo")))
+			n, err := c.Read(b)
+			Expect(err).ToNot(HaveOccurred())
+			Expect(n).To(BeZero())
+		})
+
+		It("allows more Read calls after unblocking", func() {
+			stream.Write([]byte("foobar"))
+			b := make([]byte, 3)
+			_, err := c.Read(b)
+			Expect(err).ToNot(HaveOccurred())
+			c.UnblockRead()
+			_, err = c.Read(b)
+			Expect(err).ToNot(HaveOccurred())
+			Expect(b).To(Equal([]byte("bar")))
+		})
+	})
+})