def test_downgrade():
patchrepo = PatchRepository()
patch1 = Patch('patch1', depends_on_names=[('patch2', False)])
patch2 = Patch('patch2', depends_on_names=[('patch3', False),
('patch4', False)])
patch3 = Patch('patch3', depends_on_names=[('patch4', False),
('patch5', False)])
patch4 = Patch('patch4', depends_on_names=[])
patch5 = Patch('patch5', depends_on_names=[])
patchrepo.add_patches(patch1, patch2, patch3, patch4, patch5)
patchrepo.resolve_dependencies()
applied_patches = [patch1, patch2, patch3, patch4, patch5]
plan = generate_downgrade_plan(applied_patches=applied_patches,
to_be_removed_patches=[patch1])
eq_(plan, [patch1])
plan = generate_downgrade_plan(applied_patches=applied_patches,
to_be_removed_patches=[patch3])
eq_(plan, [patch1, patch2, patch3])
plan = generate_downgrade_plan(applied_patches=applied_patches,
to_be_removed_patches=[patch5])
eq_(plan, [patch1, patch2, patch3, patch5])
plan = generate_downgrade_plan(applied_patches=applied_patches,
to_be_removed_patches=[patch5, patch4])
assert plan == [patch1, patch2, patch3, patch4, patch5] or \
plan == [patch1, patch2, patch3, patch5, patch4]
def test_complex_upgrade_from_empty():
patchrepo = PatchRepository()
patch1 = Patch('patch1', depends_on_names=[('patch2', False)])
patch2 = Patch('patch2', depends_on_names=[('patch3', False), ('patch4', False)])
patch3 = Patch('patch3', depends_on_names=[('patch4', False), ('patch5', False)])
patch4 = Patch('patch4', depends_on_names=[])
patch5 = Patch('patch5', depends_on_names=[])
patchrepo.add_patches(patch1, patch2, patch3, patch4, patch5)
patchrepo.resolve_dependencies()
plan = generate_upgrade_plan(applied_patches=[],
to_be_applied_patches=[patch1])
eq_(plan, [patch4, patch5, patch3, patch2, patch1])
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]))