ric/MailHog
1
0
Fork 0
mirror of https://gitlab.com/ric_harvey/MailHog.git synced 2024-11-23 22:34:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
MailHog/mime_test.go

160 lines
6.4 KiB
Go
Raw Normal View History

MIME detection and add some error responses
2014-04-21 11:14:47 +00:00
package main
import (
Add in-memory storage, clean up config
2014-04-23 23:22:50 +00:00
"github.com/ian-kent/MailHog/mailhog/config"
MIME detection and add some error responses
2014-04-21 11:14:47 +00:00
"github.com/ian-kent/MailHog/mailhog/storage"
"github.com/stretchr/testify/assert"
"net"
"regexp"
"strings"
"testing"
)
// FIXME requires a running instance of MailHog
func TestBasicMIMEHappyPath(t *testing.T) {
buf := make([]byte, 1024)
// Open a connection
conn, err := net.Dial("tcp", "127.0.0.1:1025")
assert.Nil(t, err)
// Read the greeting
n, err := conn.Read(buf)
assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "220 mailhog.example ESMTP Go-MailHog\n")
// Send EHLO
_, err = conn.Write([]byte("EHLO localhost\r\n"))
assert.Nil(t, err)
// Read the response
n, err = conn.Read(buf)
assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "250-Hello localhost\n")
n, err = conn.Read(buf)
assert.Nil(t, err)
Add basic AUTH support (RFC4954)
2014-04-22 20:19:48 +00:00
assert.Equal(t, string(buf[0:n]), "250-PIPELINING\n")
n, err = conn.Read(buf)
assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "250 AUTH EXTERNAL CRAM-MD5 LOGIN PLAIN\n")
MIME detection and add some error responses
2014-04-21 11:14:47 +00:00
// Send MAIL
_, err = conn.Write([]byte("MAIL From:<nobody@mailhog.example>\r\n"))
assert.Nil(t, err)
// Read the response
n, err = conn.Read(buf)
assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "250 Sender nobody@mailhog.example ok\n")
// Send RCPT
_, err = conn.Write([]byte("RCPT To:<someone@mailhog.example>\r\n"))
assert.Nil(t, err)
// Read the response
n, err = conn.Read(buf)
assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "250 Recipient someone@mailhog.example ok\n")
// Send DATA
_, err = conn.Write([]byte("DATA\r\n"))
assert.Nil(t, err)
// Read the response
n, err = conn.Read(buf)
assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "354 End data with <CR><LF>.<CR><LF>\n")
// Send the message
content := "Content-Type: multipart/alternative; boundary=\"--mailhog-test-boundary\"\r\n"
content += "Content-Length: 220\r\n"
content += "From: Nobody <nobody@mailhog.example>\r\n"
content += "To: Someone <someone@mailhog.example>\r\n"
content += "Subject: Example message\r\n"
content += "MIME-Version: 1.0\r\n"
content += "\r\n"
mime support and html email rendering
2014-04-21 21:32:34 +00:00
content += "----mailhog-test-boundary\r\n"
MIME detection and add some error responses
2014-04-21 11:14:47 +00:00
content += "Content-Type: text/plain\r\n"
content += "\r\n"
content += "Hi there :)\r\n"
mime support and html email rendering
2014-04-21 21:32:34 +00:00
content += "----mailhog-test-boundary\r\n"
MIME detection and add some error responses
2014-04-21 11:14:47 +00:00
content += "Content-Type: text/html\r\n"
content += "\r\n"
content += "<html>\r\n"
content += " <head>\r\n"
content += " <title>Example message</title>\r\n"
content += " </head>\r\n"
content += " <body>\r\n"
content += " <p style=\"font-weight: bold; color: #ff0000; text-decoration: underline\">\r\n"
content += " Hi there :)\r\n"
content += " </p>\r\n"
content += " </body>\r\n"
content += "</html>\r\n"
mime support and html email rendering
2014-04-21 21:32:34 +00:00
content += "----mailhog-test-boundary\r\n"
MIME detection and add some error responses
2014-04-21 11:14:47 +00:00
content += ".\r\n"
_, err = conn.Write([]byte(content))
assert.Nil(t, err)
// Read the response
n, err = conn.Read(buf)
assert.Nil(t, err)
r, _ := regexp.Compile("250 Ok: queued as ([0-9a-f]+)\n")
match := r.FindStringSubmatch(string(buf[0:n]))
assert.NotNil(t, match)
// Send QUIT
_, err = conn.Write([]byte("QUIT\r\n"))
assert.Nil(t, err)
// Read the response
n, err = conn.Read(buf)
assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "221 Bye\n")
Add in-memory storage, clean up config
2014-04-23 23:22:50 +00:00
s := storage.CreateMongoDB(config.DefaultConfig())
MIME detection and add some error responses
2014-04-21 11:14:47 +00:00
message, err := s.Load(match[1])
assert.Nil(t, err)
assert.NotNil(t, message)
assert.Equal(t, message.From.Domain, "mailhog.example", "sender domain is mailhog.example")
assert.Equal(t, message.From.Mailbox, "nobody", "sender mailbox is nobody")
assert.Equal(t, message.From.Params, "", "sender params is empty")
assert.Equal(t, len(message.From.Relays), 0, "sender has no relays")
assert.Equal(t, len(message.To), 1, "message has 1 recipient")
assert.Equal(t, message.To[0].Domain, "mailhog.example", "recipient domain is mailhog.example")
assert.Equal(t, message.To[0].Mailbox, "someone", "recipient mailbox is someone")
assert.Equal(t, message.To[0].Params, "", "recipient params is empty")
assert.Equal(t, len(message.To[0].Relays), 0, "recipient has no relays")
assert.Equal(t, len(message.Content.Headers), 9, "message has 7 headers")
assert.Equal(t, message.Content.Headers["Content-Type"], []string{"multipart/alternative; boundary=\"--mailhog-test-boundary\""}, "Content-Type header is multipart/alternative; boundary=\"--mailhog-test-boundary\"")
assert.Equal(t, message.Content.Headers["Subject"], []string{"Example message"}, "Subject header is Example message")
assert.Equal(t, message.Content.Headers["Content-Length"], []string{"220"}, "Content-Length is 220")
assert.Equal(t, message.Content.Headers["To"], []string{"Someone <someone@mailhog.example>"}, "To is Someone <someone@mailhog.example>")
assert.Equal(t, message.Content.Headers["From"], []string{"Nobody <nobody@mailhog.example>"}, "From is Nobody <nobody@mailhog.example>")
assert.True(t, strings.HasPrefix(message.Content.Headers["Received"][0], "from localhost by mailhog.example (Go-MailHog)\r\n id "+match[1]+"@mailhog.example; "), "Received header is correct")
assert.Equal(t, message.Content.Headers["Return-Path"], []string{"<nobody@mailhog.example>"}, "Return-Path is <nobody@mailhog.example>")
assert.Equal(t, message.Content.Headers["Message-ID"], []string{match[1] + "@mailhog.example"}, "Message-ID is "+match[1]+"@mailhog.example")
mime support and html email rendering
2014-04-21 21:32:34 +00:00
expected := "----mailhog-test-boundary\r\nContent-Type: text/plain\r\n\r\nHi there :)\r\n----mailhog-test-boundary\r\nContent-Type: text/html\r\n\r\n<html>\r\n <head>\r\n <title>Example message</title>\r\n </head>\r\n <body>\r\n <p style=\"font-weight: bold; color: #ff0000; text-decoration: underline\">\r\n Hi there :)\r\n </p>\r\n </body>\r\n</html>\r\n----mailhog-test-boundary"
MIME detection and add some error responses
2014-04-21 11:14:47 +00:00
assert.Equal(t, message.Content.Body, expected, "message has correct body")
mime support and html email rendering
2014-04-21 21:32:34 +00:00
assert.NotNil(t, message.MIME)
assert.Equal(t, len(message.MIME.Parts), 2)
plain := message.MIME.Parts[0]
assert.NotNil(t, plain)
assert.Equal(t, plain.Headers["Content-Type"], []string{"text/plain"}, "Content-Type header is text/plain")
assert.Equal(t, plain.Body, "Hi there :)", "plain text body is correct")
html := message.MIME.Parts[1]
assert.NotNil(t, html)
assert.Equal(t, html.Headers["Content-Type"], []string{"text/html"}, "Content-Type header is text/html")
expected = "<html>\r\n <head>\r\n <title>Example message</title>\r\n </head>\r\n <body>\r\n <p style=\"font-weight: bold; color: #ff0000; text-decoration: underline\">\r\n Hi there :)\r\n </p>\r\n </body>\r\n</html>"
assert.Equal(t, html.Body, expected, "html body is correct")
MIME detection and add some error responses
2014-04-21 11:14:47 +00:00
}
Reference in a new issue Copy permalink