copilot-swe-agent[bot]
0eb29fd62a
Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
"""
|
|
Tests for app/celery_worker.py
|
|
|
|
This module tests the Celery worker configuration, task imports, and beat schedule.
|
|
"""
|
|
|
|
import pytest
|
|
from unittest.mock import MagicMock, patch, PropertyMock
|
|
from celery.schedules import crontab
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestCeleryWorkerConfig:
|
|
"""Test Celery worker configuration."""
|
|
|
|
def test_test_task_function(self):
|
|
"""Test the test_task function returns expected value."""
|
|
from app.celery_worker import test_task
|
|
|
|
result = test_task()
|
|
assert result == "Celery is working!"
|
|
|
|
def test_celery_instance_exists(self):
|
|
"""Test that celery instance exists in module."""
|
|
from app import celery_worker
|
|
|
|
assert hasattr(celery_worker, 'celery')
|
|
assert celery_worker.celery is not None
|
|
|
|
def test_task_routes_exists(self):
|
|
"""Test that task routes configuration exists."""
|
|
from app import celery_worker
|
|
|
|
# Task routes should be configured
|
|
assert hasattr(celery_worker.celery.conf, 'task_routes')
|
|
|
|
def test_all_task_imports_successful(self):
|
|
"""Test that all task modules are imported successfully."""
|
|
# Just import the module to verify no import errors
|
|
from app import celery_worker
|
|
|
|
# Module imported successfully
|
|
assert celery_worker is not None
|
|
|
|
|
|
@pytest.mark.unit
|
|
class TestBeatScheduleConfiguration:
|
|
"""Test Celery beat schedule configuration."""
|
|
|
|
def test_beat_schedule_structure(self):
|
|
"""Test that beat schedule has expected structure."""
|
|
from app.celery_worker import celery
|
|
|
|
# Beat schedule should be a dictionary
|
|
assert isinstance(celery.conf.beat_schedule, dict)
|
|
|
|
# Should include credential check tasks
|
|
assert 'check-credentials-regularly' in celery.conf.beat_schedule
|
|
assert 'check-credentials-daily' in celery.conf.beat_schedule
|
|
assert 'monitor-stalled-steps' in celery.conf.beat_schedule
|
|
|
|
def test_credential_check_schedule(self):
|
|
"""Test credential check schedule configuration."""
|
|
from app.celery_worker import celery
|
|
|
|
schedule = celery.conf.beat_schedule.get('check-credentials-regularly')
|
|
assert schedule is not None
|
|
assert schedule['task'] == 'app.tasks.check_credentials.check_credentials'
|
|
assert 'schedule' in schedule
|
|
assert schedule['options']['expires'] == 240
|
|
|
|
def test_daily_credential_check_schedule(self):
|
|
"""Test daily credential check schedule."""
|
|
from app.celery_worker import celery
|
|
|
|
schedule = celery.conf.beat_schedule.get('check-credentials-daily')
|
|
assert schedule is not None
|
|
assert schedule['task'] == 'app.tasks.check_credentials.check_credentials'
|
|
assert 'schedule' in schedule
|
|
assert schedule['options']['expires'] == 3600
|
|
|
|
def test_monitor_stalled_steps_schedule(self):
|
|
"""Test monitor stalled steps schedule."""
|
|
from app.celery_worker import celery
|
|
|
|
schedule = celery.conf.beat_schedule.get('monitor-stalled-steps')
|
|
assert schedule is not None
|
|
assert schedule['task'] == 'app.tasks.monitor_stalled_steps.monitor_stalled_steps'
|
|
assert 'schedule' in schedule
|
|
assert schedule['options']['expires'] == 55
|
|
|
|
def test_no_none_entries_in_beat_schedule(self):
|
|
"""Test that None entries are filtered from beat schedule."""
|
|
from app.celery_worker import celery
|
|
|
|
# No None values in beat schedule
|
|
for key, value in celery.conf.beat_schedule.items():
|
|
assert value is not None, f"Beat schedule entry '{key}' should not be None"
|