e009eed875
Model migration tests were not run because of incorrect import. Run them in unit test mode instead of functional jobs. Also fix existing migrations and models which are not in sync. Change-Id: I4f7362972929308201776214fe01c007f14a6580
127 lines
3.9 KiB
Python
127 lines
3.9 KiB
Python
# Copyright 2015 VMware, Inc.
|
|
#
|
|
# All Rights Reserved.
|
|
#
|
|
# 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_config import cfg
|
|
|
|
from neutron.db.migration.alembic_migrations import external
|
|
from neutron.db.migration import cli as migration
|
|
from neutron.tests.functional.db import test_migrations
|
|
from neutron.tests.unit import testlib_api
|
|
|
|
from vmware_nsx.db.migration import alembic_migrations
|
|
from vmware_nsx.db.migration.models import head
|
|
|
|
#TODO(abhiraut): Remove this list from here once *aaS repos forms its
|
|
# own list.
|
|
# Add *aaS tables to EXTERNAL_TABLES since they should not be
|
|
# tested.
|
|
LBAAS_TABLES = {
|
|
'nsxv_edge_monitor_mappings',
|
|
'nsxv_edge_pool_mappings',
|
|
'nsxv_edge_vip_mappings',
|
|
|
|
# LBaaS v2 tables
|
|
'lbaas_healthmonitors',
|
|
'lbaas_l7policies',
|
|
'lbaas_l7rules',
|
|
'lbaas_listeners',
|
|
'lbaas_loadbalancer_statistics',
|
|
'lbaas_loadbalanceragentbindings',
|
|
'lbaas_loadbalancers',
|
|
'lbaas_members',
|
|
'lbaas_pools',
|
|
'lbaas_sessionpersistences',
|
|
'lbaas_sni',
|
|
}
|
|
|
|
L2GW_TABLES = {
|
|
'l2gw_alembic_version',
|
|
'physical_locators',
|
|
'physical_switches',
|
|
'physical_ports',
|
|
'logical_switches',
|
|
'ucast_macs_locals',
|
|
'ucast_macs_remotes',
|
|
'vlan_bindings',
|
|
'l2gatewayconnections',
|
|
'l2gatewayinterfaces',
|
|
'l2gatewaydevices',
|
|
'l2gateways',
|
|
'pending_ucast_macs_remotes'
|
|
}
|
|
|
|
SFC_TABLES = {
|
|
'sfc_flow_classifier_l7_parameters',
|
|
'sfc_flow_classifiers',
|
|
'sfc_port_chain_parameters',
|
|
'sfc_service_function_params',
|
|
'sfc_port_pair_group_params',
|
|
'sfc_chain_classifier_associations',
|
|
'sfc_port_pairs',
|
|
'sfc_chain_group_associations',
|
|
'sfc_port_pair_groups',
|
|
'sfc_port_chains',
|
|
'sfc_uuid_intid_associations',
|
|
'sfc_path_port_associations',
|
|
'sfc_portpair_details',
|
|
'sfc_path_nodes',
|
|
}
|
|
|
|
TAAS_TABLES = {
|
|
'tap_services',
|
|
'tap_flows',
|
|
'tap_id_associations',
|
|
}
|
|
|
|
# EXTERNAL_TABLES should contain all names of tables that are not related to
|
|
# current repo.
|
|
EXTERNAL_TABLES = (set(external.TABLES) | LBAAS_TABLES |
|
|
L2GW_TABLES | SFC_TABLES | TAAS_TABLES)
|
|
|
|
|
|
class _TestModelsMigrationsFoo(test_migrations._TestModelsMigrations):
|
|
|
|
def db_sync(self, engine):
|
|
cfg.CONF.set_override('connection', engine.url, group='database')
|
|
for conf in migration.get_alembic_configs():
|
|
self.alembic_config = conf
|
|
self.alembic_config.neutron_config = cfg.CONF
|
|
migration.do_alembic_command(conf, 'upgrade', 'heads')
|
|
|
|
def get_metadata(self):
|
|
return head.get_metadata()
|
|
|
|
def include_object(self, object_, name, type_, reflected, compare_to):
|
|
if type_ == 'table' and (name.startswith('alembic') or
|
|
name == alembic_migrations.VERSION_TABLE or
|
|
name in EXTERNAL_TABLES):
|
|
return False
|
|
if type_ == 'index' and reflected and name.startswith("idx_autoinc_"):
|
|
return False
|
|
return True
|
|
|
|
|
|
class TestModelsMigrationsMysql(testlib_api.MySQLTestCaseMixin,
|
|
_TestModelsMigrationsFoo,
|
|
testlib_api.SqlTestCaseLight):
|
|
pass
|
|
|
|
|
|
class TestModelsMigrationsPsql(testlib_api.PostgreSQLTestCaseMixin,
|
|
_TestModelsMigrationsFoo,
|
|
testlib_api.SqlTestCaseLight):
|
|
pass
|