mirror of
https://github.com/1f349/lotus.git
synced 2024-12-22 16:14:07 +00:00
72 lines
1.8 KiB
Go
72 lines
1.8 KiB
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))
|
|
}
|