60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package stats
|
|
|
|
import (
|
|
"math"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
shelpers "github.com/hashicorp/nomad/helper/stats"
|
|
"github.com/hashicorp/nomad/helper/testlog"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCpuStatsPercent(t *testing.T) {
|
|
cs := NewCpuStats()
|
|
cs.Percent(79.7)
|
|
time.Sleep(1 * time.Second)
|
|
percent := cs.Percent(80.69)
|
|
expectedPercent := 98.00
|
|
if percent < expectedPercent && percent > (expectedPercent+1.00) {
|
|
t.Fatalf("expected: %v, actual: %v", expectedPercent, percent)
|
|
}
|
|
}
|
|
|
|
func TestHostStats_CPU(t *testing.T) {
|
|
assert := assert.New(t)
|
|
assert.Nil(shelpers.Init())
|
|
|
|
logger := testlog.Logger(t)
|
|
cwd, err := os.Getwd()
|
|
assert.Nil(err)
|
|
hs := NewHostStatsCollector(logger, cwd)
|
|
|
|
// Collect twice so we can calculate percents we need to generate some work
|
|
// so that the cpu values change
|
|
assert.Nil(hs.Collect())
|
|
total := 0
|
|
for i := 1; i < 1000000000; i++ {
|
|
total *= i
|
|
total = total % i
|
|
}
|
|
assert.Nil(hs.Collect())
|
|
stats := hs.Stats()
|
|
|
|
assert.NotZero(stats.CPUTicksConsumed)
|
|
assert.NotZero(len(stats.CPU))
|
|
|
|
for _, cpu := range stats.CPU {
|
|
assert.False(math.IsNaN(cpu.Idle))
|
|
assert.False(math.IsNaN(cpu.Total))
|
|
assert.False(math.IsNaN(cpu.System))
|
|
assert.False(math.IsNaN(cpu.User))
|
|
|
|
assert.False(math.IsInf(cpu.Idle, 0))
|
|
assert.False(math.IsInf(cpu.Total, 0))
|
|
assert.False(math.IsInf(cpu.System, 0))
|
|
assert.False(math.IsInf(cpu.User, 0))
|
|
}
|
|
}
|