From fb39605b32ba568aabee3ab22cb209c417fd87dd Mon Sep 17 00:00:00 2001
From: Robert Sesek <rsesek@bluestatic.org>
Date: Sun, 7 Jun 2020 00:15:49 -0400
Subject: [PATCH] Add some basic tests for the SMTP server component.

---
 smtp_test.go | 106 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 106 insertions(+)
 create mode 100644 smtp_test.go

diff --git a/smtp_test.go b/smtp_test.go
new file mode 100644
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")
+	}
+}
-- 
2.22.5