(app)
| 57 | |
| 58 | @contextmanager |
| 59 | def patch_unlock_retry(app): |
| 60 | unlock = app.tasks['celery.chord_unlock'] |
| 61 | retry = Mock() |
| 62 | retry.return_value = Retry() |
| 63 | prev, unlock.retry = unlock.retry, retry |
| 64 | try: |
| 65 | yield unlock, retry |
| 66 | finally: |
| 67 | unlock.retry = prev |
| 68 | |
| 69 | |
| 70 | class test_unlock_chord_task(ChordCase): |