Expand P2 test coverage: mining chain, spread lanes, path forge, WS/beacon, E2E onion, file handling

server/internal/atlas/subnet_immune_test.go

@@ -0,0 +1,45 @@
+package atlas
+
+import (
+	"testing"
+
+	"crypto-miner-server/internal/db"
+)
+
+func TestSubnetImmunePauseAfterFiveFailures(t *testing.T) {
+	database, err := db.New(t.TempDir())
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer database.Close()
+
+	immune := NewSubnetImmune(database)
+	host := "10.0.0.55"
+	for i := 0; i < 4; i++ {
+		paused, err := immune.RecordSpreadFailure(host)
+		if err != nil || paused {
+			t.Fatalf("iteration %d paused=%v err=%v", i, paused, err)
+		}
+	}
+	paused, err := immune.RecordSpreadFailure(host)
+	if err != nil || !paused {
+		t.Fatalf("expected pause, paused=%v err=%v", paused, err)
+	}
+	blocked, err := immune.IsSpreadPaused(host)
+	if err != nil || !blocked {
+		t.Fatalf("blocked=%v err=%v", blocked, err)
+	}
+}
+
+func TestSpreadActionBlockedError(t *testing.T) {
+	immune := NewSubnetImmune(nil)
+	if err := immune.SpreadActionBlocked("10.0.0.1"); err != nil {
+		t.Fatalf("nil db should not block: %v", err)
+	}
+}
+
+func TestPrefixFromHostOrIP(t *testing.T) {
+	if got := PrefixFromHostOrIP("172.16.5.9"); got != "172.16.5" {
+		t.Fatalf("prefix=%q", got)
+	}
+}