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)
	}
}