basic common app packages
This commit is contained in:
292
email/gomail/smtp_test.go
Normal file
292
email/gomail/smtp_test.go
Normal file
@@ -0,0 +1,292 @@
|
||||
package gomail
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/tls"
|
||||
"io"
|
||||
"net"
|
||||
"net/smtp"
|
||||
"reflect"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
const (
|
||||
testPort = 587
|
||||
testSSLPort = 465
|
||||
)
|
||||
|
||||
var (
|
||||
testConn = &net.TCPConn{}
|
||||
testTLSConn = &tls.Conn{}
|
||||
testConfig = &tls.Config{InsecureSkipVerify: true}
|
||||
testAuth = smtp.PlainAuth("", testUser, testPwd, testHost)
|
||||
)
|
||||
|
||||
func TestDialer(t *testing.T) {
|
||||
d := NewDialer(testHost, testPort, "user", "pwd")
|
||||
testSendMail(t, d, []string{
|
||||
"Extension STARTTLS",
|
||||
"StartTLS",
|
||||
"Extension AUTH",
|
||||
"Auth",
|
||||
"Mail " + testFrom,
|
||||
"Rcpt " + testTo1,
|
||||
"Rcpt " + testTo2,
|
||||
"Data",
|
||||
"Write message",
|
||||
"Close writer",
|
||||
"Quit",
|
||||
"Close",
|
||||
})
|
||||
}
|
||||
|
||||
func TestDialerSSL(t *testing.T) {
|
||||
d := NewDialer(testHost, testSSLPort, "user", "pwd")
|
||||
testSendMail(t, d, []string{
|
||||
"Extension AUTH",
|
||||
"Auth",
|
||||
"Mail " + testFrom,
|
||||
"Rcpt " + testTo1,
|
||||
"Rcpt " + testTo2,
|
||||
"Data",
|
||||
"Write message",
|
||||
"Close writer",
|
||||
"Quit",
|
||||
"Close",
|
||||
})
|
||||
}
|
||||
|
||||
func TestDialerConfig(t *testing.T) {
|
||||
d := NewDialer(testHost, testPort, "user", "pwd")
|
||||
d.LocalName = "test"
|
||||
d.TLSConfig = testConfig
|
||||
testSendMail(t, d, []string{
|
||||
"Hello test",
|
||||
"Extension STARTTLS",
|
||||
"StartTLS",
|
||||
"Extension AUTH",
|
||||
"Auth",
|
||||
"Mail " + testFrom,
|
||||
"Rcpt " + testTo1,
|
||||
"Rcpt " + testTo2,
|
||||
"Data",
|
||||
"Write message",
|
||||
"Close writer",
|
||||
"Quit",
|
||||
"Close",
|
||||
})
|
||||
}
|
||||
|
||||
func TestDialerSSLConfig(t *testing.T) {
|
||||
d := NewDialer(testHost, testSSLPort, "user", "pwd")
|
||||
d.LocalName = "test"
|
||||
d.TLSConfig = testConfig
|
||||
testSendMail(t, d, []string{
|
||||
"Hello test",
|
||||
"Extension AUTH",
|
||||
"Auth",
|
||||
"Mail " + testFrom,
|
||||
"Rcpt " + testTo1,
|
||||
"Rcpt " + testTo2,
|
||||
"Data",
|
||||
"Write message",
|
||||
"Close writer",
|
||||
"Quit",
|
||||
"Close",
|
||||
})
|
||||
}
|
||||
|
||||
func TestDialerNoAuth(t *testing.T) {
|
||||
d := &Dialer{
|
||||
Host: testHost,
|
||||
Port: testPort,
|
||||
}
|
||||
testSendMail(t, d, []string{
|
||||
"Extension STARTTLS",
|
||||
"StartTLS",
|
||||
"Mail " + testFrom,
|
||||
"Rcpt " + testTo1,
|
||||
"Rcpt " + testTo2,
|
||||
"Data",
|
||||
"Write message",
|
||||
"Close writer",
|
||||
"Quit",
|
||||
"Close",
|
||||
})
|
||||
}
|
||||
|
||||
func TestDialerTimeout(t *testing.T) {
|
||||
d := &Dialer{
|
||||
Host: testHost,
|
||||
Port: testPort,
|
||||
}
|
||||
testSendMailTimeout(t, d, []string{
|
||||
"Extension STARTTLS",
|
||||
"StartTLS",
|
||||
"Mail " + testFrom,
|
||||
"Extension STARTTLS",
|
||||
"StartTLS",
|
||||
"Mail " + testFrom,
|
||||
"Rcpt " + testTo1,
|
||||
"Rcpt " + testTo2,
|
||||
"Data",
|
||||
"Write message",
|
||||
"Close writer",
|
||||
"Quit",
|
||||
"Close",
|
||||
})
|
||||
}
|
||||
|
||||
type mockClient struct {
|
||||
t *testing.T
|
||||
config *tls.Config
|
||||
addr string
|
||||
want []string
|
||||
i int
|
||||
timeout bool
|
||||
}
|
||||
|
||||
func (c *mockClient) Hello(localName string) error {
|
||||
c.do("Hello " + localName)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *mockClient) Extension(ext string) (bool, string) {
|
||||
c.do("Extension " + ext)
|
||||
return true, ""
|
||||
}
|
||||
|
||||
func (c *mockClient) StartTLS(config *tls.Config) error {
|
||||
assertConfig(c.t, config, c.config)
|
||||
c.do("StartTLS")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *mockClient) Auth(a smtp.Auth) error {
|
||||
if !reflect.DeepEqual(a, testAuth) {
|
||||
c.t.Errorf("Invalid auth, got %#v, want %#v", a, testAuth)
|
||||
}
|
||||
c.do("Auth")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *mockClient) Mail(from string) error {
|
||||
c.do("Mail " + from)
|
||||
if c.timeout {
|
||||
c.timeout = false
|
||||
return io.EOF
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *mockClient) Rcpt(to string) error {
|
||||
c.do("Rcpt " + to)
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *mockClient) Data() (io.WriteCloser, error) {
|
||||
c.do("Data")
|
||||
return &mockWriter{c: c, want: testMsg}, nil
|
||||
}
|
||||
|
||||
func (c *mockClient) Quit() error {
|
||||
c.do("Quit")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *mockClient) Close() error {
|
||||
c.do("Close")
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *mockClient) do(cmd string) {
|
||||
if c.i >= len(c.want) {
|
||||
c.t.Fatalf("Invalid command %q", cmd)
|
||||
}
|
||||
|
||||
if cmd != c.want[c.i] {
|
||||
c.t.Fatalf("Invalid command, got %q, want %q", cmd, c.want[c.i])
|
||||
}
|
||||
c.i++
|
||||
}
|
||||
|
||||
type mockWriter struct {
|
||||
want string
|
||||
c *mockClient
|
||||
buf bytes.Buffer
|
||||
}
|
||||
|
||||
func (w *mockWriter) Write(p []byte) (int, error) {
|
||||
if w.buf.Len() == 0 {
|
||||
w.c.do("Write message")
|
||||
}
|
||||
w.buf.Write(p)
|
||||
return len(p), nil
|
||||
}
|
||||
|
||||
func (w *mockWriter) Close() error {
|
||||
compareBodies(w.c.t, w.buf.String(), w.want)
|
||||
w.c.do("Close writer")
|
||||
return nil
|
||||
}
|
||||
|
||||
func testSendMail(t *testing.T, d *Dialer, want []string) {
|
||||
doTestSendMail(t, d, want, false)
|
||||
}
|
||||
|
||||
func testSendMailTimeout(t *testing.T, d *Dialer, want []string) {
|
||||
doTestSendMail(t, d, want, true)
|
||||
}
|
||||
|
||||
func doTestSendMail(t *testing.T, d *Dialer, want []string, timeout bool) {
|
||||
testClient := &mockClient{
|
||||
t: t,
|
||||
want: want,
|
||||
addr: addr(d.Host, d.Port),
|
||||
config: d.TLSConfig,
|
||||
timeout: timeout,
|
||||
}
|
||||
|
||||
netDialTimeout = func(network, address string, _ time.Duration) (net.Conn, error) {
|
||||
if network != "tcp" {
|
||||
t.Errorf("Invalid network, got %q, want tcp", network)
|
||||
}
|
||||
if address != testClient.addr {
|
||||
t.Errorf("Invalid address, got %q, want %q",
|
||||
address, testClient.addr)
|
||||
}
|
||||
return testConn, nil
|
||||
}
|
||||
|
||||
tlsClient = func(conn net.Conn, config *tls.Config) *tls.Conn {
|
||||
if conn != testConn {
|
||||
t.Errorf("Invalid conn, got %#v, want %#v", conn, testConn)
|
||||
}
|
||||
assertConfig(t, config, testClient.config)
|
||||
return testTLSConn
|
||||
}
|
||||
|
||||
smtpNewClient = func(_ net.Conn, host string) (smtpClient, error) {
|
||||
if host != testHost {
|
||||
t.Errorf("Invalid host, got %q, want %q", host, testHost)
|
||||
}
|
||||
return testClient, nil
|
||||
}
|
||||
|
||||
if err := d.DialAndSend(getTestMessage()); err != nil {
|
||||
t.Error(err)
|
||||
}
|
||||
}
|
||||
|
||||
func assertConfig(t *testing.T, got, want *tls.Config) {
|
||||
if want == nil {
|
||||
want = &tls.Config{ServerName: testHost}
|
||||
}
|
||||
if got.ServerName != want.ServerName {
|
||||
t.Errorf("Invalid field ServerName in config, got %q, want %q", got.ServerName, want.ServerName)
|
||||
}
|
||||
if got.InsecureSkipVerify != want.InsecureSkipVerify {
|
||||
t.Errorf("Invalid field InsecureSkipVerify in config, got %v, want %v", got.InsecureSkipVerify, want.InsecureSkipVerify)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user