49 lines
1.6 KiB
Go
49 lines
1.6 KiB
Go
|
package owl_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"h4kor/owl-blogs"
|
||
|
|
"h4kor/owl-blogs/test/assertions"
|
||
|
|
"net/http"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetRedirctUrisLink(t *testing.T) {
|
||
|
|
html := []byte("<link rel=\"redirect_uri\" href=\"http://example.com/redirect\" />")
|
||
|
|
parser := &owl.OwlHtmlParser{}
|
||
|
|
uris, err := parser.GetRedirctUris(constructResponse(html))
|
||
|
|
|
||
|
|
assertions.AssertNoError(t, err, "Unable to parse feed")
|
||
|
|
|
||
|
|
assertions.AssertArrayContains(t, uris, "http://example.com/redirect")
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRedirctUrisLinkMultiple(t *testing.T) {
|
||
|
|
html := []byte(`
|
||
|
|
<link rel="redirect_uri" href="http://example.com/redirect1" />
|
||
|
|
<link rel="redirect_uri" href="http://example.com/redirect2" />
|
||
|
|
<link rel="redirect_uri" href="http://example.com/redirect3" />
|
||
|
|
<link rel="foo" href="http://example.com/redirect4" />
|
||
|
|
<link href="http://example.com/redirect5" />
|
||
|
|
`)
|
||
|
|
parser := &owl.OwlHtmlParser{}
|
||
|
|
uris, err := parser.GetRedirctUris(constructResponse(html))
|
||
|
|
|
||
|
|
assertions.AssertNoError(t, err, "Unable to parse feed")
|
||
|
|
|
||
|
|
assertions.AssertArrayContains(t, uris, "http://example.com/redirect1")
|
||
|
|
assertions.AssertArrayContains(t, uris, "http://example.com/redirect2")
|
||
|
|
assertions.AssertArrayContains(t, uris, "http://example.com/redirect3")
|
||
|
|
assertions.AssertLen(t, uris, 3)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGetRedirectUrisLinkHeader(t *testing.T) {
|
||
|
|
html := []byte("")
|
||
|
|
parser := &owl.OwlHtmlParser{}
|
||
|
|
resp := constructResponse(html)
|
||
|
|
resp.Header = http.Header{"Link": []string{"<http://example.com/redirect>; rel=\"redirect_uri\""}}
|
||
|
|
uris, err := parser.GetRedirctUris(resp)
|
||
|
|
|
||
|
|
assertions.AssertNoError(t, err, "Unable to parse feed")
|
||
|
|
assertions.AssertArrayContains(t, uris, "http://example.com/redirect")
|
||
|
|
}
|