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.

mail_test.go 2.7KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687
  1. // Copyright 2019 The Gitea Authors. All rights reserved.
  2. // Use of this source code is governed by a MIT-style
  3. // license that can be found in the LICENSE file.
  4. package models
  5. import (
  6. "html/template"
  7. "testing"
  8. "code.gitea.io/gitea/modules/setting"
  9. "github.com/stretchr/testify/assert"
  10. )
  11. const tmpl = `
  12. <!DOCTYPE html>
  13. <html>
  14. <head>
  15. <meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
  16. <title>{{.Subject}}</title>
  17. </head>
  18. <body>
  19. <p>{{.Body}}</p>
  20. <p>
  21. ---
  22. <br>
  23. <a href="{{.Link}}">View it on Gitea</a>.
  24. </p>
  25. </body>
  26. </html>
  27. `
  28. func TestComposeIssueCommentMessage(t *testing.T) {
  29. assert.NoError(t, PrepareTestDatabase())
  30. var MailService setting.Mailer
  31. MailService.From = "test@gitea.com"
  32. setting.MailService = &MailService
  33. doer := AssertExistsAndLoadBean(t, &User{ID: 2}).(*User)
  34. repo := AssertExistsAndLoadBean(t, &Repository{ID: 1, Owner: doer}).(*Repository)
  35. issue := AssertExistsAndLoadBean(t, &Issue{ID: 1, Repo: repo, Poster: doer}).(*Issue)
  36. comment := AssertExistsAndLoadBean(t, &Comment{ID: 2, Issue: issue}).(*Comment)
  37. email := template.Must(template.New("issue/comment").Parse(tmpl))
  38. InitMailRender(email)
  39. tos := []string{"test@gitea.com", "test2@gitea.com"}
  40. msg := composeIssueCommentMessage(issue, doer, "test body", comment, mailIssueComment, tos, "issue comment")
  41. subject := msg.GetHeader("Subject")
  42. inreplyTo := msg.GetHeader("In-Reply-To")
  43. references := msg.GetHeader("References")
  44. assert.Equal(t, subject[0], "Re: "+issue.mailSubject(), "Comment reply subject should contain Re:")
  45. assert.Equal(t, inreplyTo[0], "<user2/repo1/issues/1@localhost>", "In-Reply-To header doesn't match")
  46. assert.Equal(t, references[0], "<user2/repo1/issues/1@localhost>", "References header doesn't match")
  47. }
  48. func TestComposeIssueMessage(t *testing.T) {
  49. assert.NoError(t, PrepareTestDatabase())
  50. var MailService setting.Mailer
  51. MailService.From = "test@gitea.com"
  52. setting.MailService = &MailService
  53. doer := AssertExistsAndLoadBean(t, &User{ID: 2}).(*User)
  54. repo := AssertExistsAndLoadBean(t, &Repository{ID: 1, Owner: doer}).(*Repository)
  55. issue := AssertExistsAndLoadBean(t, &Issue{ID: 1, Repo: repo, Poster: doer}).(*Issue)
  56. email := template.Must(template.New("issue/comment").Parse(tmpl))
  57. InitMailRender(email)
  58. tos := []string{"test@gitea.com", "test2@gitea.com"}
  59. msg := composeIssueCommentMessage(issue, doer, "test body", nil, mailIssueComment, tos, "issue create")
  60. subject := msg.GetHeader("Subject")
  61. messageID := msg.GetHeader("Message-ID")
  62. assert.Equal(t, subject[0], issue.mailSubject(), "Subject not equal to issue.mailSubject()")
  63. assert.Nil(t, msg.GetHeader("In-Reply-To"))
  64. assert.Nil(t, msg.GetHeader("References"))
  65. assert.Equal(t, messageID[0], "<user2/repo1/issues/1@localhost>", "Message-ID header doesn't match")
  66. }