From b94a42ebe8c4e34e39f1a478297944488873427b Mon Sep 17 00:00:00 2001
From: Dhia Gharsallaoui <safedhia@gmail.com>
Date: Wed, 12 Oct 2022 13:36:52 +0200
Subject: [PATCH] Expose Msg attachments fields

---
 msg.go      | 10 ++++++
 msg_test.go | 94 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 104 insertions(+)

diff --git a/msg.go b/msg.go
index db019e6..521cd30 100644
--- a/msg.go
+++ b/msg.go
@@ -488,6 +488,16 @@ func (m *Msg) GetParts() []*Part {
 	return m.parts
 }
 
+// GetAttachments returns the attachments of the Msg
+func (m *Msg) GetAttachments() []*File {
+	return m.attachments
+}
+
+// SetAttachements sets the attachements of the message.
+func (m *Msg) SetAttachements(ff []*File) {
+	m.attachments = ff
+}
+
 // SetBodyString sets the body of the message.
 func (m *Msg) SetBodyString(ct ContentType, b string, o ...PartOption) {
 	buf := bytes.NewBufferString(b)
diff --git a/msg_test.go b/msg_test.go
index 017c08e..5b24251 100644
--- a/msg_test.go
+++ b/msg_test.go
@@ -13,6 +13,7 @@ import (
 	"io"
 	"net/mail"
 	"os"
+	"sort"
 	"strings"
 	"testing"
 	ttpl "text/template"
@@ -1207,6 +1208,99 @@ func TestMsg_AttachFile(t *testing.T) {
 	}
 }
 
+// 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.SetAttachements(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_AttachFromEmbedFS tests the Msg.AttachFromEmbedFS and the WithFilename FileOption method
 func TestMsg_AttachFromEmbedFS(t *testing.T) {
 	tests := []struct {