| 102 | |
| 103 | |
| 104 | def test_disable_prefetch_none_preserves_behavior(mock_app, mock_consumer): |
| 105 | mock_app.conf.worker_disable_prefetch = False |
| 106 | kwargs_with_none = {'disable_prefetch': None} |
| 107 | if 'disable_prefetch' in kwargs_with_none and kwargs_with_none['disable_prefetch'] is not None: |
| 108 | mock_app.conf.worker_disable_prefetch = kwargs_with_none.pop('disable_prefetch') |
| 109 | assert mock_app.conf.worker_disable_prefetch is False |
| 110 | assert 'disable_prefetch' in kwargs_with_none |
| 111 | original_can_consume = mock_consumer.task_consumer.channel.qos.can_consume |
| 112 | tasks_instance = Tasks(mock_consumer) |
| 113 | tasks_instance.start(mock_consumer) |
| 114 | assert mock_consumer.task_consumer.channel.qos.can_consume == original_can_consume |
| 115 | |
| 116 | |
| 117 | def test_disable_prefetch_ignored_for_non_redis_brokers(mock_app, mock_consumer): |