lotus/sendmail/sendmail_test.go

package sendmail

import (
	"bytes"
	"github.com/emersion/go-message"
	"github.com/emersion/go-message/mail"
	"github.com/stretchr/testify/assert"
	"io"
	"log"
	"os"
	"os/exec"
	"strings"
	"testing"
	"time"
)

var sendTestMessage []byte

func init() {
	var h mail.Header
	h.SetDate(time.Date(2000, time.January, 1, 0, 0, 0, 0, time.Local))
	h.SetSubject("Happy Millennium")
	h.SetAddressList("From", []*mail.Address{{Name: "Test", Address: "test@localhost"}})
	h.SetAddressList("To", []*mail.Address{{Name: "A", Address: "a@localhost"}})
	h.Set("Content-Type", "text/plain; charset=utf-8")
	entity, err := message.New(h.Header, strings.NewReader("Thanks"))
	if err != nil {
		log.Fatal(err)
	}
	out := new(bytes.Buffer)
	if entity.WriteTo(out) != nil {
		log.Fatal(err)
	}
	sendTestMessage = out.Bytes()
}

func TestSendMail_Send(t *testing.T) {
	execCommand = func(name string, arg ...string) *exec.Cmd {
		log.Println("Hello")
		cs := append([]string{"-test.run=TestSendMailHelperProcess", "--", name}, arg...)
		cmd := exec.Command(os.Args[0], cs...)
		cmd.Env = []string{"GO_WANT_HELPER_PROCESS=1"}
		log.Println(cmd.Path)
		return cmd
	}
	defer func() { execCommand = exec.Command }()

	m := &Mail{From: &mail.Address{Address: "test@localhost"}, Body: sendTestMessage}

	s := &SendMail{SendMailCommand: "/tmp/sendmailXXXXX"}
	assert.NoError(t, s.Send(m))
}

func TestSendMailHelperProcess(t *testing.T) {
	if os.Getenv("GO_WANT_HELPER_PROCESS") != "1" {
		return
	}
	log.Println("This is a test")
	all, err := io.ReadAll(os.Stdin)
	if err != nil {
		panic(err)
	}
	assert.Equal(t, strings.ReplaceAll(`Mime-Version: 1.0
Content-Type: text/plain; charset=utf-8
To: "A" <a@localhost>
From: "Test" <test@localhost>
Subject: Happy Millennium
Date: Sat, 01 Jan 2000 00:00:00 +0000

Thanks`, "\n", "\r\n"), string(all))
}