def test_image_and_flavor_properties_as_ids(self):
node_props = {
'image': 'some-image-id',
'flavor': 'some-flavor-id'
}
with mock.patch('nova_plugin.server.ctx',
self._get_mock_ctx_with_node_properties(node_props)):
nova_client = self._get_mocked_nova_client()
serv = {}
server._handle_image_or_flavor(serv, nova_client, 'image')
server._handle_image_or_flavor(serv, nova_client, 'flavor')
self.assertEquals('some-image-id', serv.get('image'))
self.assertEquals('some-flavor-id', serv.get('flavor'))
def test_image_name_and_flavor_name_override_on_properties(self):
node_props = {
'image': 'properties-image-id',
'flavor': 'properties-flavor-id'
}
with mock.patch('nova_plugin.server.ctx',
self._get_mock_ctx_with_node_properties(node_props)):
nova_client = self._get_mocked_nova_client()
serv = {}
serv['image_name'] = 'some-image-name'
serv['flavor_name'] = 'some-flavor-name'
server._handle_image_or_flavor(serv, nova_client, 'image')
server._handle_image_or_flavor(serv, nova_client, 'flavor')
self.assertEquals('some-image-id', serv.get('image'))
self.assertNotIn('image_name', serv)
self.assertEquals('some-flavor-id', serv.get('flavor'))
self.assertNotIn('flavor_name', serv)
