def test_apply_patches(self): redis = self.mox.CreateMockAnything() self.mox.StubOutWithMock(patch.db, 'redis_conn') patch.db.redis_conn().AndReturn(self.context(redis)) self.mox.StubOutWithMock(patch.orm, 'get_lock') patch.orm.get_lock('key.LOCK', redis).AndReturn(self.context()) self.mox.StubOutWithMock(patch, '_apply_patches') patch._apply_patches(redis, 'key', 'dir', 'mod') self.mox.ReplayAll() patch.apply_patches('key', 'dir', 'mod')
def apply_db_patches(): from shinkansen.orm import patch patch.apply_patches()