<s> import alembic <EOL> from nailgun import db <EOL> from nailgun . db . migration import make_alembic_config_from_extension <EOL> from nailgun . test import base <EOL> from . . import extension <EOL> _test_revision = '<STR_LIT>' <EOL> def setup_module ( module ) : <EOL> alembic_config = make_alembic_config_from_extension ( <EOL> extension . ClusterUpgradeExtension ) <EOL> db . dropdb ( ) <EOL> alembic . command . upgrade ( alembic_config , _test_revision ) <EOL> class TestAddRelations ( base . BaseAlembicMigrationTest ) : <EOL> def test_works_without_core_migrations ( self ) : <EOL> columns = [ <EOL> t . name for t in <EOL> self . meta . tables [ '<STR_LIT>' ] . columns ] <EOL> self . assertItemsEqual ( columns , [ <EOL> '<STR_LIT:id>' , <EOL> '<STR_LIT>' , <EOL> '<STR_LIT>' , <EOL> ] ) </s>
