From: Tom Hromatka <tom.hromatka@oracle.com>
Date: Wed, 28 Oct 2020 15:30:55 +0000 (+0000)
Subject: ftests: Add support for cgclassify to Cgroup class
X-Git-Tag: v2.0.3~11^2^2~31^2~6
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=4b48319f989dd52388ec1ee80eb418c26940bfb9;p=thirdparty%2Flibcgroup.git

ftests: Add support for cgclassify to Cgroup class

Add support for cgclassify to Cgroup class.

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

diff --git a/ftests/cgroup.py b/ftests/cgroup.py
index 6f7042d6..ec5c6821 100644
--- a/ftests/cgroup.py
+++ b/ftests/cgroup.py
@@ -194,3 +194,24 @@ class Cgroup(Enum):
                                 return Cgroup.CGROUP_V2
 
         return Cgroup.CGROUP_UNK
+
+    @staticmethod
+    def classify(config, controller, cgname, pid_list, sticky=False,
+                 cancel_sticky=False, in_container=True):
+        cmd = list()
+        cmd.append(Cgroup.build_cmd_path(in_container, 'cgclassify'))
+        cmd.append('-g')
+        cmd.append('{}:{}'.format(controller, cgname))
+
+        if isinstance(pid_list, str):
+            cmd.append(pid_list)
+        elif isinstance(pid_list, int):
+            cmd.append(str(pid_list))
+        elif isinstance(pid_list, list):
+            for pid in pid_list:
+                cmd.append(pid)
+
+        if in_container:
+            config.container.run(cmd)
+        else:
+            Run.run(cmd)