296 lines
12 KiB
Python
296 lines
12 KiB
Python
# Copyright 2013 IBM Corp.
|
|
#
|
|
# 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.
|
|
|
|
import mock
|
|
from oslo_utils import timeutils
|
|
|
|
from nova import context
|
|
from nova.db import api as db
|
|
from nova import exception
|
|
from nova import objects
|
|
from nova.objects import migration
|
|
from nova.tests.unit import fake_instance
|
|
from nova.tests.unit.objects import test_objects
|
|
from nova.tests import uuidsentinel
|
|
|
|
|
|
NOW = timeutils.utcnow().replace(microsecond=0)
|
|
|
|
|
|
def fake_db_migration(**updates):
|
|
db_instance = {
|
|
'created_at': NOW,
|
|
'updated_at': None,
|
|
'deleted_at': None,
|
|
'deleted': False,
|
|
'id': 123,
|
|
'uuid': uuidsentinel.migration,
|
|
'source_compute': 'compute-source',
|
|
'dest_compute': 'compute-dest',
|
|
'source_node': 'node-source',
|
|
'dest_node': 'node-dest',
|
|
'dest_host': 'host-dest',
|
|
'old_instance_type_id': 42,
|
|
'new_instance_type_id': 84,
|
|
'instance_uuid': 'fake-uuid',
|
|
'status': 'migrating',
|
|
'migration_type': 'resize',
|
|
'hidden': False,
|
|
'memory_total': 123456,
|
|
'memory_processed': 12345,
|
|
'memory_remaining': 111111,
|
|
'disk_total': 234567,
|
|
'disk_processed': 23456,
|
|
'disk_remaining': 211111,
|
|
}
|
|
|
|
if updates:
|
|
db_instance.update(updates)
|
|
return db_instance
|
|
|
|
|
|
class _TestMigrationObject(object):
|
|
@mock.patch.object(db, 'migration_get')
|
|
def test_get_by_id(self, mock_get):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
mock_get.return_value = fake_migration
|
|
mig = migration.Migration.get_by_id(ctxt, fake_migration['id'])
|
|
self.compare_obj(mig, fake_migration)
|
|
mock_get.assert_called_once_with(ctxt, fake_migration['id'])
|
|
|
|
@mock.patch.object(db, 'migration_get_by_instance_and_status')
|
|
def test_get_by_instance_and_status(self, mock_get):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
mock_get.return_value = fake_migration
|
|
mig = migration.Migration.get_by_instance_and_status(
|
|
ctxt, fake_migration['id'], 'migrating')
|
|
self.compare_obj(mig, fake_migration)
|
|
mock_get.assert_called_once_with(ctxt,
|
|
fake_migration['id'],
|
|
'migrating')
|
|
|
|
@mock.patch('nova.db.api.migration_get_in_progress_by_instance')
|
|
def test_get_in_progress_by_instance(self, m_get_mig):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
db_migrations = [fake_migration, dict(fake_migration, id=456)]
|
|
|
|
m_get_mig.return_value = db_migrations
|
|
migrations = migration.MigrationList.get_in_progress_by_instance(
|
|
ctxt, fake_migration['instance_uuid'])
|
|
|
|
self.assertEqual(2, len(migrations))
|
|
for index, db_migration in enumerate(db_migrations):
|
|
self.compare_obj(migrations[index], db_migration)
|
|
|
|
@mock.patch.object(db, 'migration_create')
|
|
def test_create(self, mock_create):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
mock_create.return_value = fake_migration
|
|
mig = migration.Migration(context=ctxt)
|
|
mig.source_compute = 'foo'
|
|
mig.migration_type = 'resize'
|
|
mig.uuid = uuidsentinel.migration
|
|
mig.create()
|
|
self.assertEqual(fake_migration['dest_compute'], mig.dest_compute)
|
|
self.assertIn('uuid', mig)
|
|
mock_create.assert_called_once_with(ctxt,
|
|
{'source_compute': 'foo',
|
|
'migration_type': 'resize',
|
|
'uuid': uuidsentinel.migration})
|
|
|
|
@mock.patch.object(db, 'migration_create')
|
|
def test_recreate_fails(self, mock_create):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
mock_create.return_value = fake_migration
|
|
mig = migration.Migration(context=ctxt)
|
|
mig.source_compute = 'foo'
|
|
mig.migration_type = 'resize'
|
|
mig.uuid = uuidsentinel.migration
|
|
mig.create()
|
|
self.assertRaises(exception.ObjectActionError, mig.create)
|
|
mock_create.assert_called_once_with(ctxt,
|
|
{'source_compute': 'foo',
|
|
'migration_type': 'resize',
|
|
'uuid': uuidsentinel.migration})
|
|
|
|
def test_create_fails_migration_type(self):
|
|
ctxt = context.get_admin_context()
|
|
mig = migration.Migration(context=ctxt,
|
|
old_instance_type_id=42,
|
|
new_instance_type_id=84)
|
|
mig.source_compute = 'foo'
|
|
self.assertRaises(exception.ObjectActionError, mig.create)
|
|
|
|
@mock.patch.object(db, 'migration_update')
|
|
def test_save(self, mock_update):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
mock_update.return_value = fake_migration
|
|
mig = migration.Migration(context=ctxt)
|
|
mig.id = 123
|
|
mig.source_compute = 'foo'
|
|
mig.save()
|
|
self.assertEqual(fake_migration['dest_compute'], mig.dest_compute)
|
|
mock_update.assert_called_once_with(ctxt, 123,
|
|
{'source_compute': 'foo'})
|
|
|
|
@mock.patch.object(db, 'instance_get_by_uuid')
|
|
def test_instance(self, mock_get):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
fake_inst = fake_instance.fake_db_instance()
|
|
mock_get.return_value = fake_inst
|
|
mig = migration.Migration._from_db_object(ctxt,
|
|
migration.Migration(),
|
|
fake_migration)
|
|
mig._context = ctxt
|
|
self.assertEqual(mig.instance.host, fake_inst['host'])
|
|
mock_get.assert_called_once_with(ctxt,
|
|
fake_migration['instance_uuid'],
|
|
columns_to_join=['info_cache',
|
|
'security_groups'])
|
|
|
|
def test_instance_setter(self):
|
|
migration = objects.Migration(instance_uuid=uuidsentinel.instance)
|
|
inst = objects.Instance(uuid=uuidsentinel.instance)
|
|
with mock.patch('nova.objects.Instance.get_by_uuid') as mock_get:
|
|
migration.instance = inst
|
|
migration.instance
|
|
self.assertFalse(mock_get.called)
|
|
self.assertEqual(inst, migration._cached_instance)
|
|
self.assertEqual(inst, migration.instance)
|
|
|
|
@mock.patch.object(db, 'migration_get_unconfirmed_by_dest_compute')
|
|
def test_get_unconfirmed_by_dest_compute(self, mock_get):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
db_migrations = [fake_migration, dict(fake_migration, id=456)]
|
|
mock_get.return_value = db_migrations
|
|
migrations = (
|
|
migration.MigrationList.get_unconfirmed_by_dest_compute(
|
|
ctxt, 'window', 'foo', use_slave=False))
|
|
self.assertEqual(2, len(migrations))
|
|
for index, db_migration in enumerate(db_migrations):
|
|
self.compare_obj(migrations[index], db_migration)
|
|
mock_get.assert_called_once_with(ctxt, 'window', 'foo')
|
|
|
|
@mock.patch.object(db, 'migration_get_in_progress_by_host_and_node')
|
|
def test_get_in_progress_by_host_and_node(self, mock_get):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
db_migrations = [fake_migration, dict(fake_migration, id=456)]
|
|
mock_get.return_value = db_migrations
|
|
migrations = (
|
|
migration.MigrationList.get_in_progress_by_host_and_node(
|
|
ctxt, 'host', 'node'))
|
|
self.assertEqual(2, len(migrations))
|
|
for index, db_migration in enumerate(db_migrations):
|
|
self.compare_obj(migrations[index], db_migration)
|
|
mock_get.assert_called_once_with(ctxt, 'host', 'node')
|
|
|
|
@mock.patch.object(db, 'migration_get_all_by_filters')
|
|
def test_get_by_filters(self, mock_get):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
db_migrations = [fake_migration, dict(fake_migration, id=456)]
|
|
filters = {'foo': 'bar'}
|
|
mock_get.return_value = db_migrations
|
|
migrations = migration.MigrationList.get_by_filters(ctxt, filters)
|
|
self.assertEqual(2, len(migrations))
|
|
for index, db_migration in enumerate(db_migrations):
|
|
self.compare_obj(migrations[index], db_migration)
|
|
mock_get.assert_called_once_with(ctxt, filters,
|
|
sort_dirs=None, sort_keys=None,
|
|
limit=None, marker=None)
|
|
|
|
def test_migrate_old_resize_record(self):
|
|
db_migration = dict(fake_db_migration(), migration_type=None)
|
|
with mock.patch('nova.db.api.migration_get') as fake_get:
|
|
fake_get.return_value = db_migration
|
|
mig = objects.Migration.get_by_id(context.get_admin_context(), 1)
|
|
self.assertTrue(mig.obj_attr_is_set('migration_type'))
|
|
self.assertEqual('resize', mig.migration_type)
|
|
|
|
def test_migrate_old_migration_record(self):
|
|
db_migration = dict(
|
|
fake_db_migration(), migration_type=None,
|
|
old_instance_type_id=1, new_instance_type_id=1)
|
|
with mock.patch('nova.db.api.migration_get') as fake_get:
|
|
fake_get.return_value = db_migration
|
|
mig = objects.Migration.get_by_id(context.get_admin_context(), 1)
|
|
self.assertTrue(mig.obj_attr_is_set('migration_type'))
|
|
self.assertEqual('migration', mig.migration_type)
|
|
|
|
def test_migrate_unset_type_resize(self):
|
|
mig = objects.Migration(old_instance_type_id=1,
|
|
new_instance_type_id=2)
|
|
self.assertEqual('resize', mig.migration_type)
|
|
self.assertTrue(mig.obj_attr_is_set('migration_type'))
|
|
|
|
def test_migrate_unset_type_migration(self):
|
|
mig = objects.Migration(old_instance_type_id=1,
|
|
new_instance_type_id=1)
|
|
self.assertEqual('migration', mig.migration_type)
|
|
self.assertTrue(mig.obj_attr_is_set('migration_type'))
|
|
|
|
@mock.patch('nova.db.api.migration_get_by_id_and_instance')
|
|
def test_get_by_id_and_instance(self, fake_get):
|
|
ctxt = context.get_admin_context()
|
|
fake_migration = fake_db_migration()
|
|
fake_get.return_value = fake_migration
|
|
migration = objects.Migration.get_by_id_and_instance(ctxt, '1', '1')
|
|
self.compare_obj(migration, fake_migration)
|
|
|
|
def test_create_uuid_on_load(self):
|
|
values = {'source_compute': 'src',
|
|
'dest_compute': 'dst',
|
|
'source_node': 'srcnode',
|
|
'dest_node': 'dstnode',
|
|
'instance_uuid': 'fake',
|
|
'status': 'faking',
|
|
'migration_type': 'migration',
|
|
'created_at': None,
|
|
'deleted_at': None,
|
|
'updated_at': None}
|
|
db_mig = db.migration_create(self.context, values)
|
|
mig = objects.Migration.get_by_id(self.context, db_mig.id)
|
|
self.assertIn('uuid', mig)
|
|
uuid = mig.uuid
|
|
# Make sure that it was saved and we get the same one back
|
|
mig = objects.Migration.get_by_id(self.context, db_mig.id)
|
|
self.assertEqual(uuid, mig.uuid)
|
|
|
|
def test_is_same_host(self):
|
|
same_host = objects.Migration(source_compute='fake-host',
|
|
dest_compute='fake-host')
|
|
diff_host = objects.Migration(source_compute='fake-host1',
|
|
dest_compute='fake-host2')
|
|
self.assertTrue(same_host.is_same_host())
|
|
self.assertFalse(diff_host.is_same_host())
|
|
|
|
|
|
class TestMigrationObject(test_objects._LocalTest,
|
|
_TestMigrationObject):
|
|
pass
|
|
|
|
|
|
class TestRemoteMigrationObject(test_objects._RemoteTest,
|
|
_TestMigrationObject):
|
|
pass
|