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
'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',