@ -4,18 +4,23 @@
package integration
import (
"bytes"
"io"
"net/http"
"net/url"
"strings"
"testing"
repo_model "code.gitea.io/gitea/models/repo"
"code.gitea.io/gitea/models/unittest"
user_model "code.gitea.io/gitea/models/user"
"code.gitea.io/gitea/modules/markup"
"code.gitea.io/gitea/modules/markup/external"
"code.gitea.io/gitea/modules/setting"
"code.gitea.io/gitea/modules/test"
"code.gitea.io/gitea/tests"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestExternalMarkupRenderer ( t * testing . T ) {
@ -25,36 +30,52 @@ func TestExternalMarkupRenderer(t *testing.T) {
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 ) )
onGiteaRun ( t , func ( t * testing . T , _ * url . URL ) {
t . Run ( "RenderNoSanitizer" , func ( t * testing . T ) {
user2 := unittest . AssertExistsAndLoadBean ( t , & user_model . User { ID : 2 } )
repo1 := unittest . AssertExistsAndLoadBean ( t , & repo_model . Repository { ID : 1 } )
_ , err := createFile ( user2 , repo1 , "file.no-sanitizer" , "master" , ` any content ` )
require . NoError ( t , err )
req := NewRequest ( t , "GET" , "/user2/repo1/src/branch/master/file.no-sanitizer" )
resp := MakeRequest ( t , req , http . StatusOK )
doc := NewHTMLParser ( t , resp . Body )
div := doc . Find ( "div.file-view" )
data , err := div . Html ( )
assert . NoError ( t , err )
assert . Equal ( t , ` <script>window.alert("hi")</script> ` , strings . TrimSpace ( data ) )
} )
} )
t . Run ( "RenderContentDirectly" , func ( t * testing . T ) {
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" ) )
doc := NewHTMLParser ( t , resp . Body )
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 ( "any-file.html" ) . ( * external . Renderer )
defer test . MockVariableValue ( & r . RenderContentMode , setting . RenderContentModeIframe ) ( )
t . Run ( "RenderContentInIFrame" , func ( t * testing . T ) {
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" ) )
doc := NewHTMLParser ( t , resp . Body )
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 ) )
} )
}