mirror of
https://gitlab.com/ric_harvey/MailHog.git
synced 2024-11-23 22:34:04 +00:00
115 lines
4 KiB
Go
115 lines
4 KiB
Go
package main
|
|
|
|
import (
|
|
"testing"
|
|
"github.com/stretchr/testify/assert"
|
|
"net"
|
|
"strings"
|
|
"regexp"
|
|
"github.com/ian-kent/MailHog/mailhog/storage"
|
|
)
|
|
|
|
// FIXME requires a running instance of MailHog
|
|
|
|
func TestBasicHappyPath(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")
|
|
|
|
// 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: text/plain\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 += "\r\n"
|
|
content += "Hi there :)\r\n"
|
|
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")
|
|
|
|
message, err := storage.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), 7, "message has 7 headers")
|
|
assert.Equal(t, message.Content.Headers["Content-Type"], []string{"text/plain"}, "Content-Type header is text/plain")
|
|
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")
|
|
|
|
assert.Equal(t, message.Content.Body, "Hi there :)", "message has correct body")
|
|
}
|