mirror of
https://github.com/jhillyerd/inbucket.git
synced 2025-12-17 17:47:03 +00:00
74 lines
1.7 KiB
Go
74 lines
1.7 KiB
Go
package inbucket
|
|
|
|
import (
|
|
"bufio"
|
|
"fmt"
|
|
"github.com/stretchrcom/testify/assert"
|
|
"net/mail"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestIdentifyNonMime(t *testing.T) {
|
|
msg := readMessage("non-mime.raw")
|
|
assert.False(t, IsMIMEMessage(msg), "Failed to identify non-MIME message")
|
|
}
|
|
|
|
func TestIdentifyMime(t *testing.T) {
|
|
msg := readMessage("html-mime-inline.raw")
|
|
assert.True(t, IsMIMEMessage(msg), "Failed to identify MIME message")
|
|
}
|
|
|
|
func TestParseNonMime(t *testing.T) {
|
|
msg := readMessage("non-mime.raw")
|
|
|
|
mime, err := ParseMIMEBody(msg)
|
|
if err != nil {
|
|
t.Fatalf("Failed to parse non-MIME: %v", err)
|
|
}
|
|
|
|
assert.Contains(t, mime.Text, "This is a test mailing")
|
|
}
|
|
|
|
func TestParseInlineText(t *testing.T) {
|
|
msg := readMessage("html-mime-inline.raw")
|
|
|
|
mime, err := ParseMIMEBody(msg)
|
|
if err != nil {
|
|
t.Fatalf("Failed to parse MIME: %v", err)
|
|
}
|
|
|
|
assert.Equal(t, mime.Text, "Test of HTML section")
|
|
}
|
|
|
|
func TestParseInlineHtml(t *testing.T) {
|
|
msg := readMessage("html-mime-inline.raw")
|
|
|
|
mime, err := ParseMIMEBody(msg)
|
|
if err != nil {
|
|
t.Fatalf("Failed to parse MIME: %v", err)
|
|
}
|
|
|
|
assert.Contains(t, mime.Html, "<html>")
|
|
assert.Contains(t, mime.Html, "Test of HTML section")
|
|
}
|
|
|
|
// readMessage is a test utility function to fetch a mail.Message object.
|
|
func readMessage(filename string) *mail.Message {
|
|
// Open test email for parsing
|
|
raw, err := os.Open(filepath.Join("..", "..", "test-data", filename))
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Failed to open test data: %v", err))
|
|
}
|
|
|
|
// Parse email into a mail.Message object like we do
|
|
reader := bufio.NewReader(raw)
|
|
msg, err := mail.ReadMessage(reader)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Failed to read message: %v", err))
|
|
}
|
|
|
|
return msg
|
|
}
|