def test_run_migration(self): self.setup_migrations() migration.run_migration('20100901T100000-first-one') h.assert_equal(migration.current_version(), '20100901T100000-first-one') h.assert_equal(mongo.db.foobar.find_one()['foo'], 'bar')
def test_up(self): self.setup_migrations() # Remove dud migrations os.unlink(os.path.join(self.dir, '20101002T113000-tertiary.py')) os.unlink(os.path.join(self.dir, '20100901T100001-CAPITALS.py')) migration.up() h.assert_equal(migration.current_version(), '20100901T103000-number-2') h.assert_equal(mongo.db.foobar.find_one()['foo'], 'baz')
def test_set_version(self): migration.set_version('foobar') h.assert_equal(migration.current_version(), 'foobar') migration.set_version('bazbar') h.assert_equal(migration.current_version(), 'bazbar')
def test_null_version(self): v = migration.current_version() h.assert_true(isinstance(v, migration.NullVersion), "Migration current version wasn't the null version!")