temp fix

Author: wxiaoguang

services/mailer/mail.go

@@ -75,7 +75,6 @@ func Base64InlineImages(body string, ctx *mailCommentContext) (string, error) {
 							log.Trace("attachmentSrcToDataURI not possible: %v", err) // Not an error, just skip. This is probably an image from outside the gitea instance.
 							continue
 						}
-						log.Trace("Old value of src attribute: %s, new value (first 100 characters): %s", attr.Val, dataURI[:100])
 						n.Attr[i].Val = dataURI
 						break
 					}
@@ -115,7 +114,7 @@ func AttachmentSrcToBase64DataURI(attachmentPath string, ctx *mailCommentContext
 	}
 
 	// "Doer" is theoretically not the correct permission check (as Doer created the action on which to send), but as this is batch processed the receipants can't be accessed.
-	// Therefore we check the Doer, with which we counter leaking information as a Doer brute force attack on attachments would be possible.
+	// Therefore, we check the Doer, with which we counter leaking information as a Doer brute force attack on attachments would be possible.
 	perm, err := access_model.GetUserRepoPermission(ctx, ctx.Issue.Repo, ctx.Doer)
 	if err != nil {
 		return "", err

services/mailer/mail_test.go

@@ -5,6 +5,7 @@ package mailer
 
 import (
 	"bytes"
+	"code.gitea.io/gitea/services/attachment"
 	"context"
 	"fmt"
 	"html/template"
@@ -55,23 +56,37 @@ const bodyTpl = `
 
 func prepareMailerTest(t *testing.T) (doer *user_model.User, repo *repo_model.Repository, issue *issues_model.Issue, comment *issues_model.Comment) {
 	assert.NoError(t, unittest.PrepareTestDatabase())
-	mailService := setting.Mailer{
-		From: "[email protected]",
-	}
-
-	setting.MailService = &mailService
+	setting.MailService = &setting.Mailer{From: "[email protected]"}
 	setting.Domain = "localhost"
 	setting.AppURL = "https://try.gitea.io/"
 
 	doer = unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
 	repo = unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1, Owner: doer})
 	issue = unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{ID: 1, Repo: repo, Poster: doer})
-	assert.NoError(t, issue.LoadRepo(db.DefaultContext))
 	comment = unittest.AssertExistsAndLoadBean(t, &issues_model.Comment{ID: 2, Issue: issue})
+	require.NoError(t, issue.LoadRepo(db.DefaultContext))
 	return doer, repo, issue, comment
 }
 
-func TestComposeIssueCommentMessage(t *testing.T) {
+func prepareMailerBase64Test(t *testing.T) (doer *user_model.User, repo *repo_model.Repository, issue *issues_model.Issue) {
+	user, repo, issue, comment := prepareMailerTest(t)
+	setting.MailService.Base64EmbedImages = true
+	setting.MailService.Base64EmbedImagesMaxSizePerEmail = 10 * 1024 * 1024
+	att, err := attachment.NewAttachment(t.Context(), &repo_model.Attachment{
+		UUID:       "1234",
+		RepoID:     repo.ID,
+		IssueID:    issue.ID,
+		UploaderID: user.ID,
+		CommentID:  comment.ID,
+		Name:       "test.png",
+	}, bytes.NewReader([]byte("\x89\x50\x4e\x47\x0d\x0a\x1a\x0a")), 8)
+	require.NoError(t, err)
+	issue.Content = fmt.Sprintf(`MSG-BEFORE <image src="attachments/%s"> MSG-AFTER`, att.UUID)
+	require.NoError(t, issues_model.UpdateIssueCols(t.Context(), issue, "content"))
+	return user, repo, issue
+}
+
+func TestComposeIssueCommen(t *testing.T) {
 	doer, _, issue, comment := prepareMailerTest(t)
 
 	markup.Init(&markup.RenderHelperFuncs{
@@ -455,20 +470,7 @@ func TestFromDisplayName(t *testing.T) {
 }
 
 func TestEmbedBase64ImagesInEmail(t *testing.T) {
-	doer, repo, _, _ := prepareMailerTest(t)
-	defer test.MockVariableValue(&setting.MailService.Base64EmbedImages, true)
-	defer test.MockVariableValue(&setting.MailService.Base64EmbedImagesMaxSizePerEmail, 10*1024*1024)
-
-	err := issues_model.NewIssue(t.Context(), repo, &issues_model.Issue{
-		Poster:  doer,
-		RepoID:  repo.ID,
-		Title:   "test issue attachment",
-		Content: `content including this image: <image alt="gitea.png" src="attachments/1b267670-1793-4cd0-abc1-449269b7cff9" /> with some more content behind it`,
-	}, nil, nil)
-	require.NoError(t, err)
-	issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{Title: "test issue attachment"})
-	require.NoError(t, issue.LoadRepo(t.Context()))
-
+	doer, _, issue := prepareMailerBase64Test(t)
 	subjectTemplates = texttmpl.Must(texttmpl.New("issue/new").Parse(subjectTpl))
 	bodyTemplates = template.Must(template.New("issue/new").Parse(bodyTpl))
 
@@ -480,43 +482,21 @@ func TestEmbedBase64ImagesInEmail(t *testing.T) {
 		ActionType: activities_model.ActionCreateIssue,
 		Content:    strings.ReplaceAll(issue.Content, `src="`, `src="`+setting.AppURL),
 	}, "en-US", recipients, false, "issue create")
+	require.NoError(t, err)
 
 	mailBody := msgs[0].Body
-	re := regexp.MustCompile(`(?s)<body>(.*?)</body>`)
+	re := regexp.MustCompile(`MSG-BEFORE.*MSG-AFTER`)
 	matches := re.FindStringSubmatch(mailBody)
-	if len(matches) > 1 {
-		mailBody = matches[1]
-	}
-	// check if the mail body was correctly generated
-	assert.NoError(t, err)
-	assert.Contains(t, mailBody, "content including this image")
-
-	// check if an image was embedded
-	assert.Contains(t, mailBody, "data:image/png;base64,")
-
-	// check if the image was embedded only once
-	assert.Equal(t, 1, strings.Count(mailBody, "data:image/png;base64,"))
-
-	img2InternalBase64 := "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAFAAAAAxAQMAAAB3d7wRAAAABlBMVEVgmyF6qkqITHmkAAAAAXRSTlMBN+Ho8AAAAJhJREFUKM+V0DsOwyAQBNCxXLjkCFwk0t7McDQfhS4tpQuEzWc/iaUU2eo1zC4DUMWYF3DxVKzGTXjBGb2RsjJEo6ZhN1Zj+cEgi/9hBQl3YflkkIsbo5IO5glKTuhPpavM3Hp4C7WdjEWYrL5GMkp/R+s4GPlh/CZn4MEwv9aHHiyD3ujm5X22eaMyDa5yAm+O0B1TPa1l3W2qZWMg+KgtAAAAAElFTkSuQmCC"
-
-	// check if the image was embedded correctly
-	assert.Contains(t, mailBody, img2InternalBase64)
+	require.NotEmpty(t, matches)
+	mailBody = matches[0]
+	assert.Equal(t, `MSG-BEFORE <img src="data:image/png;base64,iVBORw0KGgo="/> MSG-AFTER`, mailBody)
 }
 
 func TestEmbedBase64Images(t *testing.T) {
-	user, repo, _, _ := prepareMailerTest(t)
+	user, repo, issue := prepareMailerBase64Test(t)
 	defer test.MockVariableValue(&setting.MailService.Base64EmbedImages, true)
 	defer test.MockVariableValue(&setting.MailService.Base64EmbedImagesMaxSizePerEmail, 10*1024*1024)
 
-	err := issues_model.NewIssue(t.Context(), repo, &issues_model.Issue{
-		Poster:  user,
-		RepoID:  repo.ID,
-		Title:   "test issue attachment",
-		Content: `content including this image: <image alt="gitea.png" src="attachments/1b267670-1793-4cd0-abc1-449269b7cff9" /> with some more content behind it`,
-	}, nil, nil)
-	require.NoError(t, err)
-	issue := unittest.AssertExistsAndLoadBean(t, &issues_model.Issue{Title: "test issue attachment"})
-	require.NoError(t, issue.LoadRepo(t.Context()))
 	attachment := unittest.AssertExistsAndLoadBean(t, &repo_model.Attachment{ID: 13, IssueID: issue.ID, RepoID: repo.ID})
 	ctx := &mailCommentContext{Context: t.Context(), Issue: issue, Doer: user}