pyblackbird_cc/users/tests/test_tasks.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

import pytest
from celery.result import EagerResult

from pyblackbird_cc.users.tasks import get_users_count
from pyblackbird_cc.users.tests.factories import UserFactory

pytestmark = pytest.mark.django_db


def test_user_count(settings):
    """A basic test to execute the get_users_count Celery task."""
    batch_size = 3
    UserFactory.create_batch(batch_size)
    settings.CELERY_TASK_ALWAYS_EAGER = True
    task_result = get_users_count.delay()
    assert isinstance(task_result, EagerResult)
    assert task_result.result == batch_size