Задача 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
Вернуться на верх