From: Tom Hromatka <tom.hromatka@oracle.com>
Date: Mon, 1 Mar 2021 16:38:03 +0000 (+0000)
Subject: ftests: Add cgroup v1 cgrules test
X-Git-Tag: v2.0.3~11^2^2~25^2~1
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=36d331fbfa2a1f48484d197eeadcfe1af6d8c64d;p=thirdparty%2Flibcgroup.git

ftests: Add cgroup v1 cgrules test

Add a simple cgrulesengd tests.  Note that it cannot run within
a container.

-----------------------------------------------------------------
Test Results:
	Run Date:                          Mar 01 16:43:17
	Passed:                                  1 test(s)
	Skipped:                                 0 test(s)
	Failed:                                  0 test(s)
-----------------------------------------------------------------
Timing Results:
	Test                                    Time (sec)
	---------------------------------------------------------
	setup                                         0.00
	006-cgrules-basic_cgrules_v1.py               5.21
	teardown                                      0.00
	---------------------------------------------------------
	Total Run Time                                5.21

Signed-off-by: Tom Hromatka <tom.hromatka@oracle.com>
(cherry picked from commit eeb176b289b5ac704d27ca52996423b1887c6c3e)
---

diff --git a/ftests/006-cgrules-basic_cgrules_v1.py b/ftests/006-cgrules-basic_cgrules_v1.py
new file mode 100755
index 00000000..53dfd89c
--- /dev/null
+++ b/ftests/006-cgrules-basic_cgrules_v1.py
@@ -0,0 +1,101 @@
+#!/usr/bin/env python3
+#
+# Basic cgrules functionality test
+#
+# Copyright (c) 2021 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>.
+#
+
+from cgroup import Cgroup, CgroupVersion
+import consts
+import ftests
+import os
+from process import Process
+import sys
+
+CONTROLLER='cpu'
+PARENT_CGNAME='006cgrules'
+CHILD_CGNAME='childcg'
+
+# move all perl processes to the 006cgrules/childcg cgroup in the
+# cpu controller
+CGRULE="*:/usr/bin/perl cpu {}".format(os.path.join(PARENT_CGNAME, CHILD_CGNAME))
+
+cg = Cgroup(os.path.join(PARENT_CGNAME, CHILD_CGNAME))
+
+def prereqs(config):
+    result = consts.TEST_PASSED
+    cause = None
+
+    if config.args.container:
+        result = consts.TEST_SKIPPED
+        cause = "This test cannot be run within a container"
+        return result, cause
+
+    if CgroupVersion.get_version('cpu') != CgroupVersion.CGROUP_V1:
+        result = consts.TEST_SKIPPED
+        cause = "This test requires the cgroup v1 cpu controller"
+        return result, cause
+
+    return result, cause
+
+def setup(config):
+    Cgroup.create(config, CONTROLLER, PARENT_CGNAME)
+    Cgroup.create(config, CONTROLLER, os.path.join(PARENT_CGNAME, CHILD_CGNAME))
+
+    Cgroup.set_cgrules_conf(config, CGRULE, append=False)
+    cg.start_cgrules(config)
+
+def test(config):
+    result = consts.TEST_PASSED
+    cause = None
+
+    pid = config.process.create_process(config)
+    proc_cgroup = Process.get_cgroup(config, pid, CONTROLLER)
+
+    # proc/{pid}/cgroup alsways prepends a '/' to the cgroup path
+    if proc_cgroup != os.path.join('/', PARENT_CGNAME, CHILD_CGNAME):
+        result = consts.TEST_FAILED
+        cause = "PID {} was expected to be in cgroup {} but is in cgroup {}".format(
+                    pid, os.path.join('/', PARENT_CGNAME, CHILD_CGNAME), proc_cgroup)
+
+    return result, cause
+
+def teardown(config):
+    # destroy the child processes
+    config.process.join_children(config)
+    cg.join_children(config)
+    Cgroup.delete(config, CONTROLLER, PARENT_CGNAME, recursive=True)
+
+def main(config):
+    [result, cause] = prereqs(config)
+    if result != consts.TEST_PASSED:
+        return [result, cause]
+
+    try:
+        setup(config)
+        [result, cause] = test(config)
+    finally:
+        teardown(config)
+
+    return [result, cause]
+
+if __name__ == '__main__':
+    config = ftests.parse_args()
+    # this test was invoked directly.  run only it
+    config.args.num = int(os.path.basename(__file__).split('-')[0])
+    sys.exit(ftests.main(config))