mirror of
https://github.com/kata-containers/cgroups-rs.git
synced 2026-08-05 02:13:23 +00:00
c9d02afe33
Support to set threaded mode in cgroup v2. The premise of switching to threaded mode is that only the cgroup of cpuset, cpu and pids is supported. Fixes: #90 Signed-off-by: yaoyinnan <yaoyinnan@foxmail.com>
107 lines
3.2 KiB
Rust
107 lines
3.2 KiB
Rust
// Copyright (c) 2020 And Group
|
|
//
|
|
// SPDX-License-Identifier: Apache-2.0 or MIT
|
|
//
|
|
|
|
//! Integration tests about the hugetlb subsystem
|
|
use cgroups_rs::memory::{MemController, SetMemory};
|
|
use cgroups_rs::Controller;
|
|
use cgroups_rs::{Cgroup, MaxValue};
|
|
|
|
#[test]
|
|
fn test_disable_oom_killer() {
|
|
let h = cgroups_rs::hierarchies::auto();
|
|
let cg = Cgroup::new(h, String::from("test_disable_oom_killer")).unwrap();
|
|
{
|
|
let mem_controller: &MemController = cg.controller_of().unwrap();
|
|
|
|
// before disable
|
|
let m = mem_controller.memory_stat();
|
|
assert!(!m.oom_control.oom_kill_disable);
|
|
|
|
// now only v1
|
|
if !mem_controller.v2() {
|
|
// disable oom killer
|
|
let r = mem_controller.disable_oom_killer();
|
|
assert!(r.is_ok());
|
|
|
|
// after disable
|
|
let m = mem_controller.memory_stat();
|
|
assert!(m.oom_control.oom_kill_disable);
|
|
}
|
|
}
|
|
cg.delete().unwrap();
|
|
}
|
|
|
|
#[test]
|
|
fn set_kmem_limit_v1() {
|
|
let h = cgroups_rs::hierarchies::auto();
|
|
if h.v2() {
|
|
return;
|
|
}
|
|
|
|
let cg = Cgroup::new(h, String::from("set_kmem_limit_v1")).unwrap();
|
|
{
|
|
let mem_controller: &MemController = cg.controller_of().unwrap();
|
|
mem_controller.set_kmem_limit(1).unwrap();
|
|
}
|
|
cg.delete().unwrap();
|
|
}
|
|
|
|
#[test]
|
|
fn set_mem_v2() {
|
|
let h = cgroups_rs::hierarchies::auto();
|
|
if !h.v2() {
|
|
return;
|
|
}
|
|
|
|
let cg = Cgroup::new(h, String::from("set_mem_v2")).unwrap();
|
|
{
|
|
let mem_controller: &MemController = cg.controller_of().unwrap();
|
|
|
|
// before disable
|
|
let m = mem_controller.get_mem().unwrap();
|
|
// case 1: get default value
|
|
assert_eq!(m.low, Some(MaxValue::Value(0)));
|
|
assert_eq!(m.min, Some(MaxValue::Value(0)));
|
|
assert_eq!(m.high, Some(MaxValue::Max));
|
|
assert_eq!(m.max, Some(MaxValue::Max));
|
|
|
|
// case 2: set parts
|
|
let m = SetMemory {
|
|
low: Some(MaxValue::Value(1024 * 1024 * 2)),
|
|
high: Some(MaxValue::Value(1024 * 1024 * 1024 * 2)),
|
|
min: Some(MaxValue::Value(1024 * 1024 * 3)),
|
|
max: None,
|
|
};
|
|
let r = mem_controller.set_mem(m);
|
|
assert!(r.is_ok());
|
|
|
|
let m = mem_controller.get_mem().unwrap();
|
|
// get
|
|
assert_eq!(m.low, Some(MaxValue::Value(1024 * 1024 * 2)));
|
|
assert_eq!(m.min, Some(MaxValue::Value(1024 * 1024 * 3)));
|
|
assert_eq!(m.high, Some(MaxValue::Value(1024 * 1024 * 1024 * 2)));
|
|
assert_eq!(m.max, Some(MaxValue::Max));
|
|
|
|
// case 3: set parts
|
|
let m = SetMemory {
|
|
max: Some(MaxValue::Value(1024 * 1024 * 1024 * 2)),
|
|
min: Some(MaxValue::Value(1024 * 1024 * 4)),
|
|
high: Some(MaxValue::Max),
|
|
low: None,
|
|
};
|
|
let r = mem_controller.set_mem(m);
|
|
assert!(r.is_ok());
|
|
|
|
let m = mem_controller.get_mem().unwrap();
|
|
// get
|
|
assert_eq!(m.low, Some(MaxValue::Value(1024 * 1024 * 2)));
|
|
assert_eq!(m.min, Some(MaxValue::Value(1024 * 1024 * 4)));
|
|
assert_eq!(m.max, Some(MaxValue::Value(1024 * 1024 * 1024 * 2)));
|
|
assert_eq!(m.high, Some(MaxValue::Max));
|
|
}
|
|
|
|
cg.delete().unwrap();
|
|
}
|