diff --git a/neutron/tests/fullstack/base.py b/neutron/tests/fullstack/base.py index 012ace848c3..2e95ba4bd24 100644 --- a/neutron/tests/fullstack/base.py +++ b/neutron/tests/fullstack/base.py @@ -15,8 +15,7 @@ from oslo_config import cfg from oslo_db.sqlalchemy import test_base -from neutron.db.migration.models import head # noqa -from neutron.db import model_base +from neutron.db.migration import cli as migration from neutron.tests.common import base from neutron.tests.fullstack.resources import client as client_resource @@ -57,11 +56,13 @@ class BaseFullStackTestCase(base.MySQLTestCase): 'password': test_base.DbFixture.PASSWORD, 'db_name': self.engine.url.database}) + alembic_config = migration.get_neutron_config() + alembic_config.neutron_config = cfg.CONF self.original_conn = cfg.CONF.database.connection self.addCleanup(self._revert_connection_address) cfg.CONF.set_override('connection', conn, group='database') - model_base.BASEV2.metadata.create_all(self.engine) + migration.do_alembic_command(alembic_config, 'upgrade', 'heads') def _revert_connection_address(self): cfg.CONF.set_override('connection',