caddy/config/setup/gzip_test.go

package setup

import (
	"testing"

	"github.com/mholt/caddy/middleware/gzip"
)

func TestGzip(t *testing.T) {
	c := NewTestController(`gzip`)

	mid, err := Gzip(c)
	if err != nil {
		t.Errorf("Expected no errors, but got: %v", err)
	}
	if mid == nil {
		t.Fatal("Expected middleware, was nil instead")
	}

	handler := mid(EmptyNext)
	myHandler, ok := handler.(gzip.Gzip)
	if !ok {
		t.Fatalf("Expected handler to be type Gzip, got: %#v", handler)
	}

	if !SameNext(myHandler.Next, EmptyNext) {
		t.Error("'Next' field of handler was not set properly")
	}

	tests := []struct {
		input     string
		shouldErr bool
	}{
		{`gzip {`, true},
		{`gzip {}`, true},
		{`gzip a b`, true},
		{`gzip a {`, true},
		{`gzip { not f } `, true},
		{`gzip { not } `, true},
		{`gzip { not /file
		 ext .html
		 level 1
		} `, false},
		{`gzip { level 9 } `, false},
		{`gzip { ext } `, true},
		{`gzip { ext /f
		} `, true},
		{`gzip { not /file
		 ext .html
		 level 1
		}
		gzip`, false},
		{`gzip { not /file
		 ext .html
		 level 1
		}
		gzip { not /file1
		 ext .htm
		 level 3
		}
		`, false},
		{`gzip { mimes text/html
		}`, false},
		{`gzip { mimes text/html application/json
		}`, false},
		{`gzip { mimes text/html application/
		}`, true},
		{`gzip { mimes text/html /json
		}`, true},
		{`gzip { mimes /json text/html
		}`, true},
		{`gzip { not /file
		 ext .html
		 level 1
		 mimes text/html text/plain
		}
		gzip { not /file1
		 ext .htm
		 level 3
		 mimes text/html text/css
		}
		`, false},
	}
	for i, test := range tests {
		c := NewTestController(test.input)
		_, err := gzipParse(c)
		if test.shouldErr && err == nil {
			t.Errorf("Test %v: Expected error but found nil", i)
		} else if !test.shouldErr && err != nil {
			t.Errorf("Test %v: Expected no error but found error: %v", i, err)
		}
	}
}