mirror of
https://github.com/wneessen/go-mail.git
synced 2024-12-23 02:50:39 +01:00
Add tests for GetGenHeader and GetParts methods in Msg
Introduce comprehensive tests for the GetGenHeader and GetParts methods. These tests cover single and multiple values for headers, body parts retrieval, and edge cases such as nil values, ensuring robustness and correctness.
This commit is contained in:
parent
e808e0b972
commit
43f9ffa3af
GPG key ID:
385AC9889632126E
1 changed files with 139 additions and 0 deletions
139
msg_test.go
139
msg_test.go
|
|
@ -5,6 +5,7 @@
|
|||
package mail
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
|
|
@ -3068,6 +3069,144 @@ func TestMsg_GetBccString(t *testing.T) {
|
|||
})
|
||||
}
|
||||
|
||||
func TestMsg_GetGenHeader(t *testing.T) {
|
||||
t.Run("GetGenHeader with single value", func(t *testing.T) {
|
||||
message := NewMsg()
|
||||
if message == nil {
|
||||
t.Fatal("message is nil")
|
||||
}
|
||||
for _, tt := range genHeaderTests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
message.SetGenHeader(tt.header, "test")
|
||||
values := message.GetGenHeader(tt.header)
|
||||
if len(values) != 1 {
|
||||
t.Errorf("GetGenHeader: expected 1 value, got: %d", len(values))
|
||||
}
|
||||
if values[0] != "test" {
|
||||
t.Errorf("GetGenHeader: expected value not returned. Want: %s, got: %s",
|
||||
"test", values[0])
|
||||
}
|
||||
})
|
||||
}
|
||||
})
|
||||
t.Run("GetGenHeader with multiple values", func(t *testing.T) {
|
||||
message := NewMsg()
|
||||
if message == nil {
|
||||
t.Fatal("message is nil")
|
||||
}
|
||||
for _, tt := range genHeaderTests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
message.SetGenHeader(tt.header, "test", "foobar")
|
||||
values := message.GetGenHeader(tt.header)
|
||||
if len(values) != 2 {
|
||||
t.Errorf("GetGenHeader: expected 1 value, got: %d", len(values))
|
||||
}
|
||||
if values[0] != "test" {
|
||||
t.Errorf("GetGenHeader: expected value not returned. Want: %s, got: %s",
|
||||
"test", values[0])
|
||||
}
|
||||
if values[1] != "foobar" {
|
||||
t.Errorf("GetGenHeader: expected value not returned. Want: %s, got: %s",
|
||||
"foobar", values[1])
|
||||
}
|
||||
})
|
||||
}
|
||||
})
|
||||
t.Run("GetGenHeader with nil", func(t *testing.T) {
|
||||
message := NewMsg()
|
||||
if message == nil {
|
||||
t.Fatal("message is nil")
|
||||
}
|
||||
for _, tt := range genHeaderTests {
|
||||
t.Run(tt.name, func(t *testing.T) {
|
||||
message.SetGenHeader(tt.header)
|
||||
values := message.GetGenHeader(tt.header)
|
||||
if len(values) != 0 {
|
||||
t.Errorf("GetGenHeader: expected 1 value, got: %d", len(values))
|
||||
}
|
||||
})
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func TestMsg_GetParts(t *testing.T) {
|
||||
t.Run("GetParts with single part", func(t *testing.T) {
|
||||
message := NewMsg()
|
||||
if message == nil {
|
||||
t.Fatal("message is nil")
|
||||
}
|
||||
message.SetBodyString(TypeTextPlain, "this is a test body")
|
||||
parts := message.GetParts()
|
||||
if len(parts) != 1 {
|
||||
t.Fatalf("GetParts: expected 1 part, got: %d", len(parts))
|
||||
}
|
||||
if parts[0] == nil {
|
||||
t.Fatalf("GetParts: expected part, got nil")
|
||||
}
|
||||
if parts[0].contentType != TypeTextPlain {
|
||||
t.Errorf("GetParts: expected contentType to be TypeTextPlain, got: %s", parts[0].contentType)
|
||||
}
|
||||
messageBuf := bytes.NewBuffer(nil)
|
||||
_, err := parts[0].writeFunc(messageBuf)
|
||||
if err != nil {
|
||||
t.Errorf("GetParts: writeFunc failed: %s", err)
|
||||
}
|
||||
if !strings.EqualFold(messageBuf.String(), "this is a test body") {
|
||||
t.Errorf("GetParts: expected message body to be %s, got: %s", "this is a test body",
|
||||
messageBuf.String())
|
||||
}
|
||||
})
|
||||
t.Run("GetParts with multiple parts", func(t *testing.T) {
|
||||
message := NewMsg()
|
||||
if message == nil {
|
||||
t.Fatal("message is nil")
|
||||
}
|
||||
message.SetBodyString(TypeTextPlain, "this is a test body")
|
||||
message.AddAlternativeString(TypeTextHTML, "<p>This is HTML</p>")
|
||||
parts := message.GetParts()
|
||||
if len(parts) != 2 {
|
||||
t.Fatalf("GetParts: expected 2 parts, got: %d", len(parts))
|
||||
}
|
||||
if parts[0] == nil || parts[1] == nil {
|
||||
t.Fatalf("GetParts: expected parts, got nil")
|
||||
}
|
||||
if parts[0].contentType != TypeTextPlain {
|
||||
t.Errorf("GetParts: expected contentType to be TypeTextPlain, got: %s", parts[0].contentType)
|
||||
}
|
||||
if parts[1].contentType != TypeTextHTML {
|
||||
t.Errorf("GetParts: expected contentType to be TypeTextHTML, got: %s", parts[1].contentType)
|
||||
}
|
||||
messageBuf := bytes.NewBuffer(nil)
|
||||
_, err := parts[0].writeFunc(messageBuf)
|
||||
if err != nil {
|
||||
t.Errorf("GetParts: writeFunc failed: %s", err)
|
||||
}
|
||||
if !strings.EqualFold(messageBuf.String(), "this is a test body") {
|
||||
t.Errorf("GetParts: expected message body to be %s, got: %s", "this is a test body",
|
||||
messageBuf.String())
|
||||
}
|
||||
messageBuf.Reset()
|
||||
_, err = parts[1].writeFunc(messageBuf)
|
||||
if err != nil {
|
||||
t.Errorf("GetParts: writeFunc failed: %s", err)
|
||||
}
|
||||
if !strings.EqualFold(messageBuf.String(), "<p>This is HTML</p>") {
|
||||
t.Errorf("GetParts: expected message body to be %s, got: %s", "<p>This is HTML</p>",
|
||||
messageBuf.String())
|
||||
}
|
||||
})
|
||||
t.Run("GetParts with no parts", func(t *testing.T) {
|
||||
message := NewMsg()
|
||||
if message == nil {
|
||||
t.Fatal("message is nil")
|
||||
}
|
||||
parts := message.GetParts()
|
||||
if len(parts) != 0 {
|
||||
t.Fatalf("GetParts: expected no parts, got: %d", len(parts))
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
// checkAddrHeader verifies the correctness of an AddrHeader in a Msg based on the provided criteria.
|
||||
// It checks whether the AddrHeader contains the correct address, name, and number of fields.
|
||||
func checkAddrHeader(t *testing.T, message *Msg, header AddrHeader, fn string, field, wantFields int,
|
||||
|
|
|
|||
Loading…
Reference in a new issue