def test_uninit_other_repo(self):
self.driver.init_repo()
other_patchrepo = PatchRepository(repo_name='test_repo2')
other_driver = Driver(self.sess, other_patchrepo)
other_driver.init_repo()
self.assert_tables_exist(['migrate_repositories',
'migrate_applied_patches'])
dbrepos = self.sess.query(Repository.repository_name).all()
eq_(set(dbrepos), set([('test_repo',), ('test_repo2',)]))
self.driver.uninit_repo()
self.assert_tables_exist(['migrate_repositories',
'migrate_applied_patches'])
dbrepos = self.sess.query(Repository.repository_name).all()
eq_(dbrepos, [('test_repo2',)])
other_driver.uninit_repo()
self.assert_tables_not_exist(['migrate_repositories',
'migrate_applied_patches'])
class TestDriver(object):
def __init__(self):
self.engine = None
self.sess = None
self.driver = None
self.patchrepo = None
self.patch1 = None
self.patch2 = None
self.patch3 = None
def setUp(self):
self.engine = create_engine('sqlite://',
listeners=[SQLiteForeignKeysListener()])
Session = sessionmaker(bind=self.engine)
self.sess = Session()
self.patchrepo = PatchRepository(repo_name='test_repo')
self.driver = Driver(self.sess, self.patchrepo)
self.patch1 = None
self.patch2 = None
self.patch3 = None
def init_repo(self):
self.driver.init_repo()
self.patch1 = Patch('patch1', depends_on_names=[('patch2', False)],
upgrade_sql='CREATE TABLE t1(a integer);\n' + \
'CREATE TABLE t12(a integer);',
downgrade_sql='DROP TABLE t1; DROP TABLE t12;')
self.patch2 = Patch('patch2', depends_on_names=[('patch3', False)],
upgrade_sql='CREATE TABLE t2(a integer);',
downgrade_sql='DROP TABLE t2;')
self.patch3 = Patch('patch3', depends_on_names=[],
upgrade_sql='CREATE TABLE t3(a integer);',
downgrade_sql='DROP TABLE t3;')
self.patchrepo.add_patches(self.patch1, self.patch2, self.patch3)
self.patchrepo.resolve_dependencies()
def assert_table_exists(self, table_name):
assert table_exists(self.sess, table_name), \
"Expected table '%s' to exist" % (table_name)
def assert_tables_exist(self, table_names):
for table_name in table_names:
self.assert_table_exists(table_name)
def assert_table_not_exists(self, table_name):
assert not table_exists(self.sess, table_name), \
"Expected table '%s' to NOT exist" % (table_name)
def assert_tables_not_exist(self, table_names):
for table_name in table_names:
self.assert_table_not_exists(table_name)
def test_init(self):
self.assert_tables_not_exist(['migrate_repositories',
'migrate_applied_patches'])
self.driver.init_repo()
dbrepos = self.sess.query(Repository.repository_name).all()
eq_(dbrepos, [('test_repo',)])
self.assert_tables_exist(['migrate_repositories',
'migrate_applied_patches'])
def test_uninit(self):
self.driver.init_repo()
self.assert_tables_exist(['migrate_repositories',
'migrate_applied_patches'])
self.driver.uninit_repo()
self.assert_tables_not_exist(['migrate_repositories',
'migrate_applied_patches'])
def test_uninit_with_patches(self):
self.init_repo()
self.assert_tables_exist(['migrate_repositories',
'migrate_applied_patches'])
self.driver.upgrade_patches(patches=[self.patch3], execute_sql=False)
self.sess.flush()
self.driver.uninit_repo()
self.assert_tables_not_exist(['migrate_repositories',
'migrate_applied_patches'])
def test_uninit_other_repo(self):
self.driver.init_repo()
other_patchrepo = PatchRepository(repo_name='test_repo2')
other_driver = Driver(self.sess, other_patchrepo)
other_driver.init_repo()
self.assert_tables_exist(['migrate_repositories',
'migrate_applied_patches'])
dbrepos = self.sess.query(Repository.repository_name).all()
eq_(set(dbrepos), set([('test_repo',), ('test_repo2',)]))
self.driver.uninit_repo()
self.assert_tables_exist(['migrate_repositories',
'migrate_applied_patches'])
dbrepos = self.sess.query(Repository.repository_name).all()
eq_(dbrepos, [('test_repo2',)])
other_driver.uninit_repo()
self.assert_tables_not_exist(['migrate_repositories',
'migrate_applied_patches'])
def test_upgrade_to_patch(self):
self.init_repo()
eq_(self.driver.applied_patches, [])
self.assert_tables_not_exist(['t1', 't12', 't2', 't3'])
self.driver.upgrade_patches([self.patch2])
eq_(set(self.driver.applied_patches), set([self.patch2, self.patch3]))
self.assert_tables_not_exist(['t1', 't12'])
self.assert_tables_exist(['t2', 't3'])
def test_upgrade_from_empty(self):
self.init_repo()
eq_(self.driver.applied_patches, [])
self.assert_tables_not_exist(['t1', 't12', 't2', 't3'])
self.driver.upgrade()
eq_(set(self.driver.applied_patches), set([self.patch1, self.patch2,
self.patch3]))
self.assert_tables_exist(['t1', 't12', 't2', 't3'])
def test_downgrade_patch(self):
self.init_repo()
self.driver.upgrade_patches([self.patch1, self.patch2, self.patch3])
eq_(self.driver.applied_patches, [self.patch1, self.patch2,
self.patch3])
self.assert_tables_exist(['t1', 't12', 't2', 't3'])
self.driver.downgrade_patches([self.patch2])
eq_(set(self.driver.applied_patches), set([self.patch3]))
self.assert_tables_not_exist(['t1', 't12', 't2'])
self.assert_tables_exist(['t3'])
def test_downgrade(self):
self.init_repo()
self.driver.upgrade_patches([self.patch1, self.patch2, self.patch3])
eq_(self.driver.applied_patches, [self.patch1, self.patch2,
self.patch3])
self.assert_tables_exist(['t1', 't12', 't2', 't3'])
self.driver.downgrade()
eq_(set(self.driver.applied_patches), set([]))
self.assert_tables_not_exist(['t1', 't12', 't2', 't3'])
def test_empty_upgrade(self):
self.driver.init_repo()
self.patch1 = Patch('patch1', depends_on_names=[('patch2', False)])
self.patch2 = Patch('patch2', depends_on_names=[('patch3', False)])
self.patch3 = Patch('patch3', depends_on_names=[])
self.patchrepo.add_patches(self.patch1, self.patch2, self.patch3)
self.patchrepo.resolve_dependencies()
self.driver.upgrade()
eq_(self.driver.applied_patches, [self.patch1, self.patch2,
self.patch3])
self.driver.downgrade()
eq_(set(self.driver.applied_patches), set([]))
def test_calculate_minimal_deps(self):
self.init_repo()
patch4 = Patch('patch4', depends_on_names=[('patch5', False)])
patch5 = Patch('patch5', depends_on_names=[])
self.patchrepo.add_patches(patch4, patch5)
self.patchrepo.resolve_dependencies()
eq_(set(self.driver.calculate_minimal_deps(patches=[self.patch1,
self.patch2, self.patch3, patch4, patch5])), set([self.patch1,
patch4]))