ric/MailHog
1
0
Fork 0
mirror of https://gitlab.com/ric_harvey/MailHog.git synced 2025-03-15 13:04:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

go fmt

Browse source
This commit is contained in:
Ian Kent 2014-04-20 17:12:32 +01:00
parent ecf4d071c2
commit 9868f1f7f6
2 changed files with 26 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
main.go
View file

@ -2,12 +2,12 @@ package main
import ( import (
"flag" "flag"
"log"
"net"
"os"
"github.com/ian-kent/MailHog/mailhog" "github.com/ian-kent/MailHog/mailhog"
"github.com/ian-kent/MailHog/mailhog/http" "github.com/ian-kent/MailHog/mailhog/http"
"github.com/ian-kent/MailHog/mailhog/smtp" "github.com/ian-kent/MailHog/mailhog/smtp"
"log"
"net"
"os"
) )
var conf *mailhog.Config var conf *mailhog.Config
@ -28,10 +28,10 @@ func config() {
conf = &mailhog.Config{ conf = &mailhog.Config{
SMTPBindAddr: smtpbindaddr, SMTPBindAddr: smtpbindaddr,
HTTPBindAddr: httpbindaddr, HTTPBindAddr: httpbindaddr,
Hostname: hostname, Hostname: hostname,
MongoUri: mongouri, MongoUri: mongouri,
MongoDb: mongodb, MongoDb: mongodb,
MongoColl: mongocoll, MongoColl: mongocoll,
} }
} }
@ -40,13 +40,13 @@ func main() {
exitCh = make(chan int) exitCh = make(chan int)
go web_listen() go web_listen()
go smtp_listen() go smtp_listen()
for { for {
select { select {
case <-exitCh: case <-exitCh:
log.Printf("Received exit signal") log.Printf("Received exit signal")
os.Exit(0) os.Exit(0)
} }
} }
} }
@ -56,7 +56,7 @@ func web_listen() {
http.Start(exitCh, conf) http.Start(exitCh, conf)
} }
func smtp_listen() (*net.TCPListener) { func smtp_listen() *net.TCPListener {
log.Printf("[SMTP] Binding to address: %s\n", conf.SMTPBindAddr) log.Printf("[SMTP] Binding to address: %s\n", conf.SMTPBindAddr)
ln, err := net.Listen("tcp", conf.SMTPBindAddr) ln, err := net.Listen("tcp", conf.SMTPBindAddr)
if err != nil { if err != nil {

28
main_test.go
View file

@ -1,13 +1,13 @@
package main package main
import ( import (
"testing"
"github.com/stretchr/testify/assert"
"net"
"strings"
"regexp"
"github.com/ian-kent/MailHog/mailhog" "github.com/ian-kent/MailHog/mailhog"
"github.com/ian-kent/MailHog/mailhog/storage" "github.com/ian-kent/MailHog/mailhog/storage"
"github.com/stretchr/testify/assert"
"net"
"regexp"
"strings"
"testing"
) )
// FIXME requires a running instance of MailHog // FIXME requires a running instance of MailHog
@ -21,7 +21,7 @@ func TestBasicHappyPath(t *testing.T) {
// Read the greeting // Read the greeting
n, err := conn.Read(buf) n, err := conn.Read(buf)
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "220 mailhog.example ESMTP Go-MailHog\n") assert.Equal(t, string(buf[0:n]), "220 mailhog.example ESMTP Go-MailHog\n")
// Send EHLO // Send EHLO
@ -30,7 +30,7 @@ func TestBasicHappyPath(t *testing.T) {
// Read the response // Read the response
n, err = conn.Read(buf) n, err = conn.Read(buf)
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "250 Hello localhost\n") assert.Equal(t, string(buf[0:n]), "250 Hello localhost\n")
// Send MAIL // Send MAIL
@ -39,7 +39,7 @@ func TestBasicHappyPath(t *testing.T) {
// Read the response // Read the response
n, err = conn.Read(buf) n, err = conn.Read(buf)
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "250 Sender nobody@mailhog.example ok\n") assert.Equal(t, string(buf[0:n]), "250 Sender nobody@mailhog.example ok\n")
// Send RCPT // Send RCPT
@ -48,7 +48,7 @@ func TestBasicHappyPath(t *testing.T) {
// Read the response // Read the response
n, err = conn.Read(buf) n, err = conn.Read(buf)
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "250 Recipient someone@mailhog.example ok\n") assert.Equal(t, string(buf[0:n]), "250 Recipient someone@mailhog.example ok\n")
// Send DATA // Send DATA
@ -57,7 +57,7 @@ func TestBasicHappyPath(t *testing.T) {
// Read the response // Read the response
n, err = conn.Read(buf) n, err = conn.Read(buf)
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "354 End data with <CR><LF>.<CR><LF>\n") assert.Equal(t, string(buf[0:n]), "354 End data with <CR><LF>.<CR><LF>\n")
// Send the message // Send the message
@ -73,7 +73,7 @@ func TestBasicHappyPath(t *testing.T) {
// Read the response // Read the response
n, err = conn.Read(buf) n, err = conn.Read(buf)
assert.Nil(t, err) assert.Nil(t, err)
r, _ := regexp.Compile("250 Ok: queued as ([0-9a-f]+)\n") r, _ := regexp.Compile("250 Ok: queued as ([0-9a-f]+)\n")
match := r.FindStringSubmatch(string(buf[0:n])) match := r.FindStringSubmatch(string(buf[0:n]))
assert.NotNil(t, match) assert.NotNil(t, match)
@ -84,7 +84,7 @@ func TestBasicHappyPath(t *testing.T) {
// Read the response // Read the response
n, err = conn.Read(buf) n, err = conn.Read(buf)
assert.Nil(t, err) assert.Nil(t, err)
assert.Equal(t, string(buf[0:n]), "221 Bye\n") assert.Equal(t, string(buf[0:n]), "221 Bye\n")
message, err := storage.Load(mailhog.DefaultConfig(), match[1]) message, err := storage.Load(mailhog.DefaultConfig(), match[1])
@ -108,9 +108,9 @@ func TestBasicHappyPath(t *testing.T) {
assert.Equal(t, message.Content.Headers["Content-Length"], []string{"220"}, "Content-Length is 220") 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["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.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.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["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.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") assert.Equal(t, message.Content.Body, "Hi there :)", "message has correct body")
} }