def test__delete_environment(self):
fuel_scenario = utils.FuelScenario()
fuel_scenario.admin_clients = self.admin_clients
fuel_scenario._delete_environment(42, 33)
tmp_mock = fuel_scenario.admin_clients("fuel")
tmp_mock.environment.delete.assert_called_once_with(42, 33)
def test__remove_nodes_error(self, mock_fuel_client):
mock_tmp = mock_fuel_client.fuelclient_module.objects
mock_env = mock_tmp.environment.Environment
mock_env.return_value = mock.Mock()
mock_env.return_value.unassign.side_effect = BaseException
fscen = utils.FuelScenario()
self.assertRaises(RuntimeError, fscen._remove_node, "1", "2")
def test__add_nodes_error(self):
fscen = utils.FuelScenario()
fscen.admin_clients = mock.Mock()
tmp_mock = fscen.admin_clients.return_value.environment.client
tmp_mock.add_nodes.side_effect = BaseException
self.assertRaises(RuntimeError, fscen._add_node, "1", "42",
node_roles="some_role")
def test__remove_nodes(self, mock_fuel_client):
mock_tmp = mock_fuel_client.fuelclient_module.objects
mock_env = mock_tmp.environment.Environment
mock_env.return_value = mock.Mock()
fscen = utils.FuelScenario()
fscen._remove_node("1", "2")
mock_env.assert_called_once_with("1")
mock_env.return_value.unassign.assert_called_once_with(["2"])
def test__list_environments(self):
scenario = utils.FuelScenario(self.context)
self.assertEqual(
scenario._list_environments(),
self.admin_clients("fuel").environment.list.return_value)
self.admin_clients("fuel").environment.list.assert_called_once_with()
self._test_atomic_action_timer(scenario.atomic_actions(),
"fuel.list_environments")
def test__node_is_assigned(self):
fscen = utils.FuelScenario()
fscen.admin_clients = mock.Mock()
fscen.admin_clients.return_value.node.get_by_id.return_value = {
"id": "id1", "cluster": "some_id"}
self.assertTrue(fscen._node_is_assigned("id1"))
fscen.admin_clients.return_value.node.get_by_id.return_value[
"cluster"] = ""
self.assertFalse(fscen._node_is_assigned("id2"))
def test__list_node_ids(self):
fscen = utils.FuelScenario()
fscen.admin_clients = mock.Mock()
fscen.admin_clients.return_value.node.get_all.return_value = [
{"id": "id1"}, {"id": "id2"}]
res = fscen._list_node_ids("env")
self.assertEqual(["id1", "id2"], res)
tmp_mock = fscen.admin_clients.return_value.node.get_all
tmp_mock.assert_called_once_with(environment_id="env")
def test__list_environments(self):
self.admin_clients("fuel").environment.list.return_value = [{
"name":
"some_name1"
}, {
"name":
"rally_name2"
}]
scenario = utils.FuelScenario(self.context)
self.assertEqual([{
"name": "rally_name2"
}], scenario._list_environments())
self.admin_clients("fuel").environment.list.assert_called_once_with()
self._test_atomic_action_timer(scenario.atomic_actions(),
"fuel.list_environments")
def test__create_environment(self):
self.admin_clients("fuel").environment.create.return_value = {"id": 42}
fuel_scenario = utils.FuelScenario()
fuel_scenario.admin_clients = self.admin_clients
fuel_scenario.generate_random_name = mock.Mock()
result = fuel_scenario._create_environment()
self.assertEqual(
self.admin_clients("fuel").environment.create.return_value["id"],
result)
tmp_mck = self.admin_clients("fuel").environment.create
tmp_mck.assert_called_once_with(
fuel_scenario.generate_random_name.return_value, 1, "neutron",
"ha_compact", "vlan")
def setup(self):
"""Create Fuel environments, using the broker pattern."""
self.context.setdefault("fuel", {})
self.context["fuel"].setdefault("environments", [])
threads = self.config["resource_management_workers"]
LOG.debug("Creating %(envs)d environments using %(threads)s threads" %
{"envs": self.config["environments"],
"threads": threads})
self.fscenario = fuel_utils.FuelScenario(self.context)
self.context["fuel"]["environments"] = self._create_envs()
if len(self.context[
"fuel"]["environments"]) != self.config["environments"]:
raise exceptions.ContextSetupFailure(
ctx_name=self.get_name(),
msg=_("failed to create the requested"
" number of environments."))
def test__get_free_node_id_exception(self, mock__list_node_ids,
mock__node_is_assigned):
self.assertRaises(RuntimeError, utils.FuelScenario()._get_free_node_id)
def test__get_free_node_id(self, mock__list_node_ids,
mock__node_is_assigned):
node_id = utils.FuelScenario()._get_free_node_id()
self.assertIn(node_id, mock__list_node_ids.return_value)
def test__add_nodes(self):
fscen = utils.FuelScenario()
fscen.admin_clients = mock.Mock()
fscen._add_node("1", ["42"], node_roles=["some_role"])
tmp_mock = fscen.admin_clients.return_value.environment.client
tmp_mock.add_nodes.assert_called_once_with("1", ["42"], ["some_role"])
