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.
130 lines
3.2 KiB
Go
130 lines
3.2 KiB
Go
package msgpipeline
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/foxcpp/maddy/internal/modify"
|
|
"github.com/foxcpp/maddy/internal/module"
|
|
"github.com/foxcpp/maddy/internal/testutils"
|
|
)
|
|
|
|
type multipleErrs map[string]error
|
|
|
|
func (m multipleErrs) SetStatus(rcptTo string, err error) {
|
|
m[rcptTo] = err
|
|
}
|
|
|
|
func TestMsgPipeline_BodyNonAtomic(t *testing.T) {
|
|
err := errors.New("go away")
|
|
|
|
target := testutils.Target{
|
|
PartialBodyErr: map[string]error{
|
|
"tester@example.org": err,
|
|
},
|
|
}
|
|
d := MsgPipeline{
|
|
msgpipelineCfg: msgpipelineCfg{
|
|
perSource: map[string]sourceBlock{},
|
|
defaultSource: sourceBlock{
|
|
perRcpt: map[string]*rcptBlock{},
|
|
defaultRcpt: &rcptBlock{
|
|
targets: []module.DeliveryTarget{&target},
|
|
},
|
|
},
|
|
},
|
|
Log: testutils.Logger(t, "msgpipeline"),
|
|
}
|
|
|
|
c := multipleErrs{}
|
|
testutils.DoTestDeliveryNonAtomic(t, c, &d, "sender@example.org", []string{"tester@example.org", "tester2@example.org"})
|
|
|
|
if c["tester@example.org"] == nil {
|
|
t.Fatalf("no error for tester@example.org")
|
|
}
|
|
if c["tester@example.org"].Error() != err.Error() {
|
|
t.Errorf("wrong error for tester@example.org: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestMsgPipeline_BodyNonAtomic_ModifiedRcpt(t *testing.T) {
|
|
err := errors.New("go away")
|
|
|
|
target := testutils.Target{
|
|
PartialBodyErr: map[string]error{
|
|
"tester-alias@example.org": err,
|
|
},
|
|
}
|
|
d := MsgPipeline{
|
|
msgpipelineCfg: msgpipelineCfg{
|
|
globalModifiers: modify.Group{
|
|
Modifiers: []module.Modifier{
|
|
testutils.Modifier{
|
|
InstName: "test_modifier",
|
|
RcptTo: map[string]string{
|
|
"tester@example.org": "tester-alias@example.org",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
perSource: map[string]sourceBlock{},
|
|
defaultSource: sourceBlock{
|
|
perRcpt: map[string]*rcptBlock{},
|
|
defaultRcpt: &rcptBlock{
|
|
targets: []module.DeliveryTarget{&target},
|
|
},
|
|
},
|
|
},
|
|
Log: testutils.Logger(t, "msgpipeline"),
|
|
}
|
|
|
|
c := multipleErrs{}
|
|
testutils.DoTestDeliveryNonAtomic(t, c, &d, "sender@example.org", []string{"tester@example.org"})
|
|
|
|
if c["tester@example.org"] == nil {
|
|
t.Fatalf("no error for tester@example.org")
|
|
}
|
|
if c["tester@example.org"].Error() != err.Error() {
|
|
t.Errorf("wrong error for tester@example.org: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestMsgPipeline_BodyNonAtomic_ExpandAtomic(t *testing.T) {
|
|
err := errors.New("go away")
|
|
|
|
target, target2 := testutils.Target{
|
|
PartialBodyErr: map[string]error{
|
|
"tester@example.org": err,
|
|
},
|
|
}, testutils.Target{
|
|
BodyErr: err,
|
|
}
|
|
d := MsgPipeline{
|
|
msgpipelineCfg: msgpipelineCfg{
|
|
perSource: map[string]sourceBlock{},
|
|
defaultSource: sourceBlock{
|
|
perRcpt: map[string]*rcptBlock{},
|
|
defaultRcpt: &rcptBlock{
|
|
targets: []module.DeliveryTarget{&target, &target2},
|
|
},
|
|
},
|
|
},
|
|
Log: testutils.Logger(t, "msgpipeline"),
|
|
}
|
|
|
|
c := multipleErrs{}
|
|
testutils.DoTestDeliveryNonAtomic(t, c, &d, "sender@example.org", []string{"tester@example.org", "tester2@example.org"})
|
|
|
|
if c["tester@example.org"] == nil {
|
|
t.Fatalf("no error for tester@example.org")
|
|
}
|
|
if c["tester@example.org"].Error() != err.Error() {
|
|
t.Errorf("wrong error for tester@example.org: %v", err)
|
|
}
|
|
if c["tester2@example.org"] == nil {
|
|
t.Fatalf("no error for tester@example.org")
|
|
}
|
|
if c["tester2@example.org"].Error() != err.Error() {
|
|
t.Errorf("wrong error for tester@example.org: %v", err)
|
|
}
|
|
}
|