70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package httpgetter
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type roundTripFunc func(*http.Request) (*http.Response, error)
|
|
|
|
func (f roundTripFunc) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
return f(req)
|
|
}
|
|
|
|
func TestGetHTMLMeta(t *testing.T) {
|
|
originalHTTPClient := httpClient
|
|
t.Cleanup(func() {
|
|
httpClient = originalHTTPClient
|
|
})
|
|
|
|
httpClient = &http.Client{
|
|
Transport: roundTripFunc(func(req *http.Request) (*http.Response, error) {
|
|
require.Equal(t, "http://93.184.216.34/article", req.URL.String())
|
|
return &http.Response{
|
|
StatusCode: http.StatusOK,
|
|
Header: http.Header{"Content-Type": []string{"text/html; charset=utf-8"}},
|
|
Body: io.NopCloser(strings.NewReader(`<!doctype html>
|
|
<html>
|
|
<head>
|
|
<title>Fallback title</title>
|
|
<meta name="description" content="Fallback description">
|
|
<meta property="og:title" content="Open Graph title">
|
|
<meta property="og:description" content="Open Graph description">
|
|
<meta property="og:image" content="https://example.com/cover.png">
|
|
</head>
|
|
<body>ignored</body>
|
|
</html>`)),
|
|
Request: req,
|
|
}, nil
|
|
}),
|
|
}
|
|
|
|
metadata, err := GetHTMLMeta("http://93.184.216.34/article")
|
|
require.NoError(t, err)
|
|
require.Equal(t, HTMLMeta{
|
|
Title: "Open Graph title",
|
|
Description: "Open Graph description",
|
|
Image: "https://example.com/cover.png",
|
|
}, *metadata)
|
|
}
|
|
|
|
func TestGetHTMLMetaForInternal(t *testing.T) {
|
|
// test for internal IP
|
|
if _, err := GetHTMLMeta("http://192.168.0.1"); !errors.Is(err, ErrInternalIP) {
|
|
t.Errorf("Expected error for internal IP, got %v", err)
|
|
}
|
|
|
|
// test for resolved internal IP
|
|
if _, err := GetHTMLMeta("http://localhost"); !errors.Is(err, ErrInternalIP) {
|
|
t.Errorf("Expected error for resolved internal IP, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestHTTPClientHasTimeout(t *testing.T) {
|
|
require.NotZero(t, httpClient.Timeout)
|
|
}
|