aw/linkheader
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adds String method for Link

Browse Source
This commit is contained in:
Tom Hudson 2016-06-17 00:18:15 +01:00
parent 6953a30d44
commit 7750da0170
2 changed files with 44 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
main.go
View File

@ -33,6 +33,20 @@ func (l Link) Param(key string) (string, error) {
return "", fmt.Errorf("Could not find param '%s'", key) return "", fmt.Errorf("Could not find param '%s'", key)
} }
// String returns the string representation of a Link header
// for use in HTTP responses
func (l Link) String() string {
p := make([]string, len(l.Params))
for k, v := range l.Params {
p = append(p, fmt.Sprintf("%s=\"%s\"", k, v))
}
if l.Rel != "" {
p = append(p, fmt.Sprintf("%s=\"%s\"", "rel", l.Rel))
}
return fmt.Sprintf("<%s>; %s", l.URL, strings.Join(p, "; "))
}
// Links is a slice of Link structs // Links is a slice of Link structs
type Links []Link type Links []Link

32
main_test.go
View File

@ -66,7 +66,7 @@ func TestLinkMethods(t *testing.T) {
} }
func testLinksMethods(t *testing.T) { func TestLinksMethods(t *testing.T) {
header := "<https://api.github.com/user/9287/repos?page=3&per_page=100>; rel=\"next\", " + header := "<https://api.github.com/user/9287/repos?page=3&per_page=100>; rel=\"next\", " +
"<https://api.github.com/user/9287/repos?page=1&per_page=100>; rel=\"stylesheet\"; pet=\"cat\", " + "<https://api.github.com/user/9287/repos?page=1&per_page=100>; rel=\"stylesheet\"; pet=\"cat\", " +
"<https://api.github.com/user/9287/repos?page=5&per_page=100>; rel=\"stylesheet\"" "<https://api.github.com/user/9287/repos?page=5&per_page=100>; rel=\"stylesheet\""
@ -91,7 +91,7 @@ func testLinksMethods(t *testing.T) {
} }
func testParseMultiple(t *testing.T) { func TestParseMultiple(t *testing.T) {
headers := []string{ headers := []string{
"<https://api.github.com/user/58276/repos?page=2>; rel=\"next\"", "<https://api.github.com/user/58276/repos?page=2>; rel=\"next\"",
"<https://api.github.com/user/58276/repos?page=2>; rel=\"last\"", "<https://api.github.com/user/58276/repos?page=2>; rel=\"last\"",
@ -103,3 +103,31 @@ func testParseMultiple(t *testing.T) {
t.Errorf("Should have returned 2 links") t.Errorf("Should have returned 2 links")
} }
} }
func TestLinkToString(t *testing.T) {
l := Link{
URL: "http://example.com/page/2",
Rel: "next",
}
have := l.String()
want := "<http://example.com/page/2>; rel=\"next\""
if have != want {
t.Errorf("Want `%s`; have `%s`", want, have)
}
parsed := Parse(have)
if len(parsed) != 1 {
t.Errorf("Expected only 1 link")
}
if parsed[0].URL != l.URL {
t.Errorf("Re-parsed link header should have matching URL, but has `%s`", parsed[0].URL)
}
if parsed[0].Rel != l.Rel {
t.Errorf("Re-parsed link header should have matching rel, but has `%s`", parsed[0].Rel)
}
}