// SPDX-FileCopyrightText: 2022-2023 The go-mail Authors // // SPDX-License-Identifier: MIT package mail import ( "bufio" "bytes" "embed" "errors" "fmt" htpl "html/template" "io" "net/mail" "os" "sort" "strings" "testing" ttpl "text/template" "time" ) //go:embed README.md var efs embed.FS // TestNewMsg tests the NewMsg method func TestNewMsg(t *testing.T) { m := NewMsg() var err error if m.encoding != EncodingQP { err = fmt.Errorf("default encoding is not Quoted-Prinable") } if m.charset != CharsetUTF8 { err = fmt.Errorf("default charset is not UTF-8") } if err != nil { t.Errorf("NewMsg() failed: %s", err) return } } // TestNewMsgCharset tests WithCharset and Msg.SetCharset func TestNewMsgCharset(t *testing.T) { tests := []struct { name string value Charset want Charset }{ {"charset is UTF-7", CharsetUTF7, "UTF-7"}, {"charset is UTF-8", CharsetUTF8, "UTF-8"}, {"charset is US-ASCII", CharsetASCII, "US-ASCII"}, {"charset is ISO-8859-1", CharsetISO88591, "ISO-8859-1"}, {"charset is ISO-8859-2", CharsetISO88592, "ISO-8859-2"}, {"charset is ISO-8859-3", CharsetISO88593, "ISO-8859-3"}, {"charset is ISO-8859-4", CharsetISO88594, "ISO-8859-4"}, {"charset is ISO-8859-5", CharsetISO88595, "ISO-8859-5"}, {"charset is ISO-8859-6", CharsetISO88596, "ISO-8859-6"}, {"charset is ISO-8859-7", CharsetISO88597, "ISO-8859-7"}, {"charset is ISO-8859-9", CharsetISO88599, "ISO-8859-9"}, {"charset is ISO-8859-13", CharsetISO885913, "ISO-8859-13"}, {"charset is ISO-8859-14", CharsetISO885914, "ISO-8859-14"}, {"charset is ISO-8859-15", CharsetISO885915, "ISO-8859-15"}, {"charset is ISO-8859-16", CharsetISO885916, "ISO-8859-16"}, {"charset is ISO-2022-JP", CharsetISO2022JP, "ISO-2022-JP"}, {"charset is ISO-2022-KR", CharsetISO2022KR, "ISO-2022-KR"}, {"charset is windows-1250", CharsetWindows1250, "windows-1250"}, {"charset is windows-1251", CharsetWindows1251, "windows-1251"}, {"charset is windows-1252", CharsetWindows1252, "windows-1252"}, {"charset is windows-1255", CharsetWindows1255, "windows-1255"}, {"charset is windows-1256", CharsetWindows1256, "windows-1256"}, {"charset is KOI8-R", CharsetKOI8R, "KOI8-R"}, {"charset is KOI8-U", CharsetKOI8U, "KOI8-U"}, {"charset is Big5", CharsetBig5, "Big5"}, {"charset is GB18030", CharsetGB18030, "GB18030"}, {"charset is GB2312", CharsetGB2312, "GB2312"}, {"charset is TIS-620", CharsetTIS620, "TIS-620"}, {"charset is EUC-KR", CharsetEUCKR, "EUC-KR"}, {"charset is Shift_JIS", CharsetShiftJIS, "Shift_JIS"}, {"charset is GBK", CharsetGBK, "GBK"}, {"charset is Unknown", CharsetUnknown, "Unknown"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m := NewMsg(WithCharset(tt.value), nil) if m.charset != tt.want { t.Errorf("WithCharset() failed. Expected: %s, got: %s", tt.want, m.charset) } m.SetCharset(CharsetUTF8) if m.charset != CharsetUTF8 { t.Errorf("SetCharset() failed. Expected: %s, got: %s", CharsetUTF8, m.charset) } m.SetCharset(tt.value) if m.charset != tt.want { t.Errorf("SetCharset() failed. Expected: %s, got: %s", tt.want, m.charset) } }) } } // TestNewMsgWithCharset tests WithEncoding and Msg.SetEncoding func TestNewMsgWithEncoding(t *testing.T) { tests := []struct { name string value Encoding want Encoding }{ {"encoding is Quoted-Printable", EncodingQP, "quoted-printable"}, {"encoding is Base64", EncodingB64, "base64"}, {"encoding is Unencoded 8-Bit", NoEncoding, "8bit"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m := NewMsg(WithEncoding(tt.value)) if m.encoding != tt.want { t.Errorf("WithEncoding() failed. Expected: %s, got: %s", tt.want, m.encoding) } m.SetEncoding(NoEncoding) if m.encoding != NoEncoding { t.Errorf("SetEncoding() failed. Expected: %s, got: %s", NoEncoding, m.encoding) } m.SetEncoding(tt.want) if m.encoding != tt.want { t.Errorf("SetEncoding() failed. Expected: %s, got: %s", tt.want, m.encoding) } }) } } // TestNewMsgWithMIMEVersion tests WithMIMEVersion and Msg.SetMIMEVersion func TestNewMsgWithMIMEVersion(t *testing.T) { tests := []struct { name string value MIMEVersion want MIMEVersion }{ {"MIME version is 1.0", MIME10, "1.0"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m := NewMsg(WithMIMEVersion(tt.value)) if m.mimever != tt.want { t.Errorf("WithMIMEVersion() failed. Expected: %s, got: %s", tt.want, m.mimever) } m.mimever = "" m.SetMIMEVersion(tt.value) if m.mimever != tt.want { t.Errorf("SetMIMEVersion() failed. Expected: %s, got: %s", tt.want, m.mimever) } }) } } // TestNewMsgWithBoundary tests WithBoundary and Msg.SetBoundary func TestNewMsgWithBoundary(t *testing.T) { tests := []struct { name string value string }{ {"boundary is test123", "test123"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m := NewMsg(WithBoundary(tt.value)) if m.boundary != tt.value { t.Errorf("WithBoundary() failed. Expected: %s, got: %s", tt.value, m.boundary) } m.boundary = "" m.SetBoundary(tt.value) if m.boundary != tt.value { t.Errorf("SetBoundary() failed. Expected: %s, got: %s", tt.value, m.boundary) } }) } } // TestNewMsg_WithPGPType tests WithPGPType option func TestNewMsg_WithPGPType(t *testing.T) { tests := []struct { name string pt PGPType hpt bool }{ {"Not a PGP encoded message", NoPGP, false}, {"PGP encrypted message", PGPEncrypt, true}, {"PGP signed message", PGPSignature, true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m := NewMsg(WithPGPType(tt.pt)) if m.pgptype != tt.pt { t.Errorf("WithPGPType() failed. Expected: %d, got: %d", tt.pt, m.pgptype) } m.pgptype = 99 m.SetPGPType(tt.pt) if m.pgptype != tt.pt { t.Errorf("SetPGPType() failed. Expected: %d, got: %d", tt.pt, m.pgptype) } if m.hasPGPType() != tt.hpt { t.Errorf("hasPGPType() failed. Expected %t, got: %t", tt.hpt, m.hasPGPType()) } }) } } type uppercaseMiddleware struct{} func (mw uppercaseMiddleware) Handle(m *Msg) *Msg { s, ok := m.genHeader[HeaderSubject] if !ok { fmt.Println("can't find the subject header") } if s == nil || len(s) < 1 { s = append(s, "") } m.Subject(strings.ToUpper(s[0])) return m } func (mw uppercaseMiddleware) Type() MiddlewareType { return "uppercase" } type encodeMiddleware struct{} func (mw encodeMiddleware) Handle(m *Msg) *Msg { s, ok := m.genHeader[HeaderSubject] if !ok { fmt.Println("can't find the subject header") } if s == nil || len(s) < 1 { s = append(s, "") } m.Subject(strings.Replace(s[0], "a", "@", -1)) return m } func (mw encodeMiddleware) Type() MiddlewareType { return "encode" } // TestNewMsgWithMiddleware tests WithMiddleware func TestNewMsgWithMiddleware(t *testing.T) { m := NewMsg() if len(m.middlewares) != 0 { t.Errorf("empty middlewares failed. m.middlewares expected to be: empty, got: %d middleware", len(m.middlewares)) } m = NewMsg(WithMiddleware(uppercaseMiddleware{})) if len(m.middlewares) != 1 { t.Errorf("empty middlewares failed. m.middlewares expected to be: 1, got: %d middleware", len(m.middlewares)) } m = NewMsg(WithMiddleware(uppercaseMiddleware{}), WithMiddleware(encodeMiddleware{})) if len(m.middlewares) != 2 { t.Errorf("empty middlewares failed. m.middlewares expected to be: 2, got: %d middleware", len(m.middlewares)) } } // TestApplyMiddlewares tests the applyMiddlewares for the Msg object func TestApplyMiddlewares(t *testing.T) { tests := []struct { name string sub string want string }{ {"normal subject", "This is a test subject", "THIS IS @ TEST SUBJECT"}, {"subject with one middleware effect", "This is test subject", "THIS IS TEST SUBJECT"}, {"subject with one middleware effect", "This is A test subject", "THIS IS A TEST SUBJECT"}, } m := NewMsg(WithMiddleware(encodeMiddleware{}), WithMiddleware(uppercaseMiddleware{})) for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m.Subject(tt.sub) if m.genHeader[HeaderSubject] == nil { t.Errorf("Subject() method failed in applyMiddlewares() test. Generic header for subject is empty") return } m = m.applyMiddlewares(m) s, ok := m.genHeader[HeaderSubject] if !ok { t.Errorf("failed to get subject header") } if s[0] != tt.want { t.Errorf("applyMiddlewares() method failed. Expected: %s, got: %s", tt.want, s[0]) } }) } } // TestMsg_SetGenHeader tests Msg.SetGenHeader func TestMsg_SetGenHeader(t *testing.T) { tests := []struct { name string header Header values []string }{ {"set subject", HeaderSubject, []string{"This is Subject"}}, {"set content-language", HeaderContentLang, []string{"en", "de", "fr", "es"}}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m := NewMsg() m.SetGenHeader(tt.header, tt.values...) if m.genHeader[tt.header] == nil { t.Errorf("SetGenHeader() failed. Tried to set header %s, but it is empty", tt.header) return } for _, v := range tt.values { found := false for _, hv := range m.genHeader[tt.header] { if hv == v { found = true } } if !found { t.Errorf("SetGenHeader() failed. Value %s not found in header field", v) } } }) } } // TestMsg_SetGenHeaderPreformatted tests Msg.SetGenHeaderPreformatted func TestMsg_SetGenHeaderPreformatted(t *testing.T) { tests := []struct { name string header Header value string }{ {"set subject", HeaderSubject, "This is Subject"}, {"set content-language", HeaderContentLang, fmt.Sprintf("%s, %s, %s, %s", "en", "de", "fr", "es")}, {"set subject with newline", HeaderSubject, "This is Subject\r\n with 2nd line"}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m := &Msg{} m.SetGenHeaderPreformatted(tt.header, tt.value) m = NewMsg() m.SetGenHeaderPreformatted(tt.header, tt.value) if m.preformHeader[tt.header] == "" { t.Errorf("SetGenHeaderPreformatted() failed. Tried to set header %s, but it is empty", tt.header) } if m.preformHeader[tt.header] != tt.value { t.Errorf("SetGenHeaderPreformatted() failed. Expected: %q, got: %q", tt.value, m.preformHeader[tt.header]) } buf := bytes.Buffer{} _, err := m.WriteTo(&buf) if err != nil { t.Errorf("failed to write message to memory: %s", err) return } if !strings.Contains(buf.String(), fmt.Sprintf("%s: %s%s", tt.header, tt.value, SingleNewLine)) { t.Errorf("SetGenHeaderPreformatted() failed. Unable to find correctly formated header in " + "mail message output") } }) } } // TestMsg_AddTo tests the Msg.AddTo method func TestMsg_AddTo(t *testing.T) { a := []string{"address1@example.com", "address2@example.com"} na := "address3@example.com" m := NewMsg() if err := m.To(a...); err != nil { t.Errorf("failed to set TO addresses: %s", err) return } if err := m.AddTo(na); err != nil { t.Errorf("AddTo failed: %s", err) return } atf := false for _, v := range m.addrHeader[HeaderTo] { if v.Address == na { atf = true } } if !atf { t.Errorf("AddTo() failed. Address %q not found in TO address slice.", na) } } // TestMsg_From tests the Msg.From and Msg.GetSender methods func TestMsg_From(t *testing.T) { a := "toni@example.com" n := "Toni Tester" na := fmt.Sprintf(`"%s" <%s>`, n, a) m := NewMsg() _, err := m.GetSender(false) if err == nil { t.Errorf("GetSender(false) without a set From address succeeded but was expected to fail") return } if err := m.From(na); err != nil { t.Errorf("failed to set FROM addresses: %s", err) return } gs, err := m.GetSender(false) if err != nil { t.Errorf("GetSender(false) failed: %s", err) return } if gs != a { t.Errorf("From() failed. Expected: %s, got: %s", a, gs) return } gs, err = m.GetSender(true) if err != nil { t.Errorf("GetSender(true) failed: %s", err) return } if gs != na { t.Errorf("From() failed. Expected: %s, got: %s", na, gs) return } } // TestMsg_EnvelopeFrom tests the Msg.EnvelopeFrom and Msg.GetSender methods func TestMsg_EnvelopeFrom(t *testing.T) { e := "envelope@example.com" a := "toni@example.com" n := "Toni Tester" na := fmt.Sprintf(`"%s" <%s>`, n, a) ne := fmt.Sprintf(`<%s>`, e) m := NewMsg() _, err := m.GetSender(false) if err == nil { t.Errorf("GetSender(false) without a set envelope From address succeeded but was expected to fail") return } if err := m.EnvelopeFrom(e); err != nil { t.Errorf("failed to set envelope FROM addresses: %s", err) return } gs, err := m.GetSender(false) if err != nil { t.Errorf("GetSender(false) failed: %s", err) return } if gs != e { t.Errorf("From() failed. Expected: %s, got: %s", e, gs) return } if err := m.From(na); err != nil { t.Errorf("failed to set FROM addresses: %s", err) return } gs, err = m.GetSender(false) if err != nil { t.Errorf("GetSender(false) failed: %s", err) return } if gs != e { t.Errorf("From() failed. Expected: %s, got: %s", e, gs) return } gs, err = m.GetSender(true) if err != nil { t.Errorf("GetSender(true) failed: %s", err) return } if gs != ne { t.Errorf("From() failed. Expected: %s, got: %s", ne, gs) return } m.Reset() if err := m.From(na); err != nil { t.Errorf("failed to set FROM addresses: %s", err) return } gs, err = m.GetSender(false) if err != nil { t.Errorf("GetSender(true) failed: %s", err) return } if gs != a { t.Errorf("From() failed. Expected: %s, got: %s", a, gs) return } gs, err = m.GetSender(true) if err != nil { t.Errorf("GetSender(true) failed: %s", err) return } if gs != na { t.Errorf("From() failed. Expected: %s, got: %s", na, gs) return } } // TestMsg_AddToFormat tests the Msg.AddToFormat method func TestMsg_AddToFormat(t *testing.T) { a := []string{"address1@example.com", "address2@example.com"} nn := "Toni Tester" na := "address3@example.com" w := `"Toni Tester" ` m := NewMsg() if err := m.To(a...); err != nil { t.Errorf("failed to set TO addresses: %s", err) return } if err := m.AddToFormat(nn, na); err != nil { t.Errorf("AddToFormat failed: %s", err) return } atf := false for _, v := range m.addrHeader[HeaderTo] { if v.String() == w { atf = true } } if !atf { t.Errorf("AddToFormat() failed. Address %q not found in TO address slice.", w) } } // TestMsg_ToIgnoreInvalid tests the Msg.ToIgnoreInvalid method func TestMsg_ToIgnoreInvalid(t *testing.T) { a := []string{"address1@example.com", "address2@example.com"} fa := []string{"address1@example.com", "address2@example.com", "failedaddress.com"} m := NewMsg() m.ToIgnoreInvalid(a...) l := len(m.addrHeader[HeaderTo]) if l != len(a) { t.Errorf("ToIgnoreInvalid() failed. Expected %d addresses, got: %d", len(a), l) } m.ToIgnoreInvalid(fa...) l = len(m.addrHeader[HeaderTo]) if l != len(fa)-1 { t.Errorf("ToIgnoreInvalid() failed. Expected %d addresses, got: %d", len(fa)-1, l) } } // TestMsg_AddCc tests the Msg.AddCc method func TestMsg_AddCc(t *testing.T) { a := []string{"address1@example.com", "address2@example.com"} na := "address3@example.com" m := NewMsg() if err := m.Cc(a...); err != nil { t.Errorf("failed to set CC addresses: %s", err) return } if err := m.AddCc(na); err != nil { t.Errorf("AddCc failed: %s", err) return } atf := false for _, v := range m.addrHeader[HeaderCc] { if v.Address == na { atf = true } } if !atf { t.Errorf("AddCc() failed. Address %q not found in CC address slice.", na) } } // TestMsg_AddCcFormat tests the Msg.AddCcFormat method func TestMsg_AddCcFormat(t *testing.T) { a := []string{"address1@example.com", "address2@example.com"} nn := "Toni Tester" na := "address3@example.com" w := `"Toni Tester" ` m := NewMsg() if err := m.Cc(a...); err != nil { t.Errorf("failed to set CC addresses: %s", err) return } if err := m.AddCcFormat(nn, na); err != nil { t.Errorf("AddCcFormat failed: %s", err) return } atf := false for _, v := range m.addrHeader[HeaderCc] { if v.String() == w { atf = true } } if !atf { t.Errorf("AddCcFormat() failed. Address %q not found in CC address slice.", w) } } // TestMsg_CcIgnoreInvalid tests the Msg.CcIgnoreInvalid method func TestMsg_CcIgnoreInvalid(t *testing.T) { a := []string{"address1@example.com", "address2@example.com"} fa := []string{"address1@example.com", "address2@example.com", "failedaddress.com"} m := NewMsg() m.CcIgnoreInvalid(a...) l := len(m.addrHeader[HeaderCc]) if l != len(a) { t.Errorf("CcIgnoreInvalid() failed. Expected %d addresses, got: %d", len(a), l) } m.CcIgnoreInvalid(fa...) l = len(m.addrHeader[HeaderCc]) if l != len(fa)-1 { t.Errorf("CcIgnoreInvalid() failed. Expected %d addresses, got: %d", len(fa)-1, l) } } // TestMsg_AddBcc tests the Msg.AddBcc method func TestMsg_AddBcc(t *testing.T) { a := []string{"address1@example.com", "address2@example.com"} na := "address3@example.com" m := NewMsg() if err := m.Bcc(a...); err != nil { t.Errorf("failed to set BCC addresses: %s", err) return } if err := m.AddBcc(na); err != nil { t.Errorf("AddBcc failed: %s", err) return } atf := false for _, v := range m.addrHeader[HeaderBcc] { if v.Address == na { atf = true } } if !atf { t.Errorf("AddBcc() failed. Address %q not found in BCC address slice.", na) } } // TestMsg_AddBccFormat tests the Msg.AddBccFormat method func TestMsg_AddBccFormat(t *testing.T) { a := []string{"address1@example.com", "address2@example.com"} nn := "Toni Tester" na := "address3@example.com" w := `"Toni Tester" ` m := NewMsg() if err := m.Bcc(a...); err != nil { t.Errorf("failed to set BCC addresses: %s", err) return } if err := m.AddBccFormat(nn, na); err != nil { t.Errorf("AddBccFormat failed: %s", err) return } atf := false for _, v := range m.addrHeader[HeaderBcc] { if v.String() == w { atf = true } } if !atf { t.Errorf("AddBccFormat() failed. Address %q not found in BCC address slice.", w) } } // TestMsg_BccIgnoreInvalid tests the Msg.BccIgnoreInvalid method func TestMsg_BccIgnoreInvalid(t *testing.T) { a := []string{"address1@example.com", "address2@example.com"} fa := []string{"address1@example.com", "address2@example.com", "failedaddress.com"} m := NewMsg() m.BccIgnoreInvalid(a...) l := len(m.addrHeader[HeaderBcc]) if l != len(a) { t.Errorf("BccIgnoreInvalid() failed. Expected %d addresses, got: %d", len(a), l) } m.BccIgnoreInvalid(fa...) l = len(m.addrHeader[HeaderBcc]) if l != len(fa)-1 { t.Errorf("BccIgnoreInvalid() failed. Expected %d addresses, got: %d", len(fa)-1, l) } } // TestMsg_SetBulk tests the Msg.SetBulk method func TestMsg_SetBulk(t *testing.T) { m := NewMsg() m.SetBulk() if m.genHeader[HeaderPrecedence] == nil { t.Errorf("SetBulk() failed. Precedence header is nil") return } if m.genHeader[HeaderPrecedence][0] != "bulk" { t.Errorf("SetBulk() failed. Expected Precedence header: %q, got: %q", "bulk", m.genHeader[HeaderPrecedence][0]) } if m.genHeader[HeaderXAutoResponseSuppress] == nil { t.Errorf("SetBulk() failed. X-Auto-Response-Suppress header is nil") return } if m.genHeader[HeaderXAutoResponseSuppress][0] != "All" { t.Errorf("SetBulk() failed. Expected X-Auto-Response-Suppress header: %q, got: %q", "All", m.genHeader[HeaderXAutoResponseSuppress][0]) } } // TestMsg_SetDate tests the Msg.SetDate and Msg.SetDateWithValue method func TestMsg_SetDate(t *testing.T) { m := NewMsg() m.SetDate() if m.genHeader[HeaderDate] == nil { t.Errorf("SetDate() failed. Date header is nil") return } d, ok := m.genHeader[HeaderDate] if !ok { t.Errorf("failed to get date header") return } _, err := time.Parse(time.RFC1123Z, d[0]) if err != nil { t.Errorf("failed to parse time in date header: %s", err) } m.genHeader = nil m.genHeader = make(map[Header][]string) now := time.Now() m.SetDateWithValue(now) if m.genHeader[HeaderDate] == nil { t.Errorf("SetDateWithValue() failed. Date header is nil") return } d, ok = m.genHeader[HeaderDate] if !ok { t.Errorf("failed to get date header") return } pt, err := time.Parse(time.RFC1123Z, d[0]) if err != nil { t.Errorf("failed to parse time in date header: %s", err) } if pt.Unix() != now.Unix() { t.Errorf("SetDateWithValue() failed. Expected time: %d, got: %d", now.Unix(), pt.Unix()) } } // TestMsg_SetMessageIDWIthValue tests the Msg.SetMessageIDWithValue and Msg.SetMessageID methods func TestMsg_SetMessageIDWithValue(t *testing.T) { m := NewMsg() m.SetMessageID() if m.genHeader[HeaderMessageID] == nil { t.Errorf("SetMessageID() failed. MessageID header is nil") return } if m.genHeader[HeaderMessageID][0] == "" { t.Errorf("SetMessageID() failed. Expected value, got: empty") return } if _, ok := m.genHeader[HeaderMessageID]; ok { m.genHeader[HeaderMessageID] = nil } v := "This.is.a.message.id" vf := "" m.SetMessageIDWithValue(v) if m.genHeader[HeaderMessageID] == nil { t.Errorf("SetMessageIDWithValue() failed. MessageID header is nil") return } if m.genHeader[HeaderMessageID][0] != vf { t.Errorf("SetMessageIDWithValue() failed. Expected: %s, got: %s", vf, m.genHeader[HeaderMessageID][0]) return } } // TestMsg_SetMessageIDRandomness tests the randomness of Msg.SetMessageID methods func TestMsg_SetMessageIDRandomness(t *testing.T) { var mids []string m := NewMsg() for i := 0; i < 50_000; i++ { m.SetMessageID() mid := m.GetMessageID() mids = append(mids, mid) } c := make(map[string]int) for i := range mids { c[mids[i]]++ } for k, v := range c { if v > 1 { t.Errorf("MessageID randomness not given. MessageID %q was generated %d times", k, v) } } } func TestMsg_GetMessageID(t *testing.T) { expected := "this.is.a.message.id" msg := NewMsg() msg.SetMessageIDWithValue(expected) val := msg.GetMessageID() if !strings.EqualFold(val, fmt.Sprintf("<%s>", expected)) { t.Errorf("GetMessageID() failed. Expected: %s, got: %s", fmt.Sprintf("<%s>", expected), val) } msg.genHeader[HeaderMessageID] = nil val = msg.GetMessageID() if val != "" { t.Errorf("GetMessageID() failed. Expected empty string, got: %s", val) } } // TestMsg_FromFormat tests the FromFormat and EnvelopeFrom methods for the Msg object func TestMsg_FromFormat(t *testing.T) { tests := []struct { tname string name string addr string want string fail bool }{ { "valid name and addr", "Toni Tester", "tester@example.com", `"Toni Tester" `, false, }, { "no name with valid addr", "", "tester@example.com", ``, false, }, { "valid name with invalid addr", "Toni Tester", "@example.com", ``, true, }, } m := NewMsg() for _, tt := range tests { t.Run(tt.tname, func(t *testing.T) { if err := m.FromFormat(tt.name, tt.addr); err != nil && !tt.fail { t.Errorf("failed to FromFormat(): %s", err) return } if err := m.EnvelopeFromFormat(tt.name, tt.addr); err != nil && !tt.fail { t.Errorf("failed to EnvelopeFromFormat(): %s", err) return } var fa *mail.Address f, ok := m.addrHeader[HeaderFrom] if ok && len(f) > 0 { fa = f[0] } if (!ok || len(f) == 0) && !tt.fail { t.Errorf(`valid from address expected, but "From:" field is empty`) return } if tt.fail && len(f) > 0 { t.Errorf("FromFormat() was supposed to failed but got value: %s", fa.String()) return } if !tt.fail && fa.String() != tt.want { t.Errorf("wrong result for FromFormat(). Want: %s, got: %s", tt.want, fa.String()) } m.addrHeader[HeaderFrom] = nil }) } } func TestMsg_GetRecipients(t *testing.T) { a := []string{"to@example.com", "cc@example.com", "bcc@example.com"} m := NewMsg() _, err := m.GetRecipients() if err == nil { t.Errorf("GetRecipients() succeeded but was expected to fail") return } if err := m.AddTo(a[0]); err != nil { t.Errorf("AddTo() failed: %s", err) return } if err := m.AddCc(a[1]); err != nil { t.Errorf("AddCc() failed: %s", err) return } if err := m.AddBcc(a[2]); err != nil { t.Errorf("AddBcc() failed: %s", err) return } al, err := m.GetRecipients() if err != nil { t.Errorf("GetRecipients() failed: %s", err) return } tf, cf, bf := false, false, false for _, r := range al { if r == a[0] { tf = true } if r == a[1] { cf = true } if r == a[2] { bf = true } } if !tf { t.Errorf("GetRecipients() failed. Expected to address %s but was not found", a[0]) return } if !cf { t.Errorf("GetRecipients() failed. Expected cc address %s but was not found", a[1]) return } if !bf { t.Errorf("GetRecipients() failed. Expected bcc address %s but was not found", a[2]) return } } // TestMsg_ReplyTo tests the Msg.ReplyTo and Msg.ReplyToFormat methods func TestMsg_ReplyTo(t *testing.T) { tests := []struct { tname string name string addr string want string sf bool }{ { "valid name and addr", "Toni Tester", "tester@example.com", `"Toni Tester" `, false, }, { "no name with valid addr", "", "tester@example.com", ``, false, }, { "valid name with invalid addr", "Toni Tester", "@example.com", ``, true, }, } m := NewMsg() for _, tt := range tests { t.Run(tt.tname, func(t *testing.T) { if err := m.ReplyTo(tt.want); err != nil && !tt.sf { t.Errorf("ReplyTo() method failed: %s", err) } if !tt.sf { rt, ok := m.genHeader[HeaderReplyTo] if !ok { t.Errorf("ReplyTo() failed: ReplyTo generic header not set") return } if len(rt) <= 0 { t.Errorf("ReplyTo() failed: length of generic ReplyTo header is zero or less than zero") return } if rt[0] != tt.want { t.Errorf("ReplyTo() failed: expected value: %s, got: %s", tt.want, rt[0]) } } m.genHeader = nil m.genHeader = make(map[Header][]string) if err := m.ReplyToFormat(tt.name, tt.addr); err != nil && !tt.sf { t.Errorf("ReplyToFormat() method failed: %s", err) } if !tt.sf { rt, ok := m.genHeader[HeaderReplyTo] if !ok { t.Errorf("ReplyTo() failed: ReplyTo generic header not set") return } if len(rt) <= 0 { t.Errorf("ReplyTo() failed: length of generic ReplyTo header is zero or less than zero") return } if rt[0] != tt.want { t.Errorf("ReplyTo() failed: expected value: %s, got: %s", tt.want, rt[0]) } } m.genHeader = nil m.genHeader = make(map[Header][]string) }) } } // TestMsg_Subject tests the Msg.Subject method func TestMsg_Subject(t *testing.T) { tests := []struct { name string sub string want string }{ {"normal subject", "This is a test subject", "This is a test subject"}, { "subject with umlauts", "This is a test subject with umlauts: üäöß", "=?UTF-8?q?This_is_a_test_subject_with_umlauts:_=C3=BC=C3=A4=C3=B6=C3=9F?=", }, { "subject with emoji", "This is a test subject with emoji: 📧", "=?UTF-8?q?This_is_a_test_subject_with_emoji:_=F0=9F=93=A7?=", }, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m.Subject(tt.sub) s, ok := m.genHeader[HeaderSubject] if !ok || len(s) <= 0 { t.Errorf("Subject() method failed. Generic header for Subject is empty") return } if s[0] != tt.want { t.Errorf("Subject() method failed. Expected: %s, got: %s", tt.want, s[0]) } }) } } // TestMsg_SetImportance tests the Msg.SetImportance method func TestMsg_SetImportance(t *testing.T) { tests := []struct { name string imp Importance wantns string xprio string want string sf bool }{ {"Importance: Non-Urgent", ImportanceNonUrgent, "0", "5", "non-urgent", false}, {"Importance: Low", ImportanceLow, "0", "5", "low", false}, {"Importance: Normal", ImportanceNormal, "", "", "", true}, {"Importance: High", ImportanceHigh, "1", "1", "high", false}, {"Importance: Urgent", ImportanceUrgent, "1", "1", "urgent", false}, {"Importance: Unknown", 9, "", "", "", true}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m.SetImportance(tt.imp) hi, ok := m.genHeader[HeaderImportance] if (!ok || len(hi) <= 0) && !tt.sf { t.Errorf("SetImportance() method failed. Generic header for Importance is empty") } hp, ok := m.genHeader[HeaderPriority] if (!ok || len(hp) <= 0) && !tt.sf { t.Errorf("SetImportance() method failed. Generic header for Priority is empty") } hx, ok := m.genHeader[HeaderXPriority] if (!ok || len(hx) <= 0) && !tt.sf { t.Errorf("SetImportance() method failed. Generic header for X-Priority is empty") } hm, ok := m.genHeader[HeaderXMSMailPriority] if (!ok || len(hm) <= 0) && !tt.sf { t.Errorf("SetImportance() method failed. Generic header for X-MS-XPriority is empty") } if !tt.sf { if hi[0] != tt.want { t.Errorf("SetImportance() method failed. Expected Imporance: %s, got: %s", tt.want, hi[0]) } if hp[0] != tt.wantns { t.Errorf("SetImportance() method failed. Expected Priority: %s, got: %s", tt.want, hp[0]) } if hx[0] != tt.xprio { t.Errorf("SetImportance() method failed. Expected X-Priority: %s, got: %s", tt.want, hx[0]) } if hm[0] != tt.wantns { t.Errorf("SetImportance() method failed. Expected X-MS-Priority: %s, got: %s", tt.wantns, hm[0]) } } m.genHeader = nil m.genHeader = make(map[Header][]string) }) } } // TestMsg_SetOrganization tests the Msg.SetOrganization method func TestMsg_SetOrganization(t *testing.T) { tests := []struct { name string org string }{ {"Org: testcorp", "testcorp"}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m.SetOrganization(tt.org) o, ok := m.genHeader[HeaderOrganization] if !ok || len(o) <= 0 { t.Errorf("SetOrganization() method failed. Generic header for Organization is empty") return } if o[0] != tt.org { t.Errorf("SetOrganization() method failed. Expected: %s, got: %s", tt.org, o[0]) } }) } } // TestMsg_SetUserAgent tests the Msg.SetUserAgent method func TestMsg_SetUserAgent(t *testing.T) { tests := []struct { name string ua string }{ {"UA: Testmail 1.0", "Testmailer 1.0"}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m.SetUserAgent(tt.ua) xm, ok := m.genHeader[HeaderXMailer] if !ok || len(xm) <= 0 { t.Errorf("SetUserAgent() method failed. Generic header for X-Mailer is empty") return } ua, ok := m.genHeader[HeaderUserAgent] if !ok || len(ua) <= 0 { t.Errorf("SetUserAgent() method failed. Generic header for UserAgent is empty") return } if xm[0] != tt.ua { t.Errorf("SetUserAgent() method failed. Expected X-Mailer: %s, got: %s", tt.ua, xm[0]) } if ua[0] != tt.ua { t.Errorf("SetUserAgent() method failed. Expected User-Agent: %s, got: %s", tt.ua, ua[0]) } }) } } // TestMsg_RequestMDN tests the different RequestMDN* related methods of Msg func TestMsg_RequestMDN(t *testing.T) { n := "Toni Tester" n2 := "Melanie Tester" v := "toni.tester@example.com" v2 := "melanie.tester@example.com" iv := "testertest.tld" vl := []string{v, v2} m := NewMsg() // Single valid address if err := m.RequestMDNTo(v); err != nil { t.Errorf("RequestMDNTo with a single valid address failed: %s", err) } if val := m.genHeader[HeaderDispositionNotificationTo]; len(val) > 1 { if val[0] != fmt.Sprintf("<%s>", v) { t.Errorf("RequestMDNTo with a single valid address failed. Expected: %s, got: %s", v, val[0]) } } m.Reset() // Multiples valid addresses if err := m.RequestMDNTo(vl...); err != nil { t.Errorf("RequestMDNTo with a multiple valid address failed: %s", err) } if val := m.genHeader[HeaderDispositionNotificationTo]; len(val) > 0 { if val[0] != fmt.Sprintf("<%s>", v) { t.Errorf("RequestMDNTo with a multiple valid addresses failed. Expected 0: %s, got 0: %s", v, val[0]) } } if val := m.genHeader[HeaderDispositionNotificationTo]; len(val) > 1 { if val[1] != fmt.Sprintf("<%s>", v2) { t.Errorf("RequestMDNTo with a multiple valid addresses failed. Expected 1: %s, got 1: %s", v2, val[1]) } } m.Reset() // Invalid address if err := m.RequestMDNTo(iv); err == nil { t.Errorf("RequestMDNTo with an invalid address was supposed to failed, but didn't") } m.Reset() // Single valid addresses + AddTo if err := m.RequestMDNTo(v); err != nil { t.Errorf("RequestMDNTo with a single valid address failed: %s", err) } if err := m.RequestMDNAddTo(v2); err != nil { t.Errorf("RequestMDNAddTo with a valid address failed: %s", err) } if val := m.genHeader[HeaderDispositionNotificationTo]; len(val) > 1 { if val[1] != fmt.Sprintf("<%s>", v2) { t.Errorf("RequestMDNTo with a multiple valid addresses failed. Expected 1: %s, got 1: %s", v2, val[1]) } } m.Reset() // Single valid address formated + AddToFromat if err := m.RequestMDNToFormat(n, v); err != nil { t.Errorf("RequestMDNToFormat with a single valid address failed: %s", err) } if val := m.genHeader[HeaderDispositionNotificationTo]; len(val) > 0 { if val[0] != fmt.Sprintf(`"%s" <%s>`, n, v) { t.Errorf(`RequestMDNToFormat with a single valid address failed. Expected: "%s" <%s>, got: %s`, n, v, val[0]) } } if err := m.RequestMDNAddToFormat(n2, v2); err != nil { t.Errorf("RequestMDNAddToFormat with a valid address failed: %s", err) } if val := m.genHeader[HeaderDispositionNotificationTo]; len(val) > 1 { if val[1] != fmt.Sprintf(`"%s" <%s>`, n2, v2) { t.Errorf(`RequestMDNAddToFormat with a single valid address failed. Expected: "%s" <%s>, got: %s`, n2, v2, val[1]) } } m.Reset() // Invalid formated address if err := m.RequestMDNToFormat(n, iv); err == nil { t.Errorf("RequestMDNToFormat with an invalid address was supposed to failed, but didn't") } // Invalid address AddTo + AddToFormat if err := m.RequestMDNAddTo(iv); err == nil { t.Errorf("RequestMDNAddTo with an invalid address was supposed to failed, but didn't") } if err := m.RequestMDNAddToFormat(n, iv); err == nil { t.Errorf("RequestMDNAddToFormat with an invalid address was supposed to failed, but didn't") } } // TestMsg_SetBodyString tests the Msg.SetBodyString method func TestMsg_SetBodyString(t *testing.T) { tests := []struct { name string ct ContentType value string want string sf bool }{ {"Body: test", TypeTextPlain, "test", "test", false}, { "Body: with Umlauts", TypeTextHTML, "üäöß", "üäöß", false, }, {"Body: with emoji", TypeTextPlain, "📧", "📧", false}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m.SetBodyString(tt.ct, tt.value) if len(m.parts) != 1 { t.Errorf("SetBodyString() failed: no mail parts found") } part := m.parts[0] res := bytes.Buffer{} if _, err := part.writeFunc(&res); err != nil && !tt.sf { t.Errorf("WriteFunc of part failed: %s", err) } if res.String() != tt.want { t.Errorf("SetBodyString() failed. Expecteding: %s, got: %s", tt.want, res.String()) } }) } } // TestMsg_AddAlternativeString tests the Msg.AddAlternativeString method func TestMsg_AddAlternativeString(t *testing.T) { tests := []struct { name string value string want string sf bool }{ {"Body: test", "test", "test", false}, {"Body: with Umlauts", "üäöß", "üäöß", false}, {"Body: with emoji", "📧", "📧", false}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m.SetBodyString(TypeTextPlain, tt.value) if len(m.parts) != 1 { t.Errorf("AddAlternativeString() => SetBodyString() failed: no mail parts found") } m.AddAlternativeString(TypeTextHTML, tt.value) if len(m.parts) != 2 { t.Errorf("AddAlternativeString() failed: no alternative mail parts found") } apart := m.parts[1] res := bytes.Buffer{} if _, err := apart.writeFunc(&res); err != nil && !tt.sf { t.Errorf("WriteFunc of part failed: %s", err) } if res.String() != tt.want { t.Errorf("AddAlternativeString() failed. Expecteding: %s, got: %s", tt.want, res.String()) } }) } } // TestMsg_AttachFile tests the Msg.AttachFile and the WithFilename FileOption method func TestMsg_AttachFile(t *testing.T) { tests := []struct { name string file string fn string sf bool }{ {"File: README.md", "README.md", "README.md", false}, {"File: doc.go", "doc.go", "foo.go", false}, {"File: nonexisting", "", "invalid.file", true}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m.AttachFile(tt.file, WithFileName(tt.fn), nil) if len(m.attachments) != 1 && !tt.sf { t.Errorf("AttachFile() failed. Number of attachments expected: %d, got: %d", 1, len(m.attachments)) return } if !tt.sf { file := m.attachments[0] if file == nil { t.Errorf("AttachFile() failed. Attachment file pointer is nil") return } if file.Name != tt.fn { t.Errorf("AttachFile() failed. Filename of attachment expected: %s, got: %s", tt.fn, file.Name) } buf := bytes.Buffer{} if _, err := file.Writer(&buf); err != nil { t.Errorf("failed to execute WriterFunc: %s", err) return } } m.Reset() }) } } // TestMsg_GetAttachments tests the Msg.GetAttachments method func TestMsg_GetAttachments(t *testing.T) { tests := []struct { name string files []string }{ {"File: README.md", []string{"README.md"}}, {"File: doc.go", []string{"doc.go"}}, {"File: README.md and doc.go", []string{"README.md", "doc.go"}}, {"File: nonexisting", nil}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { for _, f := range tt.files { m.AttachFile(f, WithFileName(f), nil) } if len(m.attachments) != len(tt.files) { t.Errorf("AttachFile() failed. Number of attachments expected: %d, got: %d", len(tt.files), len(m.attachments)) return } ff := m.GetAttachments() if len(m.attachments) != len(ff) { t.Errorf("GetAttachments() failed. Number of attachments expected: %d, got: %d", len(m.attachments), len(ff)) return } var fn []string for _, f := range ff { fn = append(fn, f.Name) } sort.Strings(fn) sort.Strings(tt.files) for i, f := range tt.files { if f != fn[i] { t.Errorf("GetAttachments() failed. Attachment name expected: %s, got: %s", f, fn[i]) return } } m.Reset() }) } } // TestMsg_SetAttachments tests the Msg.GetAttachments method func TestMsg_SetAttachments(t *testing.T) { tests := []struct { name string attachments []string files []string }{ {"File: replace README.md with doc.go", []string{"README.md"}, []string{"doc.go"}}, {"File: add README.md with doc.go ", []string{"doc.go"}, []string{"README.md", "doc.go"}}, {"File: remove README.md and doc.go", []string{"README.md", "doc.go"}, nil}, {"File: add README.md and doc.go", nil, []string{"README.md", "doc.go"}}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { sort.Strings(tt.attachments) sort.Strings(tt.files) for _, a := range tt.attachments { m.AttachFile(a, WithFileName(a), nil) } if len(m.attachments) != len(tt.attachments) { t.Errorf("AttachFile() failed. Number of attachments expected: %d, got: %d", len(tt.files), len(m.attachments)) return } var files []*File for _, f := range tt.files { files = append(files, &File{Name: f}) } m.SetAttachments(files) if len(m.attachments) != len(files) { t.Errorf("SetAttachements() failed. Number of attachments expected: %d, got: %d", len(files), len(m.attachments)) return } for i, f := range tt.files { if f != m.attachments[i].Name { t.Errorf("SetAttachments() failed. Attachment name expected: %s, got: %s", f, m.attachments[i].Name) return } } m.Reset() }) } } // TestMsg_UnsetAllAttachments tests the Msg.UnsetAllAttachments method func TestMsg_UnsetAllAttachments(t *testing.T) { tests := []struct { name string attachments []string }{ {"File: one file", []string{"README.md"}}, {"File: two files", []string{"README.md", "doc.go"}}, {"File: nil", nil}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { var files []*File for _, f := range tt.attachments { files = append(files, &File{Name: f}) } m.SetAttachments(files) if len(m.attachments) != len(files) { t.Errorf("SetAttachements() failed. Number of attachments expected: %d, got: %d", len(files), len(m.attachments)) return } m.UnsetAllAttachments() if m.attachments != nil { t.Errorf("UnsetAllAttachments() failed. The attachments file's pointer is not nil") return } m.Reset() }) } } // TestMsg_GetEmbeds tests the Msg.GetEmbeds method func TestMsg_GetEmbeds(t *testing.T) { tests := []struct { name string files []string }{ {"File: README.md", []string{"README.md"}}, {"File: doc.go", []string{"doc.go"}}, {"File: README.md and doc.go", []string{"README.md", "doc.go"}}, {"File: nonexisting", nil}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { for _, f := range tt.files { m.EmbedFile(f, WithFileName(f), nil) } if len(m.embeds) != len(tt.files) { t.Errorf("EmbedFile() failed. Number of embedded files expected: %d, got: %d", len(tt.files), len(m.embeds)) return } ff := m.GetEmbeds() if len(m.embeds) != len(ff) { t.Errorf("GetEmbeds() failed. Number of embedded files expected: %d, got: %d", len(m.embeds), len(ff)) return } var fn []string for _, f := range ff { fn = append(fn, f.Name) } sort.Strings(fn) sort.Strings(tt.files) for i, f := range tt.files { if f != fn[i] { t.Errorf("GetEmbeds() failed. Embedded file name expected: %s, got: %s", f, fn[i]) return } } m.Reset() }) } } // TestMsg_SetEmbeds tests the Msg.GetEmbeds method func TestMsg_SetEmbeds(t *testing.T) { tests := []struct { name string embeds []string files []string }{ {"File: replace README.md with doc.go", []string{"README.md"}, []string{"doc.go"}}, {"File: add README.md with doc.go ", []string{"doc.go"}, []string{"README.md", "doc.go"}}, {"File: remove README.md and doc.go", []string{"README.md", "doc.go"}, nil}, {"File: add README.md and doc.go", nil, []string{"README.md", "doc.go"}}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { sort.Strings(tt.embeds) sort.Strings(tt.files) for _, a := range tt.embeds { m.EmbedFile(a, WithFileName(a), nil) } if len(m.embeds) != len(tt.embeds) { t.Errorf("EmbedFile() failed. Number of embedded files expected: %d, got: %d", len(tt.files), len(m.embeds)) return } var files []*File for _, f := range tt.files { files = append(files, &File{Name: f}) } m.SetEmbeds(files) if len(m.embeds) != len(files) { t.Errorf("SetEmbeds() failed. Number of embedded files expected: %d, got: %d", len(files), len(m.embeds)) return } for i, f := range tt.files { if f != m.embeds[i].Name { t.Errorf("SetEmbeds() failed. Embedded file name expected: %s, got: %s", f, m.embeds[i].Name) return } } m.Reset() }) } } // TestMsg_UnsetAllEmbeds tests the Msg.TestMsg_UnsetAllEmbeds method func TestMsg_UnsetAllEmbeds(t *testing.T) { tests := []struct { name string embeds []string }{ {"File: one file", []string{"README.md"}}, {"File: two files", []string{"README.md", "doc.go"}}, {"File: nil", nil}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { var files []*File for _, f := range tt.embeds { files = append(files, &File{Name: f}) } m.SetEmbeds(files) if len(m.embeds) != len(files) { t.Errorf("SetEmbeds() failed. Number of embedded files expected: %d, got: %d", len(files), len(m.embeds)) return } m.UnsetAllEmbeds() if m.embeds != nil { t.Errorf("UnsetAllEmbeds() failed. The embeds file's point is not nil") return } m.Reset() }) } } // TestMsg_UnsetAllParts tests the Msg.TestMsg_UnsetAllParts method func TestMsg_UnsetAllParts(t *testing.T) { tests := []struct { name string attachments []string embeds []string }{ {"File: both is exist", []string{"README.md"}, []string{"doc.go"}}, {"File: both is nil", nil, nil}, {"File: attachment exist, embed nil", []string{"README.md"}, nil}, {"File: attachment nil, embed exist", nil, []string{"README.md"}}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { var attachments []*File for _, f := range tt.attachments { attachments = append(attachments, &File{Name: f}) } m.SetAttachments(attachments) if len(m.attachments) != len(attachments) { t.Errorf("SetAttachements() failed. Number of attachments files expected: %d, got: %d", len(attachments), len(m.attachments)) return } var embeds []*File for _, f := range tt.embeds { embeds = append(embeds, &File{Name: f}) } m.SetEmbeds(embeds) if len(m.embeds) != len(embeds) { t.Errorf("SetEmbeds() failed. Number of embedded files expected: %d, got: %d", len(embeds), len(m.embeds)) return } m.UnsetAllParts() if m.attachments != nil { t.Errorf("UnsetAllParts() failed. The attachments file's point is not nil") return } if m.embeds != nil { t.Errorf("UnsetAllParts() failed. The embeds file's point is not nil") return } m.Reset() }) } } // TestMsg_AttachFromEmbedFS tests the Msg.AttachFromEmbedFS and the WithFilename FileOption method func TestMsg_AttachFromEmbedFS(t *testing.T) { tests := []struct { name string file string fn string sf bool }{ {"File: README.md", "README.md", "README.md", false}, {"File: nonexisting", "", "invalid.file", true}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { if err := m.AttachFromEmbedFS(tt.file, &efs, WithFileName(tt.fn)); err != nil && !tt.sf { t.Errorf("AttachFromEmbedFS() failed: %s", err) return } if len(m.attachments) != 1 && !tt.sf { t.Errorf("AttachFile() failed. Number of attachments expected: %d, got: %d", 1, len(m.attachments)) return } if !tt.sf { file := m.attachments[0] if file == nil { t.Errorf("AttachFile() failed. Attachment file pointer is nil") return } if file.Name != tt.fn { t.Errorf("AttachFile() failed. Filename of attachment expected: %s, got: %s", tt.fn, file.Name) } buf := bytes.Buffer{} if _, err := file.Writer(&buf); err != nil { t.Errorf("failed to execute WriterFunc: %s", err) return } } m.Reset() }) } } // TestMsg_AttachFileBrokenFunc tests WriterFunc of the Msg.AttachFile method func TestMsg_AttachFileBrokenFunc(t *testing.T) { m := NewMsg() m.AttachFile("README.md") if len(m.attachments) != 1 { t.Errorf("AttachFile() failed. Number of attachments expected: %d, got: %d", 1, len(m.attachments)) return } file := m.attachments[0] if file == nil { t.Errorf("AttachFile() failed. Attachment file pointer is nil") return } file.Writer = func(io.Writer) (int64, error) { return 0, fmt.Errorf("failing intentionally") } buf := bytes.Buffer{} if _, err := file.Writer(&buf); err == nil { t.Errorf("execute WriterFunc did not fail, but was expected to fail") } } // TestMsg_AttachReader tests the Msg.AttachReader method func TestMsg_AttachReader(t *testing.T) { m := NewMsg() ts := "This is a test string" rbuf := bytes.Buffer{} rbuf.WriteString(ts) r := bufio.NewReader(&rbuf) if err := m.AttachReader("testfile.txt", r); err != nil { t.Errorf("AttachReader() failed. Expected no error, got: %s", err.Error()) return } if len(m.attachments) != 1 { t.Errorf("AttachReader() failed. Number of attachments expected: %d, got: %d", 1, len(m.attachments)) return } file := m.attachments[0] if file == nil { t.Errorf("AttachReader() failed. Attachment file pointer is nil") return } if file.Name != "testfile.txt" { t.Errorf("AttachReader() failed. Expected file name: %s, got: %s", "testfile.txt", file.Name) } wbuf := bytes.Buffer{} if _, err := file.Writer(&wbuf); err != nil { t.Errorf("execute WriterFunc failed: %s", err) } if wbuf.String() != ts { t.Errorf("AttachReader() failed. Expected string: %q, got: %q", ts, wbuf.String()) } } // TestMsg_EmbedFile tests the Msg.EmbedFile and the WithFilename FileOption method func TestMsg_EmbedFile(t *testing.T) { tests := []struct { name string file string fn string sf bool }{ {"File: README.md", "README.md", "README.md", false}, {"File: doc.go", "doc.go", "foo.go", false}, {"File: nonexisting", "", "invalid.file", true}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m.EmbedFile(tt.file, WithFileName(tt.fn), nil) if len(m.embeds) != 1 && !tt.sf { t.Errorf("EmbedFile() failed. Number of embeds expected: %d, got: %d", 1, len(m.embeds)) return } if !tt.sf { file := m.embeds[0] if file == nil { t.Errorf("EmbedFile() failed. Embedded file pointer is nil") return } if file.Name != tt.fn { t.Errorf("EmbedFile() failed. Filename of embeds expected: %s, got: %s", tt.fn, file.Name) } buf := bytes.Buffer{} if _, err := file.Writer(&buf); err != nil { t.Errorf("failed to execute WriterFunc: %s", err) return } } m.Reset() }) } } // TestMsg_EmbedFromEmbedFS tests the Msg.EmbedFromEmbedFS and the WithFilename FileOption method func TestMsg_EmbedFromEmbedFS(t *testing.T) { tests := []struct { name string file string fn string sf bool }{ {"File: README.md", "README.md", "README.md", false}, {"File: nonexisting", "", "invalid.file", true}, } m := NewMsg() for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { if err := m.EmbedFromEmbedFS(tt.file, &efs, WithFileName(tt.fn)); err != nil && !tt.sf { t.Errorf("EmbedFromEmbedFS() failed: %s", err) return } if len(m.embeds) != 1 && !tt.sf { t.Errorf("EmbedFile() failed. Number of embeds expected: %d, got: %d", 1, len(m.embeds)) return } if !tt.sf { file := m.embeds[0] if file == nil { t.Errorf("EmbedFile() failed. Embedded file pointer is nil") return } if file.Name != tt.fn { t.Errorf("EmbedFile() failed. Filename of embeds expected: %s, got: %s", tt.fn, file.Name) } buf := bytes.Buffer{} if _, err := file.Writer(&buf); err != nil { t.Errorf("failed to execute WriterFunc: %s", err) return } } m.Reset() }) } } // TestMsg_EmbedFileBrokenFunc tests WriterFunc of the Msg.EmbedFile method func TestMsg_EmbedFileBrokenFunc(t *testing.T) { m := NewMsg() m.EmbedFile("README.md") if len(m.embeds) != 1 { t.Errorf("EmbedFile() failed. Number of embeds expected: %d, got: %d", 1, len(m.embeds)) return } file := m.embeds[0] if file == nil { t.Errorf("EmbedFile() failed. Embedded file pointer is nil") return } file.Writer = func(io.Writer) (int64, error) { return 0, fmt.Errorf("failing intentionally") } buf := bytes.Buffer{} if _, err := file.Writer(&buf); err == nil { t.Errorf("execute WriterFunc did not fail, but was expected to fail") } } // TestMsg_EmbedReader tests the Msg.EmbedReader method func TestMsg_EmbedReader(t *testing.T) { m := NewMsg() ts := "This is a test string" rbuf := bytes.Buffer{} rbuf.WriteString(ts) r := bufio.NewReader(&rbuf) if err := m.EmbedReader("testfile.txt", r); err != nil { t.Errorf("EmbedReader() failed. Expected no error, got: %s", err.Error()) return } if len(m.embeds) != 1 { t.Errorf("EmbedReader() failed. Number of embeds expected: %d, got: %d", 1, len(m.embeds)) return } file := m.embeds[0] if file == nil { t.Errorf("EmbedReader() failed. Embedded file pointer is nil") return } if file.Name != "testfile.txt" { t.Errorf("EmbedReader() failed. Expected file name: %s, got: %s", "testfile.txt", file.Name) } wbuf := bytes.Buffer{} if _, err := file.Writer(&wbuf); err != nil { t.Errorf("execute WriterFunc failed: %s", err) } if wbuf.String() != ts { t.Errorf("EmbedReader() failed. Expected string: %q, got: %q", ts, wbuf.String()) } } // TestMsg_hasAlt tests the hasAlt() method of the Msg func TestMsg_hasAlt(t *testing.T) { m := NewMsg() m.SetBodyString(TypeTextPlain, "Plain") m.AddAlternativeString(TypeTextHTML, "HTML") if !m.hasAlt() { t.Errorf("mail has alternative parts but hasAlt() returned true") } } // TestMsg_hasAlt tests the hasAlt() method of the Msg with active S/MIME func TestMsg_hasAltWithSMime(t *testing.T) { privateKey, certificate, intermediateCertificate, err := getDummyRSACryptoMaterial() if err != nil { t.Errorf("failed to laod dummy crypto material. Cause: %v", err) } m := NewMsg() m.SetBodyString(TypeTextPlain, "Plain") m.AddAlternativeString(TypeTextHTML, "HTML") if err := m.SignWithSMimeRSA(privateKey, certificate, intermediateCertificate); err != nil { t.Errorf("failed to init smime configuration") } if m.hasAlt() { t.Errorf("mail has alternative parts and S/MIME is active, but hasAlt() returned true") } } // TestMsg_hasSMime tests the hasSMime() method of the Msg func TestMsg_hasSMime(t *testing.T) { privateKey, certificate, intermediateCertificate, err := getDummyRSACryptoMaterial() if err != nil { t.Errorf("failed to laod dummy crypto material. Cause: %v", err) } m := NewMsg() if err := m.SignWithSMimeRSA(privateKey, certificate, intermediateCertificate); err != nil { t.Errorf("failed to init smime configuration") } m.SetBodyString(TypeTextPlain, "Plain") if !m.hasSMime() { t.Errorf("mail has smime configured but hasSMime() returned true") } } // TestMsg_hasRelated tests the hasRelated() method of the Msg func TestMsg_hasRelated(t *testing.T) { m := NewMsg() m.SetBodyString(TypeTextPlain, "Plain") m.EmbedFile("README.md") if !m.hasRelated() { t.Errorf("mail has related parts but hasRelated() returned true") } } // TestMsg_hasMixed tests the hasMixed() method of the Msg func TestMsg_hasMixed(t *testing.T) { m := NewMsg() m.SetBodyString(TypeTextPlain, "Plain") m.AttachFile("README.md") if !m.hasMixed() { t.Errorf("mail has mixed parts but hasMixed() returned true") } } // TestMsg_WriteTo tests the WriteTo() method of the Msg func TestMsg_WriteTo(t *testing.T) { m := NewMsg() m.SetBodyString(TypeTextPlain, "Plain") wbuf := bytes.Buffer{} n, err := m.WriteTo(&wbuf) if err != nil { t.Errorf("WriteTo() failed: %s", err) return } if n != int64(wbuf.Len()) { t.Errorf("WriteTo() failed: expected written byte length: %d, got: %d", n, wbuf.Len()) } } // TestMsg_WriteToSkipMiddleware tests the WriteTo() method of the Msg func TestMsg_WriteToSkipMiddleware(t *testing.T) { m := NewMsg(WithMiddleware(encodeMiddleware{}), WithMiddleware(uppercaseMiddleware{})) m.Subject("This is a test") m.SetBodyString(TypeTextPlain, "Plain") wbuf := bytes.Buffer{} n, err := m.WriteToSkipMiddleware(&wbuf, "uppercase") if err != nil { t.Errorf("WriteToSkipMiddleware() failed: %s", err) return } if n != int64(wbuf.Len()) { t.Errorf("WriteToSkipMiddleware() failed: expected written byte length: %d, got: %d", n, wbuf.Len()) } if !strings.Contains(wbuf.String(), "Subject: This is @ test") { t.Errorf("WriteToSkipMiddleware failed. Unable to find encoded subject") } wbuf2 := bytes.Buffer{} n, err = m.WriteTo(&wbuf2) if err != nil { t.Errorf("WriteTo() failed: %s", err) return } if n != int64(wbuf2.Len()) { t.Errorf("WriteTo() failed: expected written byte length: %d, got: %d", n, wbuf2.Len()) } if !strings.Contains(wbuf2.String(), "Subject: THIS IS @ TEST") { t.Errorf("WriteToSkipMiddleware failed. Unable to find encoded and upperchase subject") } } // TestMsg_WriteToWithSMIME tests the WriteTo() method of the Msg func TestMsg_WriteToWithSMIME(t *testing.T) { privateKey, certificate, intermediateCertificate, err := getDummyRSACryptoMaterial() if err != nil { t.Errorf("failed to laod dummy crypto material. Cause: %v", err) } m := NewMsg() m.Subject("This is a test") m.SetBodyString(TypeTextPlain, "Plain") if err := m.SignWithSMimeRSA(privateKey, certificate, intermediateCertificate); err != nil { t.Errorf("failed to init smime configuration") } wbuf := bytes.Buffer{} if _, err = m.WriteTo(&wbuf); err != nil { t.Errorf("WriteTo() failed: %s", err) } result := wbuf.String() boundary := result[strings.LastIndex(result, "--")-60 : strings.LastIndex(result, "--")] if strings.Count(result, boundary) != 4 { t.Errorf("WriteTo() failed. False number of boundaries found") } parts := strings.Split(result, fmt.Sprintf("--%s", boundary)) if len(parts) != 4 { t.Errorf("WriteTo() failed. False number of parts found") } preamble := parts[0] if !strings.Contains(preamble, "MIME-Version: 1.0") { t.Errorf("WriteTo() failed. Unable to find MIME-Version") } if !strings.Contains(preamble, "Subject: This is a test") { t.Errorf("WriteTo() failed. Unable to find subject") } if !strings.Contains(preamble, fmt.Sprintf("Content-Type: multipart/signed; protocol=\"application/pkcs7-signature\"; micalg=sha-256;\r\n boundary=%s", boundary)) { t.Errorf("WriteTo() failed. Unable to find Content-Type") } signedData := parts[1] if !strings.Contains(signedData, "Content-Transfer-Encoding: quoted-printable") { t.Errorf("WriteTo() failed. Unable to find Content-Transfer-Encoding") } if !strings.Contains(signedData, "Content-Type: text/plain; charset=UTF-8") { t.Errorf("WriteTo() failed. Unable to find Content-Type") } if !strings.Contains(signedData, "Plain") { t.Errorf("WriteTo() failed. Unable to find Content") } signature := parts[2] if !strings.Contains(signature, "Content-Transfer-Encoding: base64") { t.Errorf("WriteTo() failed. Unable to find Content-Transfer-Encoding") } if !strings.Contains(signature, `application/pkcs7-signature; name="smime.p7s"`) { t.Errorf("WriteTo() failed. Unable to find Content-Type") } if strings.Contains(signature, "Plain") { t.Errorf("WriteTo() failed. Unable to find signature") } } // TestMsg_WriteTo_fails tests the WriteTo() method of the Msg but with a failing body writer function func TestMsg_WriteTo_fails(t *testing.T) { m := NewMsg() m.SetBodyWriter(TypeTextPlain, func(io.Writer) (int64, error) { return 0, errors.New("failed") }) _, err := m.WriteTo(io.Discard) if err == nil { t.Errorf("WriteTo() with failing BodyWriter function was supposed to fail, but didn't") return } // NoEncoding handles the errors separately m = NewMsg(WithEncoding(NoEncoding)) m.SetBodyWriter(TypeTextPlain, func(io.Writer) (int64, error) { return 0, errors.New("failed") }) _, err = m.WriteTo(io.Discard) if err == nil { t.Errorf("WriteTo() (no encoding) with failing BodyWriter function was supposed to fail, but didn't") return } } // TestMsg_Write tests the Write() method of the Msg func TestMsg_Write(t *testing.T) { m := NewMsg() m.SetBodyString(TypeTextPlain, "Plain") wbuf := bytes.Buffer{} n, err := m.Write(&wbuf) if err != nil { t.Errorf("WriteTo() failed: %s", err) return } if n != int64(wbuf.Len()) { t.Errorf("WriteTo() failed: expected written byte length: %d, got: %d", n, wbuf.Len()) } } // TestMsg_WriteWithLongHeader tests the WriteTo() method of the Msg with a long header func TestMsg_WriteWithLongHeader(t *testing.T) { m := NewMsg() m.SetBodyString(TypeTextPlain, "Plain") m.SetGenHeader(HeaderContentLang, "de", "en", "fr", "es", "xxxx", "yyyy", "de", "en", "fr", "es", "xxxx", "yyyy", "de", "en", "fr", "es", "xxxx", "yyyy", "de", "en", "fr") m.SetGenHeader(HeaderContentID, "XXXXXXXXXXXXXXX XXXXXXXXXXXXXXX XXXXXXXXXXXXXXXXXX XXXXXXXXXXXXXXXXXXXXXX", "XXXXXXXXXXXXX XXXXXXXXXXXXXXXXXXX XXXXXXXXXXXXXXXXXXX XXXXXXXXXXXXXXXXXXXXXXXXXXX") wbuf := bytes.Buffer{} n, err := m.WriteTo(&wbuf) if err != nil { t.Errorf("WriteTo() failed: %s", err) return } if n != int64(wbuf.Len()) { t.Errorf("WriteTo() failed: expected written byte length: %d, got: %d", n, wbuf.Len()) } } // TestMsg_WriteDiffEncoding tests the WriteTo() method of the Msg with different Encoding func TestMsg_WriteDiffEncoding(t *testing.T) { tests := []struct { name string ct ContentType en Encoding alt bool wa bool we bool }{ {"Plain/QP/NoAlt/NoAttach/NoEmbed", TypeTextPlain, EncodingQP, false, false, false}, {"Plain/B64/NoAlt/NoAttach/NoEmbed", TypeTextPlain, EncodingB64, false, false, false}, {"Plain/No/NoAlt/NoAttach/NoEmbed", TypeTextPlain, NoEncoding, false, false, false}, {"HTML/QP/NoAlt/NoAttach/NoEmbed", TypeTextHTML, EncodingQP, false, false, false}, {"HTML/B64/NoAlt/NoAttach/NoEmbed", TypeTextHTML, EncodingB64, false, false, false}, {"HTML/No/NoAlt/NoAttach/NoEmbed", TypeTextHTML, NoEncoding, false, false, false}, {"Plain/QP/HTML/NoAttach/NoEmbed", TypeTextPlain, EncodingQP, true, false, false}, {"Plain/B64/HTML/NoAttach/NoEmbed", TypeTextPlain, EncodingB64, true, false, false}, {"Plain/No/HTML/NoAttach/NoEmbed", TypeTextPlain, NoEncoding, true, false, false}, {"Plain/QP/NoAlt/Attach/NoEmbed", TypeTextPlain, EncodingQP, false, true, false}, {"Plain/B64/NoAlt/Attach/NoEmbed", TypeTextPlain, EncodingB64, false, true, false}, {"Plain/No/NoAlt/Attach/NoEmbed", TypeTextPlain, NoEncoding, false, true, false}, {"Plain/QP/NoAlt/NoAttach/Embed", TypeTextPlain, EncodingQP, false, false, true}, {"Plain/B64/NoAlt/NoAttach/Embed", TypeTextPlain, EncodingB64, false, false, true}, {"Plain/No/NoAlt/NoAttach/Embed", TypeTextPlain, NoEncoding, false, false, true}, {"Plain/QP/HTML/Attach/Embed", TypeTextPlain, EncodingQP, true, true, true}, {"Plain/B64/HTML/Attach/Embed", TypeTextPlain, EncodingB64, true, true, true}, {"Plain/No/HTML/Attach/Embed", TypeTextPlain, NoEncoding, true, true, true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { m := NewMsg(WithEncoding(tt.en)) m.SetBodyString(tt.ct, tt.name) if tt.alt { m.AddAlternativeString(TypeTextHTML, fmt.Sprintf("

