go-simple-mail/attach_test.go

123 lines
3.3 KiB
Go

package mail
import (
"bytes"
"testing"
)
func checkError(t *testing.T, err error) {
if err != nil {
t.Errorf("got error: %v", err)
}
}
func checkByteSlice(t *testing.T, got, want []byte) {
if !bytes.Equal(got, want) {
t.Errorf("got: %v, want: %v", got, want)
}
}
func TestAttachments(t *testing.T) {
want := []byte("foo")
t.Run("Inline File", func(t *testing.T) {
msg := NewMSG()
msg.Attach(&File{FilePath: "testdata/foo.txt", Name: "foo", Inline: true})
checkError(t, msg.Error)
got := msg.inlines[0].Data
checkByteSlice(t, got, want)
})
t.Run("Inline Base64", func(t *testing.T) {
msg := NewMSG()
msg.Attach(&File{B64Data: "Zm9v", Name: "foo", Inline: true})
checkError(t, msg.Error)
got := msg.inlines[0].Data
checkByteSlice(t, got, want)
})
t.Run("Inline Data", func(t *testing.T) {
msg := NewMSG()
msg.Attach(&File{Data: []byte("foo"), Name: "foo", Inline: true})
checkError(t, msg.Error)
got := msg.inlines[0].Data
checkByteSlice(t, got, want)
})
t.Run("Attachment File", func(t *testing.T) {
msg := NewMSG()
msg.Attach(&File{FilePath: "testdata/foo.txt", Name: "foo"})
checkError(t, msg.Error)
got := msg.attachments[0].Data
checkByteSlice(t, got, want)
})
t.Run("Attachment Base64", func(t *testing.T) {
msg := NewMSG()
msg.Attach(&File{B64Data: "Zm9v", Name: "foo"})
checkError(t, msg.Error)
got := msg.attachments[0].Data
checkByteSlice(t, got, want)
})
t.Run("Attachment Data", func(t *testing.T) {
msg := NewMSG()
msg.Attach(&File{Data: []byte("foo"), Name: "foo"})
checkError(t, msg.Error)
got := msg.attachments[0].Data
checkByteSlice(t, got, want)
})
// DEPRECATED. TODO: Remove before launch v3
t.Run("Inline File Deprecated", func(t *testing.T) {
msg := NewMSG()
msg.AddInline("testdata/foo.txt", "foo")
checkError(t, msg.Error)
got := msg.inlines[0].Data
checkByteSlice(t, got, want)
})
t.Run("Inline Base64 Deprecated", func(t *testing.T) {
msg := NewMSG()
msg.AddInlineBase64("Zm9v", "foo", "")
checkError(t, msg.Error)
got := msg.inlines[0].Data
checkByteSlice(t, got, want)
})
t.Run("Inline Data Deprecated", func(t *testing.T) {
msg := NewMSG()
msg.AddInlineData([]byte("foo"), "foo", "")
checkError(t, msg.Error)
got := msg.inlines[0].Data
checkByteSlice(t, got, want)
})
t.Run("Attachment File Deprecated", func(t *testing.T) {
msg := NewMSG()
msg.AddAttachment("testdata/foo.txt", "foo")
checkError(t, msg.Error)
got := msg.attachments[0].Data
checkByteSlice(t, got, want)
})
t.Run("Attachment Base64 Deprecated", func(t *testing.T) {
msg := NewMSG()
msg.AddAttachmentBase64("Zm9v", "foo")
checkError(t, msg.Error)
got := msg.attachments[0].Data
checkByteSlice(t, got, want)
})
t.Run("Attachment Data Deprecated", func(t *testing.T) {
msg := NewMSG()
msg.AddAttachmentData([]byte("foo"), "foo", "")
checkError(t, msg.Error)
got := msg.attachments[0].Data
checkByteSlice(t, got, want)
})
t.Run("Inline File not name Deprecated", func(t *testing.T) {
msg := NewMSG()
msg.AddInline("testdata/foo.txt")
checkError(t, msg.Error)
got := msg.inlines[0].Data
checkByteSlice(t, got, want)
})
t.Run("Attachment File not name Deprecated", func(t *testing.T) {
msg := NewMSG()
msg.AddAttachment("testdata/foo.txt")
checkError(t, msg.Error)
got := msg.attachments[0].Data
checkByteSlice(t, got, want)
})
}