linkinfo/default_test.go

70 lines
1.5 KiB
Go
Raw Normal View History

package linkinfo
2019-10-03 23:44:38 +00:00
import (
"strings"
2019-10-03 23:44:38 +00:00
"testing"
)
func TestLinkInfoApi_detectContentType(t *testing.T) {
api := New()
contentType := api.detectContentType("http://example.com", "")
if contentType == "" {
t.Fatal("Unable to test example.com, returned empty type")
}
if idx := strings.Index(contentType, ";"); idx != -1 {
contentType = contentType[:idx]
}
if contentType != contentTypeHtml {
t.Fatal("Unexpected content type:", contentType)
2019-10-03 23:44:38 +00:00
}
}
2019-10-03 23:44:38 +00:00
func TestLinkInfoApi_defaultLinkHandler(t *testing.T) {
2019-10-03 23:44:38 +00:00
type expectedData struct {
Link string
2019-10-03 23:44:38 +00:00
Title string
Type string
2019-10-03 23:44:38 +00:00
}
testLinks := []expectedData{
{"http://example.com", "Example Domain", "text/html"},
2019-10-03 23:44:38 +00:00
{"http://techslides.com/demos/sample-videos/small.mp4", "", "video/mp4"},
}
api := New()
2019-10-03 23:44:38 +00:00
for _, link := range testLinks {
l, err := api.DefaultLinkHandler(link.Link)
2019-10-03 23:44:38 +00:00
if err != nil {
t.Fatal("Unable to retrieve link info:", err)
}
if link.Title != "" && link.Title != l.Title {
t.Fatal("Unexpected title, expected:", link.Title, "got:", l.Title)
}
if link.Type != "" && link.Type != l.ContentType {
t.Fatal("Unexpected content type, expected:", link.Type, "got:", l.ContentType)
}
}
}
func TestLinkInfoApi_retrieveHtmlLinkTitle(t *testing.T) {
ret := &LinkInfo{}
api := New()
if err := api.retrieveHtmlLinkTitle(ret, "http://example.com"); err != nil {
t.Fatal("Unable to retrieve html link title:", err)
}
if ret.Title != "Example Domain" {
t.Fatal("Unexpected title", ret.Title)
}
}