#11: implement MTA-STS, TLS-RPT & Autoconfig HTTP services

2026-07-25 10:18:03 -03:00
1 parent c0042ef207
commit 314eb67dda
@@ -0,0 +1,96 @@
+package entity
+
+import (
+	"bytes"
+	"compress/gzip"
+	"testing"
+)
+
+func TestParseDmarcXmlReport_ValidXML(t *testing.T) {
+	xmlData := `<?xml version="1.0"?>
+<feedback>
+  <report_metadata>
+    <org_name>google.com</org_name>
+    <report_id>123456</report_id>
+    <date_range>
+      <begin>1609459200</begin>
+      <end>1609545600</end>
+    </date_range>
+  </report_metadata>
+  <policy_published>
+    <domain>example.test</domain>
+  </policy_published>
+  <record>
+    <row>
+      <source_ip>198.51.100.1</source_ip>
+      <count>10</count>
+      <policy_evaluated>
+        <disposition>none</disposition>
+        <dkim>pass</dkim>
+        <spf>pass</spf>
+      </policy_evaluated>
+    </row>
+    <identifiers>
+      <header_from>example.test</header_from>
+    </identifiers>
+  </record>
+</feedback>`
+
+	report, err := ParseDmarcXmlReport([]byte(xmlData))
+	if err != nil {
+		t.Fatalf("ParseDmarcXmlReport: %v", err)
+	}
+
+	if report.OrgName != "google.com" || report.Domain != "example.test" {
+		t.Errorf("unexpected report metadata: %+v", report)
+	}
+	if len(report.Records) != 1 || report.Records[0].Count != 10 {
+		t.Errorf("unexpected records: %+v", report.Records)
+	}
+
+	// Test Gzip handling
+	var gzBuf bytes.Buffer
+	gw := gzip.NewWriter(&gzBuf)
+	gw.Write([]byte(xmlData))
+	gw.Close()
+
+	gzReport, err := ParseDmarcXmlReport(gzBuf.Bytes())
+	if err != nil || gzReport.OrgName != "google.com" {
+		t.Fatalf("ParseDmarcXmlReport gzip: %v", err)
+	}
+}
+
+func TestParseTlsRptReport_ValidJSON(t *testing.T) {
+	jsonData := `{
+		"organization-name": "Google LLC",
+		"report-id": "2026-07-25-001",
+		"date-range": {
+			"start-datetime": "2026-07-25T00:00:00Z",
+			"end-datetime": "2026-07-25T23:59:59Z"
+		},
+		"policies": [
+			{
+				"policy": {
+					"policy-type": "sts",
+					"domain": "example.test"
+				},
+				"summary": {
+					"total-successful-session-count": 100,
+					"total-failure-session-count": 0
+				}
+			}
+		]
+	}`
+
+	report, err := ParseTlsRptReport([]byte(jsonData))
+	if err != nil {
+		t.Fatalf("ParseTlsRptReport: %v", err)
+	}
+
+	if report.OrganizationName != "Google LLC" || report.Domain != "example.test" {
+		t.Errorf("unexpected tlsrpt metadata: %+v", report)
+	}
+	if len(report.Policies) != 1 || report.Policies[0].SuccessCount != 100 {
+		t.Errorf("unexpected policies: %+v", report.Policies)
+	}
+}