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!")
