smtp_sender_test.go

package mail_test

import (
	"bytes"
	"testing"

	"github.com/gobuffalo/buffalo/mail"
	"github.com/gobuffalo/buffalo/render"
	"github.com/stretchr/testify/require"
)

var sender mail.Sender
var rend *render.Engine
var smtpServer *fakeSMTPServer

const smtpPort = "2002"

func init() {
	rend = render.New(render.Options{})
	smtpServer, _ = newFakeSMTPServer(smtpPort)
	sender, _ = mail.NewSMTPSender("127.0.0.1", smtpPort, "username", "password")

	go smtpServer.Start(smtpPort)
}

func TestSendPlain(t *testing.T) {
	smtpServer.Clear()
	r := require.New(t)

	m := mail.NewMessage()
	m.From = "mark@example.com"
	m.To = []string{"something@something.com"}
	m.Subject = "Cool Message"
	m.CC = []string{"other@other.com", "my@other.com"}
	m.Bcc = []string{"secret@other.com"}

	m.AddAttachment("someFile.txt", "text/plain", bytes.NewBuffer([]byte("hello")))
	m.AddAttachment("otherFile.txt", "text/plain", bytes.NewBuffer([]byte("bye")))
	m.AddEmbedded("test.jpg", bytes.NewBuffer([]byte("not a real image")))
	m.AddBody(rend.String("Hello <%= Name %>"), render.Data{"Name": "Antonio"})
	r.Equal(m.Bodies[0].Content, "Hello Antonio")

	m.SetHeader("X-SMTPAPI", `{"send_at": 1409348513}`)

	err := sender.Send(m)
	r.Nil(err)

	lastMessage := smtpServer.LastMessage()

	r.Contains(lastMessage, "FROM:<mark@example.com>")
	r.Contains(lastMessage, "RCPT TO:<other@other.com>")
	r.Contains(lastMessage, "RCPT TO:<my@other.com>")
	r.Contains(lastMessage, "RCPT TO:<secret@other.com>")
	r.Contains(lastMessage, "Subject: Cool Message")
	r.Contains(lastMessage, "Cc: other@other.com, my@other.com")
	r.Contains(lastMessage, "Content-Type: text/plain")
	r.Contains(lastMessage, "Hello Antonio")
	r.Contains(lastMessage, "Content-Disposition: attachment; filename=\"someFile.txt\"")
	r.Contains(lastMessage, "aGVsbG8=") //base64 of the file content
	r.Contains(lastMessage, "Content-Disposition: attachment; filename=\"otherFile.txt\"")
	r.Contains(lastMessage, "Ynll") //base64 of the file content
	r.Contains(lastMessage, "Content-Disposition: inline; filename=\"test.jpg\"")
	r.Contains(lastMessage, "bm90IGEgcmVhbCBpbWFnZQ==") //base64 of the file content

	r.Contains(lastMessage, `X-SMTPAPI: {"send_at": 1409348513}`)
}