94 lines
2.7 KiB
Go
94 lines
2.7 KiB
Go
package mailyak
|
|
|
|
import (
|
|
"fmt"
|
|
"net/smtp"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
// TestMailYakStringer ensures MailYak struct conforms to the Stringer interface.
|
|
func TestMailYakStringer(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
mail := New("mail.host.com:25", smtp.PlainAuth("", "user", "pass", "mail.host.com"))
|
|
mail.From("from@example.org")
|
|
mail.FromName("From Example")
|
|
mail.To("to@example.org")
|
|
mail.Bcc("bcc1@example.org", "bcc2@example.org")
|
|
mail.Subject("Test subject")
|
|
mail.ReplyTo("replies@example.org")
|
|
mail.HTML().Set("HTML part: this is just a test.")
|
|
mail.Plain().Set("Plain text part: this is also just a test.")
|
|
mail.Attach("test.html", strings.NewReader("<html><head></head></html>"))
|
|
mail.Attach("test2.html", strings.NewReader("<html><head></head></html>"))
|
|
|
|
mail.AddHeader("Precedence", "bulk")
|
|
|
|
mail.date = "a date"
|
|
|
|
want := "&MailYak{date: \"a date\", from: \"from@example.org\", fromName: \"From Example\", html: 31 bytes, plain: 42 bytes, toAddrs: [to@example.org], bccAddrs: [bcc1@example.org bcc2@example.org], subject: \"Test subject\", Precedence: [\"bulk\"], host: \"mail.host.com:25\", attachments (2): [{filename: test.html} {filename: test2.html}], auth set: true, explicit tls: false}"
|
|
got := fmt.Sprintf("%+v", mail)
|
|
if got != want {
|
|
t.Errorf("MailYak.String() = %v, want %v", got, want)
|
|
}
|
|
}
|
|
|
|
// TestMailYakDate ensures two emails sent with the same MailYak instance use
|
|
// different (updated) date timestamps.
|
|
func TestMailYakDate(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
mail := New("mail.host.com:25", smtp.PlainAuth("", "user", "pass", "mail.host.com"))
|
|
mail.From("from@example.org")
|
|
mail.To("to@example.org")
|
|
mail.Subject("Test subject")
|
|
|
|
// send two emails at different times (discarding any errors)
|
|
_, _ = mail.MimeBuf()
|
|
dateOne := mail.date
|
|
|
|
time.Sleep(1 * time.Second)
|
|
|
|
_, _ = mail.MimeBuf()
|
|
dateTwo := mail.date
|
|
|
|
if dateOne == dateTwo {
|
|
t.Errorf("MailYak.Send(): timestamp not updated: %v", dateOne)
|
|
}
|
|
}
|
|
|
|
// TestStripNames ensures that the stripNames() method correctly
|
|
// remove the name part of a list of RFC 5322 addresses.
|
|
func TestStripNames(t *testing.T) {
|
|
addresses := []string{
|
|
"a@example.com",
|
|
"b@example.com",
|
|
"invalid1",
|
|
"John Doe <c@example.com>",
|
|
"<d@example.com>",
|
|
"invalid2",
|
|
}
|
|
|
|
expected := map[string]struct{}{
|
|
"a@example.com": {},
|
|
"b@example.com": {},
|
|
"invalid1": {},
|
|
"c@example.com": {},
|
|
"d@example.com": {},
|
|
"invalid2": {},
|
|
}
|
|
|
|
result := stripNames(addresses)
|
|
|
|
if len(result) != len(expected) {
|
|
t.Fatalf("stripNames: Expected %d addresses, got %d: \n%v", len(expected), len(result), result)
|
|
}
|
|
|
|
for _, addr := range result {
|
|
if _, ok := expected[addr]; !ok {
|
|
t.Errorf("stripNames: Address %q was not expected", addr)
|
|
}
|
|
}
|
|
}
|