mirror of
https://github.com/mjl-/mox.git
synced 2025-07-10 09:54:40 +03:00
mox!
This commit is contained in:
108
dmarcdb/db_test.go
Normal file
108
dmarcdb/db_test.go
Normal file
@ -0,0 +1,108 @@
|
||||
package dmarcdb
|
||||
|
||||
import (
|
||||
"context"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"reflect"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/mjl-/mox/dmarcrpt"
|
||||
"github.com/mjl-/mox/dns"
|
||||
"github.com/mjl-/mox/mox-"
|
||||
)
|
||||
|
||||
func TestDMARCDB(t *testing.T) {
|
||||
mox.ConfigStaticPath = "../testdata/dmarcdb/fake.conf"
|
||||
mox.Conf.Static.DataDir = "."
|
||||
|
||||
dbpath := mox.DataDirPath("dmarcrpt.db")
|
||||
os.MkdirAll(filepath.Dir(dbpath), 0770)
|
||||
defer os.Remove(dbpath)
|
||||
|
||||
if err := Init(); err != nil {
|
||||
t.Fatalf("init database: %s", err)
|
||||
}
|
||||
|
||||
feedback := &dmarcrpt.Feedback{
|
||||
ReportMetadata: dmarcrpt.ReportMetadata{
|
||||
OrgName: "google.com",
|
||||
Email: "noreply-dmarc-support@google.com",
|
||||
ExtraContactInfo: "https://support.google.com/a/answer/2466580",
|
||||
ReportID: "10051505501689795560",
|
||||
DateRange: dmarcrpt.DateRange{
|
||||
Begin: 1596412800,
|
||||
End: 1596499199,
|
||||
},
|
||||
},
|
||||
PolicyPublished: dmarcrpt.PolicyPublished{
|
||||
Domain: "example.org",
|
||||
ADKIM: "r",
|
||||
ASPF: "r",
|
||||
Policy: "reject",
|
||||
SubdomainPolicy: "reject",
|
||||
Percentage: 100,
|
||||
},
|
||||
Records: []dmarcrpt.ReportRecord{
|
||||
{
|
||||
Row: dmarcrpt.Row{
|
||||
SourceIP: "127.0.0.1",
|
||||
Count: 1,
|
||||
PolicyEvaluated: dmarcrpt.PolicyEvaluated{
|
||||
Disposition: dmarcrpt.DispositionNone,
|
||||
DKIM: dmarcrpt.DMARCPass,
|
||||
SPF: dmarcrpt.DMARCPass,
|
||||
},
|
||||
},
|
||||
Identifiers: dmarcrpt.Identifiers{
|
||||
HeaderFrom: "example.org",
|
||||
},
|
||||
AuthResults: dmarcrpt.AuthResults{
|
||||
DKIM: []dmarcrpt.DKIMAuthResult{
|
||||
{
|
||||
Domain: "example.org",
|
||||
Result: dmarcrpt.DKIMPass,
|
||||
Selector: "example",
|
||||
},
|
||||
},
|
||||
SPF: []dmarcrpt.SPFAuthResult{
|
||||
{
|
||||
Domain: "example.org",
|
||||
Result: dmarcrpt.SPFPass,
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
if err := AddReport(context.Background(), feedback, dns.Domain{ASCII: "google.com"}); err != nil {
|
||||
t.Fatalf("adding report: %s", err)
|
||||
}
|
||||
|
||||
records, err := Records(context.Background())
|
||||
if err != nil || len(records) != 1 || !reflect.DeepEqual(&records[0].Feedback, feedback) {
|
||||
t.Fatalf("records: got err %v, records %#v, expected no error, single record with feedback %#v", err, records, feedback)
|
||||
}
|
||||
|
||||
record, err := RecordID(context.Background(), records[0].ID)
|
||||
if err != nil || !reflect.DeepEqual(&record.Feedback, feedback) {
|
||||
t.Fatalf("record id: got err %v, record %#v, expected feedback %#v", err, record, feedback)
|
||||
}
|
||||
|
||||
start := time.Unix(1596412800, 0)
|
||||
end := time.Unix(1596499199, 0)
|
||||
records, err = RecordsPeriodDomain(context.Background(), start, end, "example.org")
|
||||
if err != nil || len(records) != 1 || !reflect.DeepEqual(&records[0].Feedback, feedback) {
|
||||
t.Fatalf("records: got err %v, records %#v, expected no error, single record with feedback %#v", err, records, feedback)
|
||||
}
|
||||
|
||||
records, err = RecordsPeriodDomain(context.Background(), end, end, "example.org")
|
||||
if err != nil || len(records) != 0 {
|
||||
t.Fatalf("records: got err %v, records %#v, expected no error and no records", err, records)
|
||||
}
|
||||
records, err = RecordsPeriodDomain(context.Background(), start, end, "other.example")
|
||||
if err != nil || len(records) != 0 {
|
||||
t.Fatalf("records: got err %v, records %#v, expected no error and no records", err, records)
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user