🔥 Remove Celery and Flower, they are currently not used nor recommended (#694)
This commit is contained in:

committed by
GitHub

parent
63eb8b5ef8
commit
1e256bce5d
@@ -2,26 +2,12 @@ from fastapi import APIRouter, Depends
|
||||
from pydantic.networks import EmailStr
|
||||
|
||||
from app.api.deps import get_current_active_superuser
|
||||
from app.core.celery_app import celery_app
|
||||
from app.models import Message
|
||||
from app.utils import generate_test_email, send_email
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
|
||||
@router.post(
|
||||
"/test-celery/",
|
||||
dependencies=[Depends(get_current_active_superuser)],
|
||||
status_code=201,
|
||||
)
|
||||
def test_celery(body: Message) -> Message:
|
||||
"""
|
||||
Test Celery worker.
|
||||
"""
|
||||
celery_app.send_task("app.worker.test_celery", args=[body.message])
|
||||
return Message(message="Word received")
|
||||
|
||||
|
||||
@router.post(
|
||||
"/test-email/",
|
||||
dependencies=[Depends(get_current_active_superuser)],
|
||||
|
@@ -1,5 +0,0 @@
|
||||
from celery import Celery
|
||||
|
||||
celery_app = Celery("worker", broker="amqp://guest@queue//")
|
||||
|
||||
celery_app.conf.task_routes = {"app.worker.test_celery": "main-queue"}
|
@@ -1,16 +0,0 @@
|
||||
from fastapi.testclient import TestClient
|
||||
|
||||
from app.core.config import settings
|
||||
|
||||
|
||||
def test_celery_worker_test(
|
||||
client: TestClient, superuser_token_headers: dict[str, str]
|
||||
) -> None:
|
||||
data = {"message": "test"}
|
||||
r = client.post(
|
||||
f"{settings.API_V1_STR}/utils/test-celery/",
|
||||
json=data,
|
||||
headers=superuser_token_headers,
|
||||
)
|
||||
response = r.json()
|
||||
assert response["message"] == "Word received"
|
@@ -1,33 +0,0 @@
|
||||
from unittest.mock import MagicMock
|
||||
|
||||
from pytest_mock import MockerFixture
|
||||
from sqlmodel import select
|
||||
|
||||
from app.celeryworker_pre_start import init, logger
|
||||
|
||||
|
||||
def test_init_successful_connection(mocker: MockerFixture) -> None:
|
||||
engine_mock = MagicMock()
|
||||
|
||||
session_mock = MagicMock()
|
||||
exec_mock = MagicMock(return_value=True)
|
||||
session_mock.configure_mock(**{"exec.return_value": exec_mock})
|
||||
mocker.patch("sqlmodel.Session", return_value=session_mock)
|
||||
|
||||
mocker.patch.object(logger, "info")
|
||||
mocker.patch.object(logger, "error")
|
||||
mocker.patch.object(logger, "warn")
|
||||
|
||||
try:
|
||||
init(engine_mock)
|
||||
connection_successful = True
|
||||
except Exception:
|
||||
connection_successful = False
|
||||
|
||||
assert (
|
||||
connection_successful
|
||||
), "The database connection should be successful and not raise an exception."
|
||||
|
||||
assert session_mock.exec.called_once_with(
|
||||
select(1)
|
||||
), "The session should execute a select statement once."
|
@@ -1,12 +0,0 @@
|
||||
import sentry_sdk
|
||||
|
||||
from app.core.celery_app import celery_app
|
||||
from app.core.config import settings
|
||||
|
||||
if settings.SENTRY_DSN:
|
||||
sentry_sdk.init(dsn=str(settings.SENTRY_DSN))
|
||||
|
||||
|
||||
@celery_app.task(acks_late=True)
|
||||
def test_celery(word: str) -> str:
|
||||
return f"test task return {word}"
|
Reference in New Issue
Block a user