Improved ServeFile* tests

1 files changed, 9 insertions, 2 deletions

diff --git a/fs_test.go b/fs_test.go
index f2020e7..5d79487 100644
--- a/fs_test.go
+++ b/fs_test.go
@@ -28,8 +28,9 @@ func TestFSServeFileCompressed(t *testing.T) {
 		t.Fatalf("unexpected error: %s", err)
 	}
 
-	if string(resp.Header.Peek("Content-Encoding")) != "gzip" {
-		t.Fatalf("Unexpected 'Content-Encoding' %q. Expecting %q", resp.Header.Peek("Content-Encoding"), "gzip")
+	ce := resp.Header.Peek("Content-Encoding")
+	if string(ce) != "gzip" {
+		t.Fatalf("Unexpected 'Content-Encoding' %q. Expecting %q", ce, "gzip")
 	}
 
 	body, err := resp.BodyGunzip()
@@ -49,6 +50,7 @@ func TestFSServeFileUncompressed(t *testing.T) {
 	var ctx RequestCtx
 	var req Request
 	req.SetRequestURI("http://foobar.com/baz")
+	req.Header.Set("Accept-Encoding", "gzip")
 	ctx.Init(&req, nil, nil)
 
 	ServeFileUncompressed(&ctx, "fs.go")
@@ -60,6 +62,11 @@ func TestFSServeFileUncompressed(t *testing.T) {
 		t.Fatalf("unexpected error: %s", err)
 	}
 
+	ce := resp.Header.Peek("Content-Encoding")
+	if len(ce) > 0 {
+		t.Fatalf("Unexpected 'Content-Encoding' %q", ce)
+	}
+
 	body := resp.Body()
 	expectedBody, err := getFileContents("/fs.go")
 	if err != nil {