From: Tom Hromatka <tom.hromatka@oracle.com>
Date: Thu, 16 Dec 2021 15:04:35 +0000 (-0700)
Subject: ftests: Add a test for the python bindings to cgroup_cgxget()
X-Git-Tag: v3.1.0~308^2~2^2~105^2~4
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=9ddaeb70a98d709bbc8a0ffb5dde4319cc69f45c;p=thirdparty%2Flibcgroup.git

ftests: Add a test for the python bindings to cgroup_cgxget()

Add a functional test that exercises the cgroup_cgxget() python
bindings.

-----------------------------------------------------------------
Test Results:
        Run Date:                          Oct 11 22:11:48
        Passed:                                  1 test(s)
        Skipped:                                 0 test(s)
        Failed:                                  0 test(s)
-----------------------------------------------------------------
Timing Results:
        Test                                    Time (sec)
        ---------------------------------------------------------
        setup                                         0.00
        038-pybindings-cgxget.py                      0.04
        teardown                                      0.00
        ---------------------------------------------------------
        Total Run Time                                0.04

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

diff --git a/ftests/039-pybindings-cgxget.py b/ftests/039-pybindings-cgxget.py
new file mode 100755
index 00000000..04a1f9b5
--- /dev/null
+++ b/ftests/039-pybindings-cgxget.py
@@ -0,0 +1,130 @@
+#!/usr/bin/env python3
+#
+# cgxget functionality test using the python bindings
+#
+# Copyright (c) 2021-2022 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 CgroupVersion
+from cgroup import Cgroup as CgroupCli
+from libcgroup import Cgroup, Version
+import consts
+import ftests
+import os
+import sys
+
+CONTROLLER = 'cpu'
+CGNAME = "039bindings"
+
+SETTING1 = 'cpu.shares'
+VALUE1 = '4096'
+
+SETTING2 = 'cpu.weight'
+VALUE2 = '400'
+
+def prereqs(config):
+    if config.args.container:
+        result = consts.TEST_SKIPPED
+        cause = "This test cannot be run within a container"
+        return result, cause
+
+    result = consts.TEST_PASSED
+    cause = None
+
+    return result, cause
+
+def setup(config):
+    CgroupCli.create(config, CONTROLLER, CGNAME)
+    if CgroupVersion.get_version('cpu') == CgroupVersion.CGROUP_V1:
+        CgroupCli.set(config, CGNAME, SETTING1, VALUE1)
+    else:
+        CgroupCli.set(config, CGNAME, SETTING2, VALUE2)
+
+def test(config):
+    result = consts.TEST_PASSED
+    cause = None
+
+    cg1 = Cgroup(CGNAME, Version.CGROUP_V1)
+    cg1.add_controller(CONTROLLER)
+    cg1.add_setting(SETTING1)
+
+    cg1.cgxget()
+
+    if len(cg1.controllers) != 1:
+        result = consts.TEST_FAILED
+        cause = "Controller length doesn't match, expected 1, but received {}".format(
+                len(cg1.controllers))
+        return result, cause
+
+    if len(cg1.controllers[CONTROLLER].settings) != 1:
+        result = consts.TEST_FAILED
+        cause = "Settings length doesn't match, expected 1, but received {}".format(
+                len(cg1.controllers[CONTROLLER].settings))
+        return result, cause
+
+    if cg1.controllers[CONTROLLER].settings[SETTING1] != VALUE1:
+        result = consts.TEST_FAILED
+        cause = "Expected {} = {} but received {}".format(SETTING1, VALUE1,
+                cg1.controllers[CONTROLLER].settings[SETTING1])
+        return result, cause
+
+    cg2 = Cgroup(CGNAME, Version.CGROUP_V2)
+    cg2.add_controller(CONTROLLER)
+    cg2.add_setting(SETTING2)
+
+    cg2.cgxget()
+
+    if len(cg2.controllers) != 1:
+        result = consts.TEST_FAILED
+        cause = "Controller length doesn't match, expected 1, but received {}".format(
+                len(cg2.controllers))
+        return result, cause
+
+    if len(cg2.controllers[CONTROLLER].settings) != 1:
+        result = consts.TEST_FAILED
+        cause = "Settings length doesn't match, expected 1, but received {}".format(
+                len(cg2.controllers[CONTROLLER].settings))
+        return result, cause
+
+    if cg2.controllers[CONTROLLER].settings[SETTING2] != VALUE2:
+        result = consts.TEST_FAILED
+        cause = "Expected {} = {} but received {}".format(SETTING2, VALUE2,
+                cg2.controllers[CONTROLLER].settings[SETTING2])
+        return result, cause
+
+    return result, cause
+
+def teardown(config):
+    CgroupCli.delete(config, CONTROLLER, CGNAME)
+
+def main(config):
+    [result, cause] = prereqs(config)
+    if result != consts.TEST_PASSED:
+        return [result, cause]
+
+    setup(config)
+    [result, cause] = test(config)
+    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))