cgroups-rs/tests/hugetlb.rs

//! Integration tests about the hugetlb subsystem
use cgroups::hugetlb::HugeTlbController;
use cgroups::{Cgroup, Hierarchy};
use cgroups::Controller;

use cgroups::error::ErrorKind::*;
use cgroups::error::*;

#[test]
fn test_hugetlb_sizes() {
    // now only v2
    if cgroups::hierarchies::is_cgroup2_unified_mode() {
        return
    }

    let h = cgroups::hierarchies::auto();
    let h = Box::new(&*h);
    let cg = Cgroup::new(h, String::from("test_hugetlb_sizes"));
    {
        let hugetlb_controller: &HugeTlbController = cg.controller_of().unwrap();
        let sizes = hugetlb_controller.get_sizes();

        let size = "2MB";
        assert_eq!(sizes, vec![size.to_string()]);

        let supported = hugetlb_controller.size_supported(size);
        assert_eq!(supported, true);

        assert_no_error(hugetlb_controller.failcnt(size));
        assert_no_error(hugetlb_controller.limit_in_bytes(size));
        assert_no_error(hugetlb_controller.usage_in_bytes(size));
        assert_no_error(hugetlb_controller.max_usage_in_bytes(size));
    }
    cg.delete();
}

fn assert_no_error(r: Result<u64>) {
    assert_eq!(!r.is_err(), true)
}