%s

", tt.name)) } if tt.wa { m.AttachFile("README.md") } if tt.we { m.EmbedFile("README.md") } wbuf := bytes.Buffer{} n, err := m.WriteTo(&wbuf) if err != nil { t.Errorf("WriteTo() failed: %s", err) return } if n != int64(wbuf.Len()) { t.Errorf("WriteTo() failed: expected written byte length: %d, got: %d", n, wbuf.Len()) } wbuf.Reset() }) } } // TestMsg_appendFile tests the appendFile() method of the Msg func TestMsg_appendFile(t *testing.T) { m := NewMsg() var fl []*File f := &File{ Name: "file.txt", } fl = m.appendFile(fl, f, nil) if len(fl) != 1 { t.Errorf("appendFile() failed. Expected length: %d, got: %d", 1, len(fl)) } fl = m.appendFile(fl, f, nil) if len(fl) != 2 { t.Errorf("appendFile() failed. Expected length: %d, got: %d", 2, len(fl)) } } // TestMsg_multipleWrites tests multiple executions of WriteTo on the Msg func TestMsg_multipleWrites(t *testing.T) { ts := "XXX_UNIQUE_STRING_XXX" wbuf := bytes.Buffer{} m := NewMsg() m.SetBodyString(TypeTextPlain, ts) // First WriteTo() _, err := m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), ts) { t.Errorf("first WriteTo() body does not contain unique string: %s", ts) } // Second WriteTo() wbuf.Reset() _, err = m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), ts) { t.Errorf("second WriteTo() body does not contain unique string: %s", ts) } } // TestMsg_NewReader tests the Msg.NewReader method func TestMsg_NewReader(t *testing.T) { m := NewMsg() m.SetBodyString(TypeTextPlain, "TEST123") mr := m.NewReader() if mr == nil { t.Errorf("NewReader failed: Reader is nil") } if mr.Error() != nil { t.Errorf("NewReader failed: %s", mr.Error()) } } // TestMsg_NewReader_ioCopy tests the Msg.NewReader method using io.Copy func TestMsg_NewReader_ioCopy(t *testing.T) { wbuf1 := bytes.Buffer{} wbuf2 := bytes.Buffer{} m := NewMsg() m.SetBodyString(TypeTextPlain, "TEST123") mr := m.NewReader() if mr == nil { t.Errorf("NewReader failed: Reader is nil") } // First we use WriteTo to have something to compare to _, err := m.WriteTo(&wbuf1) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } // Then we write to wbuf2 via io.Copy n, err := io.Copy(&wbuf2, mr) if err != nil { t.Errorf("failed to use io.Copy on Reader: %s", err) } if n != int64(wbuf1.Len()) { t.Errorf("message length of WriteTo and io.Copy differ. Expected: %d, got: %d", wbuf1.Len(), n) } if wbuf1.String() != wbuf2.String() { t.Errorf("message content of WriteTo and io.Copy differ") } } // TestMsg_UpdateReader tests the Msg.UpdateReader method func TestMsg_UpdateReader(t *testing.T) { m := NewMsg() m.Subject("Subject-Run 1") mr := m.NewReader() if mr == nil { t.Errorf("NewReader failed: Reader is nil") } wbuf1 := bytes.Buffer{} _, err := io.Copy(&wbuf1, mr) if err != nil { t.Errorf("io.Copy on Reader failed: %s", err) } if !strings.Contains(wbuf1.String(), "Subject: Subject-Run 1") { t.Errorf("io.Copy on Reader failed. Expected to find %q but string in Subject was not found", "Subject-Run 1") } m.Subject("Subject-Run 2") m.UpdateReader(mr) wbuf2 := bytes.Buffer{} _, err = io.Copy(&wbuf2, mr) if err != nil { t.Errorf("2nd io.Copy on Reader failed: %s", err) } if !strings.Contains(wbuf2.String(), "Subject: Subject-Run 2") { t.Errorf("io.Copy on Reader failed. Expected to find %q but string in Subject was not found", "Subject-Run 2") } } // TestMsg_SetBodyTextTemplate tests the Msg.SetBodyTextTemplate method func TestMsg_SetBodyTextTemplate(t *testing.T) { tests := []struct { name string tpl string ph string sf bool }{ {"normal text", "This is a {{.Placeholder}}", "TemplateTest", false}, {"invalid tpl", "This is a {{ foo .Placeholder}}", "TemplateTest", true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { data := struct { Placeholder string }{Placeholder: tt.ph} tpl, err := ttpl.New("test").Parse(tt.tpl) if err != nil && !tt.sf { t.Errorf("failed to render template: %s", err) return } m := NewMsg() if err := m.SetBodyTextTemplate(tpl, data); err != nil && !tt.sf { t.Errorf("failed to set template as body: %s", err) } wbuf := bytes.Buffer{} _, err = m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), tt.ph) && !tt.sf { t.Errorf("SetBodyTextTemplate failed: Body does not contain the expected tpl placeholder: %s", tt.ph) } m.Reset() }) } } // TestMsg_SetBodyHTMLTemplate tests the Msg.SetBodyHTMLTemplate method func TestMsg_SetBodyHTMLTemplate(t *testing.T) { tests := []struct { name string tpl string ph string ex string sf bool }{ {"normal HTML", "

