mirror of
https://github.com/arc53/DocsGPT.git
synced 2026-02-22 12:21:39 +00:00
some tests
This commit is contained in:
18
tests/test_celery.py
Normal file
18
tests/test_celery.py
Normal file
@@ -0,0 +1,18 @@
|
||||
from celery import Celery
|
||||
from unittest.mock import patch, MagicMock
|
||||
from application.core.settings import settings
|
||||
from application.celery import make_celery
|
||||
|
||||
|
||||
@patch('application.celery.Celery')
|
||||
def test_make_celery(mock_celery):
|
||||
# Arrange
|
||||
app_name = 'test_app_name'
|
||||
|
||||
# Act
|
||||
celery = make_celery(app_name)
|
||||
|
||||
# Assert
|
||||
mock_celery.assert_called_once_with(app_name, broker=settings.CELERY_BROKER_URL)
|
||||
celery.conf.update.assert_called_once_with(settings)
|
||||
assert celery == mock_celery.return_value
|
||||
Reference in New Issue
Block a user