mirror of https://github.com/go-gitea/gitea.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.9 KiB
60 lines
1.9 KiB
// Copyright 2022 The Gitea Authors. All rights reserved. |
|
// SPDX-License-Identifier: MIT |
|
|
|
package integration |
|
|
|
import ( |
|
"bytes" |
|
"io" |
|
"net/http" |
|
"strings" |
|
"testing" |
|
|
|
"code.gitea.io/gitea/modules/markup" |
|
"code.gitea.io/gitea/modules/markup/external" |
|
"code.gitea.io/gitea/modules/setting" |
|
"code.gitea.io/gitea/tests" |
|
|
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestExternalMarkupRenderer(t *testing.T) { |
|
defer tests.PrepareTestEnv(t)() |
|
if !setting.Database.Type.IsSQLite3() { |
|
t.Skip() |
|
return |
|
} |
|
|
|
req := NewRequest(t, "GET", "/user30/renderer/src/branch/master/README.html") |
|
resp := MakeRequest(t, req, http.StatusOK) |
|
assert.Equal(t, "text/html; charset=utf-8", resp.Header().Get("Content-Type")) |
|
|
|
bs, err := io.ReadAll(resp.Body) |
|
assert.NoError(t, err) |
|
|
|
doc := NewHTMLParser(t, bytes.NewBuffer(bs)) |
|
div := doc.Find("div.file-view") |
|
data, err := div.Html() |
|
assert.NoError(t, err) |
|
assert.Equal(t, "<div>\n\ttest external renderer\n</div>", strings.TrimSpace(data)) |
|
|
|
r := markup.GetRendererByFileName("a.html").(*external.Renderer) |
|
r.RenderContentMode = setting.RenderContentModeIframe |
|
|
|
req = NewRequest(t, "GET", "/user30/renderer/src/branch/master/README.html") |
|
resp = MakeRequest(t, req, http.StatusOK) |
|
assert.Equal(t, "text/html; charset=utf-8", resp.Header().Get("Content-Type")) |
|
bs, err = io.ReadAll(resp.Body) |
|
assert.NoError(t, err) |
|
doc = NewHTMLParser(t, bytes.NewBuffer(bs)) |
|
iframe := doc.Find("iframe") |
|
assert.Equal(t, "/user30/renderer/render/branch/master/README.html", iframe.AttrOr("src", "")) |
|
|
|
req = NewRequest(t, "GET", "/user30/renderer/render/branch/master/README.html") |
|
resp = MakeRequest(t, req, http.StatusOK) |
|
assert.Equal(t, "text/html; charset=utf-8", resp.Header().Get("Content-Type")) |
|
bs, err = io.ReadAll(resp.Body) |
|
assert.NoError(t, err) |
|
assert.Equal(t, "frame-src 'self'; sandbox allow-scripts", resp.Header().Get("Content-Security-Policy")) |
|
assert.Equal(t, "<div>\n\ttest external renderer\n</div>\n", string(bs)) |
|
}
|
|
|