github.com/zhuohuang-hust/src-cbuild@v0.0.0-20230105071821-c7aab3e7c840/mergeCode/runc/libcontainer/cgroups/fs/cpuset_test.go (about) 1 // +build linux 2 3 package fs 4 5 import ( 6 "testing" 7 ) 8 9 func TestCpusetSetCpus(t *testing.T) { 10 helper := NewCgroupTestUtil("cpuset", t) 11 defer helper.cleanup() 12 13 const ( 14 cpusBefore = "0" 15 cpusAfter = "1-3" 16 ) 17 18 helper.writeFileContents(map[string]string{ 19 "cpuset.cpus": cpusBefore, 20 }) 21 22 helper.CgroupData.config.Resources.CpusetCpus = cpusAfter 23 cpuset := &CpusetGroup{} 24 if err := cpuset.Set(helper.CgroupPath, helper.CgroupData.config); err != nil { 25 t.Fatal(err) 26 } 27 28 value, err := getCgroupParamString(helper.CgroupPath, "cpuset.cpus") 29 if err != nil { 30 t.Fatalf("Failed to parse cpuset.cpus - %s", err) 31 } 32 33 if value != cpusAfter { 34 t.Fatal("Got the wrong value, set cpuset.cpus failed.") 35 } 36 } 37 38 func TestCpusetSetMems(t *testing.T) { 39 helper := NewCgroupTestUtil("cpuset", t) 40 defer helper.cleanup() 41 42 const ( 43 memsBefore = "0" 44 memsAfter = "1" 45 ) 46 47 helper.writeFileContents(map[string]string{ 48 "cpuset.mems": memsBefore, 49 }) 50 51 helper.CgroupData.config.Resources.CpusetMems = memsAfter 52 cpuset := &CpusetGroup{} 53 if err := cpuset.Set(helper.CgroupPath, helper.CgroupData.config); err != nil { 54 t.Fatal(err) 55 } 56 57 value, err := getCgroupParamString(helper.CgroupPath, "cpuset.mems") 58 if err != nil { 59 t.Fatalf("Failed to parse cpuset.mems - %s", err) 60 } 61 62 if value != memsAfter { 63 t.Fatal("Got the wrong value, set cpuset.mems failed.") 64 } 65 }