From a6bb5e33086fc2a180666ccbc496c86ee028a140 Mon Sep 17 00:00:00 2001
From: Andrew Gaul <andrew@gaul.org>
Date: Sat, 21 Jan 2017 15:04:07 -0800
Subject: [PATCH] Add unit test for GIF transcoding

---
 compy_test.go | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/compy_test.go b/compy_test.go
index 8352b3e..5165287 100644
--- a/compy_test.go
+++ b/compy_test.go
@@ -3,7 +3,9 @@ package main
 import (
 	. "gopkg.in/check.v1"
 
+	"bytes"
 	gzipp "compress/gzip"
+	gifp "image/gif"
 	jpegp "image/jpeg"
 	pngp "image/png"
 	"io/ioutil"
@@ -35,6 +37,7 @@ func (s *CompyTest) SetUpSuite(c *C) {
 	s.server = httptest.NewServer(httpbin.GetMux())
 
 	s.proxy = proxy.New()
+	s.proxy.AddTranscoder("image/gif", &tc.Gif{})
 	s.proxy.AddTranscoder("image/jpeg", tc.NewJpeg(50))
 	s.proxy.AddTranscoder("image/png", &tc.Png{})
 	s.proxy.AddTranscoder("text/html", &tc.Zip{&tc.Identity{}, *brotli, *gzip, true})
@@ -111,6 +114,25 @@ func (s *CompyTest) TestBrotli(c *C) {
 	c.Assert(err, IsNil)
 }
 
+func (s *CompyTest) TestGif(c *C) {
+	resp, err := http.Get(s.server.URL + "/image/gif")
+	c.Assert(err, IsNil)
+	uncompressedLength, err := new(bytes.Buffer).ReadFrom(resp.Body)
+	c.Assert(err, IsNil)
+	resp.Body.Close()
+
+	resp, err = s.client.Get(s.server.URL + "/image/gif")
+	c.Assert(err, IsNil)
+	defer resp.Body.Close()
+	c.Assert(resp.StatusCode, Equals, 200)
+	c.Assert(resp.Header.Get("Content-Type"), Equals, "image/gif")
+
+	_, err = gifp.Decode(resp.Body)
+	c.Assert(err, IsNil)
+	compressedLength := resp.ContentLength
+	c.Assert(uncompressedLength > compressedLength, Equals, true)
+}
+
 func (s *CompyTest) TestJpeg(c *C) {
 	resp, err := s.client.Get(s.server.URL + "/image/jpeg")
 	c.Assert(err, IsNil)