package mail

import (
	"bytes"
	"fmt"
	"io"
	"net/mail"
	"testing"
	"time"
)

// 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)
			}
		})
	}
}

// TestMsg_SetHEader tests Msg.SetHeader
func TestMsg_SetHeader(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.SetHeader(tt.header, tt.values...)
			if m.genHeader[tt.header] == nil {
				t.Errorf("SetHeader() 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("SetHeader() failed. Value %s not found in header field", v)
				}
			}
		})
	}
}

// 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_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" <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.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" <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.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" <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.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 %q, got: %q", "bulk", m.genHeader[HeaderPrecedence][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
	}
	m.genHeader[HeaderMessageID] = nil
	v := "This.is.a.message.id"
	vf := "<This.is.a.message.id>"
	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_FromFormat tests the FromFormat() method 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" <tester@example.com>`, false},
		{"no name with valid addr", "", "tester@example.com",
			`<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
			}

			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
	}

	var 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" <tester@example.com>`, false},
		{"no name with valid addr", "", "tester@example.com",
			`<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_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, "<strong>üäöß</strong>",
			"<strong>üäöß</strong>", 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.w(&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", "<strong>üäöß</strong>", "<strong>üäöß</strong>", 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.w(&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_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) error {
		return 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")
	}
}