##############################################################################
import os
-import unittest
import errno
+
import mock
+import unittest
-from yardstick.common import constants as consts
from yardstick.benchmark.contexts import base
from yardstick.benchmark.contexts import node
+from yardstick.common import constants as consts
from yardstick.common import exceptions
+from yardstick.common import yaml_loader
class NodeContextTestCase(unittest.TestCase):
self.assertEqual(self.test_context.env, {})
self.assertEqual(self.test_context.attrs, {})
- @mock.patch('yardstick.common.utils.read_yaml_file')
+ @mock.patch.object(yaml_loader, 'read_yaml_file')
@mock.patch('{}.os.path.join'.format(PREFIX))
def test_init_negative(self, mock_path_join, read_mock):
special_path = '/foo/bar/error_file'
def test__get_physical_nodes(self):
self.test_context.init(self.attrs)
nodes = self.test_context._get_physical_nodes()
- self.assertEquals(nodes, self.test_context.nodes)
+ self.assertEqual(nodes, self.test_context.nodes)
def test__get_physical_node_for_server(self):
self.test_context.init(self.attrs)
node_collectd_options = [node for node in self.test_context.nodes
if node['name'] == 'node1'][0]['collectd']
- self.assertEquals(node_collectd_options, options)
+ self.assertEqual(node_collectd_options, options)
@mock.patch('{}.NodeContext._dispatch_script'.format(PREFIX))
def test_deploy(self, dispatch_script_mock):