diff options
Diffstat (limited to 'alphabetlearning/users/tests/test_tasks.py')
-rw-r--r-- | alphabetlearning/users/tests/test_tasks.py | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/alphabetlearning/users/tests/test_tasks.py b/alphabetlearning/users/tests/test_tasks.py new file mode 100644 index 0000000..92f0e8a --- /dev/null +++ b/alphabetlearning/users/tests/test_tasks.py @@ -0,0 +1,17 @@ +import pytest +from celery.result import EagerResult + +from alphabetlearning.users.tasks import get_users_count +from alphabetlearning.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 |