2019-07-17 15:02:42 -04:00
// Copyright 2019 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
2019-09-24 13:02:49 +08:00
package mailer
2019-07-17 15:02:42 -04:00
import (
2019-11-07 10:34:28 -03:00
"bytes"
2019-07-17 15:02:42 -04:00
"html/template"
"testing"
2019-11-07 10:34:28 -03:00
texttmpl "text/template"
2019-07-17 15:02:42 -04:00
2019-09-24 13:02:49 +08:00
"code.gitea.io/gitea/models"
2019-07-17 15:02:42 -04:00
"code.gitea.io/gitea/modules/setting"
"github.com/stretchr/testify/assert"
)
2019-11-07 10:34:28 -03:00
const subjectTpl = `
{ { . SubjectPrefix } } [ { { . Repo } } ] @ { { . Doer . Name } } # { { . Issue . Index } } - { { . Issue . Title } }
`
const bodyTpl = `
2019-07-17 15:02:42 -04:00
< ! DOCTYPE html >
< html >
< head >
< meta http - equiv = "Content-Type" content = "text/html; charset=utf-8" / >
< title > { { . Subject } } < / title >
< / head >
< body >
< p > { { . Body } } < / p >
< p >
-- -
< br >
< a href = "{{.Link}}" > View it on Gitea < / a > .
< / p >
< / body >
< / html >
`
func TestComposeIssueCommentMessage ( t * testing . T ) {
2019-09-24 13:02:49 +08:00
assert . NoError ( t , models . PrepareTestDatabase ( ) )
var mailService = setting . Mailer {
From : "test@gitea.com" ,
}
2019-07-17 15:02:42 -04:00
2019-09-24 13:02:49 +08:00
setting . MailService = & mailService
setting . Domain = "localhost"
2019-07-17 15:02:42 -04:00
2019-09-24 13:02:49 +08:00
doer := models . AssertExistsAndLoadBean ( t , & models . User { ID : 2 } ) . ( * models . User )
repo := models . AssertExistsAndLoadBean ( t , & models . Repository { ID : 1 , Owner : doer } ) . ( * models . Repository )
issue := models . AssertExistsAndLoadBean ( t , & models . Issue { ID : 1 , Repo : repo , Poster : doer } ) . ( * models . Issue )
comment := models . AssertExistsAndLoadBean ( t , & models . Comment { ID : 2 , Issue : issue } ) . ( * models . Comment )
2019-07-17 15:02:42 -04:00
2019-11-07 10:34:28 -03:00
stpl := texttmpl . Must ( texttmpl . New ( "issue/comment" ) . Parse ( subjectTpl ) )
btpl := template . Must ( template . New ( "issue/comment" ) . Parse ( bodyTpl ) )
InitMailRender ( stpl , btpl )
2019-07-17 15:02:42 -04:00
tos := [ ] string { "test@gitea.com" , "test2@gitea.com" }
2019-11-18 05:08:20 -03:00
msgs := composeIssueCommentMessages ( & mailCommentContext { Issue : issue , Doer : doer , ActionType : models . ActionCommentIssue ,
Content : "test body" , Comment : comment } , tos , false , "issue comment" )
assert . Len ( t , msgs , 2 )
2019-07-17 15:02:42 -04:00
2019-11-18 05:08:20 -03:00
mailto := msgs [ 0 ] . GetHeader ( "To" )
subject := msgs [ 0 ] . GetHeader ( "Subject" )
inreplyTo := msgs [ 0 ] . GetHeader ( "In-Reply-To" )
references := msgs [ 0 ] . GetHeader ( "References" )
2019-07-17 15:02:42 -04:00
2019-11-18 05:08:20 -03:00
assert . Len ( t , mailto , 1 , "exactly one recipient is expected in the To field" )
2019-11-07 10:34:28 -03:00
assert . Equal ( t , "Re: " , subject [ 0 ] [ : 4 ] , "Comment reply subject should contain Re:" )
assert . Equal ( t , "Re: [user2/repo1] @user2 #1 - issue1" , subject [ 0 ] )
2019-07-17 15:02:42 -04:00
assert . Equal ( t , inreplyTo [ 0 ] , "<user2/repo1/issues/1@localhost>" , "In-Reply-To header doesn't match" )
assert . Equal ( t , references [ 0 ] , "<user2/repo1/issues/1@localhost>" , "References header doesn't match" )
}
func TestComposeIssueMessage ( t * testing . T ) {
2019-09-24 13:02:49 +08:00
assert . NoError ( t , models . PrepareTestDatabase ( ) )
var mailService = setting . Mailer {
From : "test@gitea.com" ,
}
2019-07-17 15:02:42 -04:00
2019-09-24 13:02:49 +08:00
setting . MailService = & mailService
setting . Domain = "localhost"
2019-07-17 15:02:42 -04:00
2019-09-24 13:02:49 +08:00
doer := models . AssertExistsAndLoadBean ( t , & models . User { ID : 2 } ) . ( * models . User )
repo := models . AssertExistsAndLoadBean ( t , & models . Repository { ID : 1 , Owner : doer } ) . ( * models . Repository )
issue := models . AssertExistsAndLoadBean ( t , & models . Issue { ID : 1 , Repo : repo , Poster : doer } ) . ( * models . Issue )
2019-07-17 15:02:42 -04:00
2019-11-07 10:34:28 -03:00
stpl := texttmpl . Must ( texttmpl . New ( "issue/new" ) . Parse ( subjectTpl ) )
btpl := template . Must ( template . New ( "issue/new" ) . Parse ( bodyTpl ) )
InitMailRender ( stpl , btpl )
2019-07-17 15:02:42 -04:00
tos := [ ] string { "test@gitea.com" , "test2@gitea.com" }
2019-11-18 05:08:20 -03:00
msgs := composeIssueCommentMessages ( & mailCommentContext { Issue : issue , Doer : doer , ActionType : models . ActionCreateIssue ,
Content : "test body" } , tos , false , "issue create" )
assert . Len ( t , msgs , 2 )
2019-07-17 15:02:42 -04:00
2019-11-18 05:08:20 -03:00
mailto := msgs [ 0 ] . GetHeader ( "To" )
subject := msgs [ 0 ] . GetHeader ( "Subject" )
messageID := msgs [ 0 ] . GetHeader ( "Message-ID" )
2019-07-17 15:02:42 -04:00
2019-11-18 05:08:20 -03:00
assert . Len ( t , mailto , 1 , "exactly one recipient is expected in the To field" )
2019-11-07 10:34:28 -03:00
assert . Equal ( t , "[user2/repo1] @user2 #1 - issue1" , subject [ 0 ] )
2019-11-18 05:08:20 -03:00
assert . Nil ( t , msgs [ 0 ] . GetHeader ( "In-Reply-To" ) )
assert . Nil ( t , msgs [ 0 ] . GetHeader ( "References" ) )
2019-07-17 15:02:42 -04:00
assert . Equal ( t , messageID [ 0 ] , "<user2/repo1/issues/1@localhost>" , "Message-ID header doesn't match" )
}
2019-11-07 10:34:28 -03:00
func TestTemplateSelection ( t * testing . T ) {
assert . NoError ( t , models . PrepareTestDatabase ( ) )
var mailService = setting . Mailer {
From : "test@gitea.com" ,
}
setting . MailService = & mailService
setting . Domain = "localhost"
doer := models . AssertExistsAndLoadBean ( t , & models . User { ID : 2 } ) . ( * models . User )
repo := models . AssertExistsAndLoadBean ( t , & models . Repository { ID : 1 , Owner : doer } ) . ( * models . Repository )
issue := models . AssertExistsAndLoadBean ( t , & models . Issue { ID : 1 , Repo : repo , Poster : doer } ) . ( * models . Issue )
tos := [ ] string { "test@gitea.com" }
stpl := texttmpl . Must ( texttmpl . New ( "issue/default" ) . Parse ( "issue/default/subject" ) )
texttmpl . Must ( stpl . New ( "issue/new" ) . Parse ( "issue/new/subject" ) )
texttmpl . Must ( stpl . New ( "pull/comment" ) . Parse ( "pull/comment/subject" ) )
texttmpl . Must ( stpl . New ( "issue/close" ) . Parse ( "" ) ) // Must default to fallback subject
btpl := template . Must ( template . New ( "issue/default" ) . Parse ( "issue/default/body" ) )
template . Must ( btpl . New ( "issue/new" ) . Parse ( "issue/new/body" ) )
template . Must ( btpl . New ( "pull/comment" ) . Parse ( "pull/comment/body" ) )
template . Must ( btpl . New ( "issue/close" ) . Parse ( "issue/close/body" ) )
InitMailRender ( stpl , btpl )
expect := func ( t * testing . T , msg * Message , expSubject , expBody string ) {
subject := msg . GetHeader ( "Subject" )
msgbuf := new ( bytes . Buffer )
_ , _ = msg . WriteTo ( msgbuf )
wholemsg := msgbuf . String ( )
assert . Equal ( t , [ ] string { expSubject } , subject )
assert . Contains ( t , wholemsg , expBody )
}
2019-11-18 05:08:20 -03:00
msg := testComposeIssueCommentMessage ( t , & mailCommentContext { Issue : issue , Doer : doer , ActionType : models . ActionCreateIssue ,
Content : "test body" } , tos , false , "TestTemplateSelection" )
2019-11-07 10:34:28 -03:00
expect ( t , msg , "issue/new/subject" , "issue/new/body" )
comment := models . AssertExistsAndLoadBean ( t , & models . Comment { ID : 2 , Issue : issue } ) . ( * models . Comment )
2019-11-18 05:08:20 -03:00
msg = testComposeIssueCommentMessage ( t , & mailCommentContext { Issue : issue , Doer : doer , ActionType : models . ActionCommentIssue ,
Content : "test body" , Comment : comment } , tos , false , "TestTemplateSelection" )
2019-11-07 10:34:28 -03:00
expect ( t , msg , "issue/default/subject" , "issue/default/body" )
pull := models . AssertExistsAndLoadBean ( t , & models . Issue { ID : 2 , Repo : repo , Poster : doer } ) . ( * models . Issue )
comment = models . AssertExistsAndLoadBean ( t , & models . Comment { ID : 4 , Issue : pull } ) . ( * models . Comment )
2019-12-22 03:29:26 -05:00
msg = testComposeIssueCommentMessage ( t , & mailCommentContext { Issue : pull , Doer : doer , ActionType : models . ActionCommentPull ,
2019-11-18 05:08:20 -03:00
Content : "test body" , Comment : comment } , tos , false , "TestTemplateSelection" )
2019-11-07 10:34:28 -03:00
expect ( t , msg , "pull/comment/subject" , "pull/comment/body" )
2019-11-18 05:08:20 -03:00
msg = testComposeIssueCommentMessage ( t , & mailCommentContext { Issue : issue , Doer : doer , ActionType : models . ActionCloseIssue ,
Content : "test body" , Comment : comment } , tos , false , "TestTemplateSelection" )
expect ( t , msg , "Re: [user2/repo1] issue1 (#1)" , "issue/close/body" )
2019-11-07 10:34:28 -03:00
}
func TestTemplateServices ( t * testing . T ) {
assert . NoError ( t , models . PrepareTestDatabase ( ) )
var mailService = setting . Mailer {
From : "test@gitea.com" ,
}
setting . MailService = & mailService
setting . Domain = "localhost"
doer := models . AssertExistsAndLoadBean ( t , & models . User { ID : 2 } ) . ( * models . User )
repo := models . AssertExistsAndLoadBean ( t , & models . Repository { ID : 1 , Owner : doer } ) . ( * models . Repository )
issue := models . AssertExistsAndLoadBean ( t , & models . Issue { ID : 1 , Repo : repo , Poster : doer } ) . ( * models . Issue )
comment := models . AssertExistsAndLoadBean ( t , & models . Comment { ID : 2 , Issue : issue } ) . ( * models . Comment )
assert . NoError ( t , issue . LoadRepo ( ) )
expect := func ( t * testing . T , issue * models . Issue , comment * models . Comment , doer * models . User ,
actionType models . ActionType , fromMention bool , tplSubject , tplBody , expSubject , expBody string ) {
stpl := texttmpl . Must ( texttmpl . New ( "issue/default" ) . Parse ( tplSubject ) )
btpl := template . Must ( template . New ( "issue/default" ) . Parse ( tplBody ) )
InitMailRender ( stpl , btpl )
tos := [ ] string { "test@gitea.com" }
2019-11-18 05:08:20 -03:00
msg := testComposeIssueCommentMessage ( t , & mailCommentContext { Issue : issue , Doer : doer , ActionType : actionType ,
Content : "test body" , Comment : comment } , tos , fromMention , "TestTemplateServices" )
2019-11-07 10:34:28 -03:00
subject := msg . GetHeader ( "Subject" )
msgbuf := new ( bytes . Buffer )
_ , _ = msg . WriteTo ( msgbuf )
wholemsg := msgbuf . String ( )
assert . Equal ( t , [ ] string { expSubject } , subject )
assert . Contains ( t , wholemsg , "\r\n" + expBody + "\r\n" )
}
expect ( t , issue , comment , doer , models . ActionCommentIssue , false ,
"{{.SubjectPrefix}}[{{.Repo}}]: @{{.Doer.Name}} commented on #{{.Issue.Index}} - {{.Issue.Title}}" ,
"//{{.ActionType}},{{.ActionName}},{{if .IsMention}}norender{{end}}//" ,
"Re: [user2/repo1]: @user2 commented on #1 - issue1" ,
"//issue,comment,//" )
expect ( t , issue , comment , doer , models . ActionCommentIssue , true ,
"{{if .IsMention}}must render{{end}}" ,
"//subject is: {{.Subject}}//" ,
"must render" ,
"//subject is: must render//" )
expect ( t , issue , comment , doer , models . ActionCommentIssue , true ,
"{{.FallbackSubject}}" ,
"//{{.SubjectPrefix}}//" ,
"Re: [user2/repo1] issue1 (#1)" ,
"//Re: //" )
}
2019-11-18 05:08:20 -03:00
func testComposeIssueCommentMessage ( t * testing . T , ctx * mailCommentContext , tos [ ] string , fromMention bool , info string ) * Message {
msgs := composeIssueCommentMessages ( ctx , tos , fromMention , info )
assert . Len ( t , msgs , 1 )
return msgs [ 0 ]
}
