This commit adds unit tests for cgroupv2_subtree_control(). [----------] 2 tests from SubtreeControlTest [ RUN ] SubtreeControlTest.AddController [ OK ] SubtreeControlTest.AddController (0 ms) [ RUN ] SubtreeControlTest.RemoveController [ OK ] SubtreeControlTest.RemoveController (0 ms) [----------] 2 tests from SubtreeControlTest (0 ms total)
Signed-off-by: Tom Hromatka <tom.hroma...@oracle.com> --- gunit/011-cgroupv2_subtree_control.cpp | 125 +++++++++++++++++++++++++ gunit/Makefile.am | 3 +- 2 files changed, 127 insertions(+), 1 deletion(-) create mode 100644 gunit/011-cgroupv2_subtree_control.cpp diff --git a/gunit/011-cgroupv2_subtree_control.cpp b/gunit/011-cgroupv2_subtree_control.cpp new file mode 100644 index 000000000000..9edede25f68c --- /dev/null +++ b/gunit/011-cgroupv2_subtree_control.cpp @@ -0,0 +1,125 @@ +/** + * libcgroup googletest for cgroupv2_subtree_control() + * + * 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 "gtest/gtest.h" + +#include "libcgroup-internal.h" + +static const char * const PARENT_DIR = "test011cgroup/"; +static const char * const SUBTREE_FILE = "cgroup.subtree_control"; + + +class SubtreeControlTest : public ::testing::Test { + protected: + + void SetUp() override { + char tmp_path[FILENAME_MAX]; + int ret, i; + FILE *f; + + ret = mkdir(PARENT_DIR, S_IRWXU | S_IRWXG | S_IRWXO); + ASSERT_EQ(ret, 0); + + memset(tmp_path, 0, sizeof(tmp_path)); + ret = snprintf(tmp_path, FILENAME_MAX - 1, "%s%s", + PARENT_DIR, SUBTREE_FILE); + ASSERT_GT(ret, 0); + + f = fopen(tmp_path, "w"); + fclose(f); + } + + /* + * 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; + + ret = rmrf(PARENT_DIR); + ASSERT_EQ(ret, 0); + } +}; + +TEST_F(SubtreeControlTest, AddController) +{ + char tmp_path[FILENAME_MAX], buf[4092]; + char ctrlr_name[] = "cpu"; + int ret; + FILE *f; + + memset(tmp_path, 0, sizeof(tmp_path)); + ret = snprintf(tmp_path, FILENAME_MAX - 1, "%s%s", + PARENT_DIR, SUBTREE_FILE); + ASSERT_GT(ret, 0); + + /* erase the contents of the file */ + f = fopen(tmp_path, "w"); + fclose(f); + + ret = cgroupv2_subtree_control(PARENT_DIR, ctrlr_name, true); + ASSERT_EQ(ret, 0); + + f = fopen(tmp_path, "r"); + ASSERT_NE(f, nullptr); + + while (fgets(buf, sizeof(buf), f)) + ASSERT_STREQ(buf, "+cpu"); + fclose(f); +} + +TEST_F(SubtreeControlTest, RemoveController) +{ + char tmp_path[FILENAME_MAX], buf[4092]; + char ctrlr_name[] = "memory"; + int ret; + FILE *f; + + memset(tmp_path, 0, sizeof(tmp_path)); + ret = snprintf(tmp_path, FILENAME_MAX - 1, "%s%s", + PARENT_DIR, SUBTREE_FILE); + ASSERT_GT(ret, 0); + + /* erase the contents of the file */ + f = fopen(tmp_path, "w"); + fclose(f); + + ret = cgroupv2_subtree_control(PARENT_DIR, ctrlr_name, false); + ASSERT_EQ(ret, 0); + + f = fopen(tmp_path, "r"); + ASSERT_NE(f, nullptr); + + while (fgets(buf, sizeof(buf), f)) + ASSERT_STREQ(buf, "-memory"); + fclose(f); +} diff --git a/gunit/Makefile.am b/gunit/Makefile.am index 43daba32579b..bc88985681c6 100644 --- a/gunit/Makefile.am +++ b/gunit/Makefile.am @@ -47,6 +47,7 @@ gtest_SOURCES = gtest.cpp \ 007-cgroup_process_v1_mount.cpp \ 008-cgroup_process_v2_mount.cpp \ 009-cgroup_set_values_recursive.cpp \ - 010-cgroup_chown_chmod_tasks.cpp + 010-cgroup_chown_chmod_tasks.cpp \ + 011-cgroupv2_subtree_control.cpp gtest_LDFLAGS = -L$(top_builddir)/googletest/googletest -l:libgtest.so \ -rpath $(abs_top_builddir)/googletest/googletest -- 2.25.3 _______________________________________________ Libcg-devel mailing list Libcg-devel@lists.sourceforge.net https://lists.sourceforge.net/lists/listinfo/libcg-devel
