0.29.1原版

2026-07-02 19:14:14 +08:00
commit d94008f0fb
947 changed files with 174905 additions and 0 deletions
@@ -0,0 +1,103 @@
+package v1
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+
+	"github.com/usememos/memos/internal/httpgetter"
+	v1pb "github.com/usememos/memos/proto/gen/api/v1"
+)
+
+func TestGetLinkMetadata(t *testing.T) {
+	originalFetchHTMLMeta := fetchHTMLMeta
+	t.Cleanup(func() {
+		fetchHTMLMeta = originalFetchHTMLMeta
+	})
+
+	fetchHTMLMeta = func(url string) (*httpgetter.HTMLMeta, error) {
+		require.Equal(t, "https://example.com/article", url)
+		return &httpgetter.HTMLMeta{
+			Title:       "Example title",
+			Description: "Example description",
+			Image:       "https://example.com/cover.png",
+		}, nil
+	}
+
+	metadata, err := (&APIV1Service{}).GetLinkMetadata(context.Background(), &v1pb.GetLinkMetadataRequest{
+		Url: "https://example.com/article",
+	})
+	require.NoError(t, err)
+	require.Equal(t, "https://example.com/article", metadata.Url)
+	require.Equal(t, "Example title", metadata.Title)
+	require.Equal(t, "Example description", metadata.Description)
+	require.Equal(t, "https://example.com/cover.png", metadata.Image)
+}
+
+func TestGetLinkMetadataEmptyURL(t *testing.T) {
+	_, err := (&APIV1Service{}).GetLinkMetadata(context.Background(), &v1pb.GetLinkMetadataRequest{})
+	require.Error(t, err)
+	require.Equal(t, codes.InvalidArgument, status.Code(err))
+}
+
+func TestGetLinkMetadataInternalURL(t *testing.T) {
+	_, err := (&APIV1Service{}).GetLinkMetadata(context.Background(), &v1pb.GetLinkMetadataRequest{
+		Url: "http://192.168.0.1",
+	})
+	require.Error(t, err)
+	require.Equal(t, codes.InvalidArgument, status.Code(err))
+}
+
+func TestBatchGetLinkMetadata(t *testing.T) {
+	originalFetchHTMLMeta := fetchHTMLMeta
+	t.Cleanup(func() {
+		fetchHTMLMeta = originalFetchHTMLMeta
+	})
+
+	var fetchedURLs []string
+	fetchHTMLMeta = func(url string) (*httpgetter.HTMLMeta, error) {
+		fetchedURLs = append(fetchedURLs, url)
+		return &httpgetter.HTMLMeta{
+			Title:       fmt.Sprintf("Title for %s", url),
+			Description: fmt.Sprintf("Description for %s", url),
+			Image:       fmt.Sprintf("%s/cover.png", url),
+		}, nil
+	}
+
+	response, err := (&APIV1Service{}).BatchGetLinkMetadata(context.Background(), &v1pb.BatchGetLinkMetadataRequest{
+		Urls: []string{
+			"https://example.com/one",
+			"https://example.com/two",
+		},
+	})
+	require.NoError(t, err)
+	require.Equal(t, []string{"https://example.com/one", "https://example.com/two"}, fetchedURLs)
+	require.Len(t, response.LinkMetadata, 2)
+	require.Equal(t, "https://example.com/one", response.LinkMetadata[0].Url)
+	require.Equal(t, "Title for https://example.com/one", response.LinkMetadata[0].Title)
+	require.Equal(t, "https://example.com/two", response.LinkMetadata[1].Url)
+	require.Equal(t, "Title for https://example.com/two", response.LinkMetadata[1].Title)
+}
+
+func TestBatchGetLinkMetadataEmptyURLs(t *testing.T) {
+	_, err := (&APIV1Service{}).BatchGetLinkMetadata(context.Background(), &v1pb.BatchGetLinkMetadataRequest{})
+	require.Error(t, err)
+	require.Equal(t, codes.InvalidArgument, status.Code(err))
+}
+
+func TestBatchGetLinkMetadataTooManyURLs(t *testing.T) {
+	urls := make([]string, maxBatchGetLinkMetadata+1)
+	for i := range urls {
+		urls[i] = fmt.Sprintf("https://example.com/%d", i)
+	}
+
+	_, err := (&APIV1Service{}).BatchGetLinkMetadata(context.Background(), &v1pb.BatchGetLinkMetadataRequest{
+		Urls: urls,
+	})
+	require.Error(t, err)
+	require.Equal(t, codes.InvalidArgument, status.Code(err))
+}