7b2d41cda8
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
46 lines
1.0 KiB
Go
46 lines
1.0 KiB
Go
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)
|
|
}
|
|
}
|