33 lines
839 B
Go
33 lines
839 B
Go
package client
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"crypto-miner-agent/vulnprobe"
|
|
)
|
|
|
|
func TestRunVulnLOTLProbeMockedContext(t *testing.T) {
|
|
SetPostureCollector(func() *PostureReport { return nil })
|
|
defer SetPostureCollector(nil)
|
|
|
|
origProbe := vulnprobeProbeHost
|
|
vulnprobeProbeHost = func(_ map[int]bool, _ string) vulnprobe.HostContext {
|
|
return vulnprobe.HostContext{
|
|
Platform: "windows",
|
|
ExchangeInstalled: true,
|
|
LastPatchDays: 150,
|
|
ListeningPorts: map[int]bool{443: true},
|
|
}
|
|
}
|
|
defer func() { vulnprobeProbeHost = origProbe }()
|
|
|
|
report := RunVulnLOTLProbe()
|
|
if report == nil || len(report.Findings) == 0 {
|
|
t.Fatal("expected vuln findings from mocked probe")
|
|
}
|
|
cached := LastVulnScan()
|
|
if cached == nil || cached.RiskScore != report.RiskScore {
|
|
t.Fatalf("cache mismatch: %+v vs %+v", cached, report)
|
|
}
|
|
}
|