Browse Source

Add a test for issue #66

Together with af6d7623, this commit solves issue #66.

Fixes #66.
master
Daan Sprenkels 5 months ago
parent
commit
03a04389ae
2 changed files with 24 additions and 3 deletions
  1. +3
    -3
      views.go
  2. +21
    -0
      views_test.go

+ 3
- 3
views.go View File

@@ -219,7 +219,7 @@ func resolveResponseContentType(r *http.Request, types []string) (string, error)
if len(types) == 0 {
return "", nil
}
acceptHeader := r.Header.Get("Accept")
acceptHeader := strings.TrimSpace(r.Header.Get("Accept"))
if acceptHeader == "" {
return types[0], nil
}
@@ -237,7 +237,7 @@ func resolveResponseContentType(r *http.Request, types []string) (string, error)
choiceParts := strings.Split(avString, ";")
mediaRange := acceptHeaderMediaRangeRegex.FindStringSubmatch(choiceParts[0])
if mediaRange == nil {
return "", fmt.Errorf("bad media-range (\"%v\")", choiceParts[0])
return "", fmt.Errorf("bad media-range ('%v')", choiceParts[0])
}
av.Type = mediaRange[1]
av.Subtype = mediaRange[2]
@@ -262,7 +262,7 @@ func resolveResponseContentType(r *http.Request, types []string) (string, error)
// Check if this parameter is still invalid in any case
acceptParams := acceptHeaderAcceptParamsRegex.FindStringSubmatchIndex(choiceParts[0])
if acceptParams == nil {
return "", fmt.Errorf("bad accept-params (\"%v\")", choiceParts[0])
return "", fmt.Errorf("bad accept-params ('%v')", choiceParts[0])
}
}
avs[i] = av


+ 21
- 0
views_test.go View File

@@ -20,6 +20,24 @@ func resolveResponseContentTypeSuccess(t *testing.T, expected string, types []st
}
}

func resolveResponseContentTypeError(t *testing.T, expected string, types []string, acceptVal string) {
r, err := http.NewRequest("HEAD", "", nil)
if err != nil {
panic(err)
}
r.Header.Set("Accept", acceptVal)
got, err := resolveResponseContentType(r, types)
if err == nil {
t.Errorf("expected an error, but got a success: '%v'", got)
}
if got != "" {
t.Errorf("error: return value should be empty, not '%v'", got)
}
if err.Error() != expected {
t.Errorf("wrong error value error: got '%v', want '%v'\n", got, expected)
}
}

func TestResolveResponseContentType(t *testing.T) {
resolveResponseContentTypeSuccess(t, "", []string{}, "text/html")
resolveResponseContentTypeSuccess(t, "text/html", []string{"text/html"}, "")
@@ -30,4 +48,7 @@ func TestResolveResponseContentType(t *testing.T) {

// Issue #17
resolveResponseContentTypeSuccess(t, "*/*", []string{"*/*"}, "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3")

// Issue #66
resolveResponseContentTypeError(t, "bad media-range ('*')", []string{"text/plain"}, " *")
}

Loading…
Cancel
Save