copilot-swe-agent[bot]
43bc58770d
Co-authored-by: christianlouis <361235+christianlouis@users.noreply.github.com>
97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
"""
|
|
Tests for app/celery_worker.py
|
|
|
|
This module tests the Celery worker configuration, task imports, and beat schedule.
|
|
"""
|
|
|
|
import pytest
|
|
|
|
|
|
@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"
|