From 34dcbd6e592dcb32a1dee68f4916f386aa43dfda Mon Sep 17 00:00:00 2001 From: luhuichun Date: Mon, 10 Aug 2015 17:22:42 +0800 Subject: [PATCH] add unit test for test_hdfs_helper Change-Id: I4743d189229b59e8998d093370cd5d0eb51afe13 --- sahara/tests/unit/service/edp/test_hdfs_helper.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/sahara/tests/unit/service/edp/test_hdfs_helper.py b/sahara/tests/unit/service/edp/test_hdfs_helper.py index 5377abae..6ebc893b 100644 --- a/sahara/tests/unit/service/edp/test_hdfs_helper.py +++ b/sahara/tests/unit/service/edp/test_hdfs_helper.py @@ -132,3 +132,14 @@ class HDFSHelperTestCase(base.SaharaTestCase): mock.call().__enter__().write_file_to(str1, mock_helper()), mock.call().__enter__().execute_command(str2, run_as_root=True), mock.call().__exit__(None, None, None)]) + + @mock.patch('six.text_type') + @mock.patch('os.open') + def test_put_file_to_hdfs(self, open_get, mock_six): + open_get.return_value = '/tmp/workflow.xml' + mock_six.return_value = 111 + helper.put_file_to_hdfs(self.cluster, open_get, 'workflow', + '/tmp', 'hdfs') + self.cluster.execute_command.assert_called_once_with( + 'sudo su - -c "hadoop dfs -copyFromLocal /tmp/workflow.111' + ' /tmp/workflow" hdfs && sudo rm -f /tmp/workflow.111')