forked from mystiq/dex
81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
package gomail
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
const (
|
|
testTo1 = "to1@example.com"
|
|
testTo2 = "to2@example.com"
|
|
testFrom = "from@example.com"
|
|
testBody = "Test message"
|
|
testMsg = "To: " + testTo1 + ", " + testTo2 + "\r\n" +
|
|
"From: " + testFrom + "\r\n" +
|
|
"Mime-Version: 1.0\r\n" +
|
|
"Date: Wed, 25 Jun 2014 17:46:00 +0000\r\n" +
|
|
"Content-Type: text/plain; charset=UTF-8\r\n" +
|
|
"Content-Transfer-Encoding: quoted-printable\r\n" +
|
|
"\r\n" +
|
|
testBody
|
|
)
|
|
|
|
type mockSender SendFunc
|
|
|
|
func (s mockSender) Send(from string, to []string, msg io.WriterTo) error {
|
|
return s(from, to, msg)
|
|
}
|
|
|
|
type mockSendCloser struct {
|
|
mockSender
|
|
close func() error
|
|
}
|
|
|
|
func (s *mockSendCloser) Close() error {
|
|
return s.close()
|
|
}
|
|
|
|
func TestSend(t *testing.T) {
|
|
s := &mockSendCloser{
|
|
mockSender: stubSend(t, testFrom, []string{testTo1, testTo2}, testMsg),
|
|
close: func() error {
|
|
t.Error("Close() should not be called in Send()")
|
|
return nil
|
|
},
|
|
}
|
|
if err := Send(s, getTestMessage()); err != nil {
|
|
t.Errorf("Send(): %v", err)
|
|
}
|
|
}
|
|
|
|
func getTestMessage() *Message {
|
|
m := NewMessage()
|
|
m.SetHeader("From", testFrom)
|
|
m.SetHeader("To", testTo1, testTo2)
|
|
m.SetBody("text/plain", testBody)
|
|
|
|
return m
|
|
}
|
|
|
|
func stubSend(t *testing.T, wantFrom string, wantTo []string, wantBody string) mockSender {
|
|
return func(from string, to []string, msg io.WriterTo) error {
|
|
if from != wantFrom {
|
|
t.Errorf("invalid from, got %q, want %q", from, wantFrom)
|
|
}
|
|
if !reflect.DeepEqual(to, wantTo) {
|
|
t.Errorf("invalid to, got %v, want %v", to, wantTo)
|
|
}
|
|
|
|
buf := new(bytes.Buffer)
|
|
_, err := msg.WriteTo(buf)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
compareBodies(t, buf.String(), wantBody)
|
|
|
|
return nil
|
|
}
|
|
}
|