mirror of
https://github.com/genuinetools/reg.git
synced 2024-08-21 06:10:23 -04:00
66 lines
1.3 KiB
Go
66 lines
1.3 KiB
Go
// +build linux
|
|
|
|
package fs
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestCpusetSetCpus(t *testing.T) {
|
|
helper := NewCgroupTestUtil("cpuset", t)
|
|
defer helper.cleanup()
|
|
|
|
const (
|
|
cpusBefore = "0"
|
|
cpusAfter = "1-3"
|
|
)
|
|
|
|
helper.writeFileContents(map[string]string{
|
|
"cpuset.cpus": cpusBefore,
|
|
})
|
|
|
|
helper.CgroupData.config.Resources.CpusetCpus = cpusAfter
|
|
cpuset := &CpusetGroup{}
|
|
if err := cpuset.Set(helper.CgroupPath, helper.CgroupData.config); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
value, err := getCgroupParamString(helper.CgroupPath, "cpuset.cpus")
|
|
if err != nil {
|
|
t.Fatalf("Failed to parse cpuset.cpus - %s", err)
|
|
}
|
|
|
|
if value != cpusAfter {
|
|
t.Fatal("Got the wrong value, set cpuset.cpus failed.")
|
|
}
|
|
}
|
|
|
|
func TestCpusetSetMems(t *testing.T) {
|
|
helper := NewCgroupTestUtil("cpuset", t)
|
|
defer helper.cleanup()
|
|
|
|
const (
|
|
memsBefore = "0"
|
|
memsAfter = "1"
|
|
)
|
|
|
|
helper.writeFileContents(map[string]string{
|
|
"cpuset.mems": memsBefore,
|
|
})
|
|
|
|
helper.CgroupData.config.Resources.CpusetMems = memsAfter
|
|
cpuset := &CpusetGroup{}
|
|
if err := cpuset.Set(helper.CgroupPath, helper.CgroupData.config); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
value, err := getCgroupParamString(helper.CgroupPath, "cpuset.mems")
|
|
if err != nil {
|
|
t.Fatalf("Failed to parse cpuset.mems - %s", err)
|
|
}
|
|
|
|
if value != memsAfter {
|
|
t.Fatal("Got the wrong value, set cpuset.mems failed.")
|
|
}
|
|
}
|