Minor fixes to tests

This commit is contained in:
Matthew Holt 2019-06-19 17:02:34 -06:00
parent 62b4553f7d
commit 05b3938556
No known key found for this signature in database
GPG key ID: 2A349DD577D586A5
2 changed files with 8 additions and 8 deletions

View file

@ -20,6 +20,7 @@ import (
"net"
"net/http"
"net/http/httptest"
"strconv"
"testing"
"github.com/mholt/caddy/caddytls"
@ -55,7 +56,7 @@ func TestRedirPlaintextHost(t *testing.T) {
},
{
Host: "foohost",
Port: HTTPSPort, // since this is the 'default' HTTPS port, should not be included in Location value
Port: strconv.Itoa(certmagic.HTTPSPort), // since this is the 'default' HTTPS port, should not be included in Location value
},
{
Host: "*.example.com",
@ -83,7 +84,7 @@ func TestRedirPlaintextHost(t *testing.T) {
if actual, expected := cfg.ListenHost, testcase.ListenHost; actual != expected {
t.Errorf("Test %d: Expected redir config to have bindhost %s but got %s", i, expected, actual)
}
if actual, expected := cfg.Addr.Port, HTTPPort; actual != expected {
if actual, expected := cfg.Addr.Port, strconv.Itoa(certmagic.HTTPPort); actual != expected {
t.Errorf("Test %d: Expected redir config to have port '%s' but got '%s'", i, expected, actual)
}

View file

@ -432,7 +432,7 @@ func TestServeHTTPFailingFS(t *testing.T) {
fsErr error
expectedStatus int
expectedErr error
expectedHeaders map[string]string
expectedHeaders map[string]struct{}
}{
{
fsErr: os.ErrNotExist,
@ -448,7 +448,7 @@ func TestServeHTTPFailingFS(t *testing.T) {
fsErr: errCustom,
expectedStatus: http.StatusServiceUnavailable,
expectedErr: errCustom,
expectedHeaders: map[string]string{"Retry-After": "5"},
expectedHeaders: map[string]struct{}{"Retry-After": {}},
},
}
@ -477,10 +477,9 @@ func TestServeHTTPFailingFS(t *testing.T) {
// check the headers - a special case for server under load
if test.expectedHeaders != nil && len(test.expectedHeaders) > 0 {
for expectedKey, expectedVal := range test.expectedHeaders {
actualVal := responseRecorder.Header().Get(expectedKey)
if expectedVal != actualVal {
t.Errorf("Test %d: Expected header %s: %s, found %s", i, expectedKey, expectedVal, actualVal)
for expectedKey := range test.expectedHeaders {
if _, ok := responseRecorder.Header()[expectedKey]; !ok {
t.Errorf("Test %d: Expected header %s, but was missing", i, expectedKey)
}
}
}