Ejemplo n.º 1
0
class TestRedisMultiprocessingLimiter(object):
    @patch("cowboycushion.multiprocessing_limiter.StrictRedis",
           mock_strict_redis_client)
    def test_setup(self):
        self.client = MultiprocessingMockClient()
        self.limited_client = RedisMultiprocessingLimiter(
            self.client, TIMEOUT, CALLS_PER_BATCH, SECONDS_PER_BATCH,
            REDIS_HOSTNAME, REDIS_PORT, REDIS_DB, POOL_SIZE)
        job = self.limited_client.do_stuff()
        assert_true(job.get())
        eq_(self.limited_client.call_count, 1)
        assert_true(self.limited_client._verify_we_can_make_call())
        self.limited_client.close()
        self.limited_client.join()
class TestRedisMultiprocessingLimiter(object):
    @patch("cowboycushion.multiprocessing_limiter.StrictRedis", mock_strict_redis_client)
    def test_setup(self):
        self.client = MultiprocessingMockClient()
        self.limited_client = RedisMultiprocessingLimiter(
            self.client,
            TIMEOUT,
            CALLS_PER_BATCH,
            SECONDS_PER_BATCH,
            REDIS_HOSTNAME,
            REDIS_PORT,
            REDIS_DB,
            POOL_SIZE
        )
        job = self.limited_client.do_stuff()
        assert_true(job.get())
        eq_(self.limited_client.call_count, 1)
        assert_true(self.limited_client._verify_we_can_make_call())
        self.limited_client.close()
        self.limited_client.join()