Skip to content

Commit

Permalink
test: add testcase for expire when set config with zero.
Browse files Browse the repository at this point in the history
  • Loading branch information
laixintao committed Sep 30, 2020
1 parent 8699e65 commit f32e536
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions t/unit/backends/test_redis.py
Expand Up @@ -712,6 +712,24 @@ def test_on_chord_part_return_no_expiry(self, restore):

self.b.expires = old_expires

@patch('celery.result.GroupResult.restore')
def test_on_chord_part_return_expire_set_to_zero(self, restore):
old_expires = self.b.expires
self.b.expires = 0
tasks = [self.create_task(i) for i in range(10)]

for i in range(10):
self.b.on_chord_part_return(tasks[i].request, states.SUCCESS, i)
assert self.b.client.zadd.call_count
self.b.client.zadd.reset_mock()
assert self.b.client.zrangebyscore.call_count
jkey = self.b.get_key_for_group('group_id', '.j')
tkey = self.b.get_key_for_group('group_id', '.t')
self.b.client.delete.assert_has_calls([call(jkey), call(tkey)])
self.b.client.expire.assert_not_called()

self.b.expires = old_expires

@patch('celery.result.GroupResult.restore')
def test_on_chord_part_return_no_expiry__unordered(self, restore):
self.app.conf.result_backend_transport_options = dict(
Expand Down

0 comments on commit f32e536

Please sign in to comment.