mirror of
https://github.com/usememos/memos.git
synced 2026-05-02 18:52:32 +00:00
Steven
94 lines
2.6 KiB
Go
94 lines
2.6 KiB
Go
package frontend
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/labstack/echo/v5"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/usememos/memos/internal/profile"
|
|
"github.com/usememos/memos/store"
|
|
teststore "github.com/usememos/memos/store/test"
|
|
)
|
|
|
|
func TestFrontendService_RobotsTXT(t *testing.T) {
|
|
ctx := context.Background()
|
|
testStore := teststore.NewTestingStore(ctx, t)
|
|
profile := &profile.Profile{
|
|
InstanceURL: "https://demo.usememos.com/",
|
|
}
|
|
|
|
e := echo.New()
|
|
NewFrontendService(profile, testStore).Serve(ctx, e)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/robots.txt", nil)
|
|
rec := httptest.NewRecorder()
|
|
e.ServeHTTP(rec, req)
|
|
|
|
require.Equal(t, http.StatusOK, rec.Code)
|
|
require.Equal(t, "text/plain; charset=UTF-8", rec.Header().Get("Content-Type"))
|
|
require.Equal(t, "User-agent: *\nAllow: /\nHost: https://demo.usememos.com\nSitemap: https://demo.usememos.com/sitemap.xml", rec.Body.String())
|
|
}
|
|
|
|
func TestFrontendService_SitemapXML(t *testing.T) {
|
|
ctx := context.Background()
|
|
testStore := teststore.NewTestingStore(ctx, t)
|
|
profile := &profile.Profile{
|
|
InstanceURL: "https://demo.usememos.com",
|
|
}
|
|
|
|
user, err := testStore.CreateUser(ctx, &store.User{
|
|
Username: "sitemap-owner",
|
|
Role: store.RoleUser,
|
|
Email: "sitemap-owner@example.com",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
_, err = testStore.CreateMemo(ctx, &store.Memo{
|
|
UID: "publicmemo",
|
|
CreatorID: user.ID,
|
|
Content: "public memo",
|
|
Visibility: store.Public,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
_, err = testStore.CreateMemo(ctx, &store.Memo{
|
|
UID: "privatememo",
|
|
CreatorID: user.ID,
|
|
Content: "private memo",
|
|
Visibility: store.Private,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
e := echo.New()
|
|
NewFrontendService(profile, testStore).Serve(ctx, e)
|
|
|
|
req := httptest.NewRequest(http.MethodGet, "/sitemap.xml", nil)
|
|
rec := httptest.NewRecorder()
|
|
e.ServeHTTP(rec, req)
|
|
|
|
require.Equal(t, http.StatusOK, rec.Code)
|
|
require.Contains(t, rec.Header().Get("Content-Type"), "application/xml")
|
|
require.Contains(t, rec.Body.String(), `<loc>https://demo.usememos.com/memos/publicmemo</loc>`)
|
|
require.NotContains(t, rec.Body.String(), "privatememo")
|
|
}
|
|
|
|
func TestFrontendService_SitemapRoutesRequireInstanceURL(t *testing.T) {
|
|
ctx := context.Background()
|
|
testStore := teststore.NewTestingStore(ctx, t)
|
|
|
|
e := echo.New()
|
|
NewFrontendService(&profile.Profile{}, testStore).Serve(ctx, e)
|
|
|
|
for _, path := range []string{"/robots.txt", "/sitemap.xml"} {
|
|
req := httptest.NewRequest(http.MethodGet, path, nil)
|
|
rec := httptest.NewRecorder()
|
|
e.ServeHTTP(rec, req)
|
|
|
|
require.Equal(t, http.StatusNotFound, rec.Code)
|
|
}
|
|
}
|