forked from mirrors/ntfy
290 lines
13 KiB
Go
290 lines
13 KiB
Go
package server
|
|
|
|
import (
|
|
"github.com/emersion/go-smtp"
|
|
"github.com/stretchr/testify/require"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestSmtpBackend_Multipart(t *testing.T) {
|
|
email := `MIME-Version: 1.0
|
|
Date: Tue, 28 Dec 2021 00:30:10 +0100
|
|
Message-ID: <CAAvm79YP0C=Rt1N=KWmSUBB87KK2rRChmdzKqF1vCwMEUiVzLQ@mail.gmail.com>
|
|
Subject: and one more
|
|
From: Phil <phil@example.com>
|
|
To: ntfy-mytopic@ntfy.sh
|
|
Content-Type: multipart/alternative; boundary="000000000000f3320b05d42915c9"
|
|
|
|
--000000000000f3320b05d42915c9
|
|
Content-Type: text/plain; charset="UTF-8"
|
|
|
|
what's up
|
|
|
|
--000000000000f3320b05d42915c9
|
|
Content-Type: text/html; charset="UTF-8"
|
|
|
|
<div dir="ltr">what's up<br clear="all"><div><br></div></div>
|
|
|
|
--000000000000f3320b05d42915c9--`
|
|
_, backend := newTestBackend(t, func(m *message) error {
|
|
require.Equal(t, "mytopic", m.Topic)
|
|
require.Equal(t, "and one more", m.Title)
|
|
require.Equal(t, "what's up", m.Message)
|
|
return nil
|
|
})
|
|
session, _ := backend.AnonymousLogin(nil)
|
|
require.Nil(t, session.Mail("phil@example.com", smtp.MailOptions{}))
|
|
require.Nil(t, session.Rcpt("ntfy-mytopic@ntfy.sh"))
|
|
require.Nil(t, session.Data(strings.NewReader(email)))
|
|
}
|
|
|
|
func TestSmtpBackend_MultipartNoBody(t *testing.T) {
|
|
email := `MIME-Version: 1.0
|
|
Date: Tue, 28 Dec 2021 01:33:34 +0100
|
|
Message-ID: <CAAvm7ABCDsi9vsuu0WTRXzZQBC8dXrDOLT8iCWdqrsmg@mail.gmail.com>
|
|
Subject: This email has a subject but no body
|
|
From: Phil <phil@example.com>
|
|
To: ntfy-emailtest@ntfy.sh
|
|
Content-Type: multipart/alternative; boundary="000000000000bcf4a405d429f8d4"
|
|
|
|
--000000000000bcf4a405d429f8d4
|
|
Content-Type: text/plain; charset="UTF-8"
|
|
|
|
|
|
|
|
--000000000000bcf4a405d429f8d4
|
|
Content-Type: text/html; charset="UTF-8"
|
|
|
|
<div dir="ltr"><br></div>
|
|
|
|
--000000000000bcf4a405d429f8d4--`
|
|
_, backend := newTestBackend(t, func(m *message) error {
|
|
require.Equal(t, "emailtest", m.Topic)
|
|
require.Equal(t, "", m.Title) // We flipped message and body
|
|
require.Equal(t, "This email has a subject but no body", m.Message)
|
|
return nil
|
|
})
|
|
session, _ := backend.AnonymousLogin(nil)
|
|
require.Nil(t, session.Mail("phil@example.com", smtp.MailOptions{}))
|
|
require.Nil(t, session.Rcpt("ntfy-emailtest@ntfy.sh"))
|
|
require.Nil(t, session.Data(strings.NewReader(email)))
|
|
}
|
|
|
|
func TestSmtpBackend_Plaintext(t *testing.T) {
|
|
email := `Date: Tue, 28 Dec 2021 00:30:10 +0100
|
|
Message-ID: <CAAvm79YP0C=Rt1N=KWmSUBB87KK2rRChmdzKqF1vCwMEUiVzLQ@mail.gmail.com>
|
|
Subject: and one more
|
|
From: Phil <phil@example.com>
|
|
To: mytopic@ntfy.sh
|
|
Content-Type: text/plain; charset="UTF-8"
|
|
|
|
what's up
|
|
`
|
|
conf, backend := newTestBackend(t, func(m *message) error {
|
|
require.Equal(t, "mytopic", m.Topic)
|
|
require.Equal(t, "and one more", m.Title)
|
|
require.Equal(t, "what's up", m.Message)
|
|
return nil
|
|
})
|
|
conf.SMTPServerAddrPrefix = ""
|
|
session, _ := backend.AnonymousLogin(nil)
|
|
require.Nil(t, session.Mail("phil@example.com", smtp.MailOptions{}))
|
|
require.Nil(t, session.Rcpt("mytopic@ntfy.sh"))
|
|
require.Nil(t, session.Data(strings.NewReader(email)))
|
|
}
|
|
|
|
func TestSmtpBackend_Plaintext_EncodedSubject(t *testing.T) {
|
|
email := `Date: Tue, 28 Dec 2021 00:30:10 +0100
|
|
Subject: =?UTF-8?B?VGhyZWUgc2FudGFzIPCfjoXwn46F8J+OhQ==?=
|
|
From: Phil <phil@example.com>
|
|
To: ntfy-mytopic@ntfy.sh
|
|
Content-Type: text/plain; charset="UTF-8"
|
|
|
|
what's up
|
|
`
|
|
_, backend := newTestBackend(t, func(m *message) error {
|
|
require.Equal(t, "Three santas 🎅🎅🎅", m.Title)
|
|
return nil
|
|
})
|
|
session, _ := backend.AnonymousLogin(nil)
|
|
require.Nil(t, session.Mail("phil@example.com", smtp.MailOptions{}))
|
|
require.Nil(t, session.Rcpt("ntfy-mytopic@ntfy.sh"))
|
|
require.Nil(t, session.Data(strings.NewReader(email)))
|
|
}
|
|
|
|
func TestSmtpBackend_Plaintext_TooLongTruncate(t *testing.T) {
|
|
email := `Date: Tue, 28 Dec 2021 00:30:10 +0100
|
|
Message-ID: <CAAvm79YP0C=Rt1N=KWmSUBB87KK2rRChmdzKqF1vCwMEUiVzLQ@mail.gmail.com>
|
|
Subject: and one more
|
|
From: Phil <phil@example.com>
|
|
To: mytopic@ntfy.sh
|
|
Content-Type: text/plain; charset="UTF-8"
|
|
|
|
you know this is a string.
|
|
it's a long string.
|
|
it's supposed to be longer than the max message length
|
|
which is 4096 bytes,
|
|
it used to be 512 bytes, but I increased that for the UnifiedPush support
|
|
the 512 bytes was a little short, some people said
|
|
but it kinda makes sense when you look at what it looks like one a phone
|
|
heck this wasn't even half of it so far.
|
|
so i'm gonna fill the rest of this with AAAAAAAAAAAAAAAAAAAAAAAAAAA
|
|
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
|
|
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAa
|
|
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
and with BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB
|
|
BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB
|
|
BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB
|
|
that should do it
|
|
`
|
|
conf, backend := newTestBackend(t, func(m *message) error {
|
|
expected := `you know this is a string.
|
|
it's a long string.
|
|
it's supposed to be longer than the max message length
|
|
which is 4096 bytes,
|
|
it used to be 512 bytes, but I increased that for the UnifiedPush support
|
|
the 512 bytes was a little short, some people said
|
|
but it kinda makes sense when you look at what it looks like one a phone
|
|
heck this wasn't even half of it so far.
|
|
so i'm gonna fill the rest of this with AAAAAAAAAAAAAAAAAAAAAAAAAAA
|
|
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
|
|
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAa
|
|
AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
......................................................................
|
|
and with BBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBBB
|
|
BBBBBBBBBBBBBBBBBBBBBBBB`
|
|
require.Equal(t, 4096, len(expected)) // Sanity check
|
|
require.Equal(t, expected, m.Message)
|
|
return nil
|
|
})
|
|
conf.SMTPServerAddrPrefix = ""
|
|
session, _ := backend.AnonymousLogin(nil)
|
|
require.Nil(t, session.Mail("phil@example.com", smtp.MailOptions{}))
|
|
require.Nil(t, session.Rcpt("mytopic@ntfy.sh"))
|
|
require.Nil(t, session.Data(strings.NewReader(email)))
|
|
}
|
|
|
|
func TestSmtpBackend_Unsupported(t *testing.T) {
|
|
email := `Date: Tue, 28 Dec 2021 00:30:10 +0100
|
|
Message-ID: <CAAvm79YP0C=Rt1N=KWmSUBB87KK2rRChmdzKqF1vCwMEUiVzLQ@mail.gmail.com>
|
|
Subject: and one more
|
|
From: Phil <phil@example.com>
|
|
To: mytopic@ntfy.sh
|
|
Content-Type: text/SOMETHINGELSE
|
|
|
|
what's up
|
|
`
|
|
conf, backend := newTestBackend(t, func(m *message) error {
|
|
return nil
|
|
})
|
|
conf.SMTPServerAddrPrefix = ""
|
|
session, _ := backend.Login(nil, "user", "pass")
|
|
require.Nil(t, session.Mail("phil@example.com", smtp.MailOptions{}))
|
|
require.Nil(t, session.Rcpt("mytopic@ntfy.sh"))
|
|
require.Equal(t, errUnsupportedContentType, session.Data(strings.NewReader(email)))
|
|
}
|
|
|
|
func newTestBackend(t *testing.T, sub subscriber) (*Config, *smtpBackend) {
|
|
conf := newTestConfig(t)
|
|
conf.SMTPServerListen = ":25"
|
|
conf.SMTPServerDomain = "ntfy.sh"
|
|
conf.SMTPServerAddrPrefix = "ntfy-"
|
|
backend := newMailBackend(conf, sub)
|
|
return conf, backend
|
|
}
|