mirror of
https://gitlab.com/ric_harvey/MailHog.git
synced 2024-11-23 22:34:04 +00:00
151 lines
3.7 KiB
Go
151 lines
3.7 KiB
Go
package main
|
|
|
|
import (
|
|
"github.com/stretchr/testify/assert"
|
|
"net"
|
|
"testing"
|
|
)
|
|
|
|
// FIXME requires a running instance of MailHog
|
|
// FIXME clean up tests, repeated conn.Read(buf) is a mess
|
|
|
|
func TestBasicSMTPAuth(t *testing.T) {
|
|
buf := make([]byte, 1024)
|
|
|
|
// Open a connection
|
|
conn, err := net.Dial("tcp", "127.0.0.1:1025")
|
|
assert.Nil(t, err)
|
|
|
|
// Read the greeting
|
|
n, err := conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "220 mailhog.example ESMTP Go-MailHog\n")
|
|
|
|
// Send EHLO
|
|
_, err = conn.Write([]byte("EHLO localhost\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "250-Hello localhost\n")
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "250-PIPELINING\n")
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "250 AUTH EXTERNAL CRAM-MD5 LOGIN PLAIN\n")
|
|
|
|
// Send AUTH
|
|
_, err = conn.Write([]byte("AUTH EXTERNAL =\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "235 Authentication successful\n")
|
|
|
|
// Send RSET and EHLO
|
|
_, err = conn.Write([]byte("RSET\r\n"))
|
|
n, err = conn.Read(buf)
|
|
_, err = conn.Write([]byte("EHLO localhost\r\n"))
|
|
n, err = conn.Read(buf)
|
|
n, err = conn.Read(buf)
|
|
n, err = conn.Read(buf)
|
|
|
|
// Send AUTH
|
|
_, err = conn.Write([]byte("AUTH PLAIN foobar\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "235 Authentication successful\n")
|
|
|
|
// Send RSET and EHLO
|
|
_, err = conn.Write([]byte("RSET\r\n"))
|
|
n, err = conn.Read(buf)
|
|
_, err = conn.Write([]byte("EHLO localhost\r\n"))
|
|
n, err = conn.Read(buf)
|
|
n, err = conn.Read(buf)
|
|
n, err = conn.Read(buf)
|
|
|
|
// Send AUTH
|
|
_, err = conn.Write([]byte("AUTH PLAIN\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "334 \n")
|
|
|
|
// Send AUTH
|
|
_, err = conn.Write([]byte("foobar\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "235 Authentication successful\n")
|
|
|
|
// Send RSET and EHLO
|
|
_, err = conn.Write([]byte("RSET\r\n"))
|
|
n, err = conn.Read(buf)
|
|
_, err = conn.Write([]byte("EHLO localhost\r\n"))
|
|
n, err = conn.Read(buf)
|
|
n, err = conn.Read(buf)
|
|
n, err = conn.Read(buf)
|
|
|
|
// Send AUTH
|
|
_, err = conn.Write([]byte("AUTH LOGIN\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "334 VXNlcm5hbWU6\n")
|
|
|
|
// Send AUTH
|
|
_, err = conn.Write([]byte("foobar\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "334 UGFzc3dvcmQ6\n")
|
|
|
|
// Send AUTH
|
|
_, err = conn.Write([]byte("foobar\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "235 Authentication successful\n")
|
|
|
|
// Send RSET and EHLO
|
|
_, err = conn.Write([]byte("RSET\r\n"))
|
|
n, err = conn.Read(buf)
|
|
_, err = conn.Write([]byte("EHLO localhost\r\n"))
|
|
n, err = conn.Read(buf)
|
|
n, err = conn.Read(buf)
|
|
n, err = conn.Read(buf)
|
|
|
|
// Send AUTH
|
|
_, err = conn.Write([]byte("AUTH CRAM-MD5\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "334 PDQxOTI5NDIzNDEuMTI4Mjg0NzJAc291cmNlZm91ci5hbmRyZXcuY211LmVkdT4=\n")
|
|
|
|
// Send AUTH
|
|
_, err = conn.Write([]byte("foobar\r\n"))
|
|
assert.Nil(t, err)
|
|
|
|
// Read the response
|
|
n, err = conn.Read(buf)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, string(buf[0:n]), "235 Authentication successful\n")
|
|
}
|