From 1a4b117f17c5cdb879b81ebbbdb3c976df5ce37b Mon Sep 17 00:00:00 2001
From: Ian Kent <email@iankent.co.uk>
Date: Thu, 10 Dec 2015 17:33:25 +0000
Subject: [PATCH] update to match changes in mailhog/smtp

---
 smtp/session.go | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/smtp/session.go b/smtp/session.go
index c87cd6b..10a220d 100644
--- a/smtp/session.go
+++ b/smtp/session.go
@@ -98,10 +98,11 @@ func (c *Session) validateSender(from string) bool {
 	return true
 }
 
-func (c *Session) acceptMessage(msg *data.Message) (id string, err error) {
-	c.logf("Storing message %s", msg.ID)
-	id, err = c.storage.Store(msg)
-	c.messageChan <- msg
+func (c *Session) acceptMessage(msg *data.SMTPMessage) (id string, err error) {
+	m := msg.Parse(c.proto.Hostname)
+	c.logf("Storing message %s", m.ID)
+	id, err = c.storage.Store(m)
+	c.messageChan <- m
 	return
 }