def test__apply_patches_no_patches(self):
redis = self.mox.CreateMockAnything()
redis.smembers('key').AndReturn([])
self.mox.StubOutWithMock(patch.glob, 'glob')
patch.glob.glob(mox.IgnoreArg()).AndReturn([])
self.mox.ReplayAll()
patch._apply_patches(redis, 'key', 'dir', 'mod')
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 test__apply_patches(self):
redis = self.mox.CreateMockAnything()
redis.smembers('key').AndReturn(['a'])
self.mox.StubOutWithMock(patch.glob, 'glob')
patch.glob.glob(mox.IgnoreArg()).AndReturn(['c', 'a', 'b'])
self.mox.StubOutWithMock(__builtin__, '__import__')
b = self.mox.CreateMockAnything()
__builtin__.__import__('mod.b', fromlist=['mod.b']).AndReturn(b)
b.main()
redis.sadd('key', 'b')
c = self.mox.CreateMockAnything()
__builtin__.__import__('mod.c', fromlist=['mod.c']).AndReturn(c)
c.main()
redis.sadd('key', 'c')
self.mox.ReplayAll()
patch._apply_patches(redis, 'key', 'dir', 'mod')