class TestRedisRetryMiddleware(TestCase): def setUp(self): self.retry = RedisRetryMiddleware(MagicMock()) self.retry.max_retry_times = 1 def test_retries(self): req = Request('http://example.com') req.meta['retry_times'] = 0 req.meta['priority'] = 70 # number of retries less than max out = self.retry._retry(req, 'stuff', MagicMock()) self.assertEqual(req.url, out.url) self.assertEqual(60, out.meta['priority']) # over max self.assertEqual(self.retry._retry(out, 'stuff', MagicMock()), None)
def setUp(self): self.retry = RedisRetryMiddleware(MagicMock()) self.retry.max_retry_times = 1