This is a {{.Placeholder}}

", "TemplateTest", "TemplateTest", false}, { "HTML with HTML", "

This is a {{.Placeholder}}

", "", "<script>alert(1)</script>", false, }, {"invalid tpl", "

This is a {{ foo .Placeholder}}

", "TemplateTest", "", true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { data := struct { Placeholder string }{Placeholder: tt.ph} tpl, err := htpl.New("test").Parse(tt.tpl) if err != nil && !tt.sf { t.Errorf("failed to render template: %s", err) return } m := NewMsg() if err := m.SetBodyHTMLTemplate(tpl, data); err != nil && !tt.sf { t.Errorf("failed to set template as body: %s", err) } wbuf := bytes.Buffer{} _, err = m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), tt.ex) && !tt.sf { t.Errorf("SetBodyTextTemplate failed: Body does not contain the expected tpl placeholder: %s", tt.ph) } m.Reset() }) } } // TestMsg_AddAlternativeTextTemplate tests the Msg.AddAlternativeTextTemplate method func TestMsg_AddAlternativeTextTemplate(t *testing.T) { tests := []struct { name string tpl string ph string sf bool }{ {"normal text", "This is a {{.Placeholder}}", "TemplateTest", false}, {"invalid tpl", "This is a {{ foo .Placeholder}}", "TemplateTest", true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { data := struct { Placeholder string }{Placeholder: tt.ph} tpl, err := ttpl.New("test").Parse(tt.tpl) if err != nil && !tt.sf { t.Errorf("failed to render template: %s", err) return } m := NewMsg() m.SetBodyString(TypeTextHTML, "") if err := m.AddAlternativeTextTemplate(tpl, data); err != nil && !tt.sf { t.Errorf("failed to set template as alternative part: %s", err) } wbuf := bytes.Buffer{} _, err = m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), tt.ph) && !tt.sf { t.Errorf("SetBodyTextTemplate failed: Body does not contain the expected tpl placeholder: %s", tt.ph) } m.Reset() }) } } // TestMsg_AddAlternativeHTMLTemplate tests the Msg.AddAlternativeHTMLTemplate method func TestMsg_AddAlternativeHTMLTemplate(t *testing.T) { tests := []struct { name string tpl string ph string ex string sf bool }{ {"normal HTML", "

