gunit: Add unit tests for cgroupv2_get_subtree_control()

This commit adds unit tests for cgroupv2_get_subtree_control()

[----------] 6 tests from GetSubtreeControlTest
[ RUN      ] GetSubtreeControlTest.SingleControllerEnabled
[       OK ] GetSubtreeControlTest.SingleControllerEnabled (0 ms)
[ RUN      ] GetSubtreeControlTest.SingleControllerNoMatch
[       OK ] GetSubtreeControlTest.SingleControllerNoMatch (0 ms)
[ RUN      ] GetSubtreeControlTest.SingleControllerNoMatch2
[       OK ] GetSubtreeControlTest.SingleControllerNoMatch2 (0 ms)
[ RUN      ] GetSubtreeControlTest.MultipleControllersEnabled
[       OK ] GetSubtreeControlTest.MultipleControllersEnabled (0 ms)
[ RUN      ] GetSubtreeControlTest.MultipleControllersEnabled2
[       OK ] GetSubtreeControlTest.MultipleControllersEnabled2 (0 ms)
[ RUN      ] GetSubtreeControlTest.MultipleControllersNoMatch
[       OK ] GetSubtreeControlTest.MultipleControllersNoMatch (0 ms)
[----------] 6 tests from GetSubtreeControlTest (0 ms total)

Signed-off-by: Tom Hromatka <tom.hromatka@oracle.com>

diff --git a/gunit/014-cgroupv2_get_subtree_control.cpp b/gunit/014-cgroupv2_get_subtree_control.cpp
new file mode 100644 (file)
index 0000000..66f3c52
--- /dev/null
+++ b/gunit/014-cgroupv2_get_subtree_control.cpp
@@ -0,0 +1,170 @@
+/**
+ * libcgroup googletest for cgroupv2_get_subtree_control()
+ *
+ * Copyright (c) 2020 Oracle and/or its affiliates.
+ * Author: Tom Hromatka <tom.hromatka@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 <fcntl.h>
+#include <ftw.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <unistd.h>
+
+#include "gtest/gtest.h"
+
+#include "libcgroup-internal.h"
+
+static const char * const PARENT_DIR = "test013cgroup/";
+static const char * const SUBTREE_FILE = "cgroup.subtree_control";
+
+
+class GetSubtreeControlTest : 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);
+       }
+
+       /*
+        * 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);
+       }
+};
+
+static void write_subtree_file(const char * const contents, ssize_t len)
+{
+       char tmp_path[FILENAME_MAX];
+       ssize_t bytes_written;
+       int ret, fd;
+
+       memset(tmp_path, 0, sizeof(tmp_path));
+       ret = snprintf(tmp_path, FILENAME_MAX - 1, "%s%s",
+                      PARENT_DIR, SUBTREE_FILE);
+       ASSERT_GT(ret, 0);
+
+       fd = open(tmp_path, O_WRONLY | O_TRUNC | O_CREAT, S_IRWXU | S_IRWXG);
+       ASSERT_GT(fd, 0);
+
+       bytes_written = write(fd, contents, len);
+       ASSERT_EQ(bytes_written, len);
+       close(fd);
+}
+
+TEST_F(GetSubtreeControlTest, SingleControllerEnabled)
+{
+       char ctrlr_name[] = "cpu";
+       char subtree_contents[] = "cpu\n";
+       bool enabled = false;
+       int ret;
+
+       write_subtree_file(subtree_contents, strlen(subtree_contents));
+
+       ret = cgroupv2_get_subtree_control(PARENT_DIR, ctrlr_name, &enabled);
+       ASSERT_EQ(ret, 0);
+       ASSERT_EQ(enabled, true);
+}
+
+TEST_F(GetSubtreeControlTest, SingleControllerNoMatch)
+{
+       char ctrlr_name[] = "cpu";
+       char subtree_contents[] = "cpuset\n";
+       bool enabled = true;
+       int ret;
+
+       write_subtree_file(subtree_contents, strlen(subtree_contents));
+
+       ret = cgroupv2_get_subtree_control(PARENT_DIR, ctrlr_name, &enabled);
+       ASSERT_EQ(ret, ECGROUPNOTMOUNTED);
+       ASSERT_EQ(enabled, false);
+}
+
+TEST_F(GetSubtreeControlTest, SingleControllerNoMatch2)
+{
+       char ctrlr_name[] = "cpuset";
+       char subtree_contents[] = "cpu\n";
+       bool enabled = true;
+       int ret;
+
+       write_subtree_file(subtree_contents, strlen(subtree_contents));
+
+       ret = cgroupv2_get_subtree_control(PARENT_DIR, ctrlr_name, &enabled);
+       ASSERT_EQ(ret, ECGROUPNOTMOUNTED);
+       ASSERT_EQ(enabled, false);
+}
+
+TEST_F(GetSubtreeControlTest, MultipleControllersEnabled)
+{
+       char ctrlr_name[] = "cpu";
+       char subtree_contents[] = "cpu cpuset io memory pids\n";
+       bool enabled = false;
+       int ret;
+
+       write_subtree_file(subtree_contents, strlen(subtree_contents));
+
+       ret = cgroupv2_get_subtree_control(PARENT_DIR, ctrlr_name, &enabled);
+       ASSERT_EQ(ret, 0);
+       ASSERT_EQ(enabled, true);
+}
+
+TEST_F(GetSubtreeControlTest, MultipleControllersEnabled2)
+{
+       char ctrlr_name[] = "pids";
+       char subtree_contents[] = "cpu cpuset io memory pids\n";
+       bool enabled = false;
+       int ret;
+
+       write_subtree_file(subtree_contents, strlen(subtree_contents));
+
+       ret = cgroupv2_get_subtree_control(PARENT_DIR, ctrlr_name, &enabled);
+       ASSERT_EQ(ret, 0);
+       ASSERT_EQ(enabled, true);
+}
+
+TEST_F(GetSubtreeControlTest, MultipleControllersNoMatch)
+{
+       char ctrlr_name[] = "network";
+       char subtree_contents[] = "cpu cpuset io memory pids\n";
+       bool enabled = true;
+       int ret;
+
+       write_subtree_file(subtree_contents, strlen(subtree_contents));
+
+       ret = cgroupv2_get_subtree_control(PARENT_DIR, ctrlr_name, &enabled);
+       ASSERT_EQ(ret, ECGROUPNOTMOUNTED);
+       ASSERT_EQ(enabled, false);
+}

diff --git a/gunit/Makefile.am b/gunit/Makefile.am
index 7ae8b3bb463dbb0d264762016fedffb196af80a3..912f2182d8d2efe1d1d730cbe51a90eee4b01315 100644 (file)
--- a/gunit/Makefile.am
+++ b/gunit/Makefile.am
@@ -50,6 +50,7 @@ gtest_SOURCES = gtest.cpp \
                010-cgroup_chown_chmod_tasks.cpp \
                011-cgroupv2_subtree_control.cpp \
                012-cgroup_create_cgroup.cpp \
-               013-cgroup_build_tasks_procs_path.cpp
+               013-cgroup_build_tasks_procs_path.cpp \
+               014-cgroupv2_get_subtree_control.cpp
 gtest_LDFLAGS = -L$(top_builddir)/googletest/googletest -l:libgtest.so \
                -rpath $(abs_top_builddir)/googletest/googletest