X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=yardstick%2Ftests%2Funit%2Fcommon%2Ftest_process.py;h=e0933c6ac1de6940178f0482d10c1418ec21df40;hb=5f11cb9f55aa0529fddfe916a2976a9aa5b4c08f;hp=1c6dfec27cf66d00338a0fb0cd8b1b95b1541b28;hpb=35b0d561b96b20042eeb68d70bf4d3268edf08fe;p=yardstick.git

diff --git a/yardstick/tests/unit/common/test_process.py b/yardstick/tests/unit/common/test_process.py
index 1c6dfec27..e0933c6ac 100644
--- a/yardstick/tests/unit/common/test_process.py
+++ b/yardstick/tests/unit/common/test_process.py
@@ -90,10 +90,11 @@ class ExecuteTestCase(unittest.TestCase):
             additional_env=self.additional_env)
         self.assertEqual(self.stdout, out)
 
-    def test_execute_exception(self):
+    @mock.patch.object(process, 'LOG')
+    def test_execute_exception(self, *args):
         self.obj.returncode = self.RET_CODE_WRONG
-        self.assertRaises(exceptions.ProcessExecutionError, process.execute,
-                          self.input_cmd, additional_env=self.additional_env)
+        with self.assertRaises(exceptions.ProcessExecutionError):
+            process.execute(self.input_cmd, additional_env=self.additional_env)
         self.obj.communicate.assert_called_once_with(None)
 
     def test_execute_with_extra_code(self):
@@ -107,7 +108,8 @@ class ExecuteTestCase(unittest.TestCase):
             additional_env=self.additional_env)
         self.assertEqual(self.stdout, out)
 
-    def test_execute_exception_no_check(self):
+    @mock.patch.object(process, 'LOG')
+    def test_execute_exception_no_check(self, *args):
         self.obj.returncode = self.RET_CODE_WRONG
         out = process.execute(self.input_cmd,
                               additional_env=self.additional_env,