This is a {{.Placeholder}}

", "TemplateTest", "TemplateTest", false}, { "HTML with HTML", "

This is a {{.Placeholder}}

", "", "<script>alert(1)</script>", false, }, {"invalid tpl", "

This is a {{ foo .Placeholder}}

", "TemplateTest", "", true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { data := struct { Placeholder string }{Placeholder: tt.ph} tpl, err := htpl.New("test").Parse(tt.tpl) if err != nil && !tt.sf { t.Errorf("failed to render template: %s", err) return } m := NewMsg() m.SetBodyString(TypeTextPlain, "") if err := m.AddAlternativeHTMLTemplate(tpl, data); err != nil && !tt.sf { t.Errorf("failed to set template as alternative part: %s", err) } wbuf := bytes.Buffer{} _, err = m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), tt.ex) && !tt.sf { t.Errorf("SetBodyTextTemplate failed: Body does not contain the expected tpl placeholder: %s", tt.ph) } m.Reset() }) } } // TestMsg_AttachTextTemplate tests the Msg.AttachTextTemplate method func TestMsg_AttachTextTemplate(t *testing.T) { tests := []struct { name string tpl string ph string ex string ac int sf bool }{ { "normal text", "This is a {{.Placeholder}}", "TemplateTest", "VGhpcyBpcyBhIFRlbXBsYXRlVGVzdA==", 1, false, }, {"invalid tpl", "This is a {{ foo .Placeholder}}", "TemplateTest", "", 0, true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { data := struct { Placeholder string }{Placeholder: tt.ph} tpl, err := ttpl.New("test").Parse(tt.tpl) if err != nil && !tt.sf { t.Errorf("failed to render template: %s", err) return } m := NewMsg() m.SetBodyString(TypeTextPlain, "This is the body") if err := m.AttachTextTemplate("attachment.txt", tpl, data); err != nil && !tt.sf { t.Errorf("failed to attach template: %s", err) } wbuf := bytes.Buffer{} _, err = m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), tt.ex) && !tt.sf { t.Errorf("SetBodyTextTemplate failed: Body does not contain the expected tpl placeholder: %s", tt.ph) } if len(m.attachments) != tt.ac { t.Errorf("wrong number of attachments. Expected: %d, got: %d", tt.ac, len(m.attachments)) } m.Reset() }) } } // TestMsg_AttachHTMLTemplate tests the Msg.AttachHTMLTemplate method func TestMsg_AttachHTMLTemplate(t *testing.T) { tests := []struct { name string tpl string ph string ex string ac int sf bool }{ { "normal HTML", "

