def test_setup_failexists(self, mock_osclients):
# Setup and mock
new_context = copy.deepcopy(self.context)
new_context["flavors"] = {}
mock_flavor_create = mock_osclients().nova().flavors.create
exception = nova_exceptions.Conflict("conflict")
mock_flavor_create.side_effect = exception
# Run
flavors_ctx = flavors.FlavorsGenerator(self.context)
flavors_ctx.setup()
# Assertions
self.assertEqual(new_context, flavors_ctx.context)
mock_osclients.assert_called_with(self.context["admin"]["endpoint"])
mock_flavor_create.assert_called_once_with(name="flavor_name",
ram=2048,
vcpus=3,
disk=10,
ephemeral=3,
swap=5)
def test_setup(self, mock_osclients):
# Setup and mock
mock_create = mock_osclients().nova().flavors.create
mock_create().to_dict.return_value = {"flavor_key": "flavor_value"}
# Run
flavors_ctx = flavors.FlavorsGenerator(self.context)
flavors_ctx.setup()
# Assertions
self.assertEqual(flavors_ctx.context["flavors"],
{"flavor_name": {
"flavor_key": "flavor_value"
}})
mock_osclients.assert_called_with(self.context["admin"]["endpoint"])
mock_create.assert_called_with(name="flavor_name",
ram=2048,
vcpus=3,
disk=10,
ephemeral=3,
swap=5)
mock_create().set_keys.assert_called_with({"key": "value"})
mock_create().to_dict.assert_called_with()
def test_cleanup(self, mock_osclients):
# Setup and mock
real_context = {
"flavors": {
"flavor_name": {
"flavor_name": "flavor_name",
"id": "flavor_name"
}
},
"admin": {
"endpoint": mock.MagicMock()
},
"task": mock.MagicMock(),
}
# Run
flavors_ctx = flavors.FlavorsGenerator(real_context)
flavors_ctx.cleanup()
# Assertions
mock_osclients.assert_called_with(real_context["admin"]["endpoint"])
mock_flavors_delete = mock_osclients().nova().flavors.delete
mock_flavors_delete.assert_called_with("flavor_name")