mix
0d98d92642
test / test (push) Waiting to run
Port 25 accepts only configured domains and listed recipients, then forwards to an upstream; the outbound path is unchanged when the flag is off. Co-authored-by: Cursor <cursoragent@cursor.com>
134 lines
2.8 KiB
Go
134 lines
2.8 KiB
Go
package validate
|
|
|
|
import "testing"
|
|
|
|
func TestNormalizeDomain(t *testing.T) {
|
|
cases := map[string]string{
|
|
" Example.COM ": "example.com",
|
|
"MAIL.Example.Org": "mail.example.org",
|
|
"example.com": "example.com",
|
|
}
|
|
for in, want := range cases {
|
|
if got := NormalizeDomain(in); got != want {
|
|
t.Errorf("NormalizeDomain(%q) = %q, want %q", in, got, want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestValidateDomainValid(t *testing.T) {
|
|
valid := []string{
|
|
"example.com",
|
|
"mail.example.com",
|
|
"a.co",
|
|
"sub-domain.example.co.uk",
|
|
"x1.y2.z3",
|
|
"1example.com",
|
|
}
|
|
for _, d := range valid {
|
|
if err := Domain(d); err != nil {
|
|
t.Errorf("Domain(%q) unexpected error: %v", d, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestValidateDomainInvalid(t *testing.T) {
|
|
invalid := []string{
|
|
"",
|
|
"localhost",
|
|
"example",
|
|
".example.com",
|
|
"example.com.",
|
|
"exa mple.com",
|
|
"example..com",
|
|
"-example.com",
|
|
"example-.com",
|
|
"example.com\n",
|
|
"exa*mple.com",
|
|
"exa_mple.com",
|
|
"Example.com",
|
|
"пример.рф",
|
|
"example.c/m",
|
|
}
|
|
for _, d := range invalid {
|
|
if err := Domain(d); err == nil {
|
|
t.Errorf("Domain(%q) = nil, want error", d)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestValidateDomainLongLabelRejected(t *testing.T) {
|
|
label := make([]byte, 64)
|
|
for i := range label {
|
|
label[i] = 'a'
|
|
}
|
|
if err := Domain(string(label) + ".com"); err == nil {
|
|
t.Error("expected error for over-long label")
|
|
}
|
|
}
|
|
|
|
func TestHostValid(t *testing.T) {
|
|
valid := []string{
|
|
"10.0.0.8",
|
|
"192.0.2.20",
|
|
"2001:db8::1",
|
|
"mail.internal.example",
|
|
"mx1",
|
|
"mail-1.lan",
|
|
}
|
|
for _, h := range valid {
|
|
if err := Host(h); err != nil {
|
|
t.Errorf("Host(%q) unexpected error: %v", h, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHostInvalid(t *testing.T) {
|
|
invalid := []string{
|
|
"",
|
|
"exa mple",
|
|
"host/name",
|
|
"host;rm",
|
|
"-bad",
|
|
"bad-",
|
|
"host\nname",
|
|
}
|
|
for _, h := range invalid {
|
|
if err := Host(h); err == nil {
|
|
t.Errorf("Host(%q) = nil, want error", h)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestNormalizeHostStripsIPv6Brackets(t *testing.T) {
|
|
if got := NormalizeHost(" [2001:DB8::1] "); got != "2001:db8::1" {
|
|
t.Errorf("NormalizeHost IPv6 = %q", got)
|
|
}
|
|
if got := NormalizeHost("Mail.Example.COM"); got != "mail.example.com" {
|
|
t.Errorf("NormalizeHost hostname = %q", got)
|
|
}
|
|
}
|
|
|
|
func TestPort(t *testing.T) {
|
|
n, err := Port("25")
|
|
if err != nil || n != 25 {
|
|
t.Fatalf("Port(25) = %d, %v", n, err)
|
|
}
|
|
for _, raw := range []string{"", "0", "65536", "abc", "-1"} {
|
|
if _, err := Port(raw); err == nil {
|
|
t.Errorf("Port(%q) = nil, want error", raw)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestMailboxInDomain(t *testing.T) {
|
|
if err := MailboxInDomain("staff@lists.example.com", "lists.example.com"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := MailboxInDomain("staff@other.com", "lists.example.com"); err == nil {
|
|
t.Fatal("expected domain mismatch error")
|
|
}
|
|
if err := MailboxInDomain("bad addr@lists.example.com", "lists.example.com"); err == nil {
|
|
t.Fatal("expected local-part error")
|
|
}
|
|
}
|