From f8cb20ade613e65a783d950fc22e8f46c8902b7c Mon Sep 17 00:00:00 2001
From: Andrew Gaul <andrew@gaul.org>
Date: Wed, 18 Jan 2017 17:16:40 -0800
Subject: [PATCH] Add unit tests for gzip

---
 compy_test.go | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/compy_test.go b/compy_test.go
index 59dbb6f..dfd3a17 100644
--- a/compy_test.go
+++ b/compy_test.go
@@ -3,6 +3,7 @@ package main
 import (
 	. "gopkg.in/check.v1"
 
+	gzipp "compress/gzip"
 	jpegp "image/jpeg"
 	"net/http"
 	"net/http/httptest"
@@ -41,7 +42,11 @@ func (s *CompyTest) SetUpSuite(c *C) {
 	}()
 
 	proxyUrl := &url.URL{Scheme: "http", Host: "localhost" + *host}
-	s.client = &http.Client{Transport: &http.Transport{Proxy: http.ProxyURL(proxyUrl)}}
+	tr := &http.Transport{
+		DisableCompression: true,
+		Proxy:              http.ProxyURL(proxyUrl),
+	}
+	s.client = &http.Client{Transport: tr}
 }
 
 func (s *CompyTest) TearDownSuite(c *C) {
@@ -56,6 +61,32 @@ func (s *CompyTest) TestHttpBin(c *C) {
 	c.Assert(resp.StatusCode, Equals, 200)
 }
 
+func (s *CompyTest) TestNoGzip(c *C) {
+	resp, err := http.Get(s.server.URL + "/html")
+	c.Assert(err, IsNil)
+	defer resp.Body.Close()
+	c.Assert(resp.StatusCode, Equals, 200)
+	c.Assert(resp.Header.Get("Content-Encoding"), Equals, "")
+
+	_, err = gzipp.NewReader(resp.Body)
+	c.Assert(err, NotNil)
+}
+
+func (s *CompyTest) TestGzip(c *C) {
+	req, err := http.NewRequest("GET", s.server.URL+"/html", nil)
+	c.Assert(err, IsNil)
+	req.Header.Add("Accept-Encoding", "gzip")
+
+	resp, err := s.client.Do(req)
+	c.Assert(err, IsNil)
+	defer resp.Body.Close()
+	c.Assert(resp.StatusCode, Equals, 200)
+	c.Assert(resp.Header.Get("Content-Encoding"), Equals, "gzip")
+
+	_, err = gzipp.NewReader(resp.Body)
+	c.Assert(err, IsNil)
+}
+
 func (s *CompyTest) TestJpeg(c *C) {
 	req, err := http.NewRequest("GET", s.server.URL+"/image/jpeg", nil)
 	c.Assert(err, IsNil)