1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
package utils_test
import (
"os/exec"
"runtime"
"strconv"
"strings"
"testing"
"github.com/ansible/receptor/pkg/utils"
)
func TestGetSysCPUCount(t *testing.T) {
got := utils.GetSysCPUCount()
if got <= 0 {
t.Errorf("Non-positive CPU count: %d\n", got)
}
if runtime.GOOS == "linux" {
commandOutput, _ := exec.Command("nproc").CombinedOutput()
commandOutputWithout := strings.TrimSpace(string(commandOutput))
want, _ := strconv.Atoi(commandOutputWithout)
if got != want {
t.Errorf("Expected CPU count: %d, got %d\n", want, got)
}
}
}
func TestGetSysMemoryMiB(t *testing.T) {
got := utils.GetSysMemoryMiB()
if got <= 0 {
t.Errorf("Non-positive Memory: %d\n", got)
}
if runtime.GOOS == "linux" {
commandOutput, _ := exec.Command("sed", "-n", "s/^MemTotal:[[:space:]]*\\([[:digit:]]*\\).*/\\1/p", "/proc/meminfo").CombinedOutput()
commandOutputWithout := strings.TrimSpace(string(commandOutput))
wantKb, _ := strconv.ParseUint(commandOutputWithout, 10, 64)
want := wantKb / 1024
if got != want {
t.Errorf("Expected Memory: %d, got %d\n", want, got)
}
}
}
|
