From ef72ac56633feff8bc6ab1320e4b8d332363505a Mon Sep 17 00:00:00 2001 From: Vitaly Gridnev Date: Fri, 16 Jun 2017 17:51:02 +0400 Subject: [PATCH] Fix wrong patch in unit tests Patch must be stopped after test is done, so fixing incorrect patch in this setUp method. Change-Id: I5bb1dc4d82e6d644207ae777a62ddec39e5a290f --- .../tests/unit/plugins/vanilla/hadoop2/test_edp_engine.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/sahara/tests/unit/plugins/vanilla/hadoop2/test_edp_engine.py b/sahara/tests/unit/plugins/vanilla/hadoop2/test_edp_engine.py index 8a338adc3d..720aab9ee2 100644 --- a/sahara/tests/unit/plugins/vanilla/hadoop2/test_edp_engine.py +++ b/sahara/tests/unit/plugins/vanilla/hadoop2/test_edp_engine.py @@ -17,7 +17,6 @@ import mock from sahara.plugins import exceptions as ex from sahara.plugins.vanilla.hadoop2 import edp_engine -from sahara.service.edp import job_utils from sahara.tests.unit import base as sahara_base @@ -28,8 +27,9 @@ class EdpOozieEngineTest(sahara_base.SaharaTestCase): def setUp(self): super(EdpOozieEngineTest, self).setUp() self.cluster = mock.Mock() - job_utils.get_plugin = mock.Mock(return_value='test_plugins') - self.engine = edp_engine.EdpOozieEngine(self.cluster) + with mock.patch('sahara.service.edp.job_utils.get_plugin', + return_value='test_plugins'): + self.engine = edp_engine.EdpOozieEngine(self.cluster) def test_get_hdfs_user(self): self.assertEqual(self.engine.get_hdfs_user(), 'hadoop')