#12: implement Antispam & Antivirus scanning pipeline (Rspamd & ClamAV)

services/protocols/internal/application/usecase/scanning_pipeline_test.go

@@ -0,0 +1,51 @@
+package usecase
+
+import (
+	"context"
+	"testing"
+
+	"lambdamail/protocols/internal/application/port"
+	"lambdamail/protocols/internal/domain/valueobject"
+)
+
+type mockScanner struct {
+	res *valueobject.ScanResult
+	err error
+}
+
+func (m *mockScanner) Scan(_ context.Context, _ port.ScanInput) (*valueobject.ScanResult, error) {
+	return m.res, m.err
+}
+
+func TestScanningPipeline_VirusRejectsImmediately(t *testing.T) {
+	s1 := &mockScanner{res: valueobject.NewVirusScanResult("EICAR-Test-Signature")}
+	s2 := &mockScanner{res: valueobject.NewCleanScanResult()}
+
+	pipeline := NewScanningPipeline(s1, s2)
+	res, err := pipeline.Scan(context.Background(), port.ScanInput{})
+	if err != nil {
+		t.Fatalf("Scan: %v", err)
+	}
+
+	if res.Verdict != valueobject.ScanVerdictVirusReject || res.VirusName != "EICAR-Test-Signature" {
+		t.Errorf("unexpected verdict: %+v", res)
+	}
+}
+
+func TestScanningPipeline_AggregatesJunkVerdict(t *testing.T) {
+	s1 := &mockScanner{res: valueobject.NewCleanScanResult()}
+	junkRes := valueobject.NewCleanScanResult()
+	junkRes.Verdict = valueobject.ScanVerdictSpamJunk
+	junkRes.Score = 7.5
+	s2 := &mockScanner{res: junkRes}
+
+	pipeline := NewScanningPipeline(s1, s2)
+	res, err := pipeline.Scan(context.Background(), port.ScanInput{})
+	if err != nil {
+		t.Fatalf("Scan: %v", err)
+	}
+
+	if res.Verdict != valueobject.ScanVerdictSpamJunk || res.Score != 7.5 {
+		t.Errorf("unexpected junk result: %+v", res)
+	}
+}