Replaces reload() and __builtin__ for Python 3.x

Uses "six.moves.reload_module" and "six.moves.builtins"
This commit is contained in:
Alessandro Pilotti
2014-09-08 15:33:44 +03:00
parent c13680eeec
commit 69afc9ab54
5 changed files with 12 additions and 7 deletions

View File

@@ -21,6 +21,7 @@ import sys
import unittest import unittest
from oslo.config import cfg from oslo.config import cfg
from six import moves
CONF = cfg.CONF CONF = cfg.CONF
@@ -37,7 +38,7 @@ class ExtendVolumesPluginTests(unittest.TestCase):
self._extend_volumes = extendvolumes.ExtendVolumesPlugin() self._extend_volumes = extendvolumes.ExtendVolumesPlugin()
def tearDown(self): def tearDown(self):
reload(sys) moves.reload_module(sys)
@mock.patch('cloudbaseinit.plugins.windows.extendvolumes' @mock.patch('cloudbaseinit.plugins.windows.extendvolumes'
'.ExtendVolumesPlugin._get_volume_index') '.ExtendVolumesPlugin._get_volume_index')

View File

@@ -18,6 +18,7 @@ import uuid
import unittest import unittest
from oslo.config import cfg from oslo.config import cfg
from six import moves
from cloudbaseinit.plugins.windows import userdatautils from cloudbaseinit.plugins.windows import userdatautils
from cloudbaseinit.tests.metadata import fake_json_response from cloudbaseinit.tests.metadata import fake_json_response
@@ -32,7 +33,7 @@ class UserDataUtilsTest(unittest.TestCase):
'2013-04-04') '2013-04-04')
def tearDown(self): def tearDown(self):
reload(uuid) moves.reload_module(uuid)
@mock.patch('re.search') @mock.patch('re.search')
@mock.patch('tempfile.gettempdir') @mock.patch('tempfile.gettempdir')

View File

@@ -19,6 +19,8 @@ import mock
import sys import sys
import unittest import unittest
from six import moves
from cloudbaseinit.plugins import base from cloudbaseinit.plugins import base
from cloudbaseinit.plugins import constants from cloudbaseinit.plugins import constants
@@ -41,7 +43,7 @@ class ConfigWinRMCertificateAuthPluginTests(unittest.TestCase):
self._certif_auth = self.winrmcert.ConfigWinRMCertificateAuthPlugin() self._certif_auth = self.winrmcert.ConfigWinRMCertificateAuthPlugin()
def tearDown(self): def tearDown(self):
reload(sys) moves.reload_module(sys)
def _test_get_credentials(self, fake_user, fake_password): def _test_get_credentials(self, fake_user, fake_password):
mock_shared_data = mock.MagicMock() mock_shared_data = mock.MagicMock()

View File

@@ -45,7 +45,7 @@ class HeatUserDataHandlerTests(unittest.TestCase):
mock_execute_user_data_script, filename): mock_execute_user_data_script, filename):
mock_part = mock.MagicMock() mock_part = mock.MagicMock()
mock_part.get_filename.return_value = filename mock_part.get_filename.return_value = filename
with mock.patch('__builtin__.open', mock.mock_open(), with mock.patch('six.moves.builtins.open', mock.mock_open(),
create=True) as handle: create=True) as handle:
response = self._heat.process(mock_part) response = self._heat.process(mock_part)
handle().write.assert_called_once_with(mock_part.get_payload()) handle().write.assert_called_once_with(mock_part.get_payload())

View File

@@ -15,10 +15,11 @@
# under the License. # under the License.
import mock import mock
import unittest
import sys import sys
import unittest
from oslo.config import cfg from oslo.config import cfg
from six import moves
from cloudbaseinit import init from cloudbaseinit import init
from cloudbaseinit.plugins import base from cloudbaseinit.plugins import base
@@ -44,8 +45,8 @@ class InitManagerTest(unittest.TestCase):
self._init = init.InitManager() self._init = init.InitManager()
def tearDown(self): def tearDown(self):
reload(sys) moves.reload_module(sys)
reload(init) moves.reload_module(init)
def _test_get_plugin_section(self, instance_id): def _test_get_plugin_section(self, instance_id):
response = self._init._get_plugins_section(instance_id=instance_id) response = self._init._get_plugins_section(instance_id=instance_id)