我创建了一个依赖于其他模块的芹菜任务。我在模型保存方法中延迟了芹菜任务,我想要通过所有没有芹菜的测试用例。以下是我的代码。
def save(self, force_insert=False, force_update=False, using=None,
update_fields=None):
super(Offers, self).save(force_insert=False, force_update=False, using=None,
update_fields=None)
change_offer_status.apply_async(args=[self.id], eta=self.valid_to, queue='credr_core_task',
routing_key='credr_core_task')
Test.py
class OfferTests(APITestCase):
authenticated_client = APIClient()
def setUp(self):
data = {
"username": "vivek",
"first_name": "Vivek",
"last_name": "Dogra",
"email": "vivk@credr.com",
"contact_number": "9834982602",
"password": "easy"
}
self.user = User.objects.create(data)
token = Token.objects.get(user__username='vivek')
self.authenticated_client.credentials(HTTP_AUTHORIZATION='Token ' + token.key)
mock_task = Mock()
mock_task.get = Mock(return_value={'success': True})
print mock_task.get() # outputs {'success': True}
with patch('offers.tasks.change_offer_status.apply_async', new=mock_task) as mocked_task:
mocked_task.return_value = True
@override_settings(CELERY_EAGER_PROPAGATES_EXCEPTIONS=True,
CELERY_ALWAYS_EAGER=True, )
def test_add_offer(self):
"""
add user address
"""
start_date = datetime.strftime(datetime.now(), "%Y-%m-%d %H:%M:%S")
end_date = datetime.strftime(datetime.now() + timedelta(days=3), "%Y-%m-%d %H:%M:%S")
data = {
"type": OfferTypeEnum.FLAT._value_,
"name": "CredR Offer",
"code": "FLAT100",
"status": OfferStatusEnum.ACTIVE._value_,
"value": "1004",
"discount_value": 1004,
"valid_from": start_date,
"valid_to": end_date,
"message": "GET FLAT 100"
}
response = self.authenticated_client.post(URL, data, format='json')
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
我在setUp方法中创建了一个模拟对象。但我得到兔子连接拒绝错误。
答案 0 :(得分:3)
我想通过所有没有芹菜的测试用例
运行测试时设置CELERY_ALWAYS_EAGER = True
。
[...]任务将在本地执行,而不是发送到队列。
在此处阅读其他配置选项也可能需要付费:http://docs.celeryproject.org/en/latest/configuration.html#celery-always-eager
答案 1 :(得分:0)
用你的话说:“ ...我想通过所有测试用例而不用芹菜。”和“ ...但是我收到RabbitMq的连接被拒绝错误”,我想您正在尝试运行测试而不会触发REAL芹菜任务。
如果是这样,则您不想模拟apply_async
。真正的要求是防止芹菜将消息发送到队列中以触发任务。因此,您可能想模拟您的celery应用程序的 send_task
。
假设您在celeryapp
模块中拥有offer
个实例。您可以尝试使用
patch('offers.celeryapp.send_task', new=mock_task)
替换
patch('offers.tasks.change_offer_status.apply_async', new=mock_task)
。
那么您的celery应用程序不应发送消息。
如果您使用的是Celery 4+(我忘记了详细版本),则需要将CELERY_ALWAYS_EAGER
更改为CELERY_TASK_ALWAYS_EAGER