150 lines
5.8 KiB
Python
150 lines
5.8 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from oslo_utils import uuidutils
|
|
|
|
from nova import context
|
|
from nova import exception
|
|
from nova.objects import cell_mapping
|
|
from nova.objects import instance_mapping
|
|
from nova import test
|
|
from nova.tests import fixtures
|
|
|
|
|
|
sample_mapping = {'instance_uuid': '',
|
|
'cell_id': 3,
|
|
'project_id': 'fake-project'}
|
|
|
|
|
|
sample_cell_mapping = {'id': 3,
|
|
'uuid': '',
|
|
'name': 'fake-cell',
|
|
'transport_url': 'rabbit:///',
|
|
'database_connection': 'mysql:///'}
|
|
|
|
|
|
def create_cell_mapping(**kwargs):
|
|
args = sample_cell_mapping.copy()
|
|
if 'uuid' not in kwargs:
|
|
args['uuid'] = uuidutils.generate_uuid()
|
|
args.update(kwargs)
|
|
ctxt = context.RequestContext('fake-user', 'fake-project')
|
|
return cell_mapping.CellMapping._create_in_db(ctxt, args)
|
|
|
|
|
|
def create_mapping(**kwargs):
|
|
args = sample_mapping.copy()
|
|
if 'instance_uuid' not in kwargs:
|
|
args['instance_uuid'] = uuidutils.generate_uuid()
|
|
args.update(kwargs)
|
|
ctxt = context.RequestContext('fake-user', 'fake-project')
|
|
return instance_mapping.InstanceMapping._create_in_db(ctxt, args)
|
|
|
|
|
|
class InstanceMappingTestCase(test.NoDBTestCase):
|
|
USES_DB_SELF = True
|
|
|
|
def setUp(self):
|
|
super(InstanceMappingTestCase, self).setUp()
|
|
self.useFixture(fixtures.Database(database='api'))
|
|
self.context = context.RequestContext('fake-user', 'fake-project')
|
|
self.mapping_obj = instance_mapping.InstanceMapping()
|
|
|
|
def test_get_by_instance_uuid(self):
|
|
cell_mapping = create_cell_mapping()
|
|
mapping = create_mapping()
|
|
db_mapping = self.mapping_obj._get_by_instance_uuid_from_db(
|
|
self.context, mapping['instance_uuid'])
|
|
for key in [key for key in self.mapping_obj.fields.keys()
|
|
if key != 'cell_mapping']:
|
|
self.assertEqual(db_mapping[key], mapping[key])
|
|
self.assertEqual(db_mapping['cell_mapping']['id'], cell_mapping['id'])
|
|
|
|
def test_get_by_instance_uuid_not_found(self):
|
|
self.assertRaises(exception.InstanceMappingNotFound,
|
|
self.mapping_obj._get_by_instance_uuid_from_db, self.context,
|
|
uuidutils.generate_uuid())
|
|
|
|
def test_save_in_db(self):
|
|
mapping = create_mapping()
|
|
cell_mapping = create_cell_mapping()
|
|
self.mapping_obj._save_in_db(self.context, mapping['instance_uuid'],
|
|
{'cell_id': cell_mapping['id']})
|
|
db_mapping = self.mapping_obj._get_by_instance_uuid_from_db(
|
|
self.context, mapping['instance_uuid'])
|
|
for key in [key for key in self.mapping_obj.fields.keys()
|
|
if key not in ['cell_id', 'cell_mapping', 'updated_at']]:
|
|
self.assertEqual(db_mapping[key], mapping[key])
|
|
self.assertEqual(db_mapping['cell_id'], cell_mapping['id'])
|
|
|
|
def test_destroy_in_db(self):
|
|
mapping = create_mapping()
|
|
self.mapping_obj._get_by_instance_uuid_from_db(self.context,
|
|
mapping['instance_uuid'])
|
|
self.mapping_obj._destroy_in_db(self.context, mapping['instance_uuid'])
|
|
self.assertRaises(exception.InstanceMappingNotFound,
|
|
self.mapping_obj._get_by_instance_uuid_from_db, self.context,
|
|
mapping['instance_uuid'])
|
|
|
|
def test_cell_id_nullable(self):
|
|
# Just ensure this doesn't raise
|
|
create_mapping(cell_id=None)
|
|
|
|
def test_modify_cell_mapping(self):
|
|
inst_mapping = instance_mapping.InstanceMapping(context=self.context)
|
|
inst_mapping.instance_uuid = uuidutils.generate_uuid()
|
|
inst_mapping.project_id = self.context.project_id
|
|
inst_mapping.cell_mapping = None
|
|
inst_mapping.create()
|
|
|
|
c_mapping = cell_mapping.CellMapping(
|
|
self.context,
|
|
uuid=uuidutils.generate_uuid(),
|
|
name="cell0",
|
|
transport_url="none:///",
|
|
database_connection="fake:///")
|
|
c_mapping.create()
|
|
|
|
inst_mapping.cell_mapping = c_mapping
|
|
inst_mapping.save()
|
|
|
|
result_mapping = instance_mapping.InstanceMapping.get_by_instance_uuid(
|
|
self.context, inst_mapping.instance_uuid)
|
|
|
|
self.assertEqual(result_mapping.cell_mapping.id,
|
|
c_mapping.id)
|
|
|
|
|
|
class InstanceMappingListTestCase(test.NoDBTestCase):
|
|
USES_DB_SELF = True
|
|
|
|
def setUp(self):
|
|
super(InstanceMappingListTestCase, self).setUp()
|
|
self.useFixture(fixtures.Database(database='api'))
|
|
self.context = context.RequestContext('fake-user', 'fake-project')
|
|
self.list_obj = instance_mapping.InstanceMappingList()
|
|
|
|
def test_get_by_project_id_from_db(self):
|
|
project_id = 'fake-project'
|
|
mappings = {}
|
|
mapping = create_mapping(project_id=project_id)
|
|
mappings[mapping['instance_uuid']] = mapping
|
|
mapping = create_mapping(project_id=project_id)
|
|
mappings[mapping['instance_uuid']] = mapping
|
|
|
|
db_mappings = self.list_obj._get_by_project_id_from_db(
|
|
self.context, project_id)
|
|
for db_mapping in db_mappings:
|
|
mapping = mappings[db_mapping.instance_uuid]
|
|
for key in instance_mapping.InstanceMapping.fields.keys():
|
|
self.assertEqual(db_mapping[key], mapping[key])
|