109 lines
2.6 KiB
Go
109 lines
2.6 KiB
Go
package cpu
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"os/exec"
|
|
"strconv"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestTimesEmpty(t *testing.T) {
|
|
orig := os.Getenv("HOST_PROC")
|
|
os.Setenv("HOST_PROC", "testdata/linux/times_empty")
|
|
_, err := Times(true)
|
|
if err != nil {
|
|
t.Error("Times(true) failed")
|
|
}
|
|
_, err = Times(false)
|
|
if err != nil {
|
|
t.Error("Times(false) failed")
|
|
}
|
|
os.Setenv("HOST_PROC", orig)
|
|
}
|
|
|
|
func TestCPUparseStatLine_424(t *testing.T) {
|
|
orig := os.Getenv("HOST_PROC")
|
|
os.Setenv("HOST_PROC", "testdata/linux/424/proc")
|
|
{
|
|
l, err := Times(true)
|
|
if err != nil || len(l) == 0 {
|
|
t.Error("Times(true) failed")
|
|
}
|
|
t.Logf("Times(true): %#v", l)
|
|
}
|
|
{
|
|
l, err := Times(false)
|
|
if err != nil || len(l) == 0 {
|
|
t.Error("Times(false) failed")
|
|
}
|
|
t.Logf("Times(false): %#v", l)
|
|
}
|
|
os.Setenv("HOST_PROC", orig)
|
|
}
|
|
|
|
func TestCPUCountsAgainstLscpu(t *testing.T) {
|
|
cmd := exec.Command("lscpu")
|
|
cmd.Env = []string{"LC_ALL=C"}
|
|
out, err := cmd.Output()
|
|
if err != nil {
|
|
if errors.Is(err, exec.ErrNotFound) {
|
|
t.Skip("no lscpu to compare with")
|
|
}
|
|
t.Errorf("error executing lscpu: %v", err)
|
|
}
|
|
var threadsPerCore, coresPerSocket, sockets int
|
|
lines := strings.Split(string(out), "\n")
|
|
for _, line := range lines {
|
|
fields := strings.Split(line, ":")
|
|
if len(fields) < 2 {
|
|
continue
|
|
}
|
|
switch fields[0] {
|
|
case "Thread(s) per core":
|
|
threadsPerCore, _ = strconv.Atoi(strings.TrimSpace(fields[1]))
|
|
case "Core(s) per socket":
|
|
coresPerSocket, _ = strconv.Atoi(strings.TrimSpace(fields[1]))
|
|
case "Socket(s)":
|
|
sockets, _ = strconv.Atoi(strings.TrimSpace(fields[1]))
|
|
}
|
|
}
|
|
if threadsPerCore == 0 || coresPerSocket == 0 || sockets == 0 {
|
|
t.Errorf("missing info from lscpu: threadsPerCore=%d coresPerSocket=%d sockets=%d", threadsPerCore, coresPerSocket, sockets)
|
|
}
|
|
expectedPhysical := coresPerSocket * sockets
|
|
expectedLogical := expectedPhysical * threadsPerCore
|
|
physical, err := Counts(false)
|
|
skipIfNotImplementedErr(t, err)
|
|
if err != nil {
|
|
t.Errorf("error %v", err)
|
|
}
|
|
logical, err := Counts(true)
|
|
skipIfNotImplementedErr(t, err)
|
|
if err != nil {
|
|
t.Errorf("error %v", err)
|
|
}
|
|
if expectedPhysical != physical {
|
|
t.Errorf("expected %v, got %v", expectedPhysical, physical)
|
|
}
|
|
if expectedLogical != logical {
|
|
t.Errorf("expected %v, got %v", expectedLogical, logical)
|
|
}
|
|
}
|
|
|
|
func TestCPUCountsLogicalAndroid_1037(t *testing.T) { // https://github.com/shirou/gopsutil/issues/1037
|
|
orig := os.Getenv("HOST_PROC")
|
|
os.Setenv("HOST_PROC", "testdata/linux/1037/proc")
|
|
defer os.Setenv("HOST_PROC", orig)
|
|
|
|
count, err := Counts(true)
|
|
if err != nil {
|
|
t.Errorf("error %v", err)
|
|
}
|
|
expected := 8
|
|
if count != expected {
|
|
t.Errorf("expected %v, got %v", expected, count)
|
|
}
|
|
}
|