This is a {{.Placeholder}}

", "TemplateTest", "PHA+VGhpcyBpcyBhIFRlbXBsYXRlVGVzdDwvcD4=", 1, false, }, { "HTML with HTML", "

This is a {{.Placeholder}}

", "", "PHA+VGhpcyBpcyBhICZsdDtzY3JpcHQmZ3Q7YWxlcnQoMSkmbHQ7L3NjcmlwdCZndDs8L3A+", 1, false, }, {"invalid tpl", "

This is a {{ foo .Placeholder}}

", "TemplateTest", "", 0, true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { data := struct { Placeholder string }{Placeholder: tt.ph} tpl, err := htpl.New("test").Parse(tt.tpl) if err != nil && !tt.sf { t.Errorf("failed to render template: %s", err) return } m := NewMsg() m.SetBodyString(TypeTextPlain, "") if err := m.AttachHTMLTemplate("attachment.html", tpl, data); err != nil && !tt.sf { t.Errorf("failed to set template as alternative part: %s", err) } wbuf := bytes.Buffer{} _, err = m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), tt.ex) && !tt.sf { t.Errorf("SetBodyTextTemplate failed: Body does not contain the expected tpl placeholder: %s", tt.ph) } if len(m.attachments) != tt.ac { t.Errorf("wrong number of attachments. Expected: %d, got: %d", tt.ac, len(m.attachments)) } m.Reset() }) } } // TestMsg_EmbedTextTemplate tests the Msg.EmbedTextTemplate method func TestMsg_EmbedTextTemplate(t *testing.T) { tests := []struct { name string tpl string ph string ex string ec int sf bool }{ { "normal text", "This is a {{.Placeholder}}", "TemplateTest", "VGhpcyBpcyBhIFRlbXBsYXRlVGVzdA==", 1, false, }, {"invalid tpl", "This is a {{ foo .Placeholder}}", "TemplateTest", "", 0, true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { data := struct { Placeholder string }{Placeholder: tt.ph} tpl, err := ttpl.New("test").Parse(tt.tpl) if err != nil && !tt.sf { t.Errorf("failed to render template: %s", err) return } m := NewMsg() m.SetBodyString(TypeTextPlain, "This is the body") if err := m.EmbedTextTemplate("attachment.txt", tpl, data); err != nil && !tt.sf { t.Errorf("failed to attach template: %s", err) } wbuf := bytes.Buffer{} _, err = m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), tt.ex) && !tt.sf { t.Errorf("SetBodyTextTemplate failed: Body does not contain the expected tpl placeholder: %s", tt.ph) } if len(m.embeds) != tt.ec { t.Errorf("wrong number of attachments. Expected: %d, got: %d", tt.ec, len(m.attachments)) } m.Reset() }) } } // TestMsg_EmbedHTMLTemplate tests the Msg.EmbedHTMLTemplate method func TestMsg_EmbedHTMLTemplate(t *testing.T) { tests := []struct { name string tpl string ph string ex string ec int sf bool }{ { "normal HTML", "

