wneessen/go-mail
1
1
Fork 0
mirror of https://github.com/wneessen/go-mail.git synced 2024-12-23 02:50:39 +01:00
Code Issues Projects Releases Packages Wiki Activity

Add tests for GetAttachments method in Msg

Browse source 
This commit introduces unit tests for the GetAttachments method in the Msg struct to ensure it correctly handles both single and multiple attachments. The tests verify the name and content of the attachments to validate the expected behavior.
This commit is contained in:
Winni Neessen 2024-10-27 10:07:05 +01:00
parent 43f9ffa3af
commit babf7b9780
Signed by: wneessen
GPG key ID: 385AC9889632126E
1 changed files with 71 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

71
msg_test.go
View file

@ -3207,6 +3207,77 @@ func TestMsg_GetParts(t *testing.T) {
})
}
func TestMsg_GetAttachments(t *testing.T) {
t.Run("GetAttachments with single attachment", func(t *testing.T) {
message := NewMsg()
if message == nil {
t.Fatal("message is nil")
}
message.AttachFile("testdata/attachment.txt")
attachments := message.GetAttachments()
if len(attachments) != 1 {
t.Fatalf("GetAttachments: expected 1 attachment, got: %d", len(attachments))
}
if attachments[0] == nil {
t.Fatalf("GetAttachments: expected attachment, got nil")
}
if attachments[0].Name != "attachment.txt" {
t.Errorf("GetAttachments: expected attachment name to be %s, got: %s", "attachment.txt",
attachments[0].Name)
}
messageBuf := bytes.NewBuffer(nil)
_, err := attachments[0].Writer(messageBuf)
if err != nil {
t.Errorf("GetAttachments: Writer func failed: %s", err)
}
if !strings.EqualFold(messageBuf.String(), "This is a test attachment\n") {
t.Errorf("GetParts: expected message body to be %s, got: %s", "This is a test attachment\n",
messageBuf.String())
}
})
t.Run("GetAttachments with multiple attachments", func(t *testing.T) {
message := NewMsg()
if message == nil {
t.Fatal("message is nil")
}
message.AttachFile("testdata/attachment.txt")
message.AttachFile("testdata/attachment.txt", WithFileName("attachment2.txt"))
attachments := message.GetAttachments()
if len(attachments) != 2 {
t.Fatalf("GetAttachments: expected 2 attachment, got: %d", len(attachments))
}
if attachments[0] == nil || attachments[1] == nil {
t.Fatalf("GetAttachments: expected attachment, got nil")
}
if attachments[0].Name != "attachment.txt" {
t.Errorf("GetAttachments: expected attachment name to be %s, got: %s", "attachment.txt",
attachments[0].Name)
}
if attachments[1].Name != "attachment2.txt" {
t.Errorf("GetAttachments: expected attachment name to be %s, got: %s", "attachment2.txt",
attachments[1].Name)
}
messageBuf := bytes.NewBuffer(nil)
_, err := attachments[0].Writer(messageBuf)
if err != nil {
t.Errorf("GetAttachments: Writer func failed: %s", err)
}
if !strings.EqualFold(messageBuf.String(), "This is a test attachment\n") {
t.Errorf("GetParts: expected message body to be %s, got: %s", "This is a test attachment\n",
messageBuf.String())
}
messageBuf.Reset()
_, err = attachments[1].Writer(messageBuf)
if err != nil {
t.Errorf("GetAttachments: Writer func failed: %s", err)
}
if !strings.EqualFold(messageBuf.String(), "This is a test attachment\n") {
t.Errorf("GetParts: expected message body to be %s, got: %s", "This is a test attachment\n",
messageBuf.String())
}
})
}
// 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,