gunit: Add unit tests for cgroupv2_subtree_control()

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.hromatka@oracle.com>

diff --git a/gunit/011-cgroupv2_subtree_control.cpp b/gunit/011-cgroupv2_subtree_control.cpp
new file mode 100644 (file)
index 0000000..9edede2
--- /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.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 <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 43daba32579b9edfaaf8e6c4f8f36c5afb4d8fa2..bc88985681c66c8f35aa569245a2b062a3884435 100644 (file)
--- 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