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

services/protocols/internal/infrastructure/clamav/clamav_adapter_test.go

@@ -0,0 +1,70 @@
+package clamav
+
+import (
+	"context"
+	"net"
+	"testing"
+
+	"lambdamail/protocols/internal/application/port"
+	"lambdamail/protocols/internal/domain/valueobject"
+)
+
+func TestClamAVAdapter_Scan_CleanAndVirus(t *testing.T) {
+	// Clean mock server
+	cleanLn, err := net.Listen("tcp", "127.0.0.1:0")
+	if err != nil {
+		t.Fatalf("listen clean server: %v", err)
+	}
+	defer cleanLn.Close()
+
+	go func() {
+		conn, err := cleanLn.Accept()
+		if err != nil {
+			return
+		}
+		defer conn.Close()
+		buf := make([]byte, 4096)
+		for {
+			_, err := conn.Read(buf)
+			if err != nil {
+				break
+			}
+		}
+		conn.Write([]byte("stream: OK\x00"))
+	}()
+
+	adapter := NewClamAVAdapter(cleanLn.Addr().String())
+	res, err := adapter.Scan(context.Background(), port.ScanInput{Payload: []byte("clean content")})
+	if err != nil || res.Verdict != valueobject.ScanVerdictClean {
+		t.Fatalf("expected clean scan, got res=%+v, err=%v", res, err)
+	}
+
+	// Virus mock server
+	virusLn, err := net.Listen("tcp", "127.0.0.1:0")
+	if err != nil {
+		t.Fatalf("listen virus server: %v", err)
+	}
+	defer virusLn.Close()
+
+	go func() {
+		conn, err := virusLn.Accept()
+		if err != nil {
+			return
+		}
+		defer conn.Close()
+		buf := make([]byte, 4096)
+		for {
+			_, err := conn.Read(buf)
+			if err != nil {
+				break
+			}
+		}
+		conn.Write([]byte("stream: EICAR-Test-Signature FOUND\x00"))
+	}()
+
+	virusAdapter := NewClamAVAdapter(virusLn.Addr().String())
+	res, err = virusAdapter.Scan(context.Background(), port.ScanInput{Payload: []byte("virus content")})
+	if err != nil || res.Verdict != valueobject.ScanVerdictVirusReject || res.VirusName != "EICAR-Test-Signature" {
+		t.Fatalf("expected virus scan, got res=%+v, err=%v", res, err)
+	}
+}