fix: Add missing commas in function calls and improve code formatting
This commit is contained in:
@@ -409,7 +409,7 @@ class TestCreditService:
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_recharge_user_credits_success(
|
||||
self, credit_service, sample_user
|
||||
self, credit_service, sample_user,
|
||||
) -> None:
|
||||
"""Test successful credit recharge for a user."""
|
||||
mock_session = credit_service.db_session_factory()
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
"""Tests for dashboard service."""
|
||||
|
||||
from datetime import UTC, datetime, timedelta
|
||||
from datetime import UTC, datetime
|
||||
from unittest.mock import AsyncMock, Mock, patch
|
||||
|
||||
import pytest
|
||||
@@ -63,7 +63,7 @@ class TestDashboardService:
|
||||
):
|
||||
"""Test getting soundboard statistics with exception."""
|
||||
mock_sound_repository.get_soundboard_statistics = AsyncMock(
|
||||
side_effect=Exception("Database error")
|
||||
side_effect=Exception("Database error"),
|
||||
)
|
||||
|
||||
with pytest.raises(Exception, match="Database error"):
|
||||
@@ -105,7 +105,7 @@ class TestDashboardService:
|
||||
):
|
||||
"""Test getting track statistics with exception."""
|
||||
mock_sound_repository.get_track_statistics = AsyncMock(
|
||||
side_effect=Exception("Database error")
|
||||
side_effect=Exception("Database error"),
|
||||
)
|
||||
|
||||
with pytest.raises(Exception, match="Database error"):
|
||||
@@ -198,7 +198,7 @@ class TestDashboardService:
|
||||
):
|
||||
"""Test getting top sounds with exception."""
|
||||
mock_sound_repository.get_top_sounds = AsyncMock(
|
||||
side_effect=Exception("Database error")
|
||||
side_effect=Exception("Database error"),
|
||||
)
|
||||
|
||||
with pytest.raises(Exception, match="Database error"):
|
||||
@@ -274,4 +274,4 @@ class TestDashboardService:
|
||||
def test_get_date_filter_unknown_period(self, dashboard_service):
|
||||
"""Test date filter for unknown period."""
|
||||
result = dashboard_service._get_date_filter("unknown_period")
|
||||
assert result is None
|
||||
assert result is None
|
||||
|
||||
@@ -25,9 +25,9 @@ class TestSchedulerService:
|
||||
@pytest.mark.asyncio
|
||||
async def test_start_scheduler(self, scheduler_service) -> None:
|
||||
"""Test starting the scheduler service."""
|
||||
with patch.object(scheduler_service.scheduler, 'add_job') as mock_add_job, \
|
||||
patch.object(scheduler_service.scheduler, 'start') as mock_start:
|
||||
|
||||
with patch.object(scheduler_service.scheduler, "add_job") as mock_add_job, \
|
||||
patch.object(scheduler_service.scheduler, "start") as mock_start:
|
||||
|
||||
await scheduler_service.start()
|
||||
|
||||
# Verify job was added
|
||||
@@ -47,7 +47,7 @@ class TestSchedulerService:
|
||||
@pytest.mark.asyncio
|
||||
async def test_stop_scheduler(self, scheduler_service) -> None:
|
||||
"""Test stopping the scheduler service."""
|
||||
with patch.object(scheduler_service.scheduler, 'shutdown') as mock_shutdown:
|
||||
with patch.object(scheduler_service.scheduler, "shutdown") as mock_shutdown:
|
||||
await scheduler_service.stop()
|
||||
mock_shutdown.assert_called_once()
|
||||
|
||||
@@ -61,7 +61,7 @@ class TestSchedulerService:
|
||||
"total_credits_added": 500,
|
||||
}
|
||||
|
||||
with patch.object(scheduler_service.credit_service, 'recharge_all_users_credits') as mock_recharge:
|
||||
with patch.object(scheduler_service.credit_service, "recharge_all_users_credits") as mock_recharge:
|
||||
mock_recharge.return_value = mock_stats
|
||||
|
||||
await scheduler_service._daily_credit_recharge()
|
||||
@@ -71,10 +71,10 @@ class TestSchedulerService:
|
||||
@pytest.mark.asyncio
|
||||
async def test_daily_credit_recharge_failure(self, scheduler_service) -> None:
|
||||
"""Test daily credit recharge task with failure."""
|
||||
with patch.object(scheduler_service.credit_service, 'recharge_all_users_credits') as mock_recharge:
|
||||
with patch.object(scheduler_service.credit_service, "recharge_all_users_credits") as mock_recharge:
|
||||
mock_recharge.side_effect = Exception("Database error")
|
||||
|
||||
# Should not raise exception, just log it
|
||||
await scheduler_service._daily_credit_recharge()
|
||||
|
||||
mock_recharge.assert_called_once()
|
||||
mock_recharge.assert_called_once()
|
||||
|
||||
Reference in New Issue
Block a user