This is a {{.Placeholder}}

", "TemplateTest", "PHA+VGhpcyBpcyBhIFRlbXBsYXRlVGVzdDwvcD4=", 1, false, }, { "HTML with HTML", "

This is a {{.Placeholder}}

", "", "PHA+VGhpcyBpcyBhICZsdDtzY3JpcHQmZ3Q7YWxlcnQoMSkmbHQ7L3NjcmlwdCZndDs8L3A+", 1, false, }, {"invalid tpl", "

This is a {{ foo .Placeholder}}

", "TemplateTest", "", 0, true}, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { data := struct { Placeholder string }{Placeholder: tt.ph} tpl, err := htpl.New("test").Parse(tt.tpl) if err != nil && !tt.sf { t.Errorf("failed to render template: %s", err) return } m := NewMsg() m.SetBodyString(TypeTextPlain, "") if err := m.EmbedHTMLTemplate("attachment.html", tpl, data); err != nil && !tt.sf { t.Errorf("failed to set template as alternative part: %s", err) } wbuf := bytes.Buffer{} _, err = m.WriteTo(&wbuf) if err != nil { t.Errorf("failed to write body to buffer: %s", err) } if !strings.Contains(wbuf.String(), tt.ex) && !tt.sf { t.Errorf("SetBodyTextTemplate failed: Body does not contain the expected tpl placeholder: %s", tt.ph) } if len(m.embeds) != tt.ec { t.Errorf("wrong number of attachments. Expected: %d, got: %d", tt.ec, len(m.attachments)) } m.Reset() }) } } // TestMsg_WriteToTempFile will test the output to temporary files func TestMsg_WriteToTempFile(t *testing.T) { m := NewMsg() _ = m.From("Toni Tester ") _ = m.To("Ellenor Tester ") m.SetBodyString(TypeTextPlain, "This is a test") f, err := m.WriteToTempFile() if err != nil { t.Errorf("failed to write message to temporary output file: %s", err) } _ = os.Remove(f) } // TestMsg_WriteToFile will test the output to a file func TestMsg_WriteToFile(t *testing.T) { f, err := os.CreateTemp("", "go-mail-test_*.eml") if err != nil { t.Errorf("failed to create temporary output file: %s", err) } defer func() { _ = f.Close() _ = os.Remove(f.Name()) }() m := NewMsg() _ = m.From("Toni Tester ") _ = m.To("Ellenor Tester ") m.SetBodyString(TypeTextPlain, "This is a test") if err := m.WriteToFile(f.Name()); err != nil { t.Errorf("failed to write to output file: %s", err) } fi, err := os.Stat(f.Name()) if err != nil { t.Errorf("failed to stat output file: %s", err) } if fi == nil { t.Errorf("received empty file handle") return } if fi.Size() <= 0 { t.Errorf("output file is expected to contain data but its size is zero") } } // TestMsg_GetGenHeader will test the GetGenHeader method of the Msg func TestMsg_GetGenHeader(t *testing.T) { m := NewMsg() m.Subject("this is a test") sa := m.GetGenHeader(HeaderSubject) if len(sa) <= 0 { t.Errorf("GetGenHeader on subject failed. Got empty slice") return } if sa[0] == "" { t.Errorf("GetGenHeader on subject failed. Got empty value") } if sa[0] != "this is a test" { t.Errorf("GetGenHeader on subject failed. Expected: %q, got: %q", "this is a test", sa[0]) } } // TestMsg_GetAddrHeader will test the Msg.GetAddrHeader method func TestMsg_GetAddrHeader(t *testing.T) { m := NewMsg() if err := m.FromFormat("Toni Sender", "sender@example.com"); err != nil { t.Errorf("failed to set FROM address: %s", err) } if err := m.AddToFormat("Toni To", "to@example.com"); err != nil { t.Errorf("failed to set TO address: %s", err) } if err := m.AddCcFormat("Toni Cc", "cc@example.com"); err != nil { t.Errorf("failed to set CC address: %s", err) } if err := m.AddBccFormat("Toni Bcc", "bcc@example.com"); err != nil { t.Errorf("failed to set BCC address: %s", err) } fh := m.GetAddrHeader(HeaderFrom) if len(fh) <= 0 { t.Errorf("GetAddrHeader on FROM failed. Got empty slice") return } if fh[0].String() == "" { t.Errorf("GetAddrHeader on FROM failed. Got empty value") } if fh[0].String() != `"Toni Sender" ` { t.Errorf("GetAddrHeader on FROM failed. Expected: %q, got: %q", `"Toni Sender" "`, fh[0].String()) } th := m.GetAddrHeader(HeaderTo) if len(th) <= 0 { t.Errorf("GetAddrHeader on TO failed. Got empty slice") return } if th[0].String() == "" { t.Errorf("GetAddrHeader on TO failed. Got empty value") } if th[0].String() != `"Toni To" ` { t.Errorf("GetAddrHeader on TO failed. Expected: %q, got: %q", `"Toni To" "`, th[0].String()) } ch := m.GetAddrHeader(HeaderCc) if len(ch) <= 0 { t.Errorf("GetAddrHeader on CC failed. Got empty slice") return } if ch[0].String() == "" { t.Errorf("GetAddrHeader on CC failed. Got empty value") } if ch[0].String() != `"Toni Cc" ` { t.Errorf("GetAddrHeader on CC failed. Expected: %q, got: %q", `"Toni Cc" "`, ch[0].String()) } bh := m.GetAddrHeader(HeaderBcc) if len(bh) <= 0 { t.Errorf("GetAddrHeader on BCC failed. Got empty slice") return } if bh[0].String() == "" { t.Errorf("GetAddrHeader on BCC failed. Got empty value") } if bh[0].String() != `"Toni Bcc" ` { t.Errorf("GetAddrHeader on BCC failed. Expected: %q, got: %q", `"Toni Bcc" "`, bh[0].String()) } } // TestMsg_GetFrom will test the Msg.GetFrom method func TestMsg_GetFrom(t *testing.T) { m := NewMsg() if err := m.FromFormat("Toni Sender", "sender@example.com"); err != nil { t.Errorf("failed to set FROM address: %s", err) } fh := m.GetFrom() if len(fh) <= 0 { t.Errorf("GetFrom failed. Got empty slice") return } if fh[0].String() == "" { t.Errorf("GetFrom failed. Got empty value") } if fh[0].String() != `"Toni Sender" ` { t.Errorf("GetFrom failed. Expected: %q, got: %q", `"Toni Sender" "`, fh[0].String()) } } // TestMsg_GetFromString will test the Msg.GetFromString method func TestMsg_GetFromString(t *testing.T) { m := NewMsg() if err := m.FromFormat("Toni Sender", "sender@example.com"); err != nil { t.Errorf("failed to set FROM address: %s", err) } fh := m.GetFromString() if len(fh) <= 0 { t.Errorf("GetFromString failed. Got empty slice") return } if fh[0] == "" { t.Errorf("GetFromString failed. Got empty value") } if fh[0] != `"Toni Sender" ` { t.Errorf("GetFromString failed. Expected: %q, got: %q", `"Toni Sender" "`, fh[0]) } } // TestMsg_GetTo will test the Msg.GetTo method func TestMsg_GetTo(t *testing.T) { m := NewMsg() if err := m.AddToFormat("Toni To", "to@example.com"); err != nil { t.Errorf("failed to set TO address: %s", err) } fh := m.GetTo() if len(fh) <= 0 { t.Errorf("GetTo failed. Got empty slice") return } if fh[0].String() == "" { t.Errorf("GetTo failed. Got empty value") } if fh[0].String() != `"Toni To" ` { t.Errorf("GetTo failed. Expected: %q, got: %q", `"Toni To" "`, fh[0].String()) } } // TestMsg_GetToString will test the Msg.GetToString method func TestMsg_GetToString(t *testing.T) { m := NewMsg() if err := m.AddToFormat("Toni To", "to@example.com"); err != nil { t.Errorf("failed to set TO address: %s", err) } fh := m.GetToString() if len(fh) <= 0 { t.Errorf("GetToString failed. Got empty slice") return } if fh[0] == "" { t.Errorf("GetToString failed. Got empty value") } if fh[0] != `"Toni To" ` { t.Errorf("GetToString failed. Expected: %q, got: %q", `"Toni To" "`, fh[0]) } } // TestMsg_GetCc will test the Msg.GetCc method func TestMsg_GetCc(t *testing.T) { m := NewMsg() if err := m.AddCcFormat("Toni Cc", "cc@example.com"); err != nil { t.Errorf("failed to set TO address: %s", err) } fh := m.GetCc() if len(fh) <= 0 { t.Errorf("GetCc failed. Got empty slice") return } if fh[0].String() == "" { t.Errorf("GetCc failed. Got empty value") } if fh[0].String() != `"Toni Cc" ` { t.Errorf("GetCc failed. Expected: %q, got: %q", `"Toni Cc" "`, fh[0].String()) } } // TestMsg_GetCcString will test the Msg.GetCcString method func TestMsg_GetCcString(t *testing.T) { m := NewMsg() if err := m.AddCcFormat("Toni Cc", "cc@example.com"); err != nil { t.Errorf("failed to set TO address: %s", err) } fh := m.GetCcString() if len(fh) <= 0 { t.Errorf("GetCcString failed. Got empty slice") return } if fh[0] == "" { t.Errorf("GetCcString failed. Got empty value") } if fh[0] != `"Toni Cc" ` { t.Errorf("GetCcString failed. Expected: %q, got: %q", `"Toni Cc" "`, fh[0]) } } // TestMsg_GetBcc will test the Msg.GetBcc method func TestMsg_GetBcc(t *testing.T) { m := NewMsg() if err := m.AddBccFormat("Toni Bcc", "bcc@example.com"); err != nil { t.Errorf("failed to set TO address: %s", err) } fh := m.GetBcc() if len(fh) <= 0 { t.Errorf("GetBcc failed. Got empty slice") return } if fh[0].String() == "" { t.Errorf("GetBcc failed. Got empty value") } if fh[0].String() != `"Toni Bcc" ` { t.Errorf("GetBcc failed. Expected: %q, got: %q", `"Toni Cc" "`, fh[0].String()) } } // TestMsg_GetBccString will test the Msg.GetBccString method func TestMsg_GetBccString(t *testing.T) { m := NewMsg() if err := m.AddBccFormat("Toni Bcc", "bcc@example.com"); err != nil { t.Errorf("failed to set TO address: %s", err) } fh := m.GetBccString() if len(fh) <= 0 { t.Errorf("GetBccString failed. Got empty slice") return } if fh[0] == "" { t.Errorf("GetBccString failed. Got empty value") } if fh[0] != `"Toni Bcc" ` { t.Errorf("GetBccString failed. Expected: %q, got: %q", `"Toni Cc" "`, fh[0]) } } // TestMsg_GetBoundary will test the Msg.GetBoundary method func TestMsg_GetBoundary(t *testing.T) { b := "random_boundary_string" m := NewMsg() if boundary := m.GetBoundary(); boundary != "" { t.Errorf("GetBoundary failed. Expected empty string, but got: %s", boundary) } m = NewMsg(WithBoundary(b)) if boundary := m.GetBoundary(); boundary != b { t.Errorf("GetBoundary failed. Expected boundary: %s, got: %s", b, boundary) } } // TestMsg_AttachEmbedReader_consecutive tests the Msg.AttachReader and Msg.EmbedReader // methods with consecutive calls to Msg.WriteTo to make sure the attachments are not // lost (see Github issue #110) func TestMsg_AttachEmbedReader_consecutive(t *testing.T) { ts1 := "This is a test string" ts2 := "Another test string" m := NewMsg() if err := m.AttachReader("attachment.txt", bytes.NewBufferString(ts1)); err != nil { t.Errorf("AttachReader() failed. Expected no error, got: %s", err.Error()) return } if err := m.EmbedReader("embedded.txt", bytes.NewBufferString(ts2)); err != nil { t.Errorf("EmbedReader() failed. Expected no error, got: %s", err.Error()) return } obuf1 := &bytes.Buffer{} obuf2 := &bytes.Buffer{} _, err := m.WriteTo(obuf1) if err != nil { t.Errorf("WriteTo to first output buffer failed: %s", err) } _, err = m.WriteTo(obuf2) if err != nil { t.Errorf("WriteTo to second output buffer failed: %s", err) } if !strings.Contains(obuf1.String(), "VGhpcyBpcyBhIHRlc3Qgc3RyaW5n") { t.Errorf("Expected file attachment string not found in first output buffer") } if !strings.Contains(obuf2.String(), "VGhpcyBpcyBhIHRlc3Qgc3RyaW5n") { t.Errorf("Expected file attachment string not found in second output buffer") } if !strings.Contains(obuf1.String(), "QW5vdGhlciB0ZXN0IHN0cmluZw==") { t.Errorf("Expected embedded file string not found in first output buffer") } if !strings.Contains(obuf2.String(), "QW5vdGhlciB0ZXN0IHN0cmluZw==") { t.Errorf("Expected embded file string not found in second output buffer") } } // TestMsg_AttachEmbedReadSeeker_consecutive tests the Msg.AttachReadSeeker and // Msg.EmbedReadSeeker methods with consecutive calls to Msg.WriteTo to make // sure the attachments are not lost (see Github issue #110) func TestMsg_AttachEmbedReadSeeker_consecutive(t *testing.T) { ts1 := []byte("This is a test string") ts2 := []byte("Another test string") m := NewMsg() m.AttachReadSeeker("attachment.txt", bytes.NewReader(ts1)) m.EmbedReadSeeker("embedded.txt", bytes.NewReader(ts2)) obuf1 := &bytes.Buffer{} obuf2 := &bytes.Buffer{} _, err := m.WriteTo(obuf1) if err != nil { t.Errorf("WriteTo to first output buffer failed: %s", err) } _, err = m.WriteTo(obuf2) if err != nil { t.Errorf("WriteTo to second output buffer failed: %s", err) } if !strings.Contains(obuf1.String(), "VGhpcyBpcyBhIHRlc3Qgc3RyaW5n") { t.Errorf("Expected file attachment string not found in first output buffer") } if !strings.Contains(obuf2.String(), "VGhpcyBpcyBhIHRlc3Qgc3RyaW5n") { t.Errorf("Expected file attachment string not found in second output buffer") } if !strings.Contains(obuf1.String(), "QW5vdGhlciB0ZXN0IHN0cmluZw==") { t.Errorf("Expected embedded file string not found in first output buffer") } if !strings.Contains(obuf2.String(), "QW5vdGhlciB0ZXN0IHN0cmluZw==") { t.Errorf("Expected embded file string not found in second output buffer") } } // TestMsg_AttachReadSeeker tests the Msg.AttachReadSeeker method func TestMsg_AttachReadSeeker(t *testing.T) { m := NewMsg() ts := []byte("This is a test string") r := bytes.NewReader(ts) m.AttachReadSeeker("testfile.txt", r) if len(m.attachments) != 1 { t.Errorf("AttachReadSeeker() failed. Number of attachments expected: %d, got: %d", 1, len(m.attachments)) return } file := m.attachments[0] if file == nil { t.Errorf("AttachReadSeeker() failed. Attachment file pointer is nil") return } if file.Name != "testfile.txt" { t.Errorf("AttachReadSeeker() failed. Expected file name: %s, got: %s", "testfile.txt", file.Name) } wbuf := bytes.Buffer{} if _, err := file.Writer(&wbuf); err != nil { t.Errorf("execute WriterFunc failed: %s", err) } if wbuf.String() != string(ts) { t.Errorf("AttachReadSeeker() failed. Expected string: %q, got: %q", ts, wbuf.String()) } } // TestMsg_EmbedReadSeeker tests the Msg.EmbedReadSeeker method func TestMsg_EmbedReadSeeker(t *testing.T) { m := NewMsg() ts := []byte("This is a test string") r := bytes.NewReader(ts) m.EmbedReadSeeker("testfile.txt", r) if len(m.embeds) != 1 { t.Errorf("EmbedReadSeeker() failed. Number of attachments expected: %d, got: %d", 1, len(m.embeds)) return } file := m.embeds[0] if file == nil { t.Errorf("EmbedReadSeeker() failed. Embedded file pointer is nil") return } if file.Name != "testfile.txt" { t.Errorf("EmbedReadSeeker() failed. Expected file name: %s, got: %s", "testfile.txt", file.Name) } wbuf := bytes.Buffer{} if _, err := file.Writer(&wbuf); err != nil { t.Errorf("execute WriterFunc failed: %s", err) } if wbuf.String() != string(ts) { t.Errorf("EmbedReadSeeker() failed. Expected string: %q, got: %q", ts, wbuf.String()) } } // TestMsg_ToFromString tests Msg.ToFromString in different scenarios func TestMsg_ToFromString(t *testing.T) { tests := []struct { n string v string w []*mail.Address sf bool }{ {"valid single address", "test@test.com", []*mail.Address{ {Name: "", Address: "test@test.com"}, }, false}, { "valid multiple addresses", "test@test.com,test2@example.com", []*mail.Address{ {Name: "", Address: "test@test.com"}, {Name: "", Address: "test2@example.com"}, }, false, }, { "valid multiple addresses with space and name", `test@test.com, "Toni Tester" `, []*mail.Address{ {Name: "", Address: "test@test.com"}, {Name: "Toni Tester", Address: "test2@example.com"}, }, false, }, { "invalid and valid multiple addresses", "test@test.com,test2#example.com", nil, true, }, } for _, tt := range tests { t.Run(tt.n, func(t *testing.T) { m := NewMsg() if err := m.ToFromString(tt.v); err != nil && !tt.sf { t.Errorf("Msg.ToFromString failed: %s", err) return } mto := m.GetTo() if len(mto) != len(tt.w) { t.Errorf("Msg.ToFromString failed, expected len: %d, got: %d", len(tt.w), len(mto)) return } for i := range mto { w := tt.w[i] g := mto[i] if w.String() != g.String() { t.Errorf("Msg.ToFromString failed, expected address: %s, got: %s", w.String(), g.String()) } } }) } } // TestMsg_CcFromString tests Msg.CcFromString in different scenarios func TestMsg_CcFromString(t *testing.T) { tests := []struct { n string v string w []*mail.Address sf bool }{ {"valid single address", "test@test.com", []*mail.Address{ {Name: "", Address: "test@test.com"}, }, false}, { "valid multiple addresses", "test@test.com,test2@example.com", []*mail.Address{ {Name: "", Address: "test@test.com"}, {Name: "", Address: "test2@example.com"}, }, false, }, { "valid multiple addresses with space and name", `test@test.com, "Toni Tester" `, []*mail.Address{ {Name: "", Address: "test@test.com"}, {Name: "Toni Tester", Address: "test2@example.com"}, }, false, }, { "invalid and valid multiple addresses", "test@test.com,test2#example.com", nil, true, }, } for _, tt := range tests { t.Run(tt.n, func(t *testing.T) { m := NewMsg() if err := m.CcFromString(tt.v); err != nil && !tt.sf { t.Errorf("Msg.CcFromString failed: %s", err) return } mto := m.GetCc() if len(mto) != len(tt.w) { t.Errorf("Msg.CcFromString failed, expected len: %d, got: %d", len(tt.w), len(mto)) return } for i := range mto { w := tt.w[i] g := mto[i] if w.String() != g.String() { t.Errorf("Msg.CcFromString failed, expected address: %s, got: %s", w.String(), g.String()) } } }) } } // TestMsg_BccFromString tests Msg.BccFromString in different scenarios func TestMsg_BccFromString(t *testing.T) { tests := []struct { n string v string w []*mail.Address sf bool }{ {"valid single address", "test@test.com", []*mail.Address{ {Name: "", Address: "test@test.com"}, }, false}, { "valid multiple addresses", "test@test.com,test2@example.com", []*mail.Address{ {Name: "", Address: "test@test.com"}, {Name: "", Address: "test2@example.com"}, }, false, }, { "valid multiple addresses with space and name", `test@test.com, "Toni Tester" `, []*mail.Address{ {Name: "", Address: "test@test.com"}, {Name: "Toni Tester", Address: "test2@example.com"}, }, false, }, { "invalid and valid multiple addresses", "test@test.com,test2#example.com", nil, true, }, } for _, tt := range tests { t.Run(tt.n, func(t *testing.T) { m := NewMsg() if err := m.BccFromString(tt.v); err != nil && !tt.sf { t.Errorf("Msg.BccFromString failed: %s", err) return } mto := m.GetBcc() if len(mto) != len(tt.w) { t.Errorf("Msg.BccFromString failed, expected len: %d, got: %d", len(tt.w), len(mto)) return } for i := range mto { w := tt.w[i] g := mto[i] if w.String() != g.String() { t.Errorf("Msg.BccFromString failed, expected address: %s, got: %s", w.String(), g.String()) } } }) } } // TestMsg_checkUserAgent tests the checkUserAgent method of the Msg func TestMsg_checkUserAgent(t *testing.T) { tests := []struct { name string noDefaultUserAgent bool genHeader map[Header][]string wantUserAgent string sf bool }{ { name: "check default user agent", noDefaultUserAgent: false, wantUserAgent: fmt.Sprintf("go-mail v%s // https://github.com/wneessen/go-mail", VERSION), sf: false, }, { name: "check no default user agent", noDefaultUserAgent: true, wantUserAgent: "", sf: true, }, { name: "check if ua and xm is already set", noDefaultUserAgent: false, genHeader: map[Header][]string{ HeaderUserAgent: {"custom UA"}, HeaderXMailer: {"custom XM"}, }, wantUserAgent: "custom UA", sf: false, }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { msg := &Msg{ noDefaultUserAgent: tt.noDefaultUserAgent, genHeader: tt.genHeader, } msg.checkUserAgent() gotUserAgent := "" if val, ok := msg.genHeader[HeaderUserAgent]; ok { gotUserAgent = val[0] // Assuming the first one is the needed value } if gotUserAgent != tt.wantUserAgent && !tt.sf { t.Errorf("UserAgent got = %v, want = %v", gotUserAgent, tt.wantUserAgent) } }) } } // TestNewMsgWithMIMEVersion tests WithMIMEVersion and Msg.SetMIMEVersion func TestNewMsgWithNoDefaultUserAgent(t *testing.T) { m := NewMsg(WithNoDefaultUserAgent()) if m.noDefaultUserAgent != true { t.Errorf("WithNoDefaultUserAgent() failed. Expected: %t, got: %t", true, false) } } // TestSignWithSMime_ValidRSAKeyPair tests WithSMimeSinging with given rsa key pair func TestSignWithSMime_ValidRSAKeyPair(t *testing.T) { privateKey, certificate, intermediateCertificate, err := getDummyRSACryptoMaterial() if err != nil { t.Errorf("failed to laod dummy crypto material. Cause: %v", err) } m := NewMsg() if err := m.SignWithSMimeRSA(privateKey, certificate, intermediateCertificate); err != nil { t.Errorf("failed to set sMime. Cause: %v", err) } if m.sMime.privateKey.rsa == nil { t.Errorf("WithSMimeSinging() - no private key is given") } if m.sMime.certificate == nil { t.Errorf("WithSMimeSinging() - no certificate is given") } } // TestSignWithSMime_ValidRSAKeyPair tests WithSMimeSinging with given ecdsa key pair func TestSignWithSMime_ValidECDSAKeyPair(t *testing.T) { privateKey, certificate, intermediateCertificate, err := getDummyECDSACryptoMaterial() if err != nil { t.Errorf("failed to laod dummy crypto material. Cause: %v", err) } m := NewMsg() if err := m.SignWithSMimeECDSA(privateKey, certificate, intermediateCertificate); err != nil { t.Errorf("failed to set sMime. Cause: %v", err) } if m.sMime.privateKey.rsa == nil { t.Errorf("WithSMimeSinging() - no private key is given") } if m.sMime.certificate == nil { t.Errorf("WithSMimeSinging() - no certificate is given") } } // TestSignWithSMime_InvalidPrivateKey tests WithSMimeSinging with given invalid private key func TestSignWithSMime_InvalidPrivateKey(t *testing.T) { m := NewMsg() err := m.SignWithSMimeRSA(nil, nil, nil) if !errors.Is(err, ErrInvalidPrivateKey) { t.Errorf("failed to pre-check SignWithSMime method values correctly: %s", err) } } // TestSignWithSMime_InvalidCertificate tests WithSMimeSinging with given invalid certificate func TestSignWithSMime_InvalidCertificate(t *testing.T) { privateKey, _, _, err := getDummyRSACryptoMaterial() if err != nil { t.Errorf("failed to laod dummy crypto material. Cause: %v", err) } m := NewMsg() err = m.SignWithSMimeRSA(privateKey, nil, nil) if !errors.Is(err, ErrInvalidCertificate) { t.Errorf("failed to pre-check SignWithSMime method values correctly: %s", err) } } // Fuzzing tests func FuzzMsg_Subject(f *testing.F) { f.Add("Testsubject") f.Add("") f.Add("This is a longer test subject.") f.Add("Let's add some umlauts: üäöß") f.Add("Or even emojis: ☝️💪👍") f.Fuzz(func(t *testing.T, data string) { m := NewMsg() m.Subject(data) m.Reset() }) } func FuzzMsg_From(f *testing.F) { f.Add("Toni Tester ") f.Add("") f.Add("mail@server.com") f.Fuzz(func(t *testing.T, data string) { m := NewMsg() if err := m.From(data); err != nil && !strings.Contains(err.Error(), "failed to parse mail address") { t.Errorf("failed set set FROM address: %s", err) } m.Reset() }) } // TestMsg_createSignaturePart tests the Msg.createSignaturePart method func TestMsg_createSignaturePart(t *testing.T) { privateKey, certificate, intermediateCertificate, err := getDummyRSACryptoMaterial() if err != nil { t.Errorf("failed to laod dummy crypto material. Cause: %v", err) } m := NewMsg() if err := m.SignWithSMimeRSA(privateKey, certificate, intermediateCertificate); err != nil { t.Errorf("failed to init smime configuration") } body := []byte("This is the body") part, err := m.createSignaturePart(EncodingQP, TypeTextPlain, CharsetUTF7, body) if err != nil { t.Errorf("createSignaturePart() method failed.") } if part.GetEncoding() != EncodingB64 { t.Errorf("createSignaturePart() method failed. Expected encoding: %s, got: %s", EncodingB64, part.GetEncoding()) } if part.GetContentType() != typeSMimeSigned { t.Errorf("createSignaturePart() method failed. Expected content type: %s, got: %s", typeSMimeSigned, part.GetContentType()) } if part.GetCharset() != CharsetUTF8 { t.Errorf("createSignaturePart() method failed. Expected charset: %s, got: %s", CharsetUTF8, part.GetCharset()) } if content, err := part.GetContent(); err != nil || len(content) == len(body) { t.Errorf("createSignaturePart() method failed. Expected content should not be equal: %s, got: %s", body, part.GetEncoding()) } } // TestMsg_signMessage tests the Msg.signMessage method func TestMsg_signMessage(t *testing.T) { privateKey, certificate, intermediateCertificate, err := getDummyRSACryptoMaterial() if err != nil { t.Errorf("failed to laod dummy crypto material. Cause: %v", err) } body := []byte("This is the body") m := NewMsg() m.SetBodyString(TypeTextPlain, string(body)) if err := m.SignWithSMimeRSA(privateKey, certificate, intermediateCertificate); err != nil { t.Errorf("failed to init smime configuration") } msg, err := m.signMessage(m) if err != nil { t.Errorf("createSignaturePart() method failed.") } parts := msg.GetParts() if len(parts) != 2 { t.Errorf("createSignaturePart() method failed. Expected 2 parts, got: %d", len(parts)) } signedPart := parts[0] if signedPart.GetEncoding() != EncodingQP { t.Errorf("createSignaturePart() method failed. Expected encoding: %s, got: %s", EncodingB64, signedPart.GetEncoding()) } if signedPart.GetContentType() != TypeTextPlain { t.Errorf("createSignaturePart() method failed. Expected content type: %s, got: %s", typeSMimeSigned, signedPart.GetContentType()) } if signedPart.GetCharset() != CharsetUTF8 { t.Errorf("createSignaturePart() method failed. Expected charset: %s, got: %s", CharsetUTF8, signedPart.GetCharset()) } if content, err := signedPart.GetContent(); err != nil || len(content) != len(body) { t.Errorf("createSignaturePart() method failed. Expected content should be equal: %s, got: %s", body, content) } signaturePart := parts[1] if signaturePart.GetEncoding() != EncodingB64 { t.Errorf("createSignaturePart() method failed. Expected encoding: %s, got: %s", EncodingB64, signaturePart.GetEncoding()) } if signaturePart.GetContentType() != typeSMimeSigned { t.Errorf("createSignaturePart() method failed. Expected content type: %s, got: %s", typeSMimeSigned, signaturePart.GetContentType()) } if signaturePart.GetCharset() != CharsetUTF8 { t.Errorf("createSignaturePart() method failed. Expected charset: %s, got: %s", CharsetUTF8, signaturePart.GetCharset()) } if content, err := signaturePart.GetContent(); err != nil || len(content) == len(body) { t.Errorf("createSignaturePart() method failed. Expected content should not be equal: %s, got: %s", body, signaturePart.GetEncoding()) } }