Add some basic tests for the SMTP server component.

diff --git a/smtp_test.go b/smtp_test.go
new file mode 100644 (file)
index 0000000..89eeb3f
--- /dev/null
+++ b/smtp_test.go
@@ -0,0 +1,106 @@
+// 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"
+)
+
+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,
+                               },
+                       },
+               },
+               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")
+       }
+}