ftests: Add controller list() support to Cgroup.classify()

Add support for passing in a list of controllers to Cgroup.classify().

Signed-off-by: Tom Hromatka <tom.hromatka@oracle.com>
Reviewed-by: Kamalesh Babulal <kamalesh.babulal@oracle.com>

diff --git a/tests/ftests/cgroup.py b/tests/ftests/cgroup.py
index 50da99e718cab71c82771335e8d7d1466bf5f599..3be9ba04e02b9350441d796b76440cc5bc84df4b 100644 (file)
--- a/tests/ftests/cgroup.py
+++ b/tests/ftests/cgroup.py
@@ -441,8 +441,15 @@ class Cgroup(object):
         if ignore_systemd:
             cmd.append('-b')
 
-        cmd.append('-g')
-        cmd.append('{}:{}'.format(controller, cgname))
+        if isinstance(controller, str):
+            cmd.append('-g')
+            cmd.append('{}:{}'.format(controller, cgname))
+        elif isinstance(controller, list):
+            controller = ','.join(controller)
+            cmd.append('-g')
+            cmd.append('{}:{}'.format(controller, cgname))
+        else:
+            raise ValueError('Unsupported controller format: {}'.format(type(controller)))
 
         if isinstance(pid_list, str):
             cmd.append(pid_list)