adding unit tests for cgroup supportw

testsuite/TestCobalt/TestComponents/test_forkers.py

@@ -19,6 +19,8 @@ import Cobalt
 import TestCobalt
 import ConfigParser
 import time
+import os
+import subprocess
 
 config_file = Cobalt.CONFIG_FILES[0]
 config_fp = open(config_file, "w")
@@ -28,8 +30,9 @@ config_fp.close()
 config = ConfigParser.ConfigParser()
 config.read(Cobalt.CONFIG_FILES)
 
-
 from mock import Mock, MagicMock, patch
+from testsuite.TestCobalt.Utilities.assert_functions import assert_match, assert_not_match
+from nose.tools import raises
 
 import Cobalt.Components.base_forker
 from Cobalt.Components.user_script_forker import UserScriptForker
@@ -119,3 +122,123 @@ class TestUserScriptForker(object):
         usf.signal(1, "SIGTERM")
         mock_kill.assert_not_called()
         mock_killpg.assert_not_called()
+
+class TestBaseChild(object):
+    '''Test Cobalt.Components.base_forker.BaseChild operations'''
+
+    #start is getting tested as a part of the overall forker test here.
+
+    def setup(self):
+        self.args = ['/usr/bin/foo', '--flag0', 'arg0', 'arg1']
+        self.child_spec = {'args': self.args,
+                      'tag': 'testsuite',
+                      'cwd': '/home/joshua',
+                      'umask': '0077',
+                      'runid': 42,
+                     }
+
+    def teardown(self):
+        Cobalt.Components.base_forker.BaseChild.id_gen = Cobalt.Data.IncrID()
+
+    def test_child_construction(self):
+        '''BaseChild: construction'''
+        child = Cobalt.Components.base_forker.BaseChild(**self.child_spec)
+        assert_match(child.args, self.args, 'Args do not match')
+        assert_match(child.tag, 'testsuite', 'Tag Mismatch')
+        assert_match(child.cwd, '/home/joshua', 'Bad cwd')
+        assert_match(child.id, 1, 'Bad id')
+        assert_match(child.umask, '0077', 'Bad umask')
+        assert_match(child.runid, 42, 'Bad runid')
+
+    def test_child_construction_set_id(self):
+        '''BaseChild: construction w/id'''
+        child = Cobalt.Components.base_forker.BaseChild(id=300, **self.child_spec)
+        assert_match(child.args, self.args, 'Args do not match')
+        assert_match(child.tag, 'testsuite', 'Tag Mismatch')
+        assert_match(child.cwd, '/home/joshua', 'Bad cwd')
+        assert_match(child.id, 300, 'Bad id')
+        assert_match(child.umask, '0077', 'Bad umask')
+        assert_match(child.runid, 42, 'Bad runid')
+
+
+    @patch('Cobalt.Components.base_forker.os.setsid')
+    @patch('Cobalt.Components.base_forker.subprocess.Popen')
+    def test_child_set_cgroup(self, mock_subprocess_popen, mock_os_setsid):
+        '''BaseChild: set cgroup'''
+        # Not calling the start() method that would invoke this due to fork/exec being really messy
+        # in this environment.  setsid should be separate
+        communicate = MagicMock(return_value=('',''))
+        mock_subprocess_popen.return_value.communicate = communicate
+        mock_subprocess_popen.return_value.returncode = 0
+        child = Cobalt.Components.base_forker.BaseChild(**self.child_spec)
+        child.use_cgroups = True
+        child.cgclassify_path = '/usr/bin/foo'
+        child.cgclassify_args = ['test', 'arguments']
+        expected_arguments = ['/usr/bin/foo', 'test', 'arguments', str(os.getpid())]
+        child.preexec_first()
+        mock_subprocess_popen.assert_called_once_with(expected_arguments, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        communicate.assert_called_once_with(None)
+
+    @patch('Cobalt.Components.base_forker.os.setsid')
+    @patch('Cobalt.Components.base_forker.subprocess.Popen')
+    def test_child_cgroup_raise_exec_error(self, mock_subprocess_popen, mock_os_setsid):
+        '''BaseChild: handle cgclassify exec failure'''
+        mock_subprocess_popen.side_effect = IOError('Fake file not found')
+        child = Cobalt.Components.base_forker.BaseChild(**self.child_spec)
+        child.use_cgroups = True
+        child.cgclassify_path = '/usr/bin/foo'
+        child.cgclassify_args = ['test', 'arguments']
+        expected_arguments = ['/usr/bin/foo', 'test', 'arguments', str(os.getpid())]
+        child.preexec_first()
+        mock_subprocess_popen.assert_called_once_with(expected_arguments, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        mock_subprocess_popen.return_value.communicate.assert_not_called()
+
+    @raises(IOError)
+    @patch('Cobalt.Components.base_forker.os.setsid')
+    @patch('Cobalt.Components.base_forker.subprocess.Popen')
+    def test_child_cgroup_raise_exec_error_fatal_flag(self, mock_subprocess_popen, mock_os_setsid):
+        '''BaseChild: handle cgclassify exec failure with fatal flag'''
+        mock_subprocess_popen.side_effect = IOError('Fake file not found')
+        child = Cobalt.Components.base_forker.BaseChild(**self.child_spec)
+        child.use_cgroups = True
+        child.cgroup_failure_fatal = True
+        child.cgclassify_path = '/usr/bin/foo'
+        child.cgclassify_args = ['test', 'arguments']
+        expected_arguments = ['/usr/bin/foo', 'test', 'arguments', str(os.getpid())]
+        child.preexec_first()
+        mock_subprocess_popen.assert_called_once_with(expected_arguments, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        mock_subprocess_popen.return_value.communicate.assert_not_called()
+
+    @patch('Cobalt.Components.base_forker.os.setsid')
+    @patch('Cobalt.Components.base_forker.subprocess.Popen')
+    def test_child_cgroup_return_nonzero(self, mock_subprocess_popen, mock_os_setsid):
+        '''BaseChild: handle nonzero returncode from cgclassify'''
+        communicate = MagicMock(return_value=('',''))
+        mock_subprocess_popen.return_value.communicate = communicate
+        mock_subprocess_popen.return_value.returncode = 1
+        child = Cobalt.Components.base_forker.BaseChild(**self.child_spec)
+        child.use_cgroups = True
+        child.cgclassify_path = '/usr/bin/foo'
+        child.cgclassify_args = ['test', 'arguments']
+        expected_arguments = ['/usr/bin/foo', 'test', 'arguments', str(os.getpid())]
+        child.preexec_first()
+        mock_subprocess_popen.assert_called_once_with(expected_arguments, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        communicate.assert_called_once_with(None)
+
+    @raises(RuntimeError)
+    @patch('Cobalt.Components.base_forker.os.setsid')
+    @patch('Cobalt.Components.base_forker.subprocess.Popen')
+    def test_child_cgroup_return_nonzero_fatal_flag(self, mock_subprocess_popen, mock_os_setsid):
+        '''BaseChild: handle nonzero returncode from cgclassify with fatal flag'''
+        communicate = MagicMock(return_value=('',''))
+        mock_subprocess_popen.return_value.communicate = communicate
+        mock_subprocess_popen.return_value.returncode = 1
+        child = Cobalt.Components.base_forker.BaseChild(**self.child_spec)
+        child.use_cgroups = True
+        child.cgroup_failure_fatal = True
+        child.cgclassify_path = '/usr/bin/foo'
+        child.cgclassify_args = ['test', 'arguments']
+        expected_arguments = ['/usr/bin/foo', 'test', 'arguments', str(os.getpid())]
+        child.preexec_first()
+        mock_subprocess_popen.assert_called_once_with(expected_arguments, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        communicate.assert_called_once_with(None)