def test_create(self): self.settings['directory'] = '/tmp' dbmigrate = DBMigrate(**self.settings) fake_file = FakeFile() dbmigrate.create('test slug', 'sql', fake_file) self.assert_(fake_file.filename.startswith('/tmp')) self.assert_(fake_file.filename.endswith('test-slug.sql')) self.assertEqual(fake_file.contents, '-- add your migration here')
def test_null_migration_after_successful_migration(self): fixtures_path = os.path.join( os.path.dirname(__file__), 'fixtures', 'initial') self.settings['directory'] = fixtures_path self.settings['out_of_order'] = False dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate() dbmigrate.migrate()
def test_current_migrations(self): fixtures_path = os.path.join( os.path.dirname(__file__), 'fixtures', 'initial') self.settings['directory'] = fixtures_path dbmigrate = DBMigrate(**self.settings) self.assertEqual( dbmigrate.current_migrations(), [( '20120115075349-create-user-table.sql', '0187aa5e13e268fc621c894a7ac4345579cf50b7' )])
def test_failing_script_migration(self): self.settings['directory'] = os.path.join( os.path.dirname(__file__), 'fixtures', 'arbitrary-scripts-failing') dbmigrate = DBMigrate(**self.settings) try: dbmigrate.migrate() except subprocess.CalledProcessError as e: self.assert_('20121019152409-script.sh' in str(e)) else: self.fail('Expected the script to fail')
def test_initial_migration(self): fixtures_path = os.path.join( os.path.dirname(__file__), 'fixtures', 'initial') self.settings['directory'] = fixtures_path dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate() # since the database is in memory we need to reach in to get it self.assertEqual( dbmigrate.engine.performed_migrations, [( '20120115075349-create-user-table.sql', '0187aa5e13e268fc621c894a7ac4345579cf50b7' )])
def test_passing_script_migration(self): self.settings['directory'] = os.path.join( os.path.dirname(__file__), 'fixtures', 'arbitrary-scripts') dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate() self.assertEqual( dbmigrate.engine.performed_migrations, [('20121019152404-initial.sql', '4485430c4b18fdbe273a845e654c66ada42d3066'), ('20121019152409-script.sh', '837a6ab019646fae8488048e20ff2651437b2fbd'), ('20121019152412-final.sql', '4485430c4b18fdbe273a845e654c66ada42d3066')])
def test_initial_migration_without_run_all(self): fixtures_path = os.path.join( os.path.dirname(__file__), 'fixtures', 'initial') self.settings['directory'] = fixtures_path self.settings['run_for_new_db'] = False dbmigrate = DBMigrate(**self.settings) migrated = dbmigrate.migrate() self.assertEqual(migrated, 'Created migrations table\nSimulated 1 migrations:\n20120115075349-create-user-table.sql') self.assertEqual( dbmigrate.engine.performed_migrations, [( '20120115075349-create-user-table.sql', '0187aa5e13e268fc621c894a7ac4345579cf50b7' )])
def test_dry_run_migration(self): fixtures_path = os.path.join( os.path.dirname(__file__), 'fixtures', 'initial') self.settings['directory'] = fixtures_path self.settings['dry_run'] = True dbmigrate = DBMigrate(**self.settings) self.assertEqual(dbmigrate.migrate(), ( """sql: -- intentionally making this imperfect so it can be migrated CREATE TABLE users ( id int PRIMARY KEY, name varchar(255), password_sha1 varchar(40) ); migration info: INSERT INTO dbmigration (filename, sha1, date) VALUES ('20120115075349-create-user-table.sql', '0187aa5e13e268fc621c894a7ac4345579cf50b7', %s());""" % dbmigrate.engine.date_func))
def test_multiple_migration_dry_run(self): fixtures_path = os.path.join( os.path.dirname(__file__), 'fixtures', 'second-run') self.settings['directory'] = fixtures_path self.settings['dry_run'] = True dbmigrate = DBMigrate(**self.settings) self.assertEqual(dbmigrate.migrate(), ( """sql: -- intentionally making this imperfect so it can be migrated CREATE TABLE users ( id int PRIMARY KEY, name varchar(255), password_sha1 varchar(40) ); migration info: INSERT INTO dbmigration (filename, sha1, date) VALUES ('20120115075349-create-user-table.sql', '0187aa5e13e268fc621c894a7ac4345579cf50b7', %(date_func)s()); sql: ALTER TABLE users ADD COLUMN email varchar(70); migration info: INSERT INTO dbmigration (filename, sha1, date) VALUES ('20120603133552-awesome.sql', '6759512e1e29b60a82b4a5587c5ea18e06b7d381', %(date_func)s());""" % {'date_func': dbmigrate.engine.date_func}))
def test_ignore_filenames_sha1_migration(self): self.settings['directory'] = os.path.join( os.path.dirname(__file__), 'fixtures', 'sha1-update-1') dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate() dbmigrate.directory = os.path.join( os.path.dirname(__file__), 'fixtures', 'sha1-update-2') dbmigrate.renamed() dbmigrate.migrate() self.assertEqual( dbmigrate.engine.performed_migrations, [('20120115075300-add-another-test-table-renamed-reordered.sql', '4aebd2514665effff5105ad568a4fbe62f567087'), ('20120115075349-create-user-table.sql', '0187aa5e13e268fc621c894a7ac4345579cf50b7')])
def test_null_dry_run_migration(self): self.settings['directory'] = os.path.join( os.path.dirname(__file__), 'fixtures', 'second-run') dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate() self.settings['dry_run'] = True dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate()
def test_multiple_migrations(self): self.settings['directory'] = os.path.join( os.path.dirname(__file__), 'fixtures', 'initial') dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate() dbmigrate.directory = os.path.join( os.path.dirname(__file__), 'fixtures', 'second-run') dbmigrate.migrate() self.assertEqual( dbmigrate.engine.performed_migrations, [('20120115075349-create-user-table.sql', '0187aa5e13e268fc621c894a7ac4345579cf50b7'), ('20120603133552-awesome.sql', '6759512e1e29b60a82b4a5587c5ea18e06b7d381')])
def test_deleted_migrations_detected(self): fixtures_path = os.path.join( os.path.dirname(__file__), 'fixtures', 'deleted-1') self.settings['directory'] = fixtures_path dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate() dbmigrate.directory = os.path.join( os.path.dirname(__file__), 'fixtures', 'deleted-2') try: dbmigrate.migrate() self.fail('Expected a ModifiedMigrationException') except ModifiedMigrationException as e: self.assertEqual( str(e), ('[20120115221757-initial.sql] migrations were ' 'deleted since they were run on this database.'))
def test_allowed_out_of_order_migration(self): fixtures_path = os.path.join( os.path.dirname(__file__), 'fixtures', 'out-of-order-1') self.settings['directory'] = fixtures_path self.settings['out_of_order'] = True dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate() dbmigrate.directory = os.path.join( os.path.dirname(__file__), 'fixtures', 'out-of-order-2') dbmigrate.migrate() self.assertEqual( dbmigrate.engine.performed_migrations, [('20120114221757-before-initial.sql', 'c7fc17564f24f7b960e9ef3f6f9130203cc87dc9'), ('20120115221757-initial.sql', '841ea60d649264965a3e8c8a955fd7aad54dad3e')])
def test_out_of_order_migration(self): fixtures_path = os.path.join( os.path.dirname(__file__), 'fixtures', 'out-of-order-1') self.settings['directory'] = fixtures_path dbmigrate = DBMigrate(**self.settings) dbmigrate.migrate() dbmigrate.directory = os.path.join( os.path.dirname(__file__), 'fixtures', 'out-of-order-2') try: dbmigrate.migrate() self.fail('Expected an OutOfOrder exception') except OutOfOrderException as e: self.assertEqual( str(e), ('[20120114221757-before-initial.sql] ' 'older than the latest performed migration'))