TODO Signed-off-by: Tom Hromatka <tom.hroma...@oracle.com> --- ftests/006-cgrules-basic_cgrules_v1.py | 101 +++++++++++++++++++++++++ 1 file changed, 101 insertions(+) create mode 100755 ftests/006-cgrules-basic_cgrules_v1.py
diff --git a/ftests/006-cgrules-basic_cgrules_v1.py b/ftests/006-cgrules-basic_cgrules_v1.py new file mode 100755 index 000000000000..53dfd89c13e6 --- /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.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>. +# + +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)) -- 2.26.2 _______________________________________________ Libcg-devel mailing list Libcg-devel@lists.sourceforge.net https://lists.sourceforge.net/lists/listinfo/libcg-devel
