Задача Celery странно ведет себя во время тестирования
Я пытаюсь проверить, что моя задача Celery обновляет модель Django. Она работает нормально, но странно ведет себя во время тестирования.
# someapp/models.py
class SomeModel(models.Model):
...
hidden = models.BooleanField(default=False)
# someapp/tasks.py
@shared_task()
def my_task(model_id):
model_instance = someapp.models.SomeModel.objects.get(id=model_id)
model_instance.hidden = True
model_instance.save()
logger.info(f'Uncovered model instance with id {model_id]')
Для проверки этого я реализовал следующий рабочий процесс:
- Я создаю объект SomeModel через фабрику-бой, потому что SomeModel зависит от нескольких моделей.
- Я присваиваю этот объект переменной model_instance .
- Я применяю задачу локально
- Я утверждаю, что model_instance.hidden равно True
Код ниже
# someapp/tests.py
@pytest.mark.django_db
@pytest.mark.celery(task_always_eager=True)
def test_celery_task_uncovers_model_instance() -> None:
SomeModelFactory.create(hidden=False)
some_model = someapp.models.SomeModel.objects.first()
assert some_model.hidden is True
my_task.apply((some_model.id, ))
assert some_model.hidden is True
возникает в последней строке. Затем я утверждаю:
assert (model_instance.pk, model_instance.hidden) == (someapp.models.SomeModel.objects.first().pk,
someapp.models.SomeModel.objects.first().hidden)
Повышает:
E assert (1, True) == (1, False)
E At index 1 diff: True != False
Наконец, я хочу проверить идентификаторы:
assert id(model_instance) == id(authapp.models.SomeModel.objects.first())
И это вызывает что-то вроде этого:
E AssertionError: assert 139938217188656 == 139938219885184
E + where 139938217188656 = id(<SomeModel: - 2022-02-01>)
E + and 139938219885184 = id(<SomeModel: - 2022-02-01>)
Почему задача не обновляет объект some_model в моем тесте?
Экземпляр был обновлен в задаче, но не в вашем тесте, вам необходимо обновить данные экземпляра из БД с помощью Model.refresh_from_db() после вызова задачи
@pytest.mark.django_db
@pytest.mark.celery(task_always_eager=True)
def test_celery_task_uncovers_model_instance() -> None:
SomeModelFactory.create(hidden=False)
some_model = someapp.models.SomeModel.objects.first()
assert some_model.hidden is False
my_task.apply((some_model.id, ))
some_model.refresh_from_db() # <<<
assert some_model.hidden is True