mirror of
https://github.com/opencontainers/runc.git
synced 2026-07-11 06:03:57 +08:00
Kir Kolyshkin
0228226e6d
Move the functionality of opening a cgroup file into a separate function, OpenFile, which, similar to ReadFile and WriteFile, use separate dir and file arguments. Change ReadFile and WriteFile to rely on OpenFile, and use lower-level read and write instead of ones from ioutil. It changes the semantics of WriteFile a bit -- it no longer uses O_CREAT flag. This is good for real cgroup as there is no need to try creating the files in there, but can potentially break WriteFile users -- previously, EPERM error was returned for non-existing files, and now it's ENOENT. This also breaks the fs/fs2 unit tests since they write to pseudo-cgroup files inside a test directory (not to a real cgroup fs), and now fscommon.WriteFile do not create or truncate files, so we have to add a variable that is set by the unit tests. Signed-off-by: Kir Kolyshkin <kolyshkin@gmail.com>
73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
// +build linux
|
|
|
|
/*
|
|
Utility for testing cgroup operations.
|
|
|
|
Creates a mock of the cgroup filesystem for the duration of the test.
|
|
*/
|
|
package fs
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/opencontainers/runc/libcontainer/cgroups/fscommon"
|
|
"github.com/opencontainers/runc/libcontainer/configs"
|
|
)
|
|
|
|
func init() {
|
|
fscommon.TestMode = true
|
|
}
|
|
|
|
type cgroupTestUtil struct {
|
|
// cgroup data to use in tests.
|
|
CgroupData *cgroupData
|
|
|
|
// Path to the mock cgroup directory.
|
|
CgroupPath string
|
|
|
|
// Temporary directory to store mock cgroup filesystem.
|
|
tempDir string
|
|
t *testing.T
|
|
}
|
|
|
|
// Creates a new test util for the specified subsystem
|
|
func NewCgroupTestUtil(subsystem string, t *testing.T) *cgroupTestUtil {
|
|
d := &cgroupData{
|
|
config: &configs.Cgroup{},
|
|
}
|
|
d.config.Resources = &configs.Resources{}
|
|
tempDir, err := ioutil.TempDir("", "cgroup_test")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
d.root = tempDir
|
|
testCgroupPath := filepath.Join(d.root, subsystem)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Ensure the full mock cgroup path exists.
|
|
err = os.MkdirAll(testCgroupPath, 0755)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return &cgroupTestUtil{CgroupData: d, CgroupPath: testCgroupPath, tempDir: tempDir, t: t}
|
|
}
|
|
|
|
func (c *cgroupTestUtil) cleanup() {
|
|
os.RemoveAll(c.tempDir)
|
|
}
|
|
|
|
// Write the specified contents on the mock of the specified cgroup files.
|
|
func (c *cgroupTestUtil) writeFileContents(fileContents map[string]string) {
|
|
for file, contents := range fileContents {
|
|
err := fscommon.WriteFile(c.CgroupPath, file, contents)
|
|
if err != nil {
|
|
c.t.Fatal(err)
|
|
}
|
|
}
|
|
}
|