implement basic integration tests with dropped packages

ref #167
2016-06-08 11:15:51 +07:00
parent d7937f8774
commit 3c2b59d20d
5 changed files with 107 additions and 11 deletions
--- a/integrationtests/drop_test.go
+++ b/integrationtests/drop_test.go
@@ -0,0 +1,94 @@
+package integrationtests
+
+import (
+	"bytes"
+	"fmt"
+	"os"
+	"os/exec"
+	"runtime"
+	"strconv"
+	"time"
+
+	_ "github.com/lucas-clemente/quic-clients" // download clients
+	"github.com/lucas-clemente/quic-go/protocol"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+	. "github.com/onsi/gomega/gexec"
+)
+
+var _ = Describe("Drop Proxy", func() {
+	var proxy *UDPProxy
+	clientPath := fmt.Sprintf(
+		"%s/src/github.com/lucas-clemente/quic-clients/client-%s-debug",
+		os.Getenv("GOPATH"),
+		runtime.GOOS,
+	)
+	proxyPort := 10001
+
+	AfterEach(func() {
+		proxy.Stop()
+		time.Sleep(time.Millisecond)
+	})
+
+	for i := range protocol.SupportedVersions {
+		version := protocol.SupportedVersions[i]
+
+		Context(fmt.Sprintf("with quic version %d", version), func() {
+			It("gets a file when many outgoing packets are dropped", func() {
+				dropper := func(p PacketNumber) bool {
+					if p <= 5 { // don't interfere with the crypto handshake
+						return false
+					}
+					return p%4 == 0
+				}
+
+				iPort, _ := strconv.Atoi(port)
+				var err error
+				proxy, err = NewUDPProxy(proxyPort, "localhost", iPort, nil, dropper)
+				Expect(err).ToNot(HaveOccurred())
+
+				command := exec.Command(
+					clientPath,
+					"--quic-version="+strconv.Itoa(int(version)),
+					"--host=127.0.0.1",
+					"--port="+strconv.Itoa(proxyPort),
+					"https://quic.clemente.io/data",
+				)
+				session, err := Start(command, nil, GinkgoWriter)
+				Expect(err).NotTo(HaveOccurred())
+				defer session.Kill()
+				Eventually(session, 4).Should(Exit(0))
+				Expect(bytes.Contains(session.Out.Contents(), data)).To(BeTrue())
+			})
+
+			It("gets a file when many incoming packets are dropped", func() {
+				dropper := func(p PacketNumber) bool {
+					if p <= 5 { // don't interfere with the crypto handshake
+						return false
+					}
+					return p%4 == 0
+				}
+
+				iPort, _ := strconv.Atoi(port)
+				var err error
+				proxy, err = NewUDPProxy(proxyPort, "localhost", iPort, dropper, nil)
+				Expect(err).ToNot(HaveOccurred())
+
+				command := exec.Command(
+					clientPath,
+					"--quic-version="+strconv.Itoa(int(version)),
+					"--host=127.0.0.1",
+					"--port="+strconv.Itoa(proxyPort),
+					"https://quic.clemente.io/data",
+				)
+				session, err := Start(command, nil, GinkgoWriter)
+				Expect(err).NotTo(HaveOccurred())
+				defer session.Kill()
+				Eventually(session, 4).Should(Exit(0))
+				Expect(bytes.Contains(session.Out.Contents(), data)).To(BeTrue())
+			})
+
+		})
+	}
+})