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  }