mirrors/maddy
1
0
Fork 0
mirror of https://github.com/foxcpp/maddy.git synced 2025-04-03 05:07:38 +03:00
Code Issues Projects Releases Packages Wiki Activity
maddy/tests/smtp_autobuffer_test.go

347 lines
10 KiB
Go
Raw Permalink Blame History

//go:build integration && cgo && !nosqlite3
// +build integration,cgo,!nosqlite3
/*
Maddy Mail Server - Composable all-in-one email server.
Copyright © 2019-2020 Max Mazurov <fox.cpp@disroot.org>, Maddy Mail Server contributors
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
package tests_test
import (
"strings"
"testing"
"time"
"github.com/foxcpp/maddy/tests"
)
func TestSMTPEndpoint_LargeMessage(tt *testing.T) {
// Send 1.44 MiB message to verify it being handled correctly
// everywhere.
// (Issue 389)
tt.Parallel()
t := tests.NewT(tt)
t.DNS(nil)
t.Port("imap")
t.Port("smtp")
t.Config(`
storage.imapsql test_store {
driver sqlite3
dsn imapsql.db
}
imap tcp://127.0.0.1:{env:TEST_PORT_imap} {
tls off
auth dummy
storage &test_store
}
smtp tcp://127.0.0.1:{env:TEST_PORT_smtp} {
hostname maddy.test
tls off
deliver_to &test_store
}
`)
t.Run(2)
defer t.Close()
imapConn := t.Conn("imap")
defer imapConn.Close()
imapConn.ExpectPattern(`\* OK *`)
imapConn.Writeln(". LOGIN testusr@maddy.test 1234")
imapConn.ExpectPattern(". OK *")
imapConn.Writeln(". SELECT INBOX")
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`. OK *`)
smtpConn := t.Conn("smtp")
defer smtpConn.Close()
smtpConn.SMTPNegotation("localhost", nil, nil)
smtpConn.Writeln("MAIL FROM:<sender@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("RCPT TO:<testusr@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("DATA")
smtpConn.ExpectPattern("354 *")
smtpConn.Writeln("From: <sender@maddy.test>")
smtpConn.Writeln("To: <testusr@maddy.test>")
smtpConn.Writeln("Subject: Hi!")
smtpConn.Writeln("")
for i := 0; i < 3000; i++ {
smtpConn.Writeln(strings.Repeat("A", 500))
}
// 3000*502 ~ 1.44 MiB not including header side
smtpConn.Writeln(".")
time.Sleep(500 * time.Millisecond)
imapConn.Writeln(". NOOP")
imapConn.ExpectPattern(`\* 1 EXISTS`)
imapConn.ExpectPattern(`\* 1 RECENT`)
imapConn.ExpectPattern(". OK *")
imapConn.Writeln(". FETCH 1 (BODY.PEEK[])")
imapConn.ExpectPattern(`\* 1 FETCH (BODY\[\] {1506312}*`)
imapConn.Expect(`Delivered-To: testusr@maddy.test`)
imapConn.Expect(`Return-Path: <sender@maddy.test>`)
imapConn.ExpectPattern(`Received: from localhost (client.maddy.test \[` + tests.DefaultSourceIP.String() + `\]) by maddy.test`)
imapConn.ExpectPattern(` (envelope-sender <sender@maddy.test>) with ESMTP id *; *`)
imapConn.ExpectPattern(` *`)
imapConn.Expect("From: <sender@maddy.test>")
imapConn.Expect("To: <testusr@maddy.test>")
imapConn.Expect("Subject: Hi!")
imapConn.Expect("")
for i := 0; i < 3000; i++ {
imapConn.Expect(strings.Repeat("A", 500))
}
imapConn.Expect(")")
imapConn.ExpectPattern(`. OK *`)
}
func TestSMTPEndpoint_FileBuffer(tt *testing.T) {
run := func(tt *testing.T, bufferOpt string) {
tt.Parallel()
t := tests.NewT(tt)
t.DNS(nil)
t.Port("imap")
t.Port("smtp")
t.Config(`
storage.imapsql test_store {
driver sqlite3
dsn imapsql.db
}
imap tcp://127.0.0.1:{env:TEST_PORT_imap} {
tls off
auth dummy
storage &test_store
}
smtp tcp://127.0.0.1:{env:TEST_PORT_smtp} {
hostname maddy.test
tls off
buffer ` + bufferOpt + `
deliver_to &test_store
}
`)
t.Run(2)
defer t.Close()
imapConn := t.Conn("imap")
defer imapConn.Close()
imapConn.ExpectPattern(`\* OK *`)
imapConn.Writeln(". LOGIN testusr@maddy.test 1234")
imapConn.ExpectPattern(". OK *")
imapConn.Writeln(". SELECT INBOX")
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`. OK *`)
smtpConn := t.Conn("smtp")
defer smtpConn.Close()
smtpConn.SMTPNegotation("localhost", nil, nil)
smtpConn.Writeln("MAIL FROM:<sender@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("RCPT TO:<testusr@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("DATA")
smtpConn.ExpectPattern("354 *")
smtpConn.Writeln("From: <sender@maddy.test>")
smtpConn.Writeln("To: <testusr@maddy.test>")
smtpConn.Writeln("Subject: Hi!")
smtpConn.Writeln("")
smtpConn.Writeln("AAAAABBBBBB")
smtpConn.Writeln(".")
smtpConn.ExpectPattern("2*")
time.Sleep(500 * time.Millisecond)
imapConn.Writeln(". NOOP")
imapConn.ExpectPattern(`\* 1 EXISTS`)
imapConn.ExpectPattern(`\* 1 RECENT`)
imapConn.ExpectPattern(". OK *")
imapConn.Writeln(". FETCH 1 (BODY.PEEK[])")
imapConn.ExpectPattern(`\* 1 FETCH (BODY\[\] {*}*`)
imapConn.Expect(`Delivered-To: testusr@maddy.test`)
imapConn.Expect(`Return-Path: <sender@maddy.test>`)
imapConn.ExpectPattern(`Received: from localhost (client.maddy.test \[` + tests.DefaultSourceIP.String() + `\]) by maddy.test`)
imapConn.ExpectPattern(` (envelope-sender <sender@maddy.test>) with ESMTP id *; *`)
imapConn.ExpectPattern(` *`)
imapConn.Expect("From: <sender@maddy.test>")
imapConn.Expect("To: <testusr@maddy.test>")
imapConn.Expect("Subject: Hi!")
imapConn.Expect("")
imapConn.Expect("AAAAABBBBBB")
imapConn.Expect(")")
imapConn.ExpectPattern(`. OK *`)
}
tt.Run("ram", func(tt *testing.T) { run(tt, "ram") })
tt.Run("fs", func(tt *testing.T) { run(tt, "fs") })
}
func TestSMTPEndpoint_Autobuffer(tt *testing.T) {
tt.Parallel()
t := tests.NewT(tt)
t.DNS(nil)
t.Port("imap")
t.Port("smtp")
t.Config(`
storage.imapsql test_store {
driver sqlite3
dsn imapsql.db
}
imap tcp://127.0.0.1:{env:TEST_PORT_imap} {
tls off
auth dummy
storage &test_store
}
smtp tcp://127.0.0.1:{env:TEST_PORT_smtp} {
hostname maddy.test
tls off
buffer auto 5b
deliver_to &test_store
}
`)
t.Run(2)
defer t.Close()
imapConn := t.Conn("imap")
defer imapConn.Close()
imapConn.ExpectPattern(`\* OK *`)
imapConn.Writeln(". LOGIN testusr@maddy.test 1234")
imapConn.ExpectPattern(". OK *")
imapConn.Writeln(". SELECT INBOX")
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`\* *`)
imapConn.ExpectPattern(`. OK *`)
smtpConn := t.Conn("smtp")
defer smtpConn.Close()
smtpConn.SMTPNegotation("localhost", nil, nil)
smtpConn.Writeln("MAIL FROM:<sender@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("RCPT TO:<testusr@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("DATA")
smtpConn.ExpectPattern("354 *")
smtpConn.Writeln("From: <sender@maddy.test>")
smtpConn.Writeln("To: <testusr@maddy.test>")
smtpConn.Writeln("Subject: Hi!")
smtpConn.Writeln("")
smtpConn.Writeln("AAAAABBBBBB")
smtpConn.Writeln(".")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("MAIL FROM:<sender@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("RCPT TO:<testusr@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("DATA")
smtpConn.ExpectPattern("354 *")
smtpConn.Writeln("From: <sender@maddy.test>")
smtpConn.Writeln("To: <testusr@maddy.test>")
smtpConn.Writeln("Subject: Hi!")
smtpConn.Writeln("")
smtpConn.Writeln(".")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("MAIL FROM:<sender@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("RCPT TO:<testusr@maddy.test>")
smtpConn.ExpectPattern("2*")
smtpConn.Writeln("DATA")
smtpConn.ExpectPattern("354 *")
smtpConn.Writeln("From: <sender@maddy.test>")
smtpConn.Writeln("To: <testusr@maddy.test>")
smtpConn.Writeln("Subject: Hi!")
smtpConn.Writeln("")
smtpConn.Writeln("AAA")
smtpConn.Writeln(".")
smtpConn.ExpectPattern("2*")
time.Sleep(500 * time.Millisecond)
imapConn.Writeln(". NOOP")
// This will break with go-imap v2 upgrade merging updates.
imapConn.ExpectPattern(`\* 3 EXISTS`)
imapConn.ExpectPattern(`\* 3 RECENT`)
imapConn.ExpectPattern(". OK *")
imapConn.Writeln(". FETCH 1:3 (BODY.PEEK[])")
imapConn.ExpectPattern(`\* 1 FETCH (BODY\[\] {*}*`)
imapConn.Expect(`Delivered-To: testusr@maddy.test`)
imapConn.Expect(`Return-Path: <sender@maddy.test>`)
imapConn.ExpectPattern(`Received: from localhost (client.maddy.test \[` + tests.DefaultSourceIP.String() + `\]) by maddy.test`)
imapConn.ExpectPattern(` (envelope-sender <sender@maddy.test>) with ESMTP id *; *`)
imapConn.ExpectPattern(` *`)
imapConn.Expect("From: <sender@maddy.test>")
imapConn.Expect("To: <testusr@maddy.test>")
imapConn.Expect("Subject: Hi!")
imapConn.Expect("")
imapConn.Expect("AAAAABBBBBB")
imapConn.Expect(")")
imapConn.ExpectPattern(`\* 2 FETCH (BODY\[\] {*}*`)
imapConn.Expect(`Delivered-To: testusr@maddy.test`)
imapConn.Expect(`Return-Path: <sender@maddy.test>`)
imapConn.ExpectPattern(`Received: from localhost (client.maddy.test \[` + tests.DefaultSourceIP.String() + `\]) by maddy.test`)
imapConn.ExpectPattern(` (envelope-sender <sender@maddy.test>) with ESMTP id *; *`)
imapConn.ExpectPattern(` *`)
imapConn.Expect("From: <sender@maddy.test>")
imapConn.Expect("To: <testusr@maddy.test>")
imapConn.Expect("Subject: Hi!")
imapConn.Expect("")
imapConn.Expect(")")
imapConn.ExpectPattern(`\* 3 FETCH (BODY\[\] {*}*`)
imapConn.Expect(`Delivered-To: testusr@maddy.test`)
imapConn.Expect(`Return-Path: <sender@maddy.test>`)
imapConn.ExpectPattern(`Received: from localhost (client.maddy.test \[` + tests.DefaultSourceIP.String() + `\]) by maddy.test`)
imapConn.ExpectPattern(` (envelope-sender <sender@maddy.test>) with ESMTP id *; *`)
imapConn.ExpectPattern(` *`)
imapConn.Expect("From: <sender@maddy.test>")
imapConn.Expect("To: <testusr@maddy.test>")
imapConn.Expect("Subject: Hi!")
imapConn.Expect("")
imapConn.Expect("AAA")
imapConn.Expect(")")
imapConn.ExpectPattern(`. OK *`)
}
Reference in a new issue View git blame Copy permalink