提交 37c8ec01 编写于 作者: J Jamie McAtamney 提交者: Shoaib Lari

Add unit test for RemoteOperation logging.

Co-authored-by: NJamie McAtamney <jmcatamney@pivotal.io>
Co-authored-by: NShoaib Lari <slari@pivotal.io>
上级 38235b4b
......@@ -10,7 +10,7 @@ from gppylib.operations.test_utils_helper import TestOperation, RaiseOperation,
RaiseOperation_Unsafe, RaiseOperation_Unpicklable, RaiseOperation_Safe, MyException, ExceptionWithArgs
from operations.unix import ListFiles
from test.unit.gp_unittest import GpTestCase, run_tests
from mock import patch, MagicMock
class UtilsTestCase(GpTestCase):
"""
......@@ -101,6 +101,17 @@ class UtilsTestCase(GpTestCase):
with self.assertRaises(Exception):
ParallelOperation([ListFiles("/tmp")], 0).run()
@patch('gppylib.commands.base.logger.debug')
@patch('pickle.loads')
@patch('gppylib.operations.utils.Command')
@patch('os.path.split', return_value = '/')
def test_RemoteOperation_logger_debug(self, mock_split, mock_cmd, mock_lods, mock_debug):
mock_cmd.run = MagicMock()
mockRemoteOperation = RemoteOperation(operation=TestOperation(), host="sdw1", dbid=2)
mockRemoteOperation.execute()
mock_debug.assert_called()
first_call = mock_debug.call_args_list[0]
self.assertTrue(first_call.startswith("Output for dbid 2 for host sdw1:"))
if __name__ == '__main__':
run_tests()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册