mirror of
https://github.com/foxcpp/maddy.git
synced 2025-04-05 14:07:38 +03:00
bf188e454f
The intention is to keep to repo root clean while the list of packages is slowly growing. Additionally, a bunch of small (~30 LoC) files in the repo root is merged into a single maddy.go file, for the same reason. Most of the internal code is moved into the internal/ directory. Go toolchain will make it impossible to import these packages from external applications. Some packages are renamed and moved into the pkg/ directory in the root. According to https://github.com/golang-standards/project-layout this is the de-facto standard to place "library code that's ok to use by external applications" in. To clearly define the purpose of top-level directories, README.md files are added to each.
100 lines
2.9 KiB
Go
100 lines
2.9 KiB
Go
package modify
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/foxcpp/maddy/internal/config"
|
|
)
|
|
|
|
func replaceAddrFromArgs(t *testing.T, modName, from, to string) *replaceAddr {
|
|
r, err := NewReplaceAddr(modName, "", nil, []string{from, to})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := r.Init(&config.Map{Block: &config.Node{}}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return r.(*replaceAddr)
|
|
}
|
|
|
|
func testReplaceAddr(t *testing.T, modName string, rewriter func(*replaceAddr, string) (string, error)) {
|
|
t.Run("plain string", func(t *testing.T) {
|
|
r := replaceAddrFromArgs(t, modName, "test@example.org", "test2@example.org")
|
|
val, err := rewriter(r, "test@example.org")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val != "test2@example.org" {
|
|
t.Fatalf("wrong result: %s != %s", val, "test2@example.org")
|
|
}
|
|
})
|
|
t.Run("plain string (case insensitive)", func(t *testing.T) {
|
|
r := replaceAddrFromArgs(t, modName, "test@EXAmple.org", "test2@example.org")
|
|
val, err := rewriter(r, "teST@exaMPLe.oRg")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val != "test2@example.org" {
|
|
t.Fatalf("wrong result: %s != %s", val, "test2@example.org")
|
|
}
|
|
})
|
|
t.Run("regexp", func(t *testing.T) {
|
|
r := replaceAddrFromArgs(t, modName, `/test@example\.org/`, "test2@example.org")
|
|
val, err := rewriter(r, "test@example.org")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val != "test2@example.org" {
|
|
t.Fatalf("wrong result: %s != %s", val, "test2@example.org")
|
|
}
|
|
})
|
|
t.Run("regexp (case insensitive)", func(t *testing.T) {
|
|
r := replaceAddrFromArgs(t, modName, `/test@EXAmple\.org/`, "test2@example.org")
|
|
val, err := rewriter(r, "teST@exaMPLe.oRg")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val != "test2@example.org" {
|
|
t.Fatalf("wrong result: %s != %s", val, "test2@example.org")
|
|
}
|
|
})
|
|
t.Run("regexp (incomplete match)", func(t *testing.T) {
|
|
r := replaceAddrFromArgs(t, modName, `/example/`, "test2@example.org")
|
|
val, err := rewriter(r, "teST@exaMPLe.oRg")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val != "teST@exaMPLe.oRg" {
|
|
t.Fatalf("wrong result: %s != %s", val, "test2@example.org")
|
|
}
|
|
})
|
|
t.Run("regexp (result expansion)", func(t *testing.T) {
|
|
r := replaceAddrFromArgs(t, modName, `/(.+)@example\.org/`, "$1@example.com")
|
|
val, err := rewriter(r, "test@example.org")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val != "test@example.com" {
|
|
t.Fatalf("wrong result: %s != %s", val, "test2@example.org")
|
|
}
|
|
})
|
|
t.Run("regexp (result expansion preserves case)", func(t *testing.T) {
|
|
r := replaceAddrFromArgs(t, modName, `/(.+)@example\.org/`, "$1@example.com")
|
|
val, err := rewriter(r, "teST@example.org")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if val != "teST@example.com" {
|
|
t.Fatalf("wrong result: %s != %s", val, "test2@example.org")
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestReplaceAddr_RewriteSender(t *testing.T) {
|
|
testReplaceAddr(t, "replace_sender", (*replaceAddr).RewriteSender)
|
|
}
|
|
|
|
func TestReplaceAddr_RewriteRcpt(t *testing.T) {
|
|
testReplaceAddr(t, "replace_rcpt", (*replaceAddr).RewriteRcpt)
|
|
}
|