This commit adds unit tests for cgroupv2_controller_enabled() [----------] 4 tests from CgroupV2ControllerEnabled [ RUN ] CgroupV2ControllerEnabled.CgroupV1Controller [ OK ] CgroupV2ControllerEnabled.CgroupV1Controller (0 ms) [ RUN ] CgroupV2ControllerEnabled.RootCgroup [ OK ] CgroupV2ControllerEnabled.RootCgroup (1 ms) [ RUN ] CgroupV2ControllerEnabled.ControllerEnabled [ OK ] CgroupV2ControllerEnabled.ControllerEnabled (0 ms) [ RUN ] CgroupV2ControllerEnabled.ControllerDisabled [ OK ] CgroupV2ControllerEnabled.ControllerDisabled (0 ms) [----------] 4 tests from CgroupV2ControllerEnabled (1 ms total)
Signed-off-by: Tom Hromatka <tom.hroma...@oracle.com> --- gunit/015-cgroupv2_controller_enabled.cpp | 185 ++++++++++++++++++++++ gunit/Makefile.am | 3 +- 2 files changed, 187 insertions(+), 1 deletion(-) create mode 100644 gunit/015-cgroupv2_controller_enabled.cpp diff --git a/gunit/015-cgroupv2_controller_enabled.cpp b/gunit/015-cgroupv2_controller_enabled.cpp new file mode 100644 index 000000000000..3c291e5a70b4 --- /dev/null +++ b/gunit/015-cgroupv2_controller_enabled.cpp @@ -0,0 +1,185 @@ +/** + * libcgroup googletest for cgroupv2_controller_enabled() + * + * Copyright (c) 2020 Oracle and/or its affiliates. + * Author: Tom Hromatka <tom.hroma...@oracle.com> + */ + +/* + * This library is free software; you can redistribute it and/or modify it + * under the terms of version 2.1 of the GNU Lesser General Public License as + * published by the Free Software Foundation. + * + * This library is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License + * for more details. + * + * You should have received a copy of the GNU Lesser General Public License + * along with this library; if not, see <http://www.gnu.org/licenses>. + */ + +#include <ftw.h> +#include <sys/stat.h> +#include <sys/types.h> + +#include "gtest/gtest.h" + +#include "libcgroup-internal.h" + +static const char * const PARENT_DIR = "test015cgroup"; +static const mode_t MODE = S_IRWXU | S_IRWXG | S_IRWXO; + +static const char * const CHILD_DIRS[] = { + "test1-v1cgroup", + "test2-rootcgroup", + "test3-ctrlrenabled", + "test4-ctrlrdisabled", +}; +static const int CHILD_DIRS_CNT = + sizeof(CHILD_DIRS) / sizeof(CHILD_DIRS[0]); + +static const char * const CONTROLLERS[] = { + "cpu", + "cpuset", + "io", + "memory", + "net_cls", + "pids", +}; +static const int CONTROLLERS_CNT = + sizeof(CONTROLLERS) / sizeof(CONTROLLERS[0]); + +static const enum cg_version_t VERSIONS[] = { + CGROUP_V2, + CGROUP_V1, + CGROUP_V2, + CGROUP_V2, + CGROUP_V2, + CGROUP_V2, +}; +static const int VERSIONS_CNT = + sizeof(VERSIONS) / sizeof(VERSIONS[0]); + +class CgroupV2ControllerEnabled : public ::testing::Test { + protected: + + void InitChildDir(const char dirname[]) + { + char tmp_path[FILENAME_MAX] = {0}; + int ret; + + /* create the directory */ + snprintf(tmp_path, FILENAME_MAX - 1, "%s/%s", + PARENT_DIR, dirname); + ret = mkdir(tmp_path, MODE); + ASSERT_EQ(ret, 0); + } + + void InitMountTable(void) + { + char tmp_path[FILENAME_MAX] = {0}; + int ret, i; + FILE *f; + + ASSERT_EQ(VERSIONS_CNT, CONTROLLERS_CNT); + + snprintf(tmp_path, FILENAME_MAX - 1, + "%s/cgroup.subtree_control", PARENT_DIR); + + f = fopen(tmp_path, "w"); + ASSERT_NE(f, nullptr); + fprintf(f, "cpu io memory pids\n"); + fclose(f); + + /* + * Artificially populate the mount table with local + * directories + */ + memset(&cg_mount_table, 0, sizeof(cg_mount_table)); + memset(&cg_namespace_table, 0, sizeof(cg_namespace_table)); + + for (i = 0; i < CONTROLLERS_CNT; i++) { + snprintf(cg_mount_table[i].name, FILENAME_MAX, + "%s", CONTROLLERS[i]); + snprintf(cg_mount_table[i].mount.path, FILENAME_MAX, + "%s", PARENT_DIR); + cg_mount_table[i].version = VERSIONS[i]; + } + } + + void SetUp() override + { + int ret, i; + + ret = mkdir(PARENT_DIR, MODE); + ASSERT_EQ(ret, 0); + + InitMountTable(); + + for (i = 0; i < CHILD_DIRS_CNT; i++) + InitChildDir(CHILD_DIRS[i]); + } + + /* + * https://stackoverflow.com/questions/5467725/how-to-delete-a-directory-and-its-contents-in-posix-c + */ + static int unlink_cb(const char *fpath, const struct stat *sb, int typeflag, + struct FTW *ftwbuf) + { + return remove(fpath); + } + + int rmrf(const char * const path) + { + return nftw(path, unlink_cb, 64, FTW_DEPTH | FTW_PHYS); + } + + void TearDown() override + { + int ret = 0; + + ret = rmrf(PARENT_DIR); + ASSERT_EQ(ret, 0); + } +}; + +TEST_F(CgroupV2ControllerEnabled, CgroupV1Controller) +{ + char ctrlr_name[] = "cpuset"; + char cg_name[] = "foo"; + int ret; + + ret = cgroupv2_controller_enabled(cg_name, ctrlr_name); + ASSERT_EQ(ret, 0); +} + +TEST_F(CgroupV2ControllerEnabled, RootCgroup) +{ + char ctrlr_name[] = "cpu"; + char cg_name[] = "/"; + int ret; + + ret = cgroupv2_controller_enabled(cg_name, ctrlr_name); + ASSERT_EQ(ret, 0); +} + +TEST_F(CgroupV2ControllerEnabled, ControllerEnabled) +{ + char ctrlr_name[] = "pids"; + char cg_name[] = "test3-ctrlrenabled"; + int ret; + + ret = cgroupv2_controller_enabled(cg_name, ctrlr_name); + ASSERT_EQ(ret, 0); +} + +TEST_F(CgroupV2ControllerEnabled, ControllerDisabled) +{ + char ctrlr_name[] = "net_cls"; + char cg_name[] = "test4-ctrlrdisabled"; + int ret; + + ret = cgroupv2_controller_enabled(cg_name, ctrlr_name); + ASSERT_EQ(ret, ECGROUPNOTMOUNTED); +} diff --git a/gunit/Makefile.am b/gunit/Makefile.am index 912f2182d8d2..6eb8ffce6d1e 100644 --- a/gunit/Makefile.am +++ b/gunit/Makefile.am @@ -51,6 +51,7 @@ gtest_SOURCES = gtest.cpp \ 011-cgroupv2_subtree_control.cpp \ 012-cgroup_create_cgroup.cpp \ 013-cgroup_build_tasks_procs_path.cpp \ - 014-cgroupv2_get_subtree_control.cpp + 014-cgroupv2_get_subtree_control.cpp \ + 015-cgroupv2_controller_enabled.cpp gtest_LDFLAGS = -L$(top_builddir)/googletest/googletest -l:libgtest.so \ -rpath $(abs_top_builddir)/googletest/googletest -- 2.25.4 _______________________________________________ Libcg-devel mailing list Libcg-devel@lists.sourceforge.net https://lists.sourceforge.net/lists/listinfo/libcg-devel
