update to match changes in mailhog/smtp

mailhog · Dec 10, 2015 · 1a4b117 · 1a4b117
1 parent 200ff5a
commit 1a4b117
Showing 1 changed file with 5 additions and 4 deletions.
diff --git 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
 }