57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
// Copyright (c) HashiCorp, Inc.
|
|
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
//go:build linux
|
|
|
|
package testutil
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/opencontainers/runc/libcontainer/cgroups"
|
|
)
|
|
|
|
// CgroupsCompatible returns true if either cgroups.v1 or cgroups.v2 is supported.
|
|
func CgroupsCompatible(t *testing.T) bool {
|
|
return cgroupsCompatibleV1(t) || cgroupsCompatibleV2(t)
|
|
}
|
|
|
|
// CgroupsCompatibleV1 skips tests unless:
|
|
// - cgroup.v1 mount point is detected
|
|
func CgroupsCompatibleV1(t *testing.T) {
|
|
if !cgroupsCompatibleV1(t) {
|
|
t.Skipf("Test requires cgroup.v1 support")
|
|
}
|
|
}
|
|
|
|
func cgroupsCompatibleV1(t *testing.T) bool {
|
|
// build tags mean this will never run outside of linux
|
|
|
|
if cgroupsCompatibleV2(t) {
|
|
t.Log("No cgroup.v1 mount point: running in cgroup.v2 mode")
|
|
return false
|
|
}
|
|
mount, err := cgroups.GetCgroupMounts(false)
|
|
if err != nil {
|
|
t.Logf("Unable to detect cgroup.v1 mount point: %v", err)
|
|
return false
|
|
}
|
|
if len(mount) == 0 {
|
|
t.Logf("No cgroup.v1 mount point: empty path")
|
|
return false
|
|
}
|
|
return true
|
|
}
|
|
|
|
// CgroupsCompatibleV2 skips tests unless:
|
|
// - cgroup.v2 unified mode is detected
|
|
func CgroupsCompatibleV2(t *testing.T) {
|
|
if !cgroupsCompatibleV2(t) {
|
|
t.Skip("Test requires cgroup.v2 support")
|
|
}
|
|
}
|
|
|
|
func cgroupsCompatibleV2(t *testing.T) bool {
|
|
return cgroups.IsCgroup2UnifiedMode()
|
|
}
|