Merge branch 'outbound'

[mailpopbox.git] / smtp_test.go

diff --git a/smtp_test.go b/smtp_test.go
index 2eef868d941de7bfa2b90138c45478c30b8b1a0f..d1239ea99a3bce649d6f04abdfef2f98ed7939c0 100644 (file)
--- a/smtp_test.go
+++ b/smtp_test.go
@@ -1,24 +1,125 @@
+// mailpopbox
+// Copyright 2020 Blue Static <https://www.bluestatic.org>
+// This program is free software licensed under the GNU General Public License,
+// version 3.0. The full text of the license can be found in LICENSE.txt.
+// SPDX-License-Identifier: GPL-3.0-only
+
 package main
 
 import (
+       "bytes"
+       "io/ioutil"
+       "net/mail"
+       "os"
+       "path/filepath"
        "testing"
+
+       "go.uber.org/zap"
+
+       "src.bluestatic.org/mailpopbox/smtp"
 )
 
-var testConfig = Config{
-       Servers: []Server{
-               Server{
-                       Domain:          "domain1.net",
-                       MailboxPassword: "d1",
+func TestVerifyAddress(t *testing.T) {
+       dir, err := ioutil.TempDir("", "maildrop")
+       if err != nil {
+               t.Errorf("Failed to create temp dir: %v", err)
+               return
+       }
+       defer os.RemoveAll(dir)
+
+       s := smtpServer{
+               config: Config{
+                       Hostname: "mx.example.com",
+                       Servers: []Server{
+                               {
+                                       Domain:       "example.com",
+                                       MaildropPath: dir,
+                               },
+                       },
                },
-               Server{
-                       Domain:          "domain2.xyz",
-                       MailboxPassword: "d2",
+               log: zap.NewNop(),
+       }
+
+       if s.VerifyAddress(mail.Address{Address: "example@example.com"}) != smtp.ReplyOK {
+               t.Errorf("Valid mailbox is not reported to be valid")
+       }
+       if s.VerifyAddress(mail.Address{Address: "mailbox@example.com"}) != smtp.ReplyOK {
+               t.Errorf("Valid mailbox is not reported to be valid")
+       }
+       if s.VerifyAddress(mail.Address{Address: "hello@other.net"}) == smtp.ReplyOK {
+               t.Errorf("Invalid mailbox reports to be valid")
+       }
+       if s.VerifyAddress(mail.Address{Address: "hello@mx.example.com"}) == smtp.ReplyOK {
+               t.Errorf("Invalid mailbox reports to be valid")
+       }
+       if s.VerifyAddress(mail.Address{Address: "unknown"}) == smtp.ReplyOK {
+               t.Errorf("Invalid mailbox reports to be valid")
+       }
+}
+
+func TestMessageDelivery(t *testing.T) {
+       dir, err := ioutil.TempDir("", "maildrop")
+       if err != nil {
+               t.Errorf("Failed to create temp dir: %v", err)
+               return
+       }
+       defer os.RemoveAll(dir)
+
+       s := smtpServer{
+               config: Config{
+                       Hostname: "mx.example.com",
+                       Servers: []Server{
+                               {
+                                       Domain:       "example.com",
+                                       MaildropPath: dir,
+                               },
+                       },
                },
-       },
+               log: zap.NewNop(),
+       }
+
+       env := smtp.Envelope{
+               MailFrom: mail.Address{Address: "sender@mail.net"},
+               RcptTo:   []mail.Address{{Address: "receive@example.com"}},
+               Data:     []byte("Hello, world"),
+               ID:       "msgid",
+       }
+
+       if rl := s.OnMessageDelivered(env); rl != nil {
+               t.Errorf("Failed to deliver message: %v", rl)
+       }
+
+       f, err := os.Open(filepath.Join(dir, "msgid.msg"))
+       if err != nil {
+               t.Errorf("Failed to open delivered message: %v", err)
+       }
+       defer f.Close()
+
+       data, err := ioutil.ReadAll(f)
+       if err != nil {
+               t.Errorf("Failed to read message: %v", err)
+       }
+
+       if !bytes.Contains(data, env.Data) {
+               t.Errorf("Could not find expected data in message")
+       }
 }
 
 func TestAuthenticate(t *testing.T) {
-       server := smtpServer{config: testConfig}
+       server := smtpServer{
+               config: Config{
+                       Servers: []Server{
+                               Server{
+                                       Domain:          "domain1.net",
+                                       MailboxPassword: "d1",
+                               },
+                               Server{
+                                       Domain:          "domain2.xyz",
+                                       MailboxPassword: "d2",
+                               },
+                       },
+               },
+       }
 
        authTests := []struct {
                authz, authc, passwd string