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 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72
|
package main
import (
"fmt"
"runtime"
"github.com/mndrix/tap-go"
"github.com/opencontainers/runtime-tools/cgroups"
"github.com/opencontainers/runtime-tools/validation/util"
)
func testBlkioCgroups(rate uint64, isEmpty bool) error {
var weight uint16 = 500
var leafWeight uint16 = 300
// It's assumed that a device /dev/sda (8:0) exists on the test system.
// The minor number must be always 0, as it's only allowed to set blkio
// weights to a whole block device /dev/sda, not to partitions like sda1.
var major int64 = 8
var minor int64
t := tap.New()
t.Header(0)
defer t.AutoPlan()
g, err := util.GetDefaultGenerator()
if err != nil {
return err
}
g.SetLinuxCgroupsPath(cgroups.AbsCgroupPath)
if !isEmpty {
g.SetLinuxResourcesBlockIOWeight(weight)
g.SetLinuxResourcesBlockIOLeafWeight(leafWeight)
}
g.AddLinuxResourcesBlockIOWeightDevice(major, minor, weight)
g.AddLinuxResourcesBlockIOLeafWeightDevice(major, minor, leafWeight)
g.AddLinuxResourcesBlockIOThrottleReadBpsDevice(major, minor, rate)
g.AddLinuxResourcesBlockIOThrottleWriteBpsDevice(major, minor, rate)
g.AddLinuxResourcesBlockIOThrottleReadIOPSDevice(major, minor, rate)
g.AddLinuxResourcesBlockIOThrottleWriteIOPSDevice(major, minor, rate)
err = util.RuntimeOutsideValidate(g, t, util.ValidateLinuxResourcesBlockIO)
if err != nil {
return err
}
return nil
}
func main() {
if "linux" != runtime.GOOS {
util.Fatal(fmt.Errorf("linux-specific cgroup test"))
}
cases := []struct {
rate uint64
isEmpty bool
}{
{102400, false},
{204800, false},
{102400, true},
{204800, true},
}
for _, c := range cases {
if err := testBlkioCgroups(c.rate, c.isEmpty); err != nil {
util.Fatal(err)
}
}
}
|