mirror of
https://github.com/usememos/memos.git
synced 2025-10-27 23:06:24 +08:00
39 lines
1 KiB
Go
39 lines
1 KiB
Go
package httpgetter
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGetHTMLMeta(t *testing.T) {
|
|
tests := []struct {
|
|
urlStr string
|
|
htmlMeta HTMLMeta
|
|
}{}
|
|
for _, test := range tests {
|
|
metadata, err := GetHTMLMeta(test.urlStr)
|
|
require.NoError(t, err)
|
|
require.Equal(t, test.htmlMeta, *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)
|
|
}
|
|
|
|
// test for redirected internal IP
|
|
// 49.232.126.226:1110 will redirects to 127.0.0.1
|
|
if _, err := GetHTMLMeta("http://49.232.126.226:1110"); !(errors.Is(err, ErrInternalIP) && strings.Contains(err.Error(), "redirect")) {
|
|
t.Errorf("Expected error for redirected internal IP, got %v", err)
|
|
}
|
|
}
|