mirror of
https://github.com/shirou/gopsutil.git
synced 2025-04-29 13:49:21 +08:00
f7e1f36418
Executing the command does the lookup if needed and returns the same error when not found, no need to do it separately.
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)
|
|
}
|
|
}
|