"""Tests for VLC player API endpoints.""" # ruff: noqa: ARG002, PLR2004, E501 from unittest.mock import AsyncMock import pytest from fastapi import FastAPI from httpx import AsyncClient from app.api.v1.sounds import get_credit_service, get_sound_repository, get_vlc_player from app.models.sound import Sound from app.models.user import User class TestVLCEndpoints: """Test VLC player API endpoints.""" @pytest.mark.asyncio async def test_play_sound_with_vlc_success( self, test_app: FastAPI, authenticated_client: AsyncClient, authenticated_user: User, ) -> None: """Test successful sound playback via VLC.""" # Set up mocks mock_vlc_service = AsyncMock() mock_repo = AsyncMock() mock_credit_service = AsyncMock() # Set up test data mock_sound = Sound( id=1, type="SDB", name="Test Sound", filename="test.mp3", duration=5000, size=1024, hash="test_hash", ) # Configure mocks mock_repo.get_by_id.return_value = mock_sound mock_credit_service.validate_and_reserve_credits.return_value = None mock_credit_service.deduct_credits.return_value = None mock_vlc_service.play_sound.return_value = True # Override dependencies test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service test_app.dependency_overrides[get_sound_repository] = lambda: mock_repo test_app.dependency_overrides[get_credit_service] = lambda: mock_credit_service try: response = await authenticated_client.post("/api/v1/sounds/vlc/play/1") assert response.status_code == 200 data = response.json() assert data["sound_id"] == 1 assert data["sound_name"] == "Test Sound" assert "Test Sound" in data["message"] # Verify service calls mock_repo.get_by_id.assert_called_once_with(1) mock_vlc_service.play_sound.assert_called_once_with(mock_sound) finally: # Clean up dependency overrides (except get_db which is needed for other tests) test_app.dependency_overrides.pop(get_vlc_player, None) test_app.dependency_overrides.pop(get_sound_repository, None) test_app.dependency_overrides.pop(get_credit_service, None) @pytest.mark.asyncio async def test_play_sound_with_vlc_sound_not_found( self, test_app: FastAPI, authenticated_client: AsyncClient, authenticated_user: User, ) -> None: """Test VLC playback when sound is not found.""" # Set up mocks mock_vlc_service = AsyncMock() mock_repo = AsyncMock() mock_credit_service = AsyncMock() # Configure mocks mock_repo.get_by_id.return_value = None # Override dependencies test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service test_app.dependency_overrides[get_sound_repository] = lambda: mock_repo test_app.dependency_overrides[get_credit_service] = lambda: mock_credit_service try: response = await authenticated_client.post("/api/v1/sounds/vlc/play/999") assert response.status_code == 404 data = response.json() assert "Sound with ID 999 not found" in data["detail"] finally: # Clean up dependency overrides (except get_db which is needed for other tests) test_app.dependency_overrides.pop(get_vlc_player, None) test_app.dependency_overrides.pop(get_sound_repository, None) test_app.dependency_overrides.pop(get_credit_service, None) @pytest.mark.asyncio async def test_play_sound_with_vlc_launch_failure( self, test_app: FastAPI, authenticated_client: AsyncClient, authenticated_user: User, ) -> None: """Test VLC playback when VLC launch fails.""" # Set up mocks mock_vlc_service = AsyncMock() mock_repo = AsyncMock() mock_credit_service = AsyncMock() # Set up test data mock_sound = Sound( id=1, type="SDB", name="Test Sound", filename="test.mp3", duration=5000, size=1024, hash="test_hash", ) # Configure mocks mock_repo.get_by_id.return_value = mock_sound mock_credit_service.validate_and_reserve_credits.return_value = None mock_credit_service.deduct_credits.return_value = None mock_vlc_service.play_sound.return_value = False # Override dependencies test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service test_app.dependency_overrides[get_sound_repository] = lambda: mock_repo test_app.dependency_overrides[get_credit_service] = lambda: mock_credit_service try: response = await authenticated_client.post("/api/v1/sounds/vlc/play/1") assert response.status_code == 500 data = response.json() assert "Failed to launch VLC for sound playback" in data["detail"] finally: # Clean up dependency overrides (except get_db which is needed for other tests) test_app.dependency_overrides.pop(get_vlc_player, None) test_app.dependency_overrides.pop(get_sound_repository, None) test_app.dependency_overrides.pop(get_credit_service, None) @pytest.mark.asyncio async def test_play_sound_with_vlc_service_exception( self, test_app: FastAPI, authenticated_client: AsyncClient, authenticated_user: User, ) -> None: """Test VLC playback when service raises an exception.""" # Set up mocks mock_vlc_service = AsyncMock() mock_repo = AsyncMock() mock_credit_service = AsyncMock() # Configure mocks mock_repo.get_by_id.side_effect = Exception("Database error") # Override dependencies test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service test_app.dependency_overrides[get_sound_repository] = lambda: mock_repo test_app.dependency_overrides[get_credit_service] = lambda: mock_credit_service try: response = await authenticated_client.post("/api/v1/sounds/vlc/play/1") assert response.status_code == 500 data = response.json() assert "Failed to play sound" in data["detail"] finally: # Clean up dependency overrides (except get_db which is needed for other tests) test_app.dependency_overrides.pop(get_vlc_player, None) test_app.dependency_overrides.pop(get_sound_repository, None) test_app.dependency_overrides.pop(get_credit_service, None) @pytest.mark.asyncio async def test_play_sound_with_vlc_unauthenticated( self, client: AsyncClient, ) -> None: """Test VLC playback without authentication.""" response = await client.post("/api/v1/sounds/vlc/play/1") assert response.status_code == 401 @pytest.mark.asyncio async def test_stop_all_vlc_instances_success( self, test_app: FastAPI, authenticated_client: AsyncClient, authenticated_user: User, ) -> None: """Test successful stopping of all VLC instances.""" # Set up mock mock_vlc_service = AsyncMock() mock_result = { "success": True, "processes_found": 3, "processes_killed": 3, "processes_remaining": 0, "message": "Killed 3 VLC processes", } mock_vlc_service.stop_all_vlc_instances.return_value = mock_result # Override dependency test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service try: response = await authenticated_client.post("/api/v1/sounds/vlc/stop-all") assert response.status_code == 200 data = response.json() assert data["success"] is True assert data["processes_found"] == 3 assert data["processes_killed"] == 3 assert data["processes_remaining"] == 0 assert "Killed 3 VLC processes" in data["message"] # Verify service call mock_vlc_service.stop_all_vlc_instances.assert_called_once() finally: # Clean up dependency override test_app.dependency_overrides.pop(get_vlc_player, None) @pytest.mark.asyncio async def test_stop_all_vlc_instances_no_processes( self, test_app: FastAPI, authenticated_client: AsyncClient, authenticated_user: User, ) -> None: """Test stopping VLC instances when none are running.""" # Set up mock mock_vlc_service = AsyncMock() mock_result = { "success": True, "processes_found": 0, "processes_killed": 0, "message": "No VLC processes found", } mock_vlc_service.stop_all_vlc_instances.return_value = mock_result # Override dependency test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service try: response = await authenticated_client.post("/api/v1/sounds/vlc/stop-all") assert response.status_code == 200 data = response.json() assert data["success"] is True assert data["processes_found"] == 0 assert data["processes_killed"] == 0 assert data["message"] == "No VLC processes found" finally: # Clean up dependency override test_app.dependency_overrides.pop(get_vlc_player, None) @pytest.mark.asyncio async def test_stop_all_vlc_instances_partial_success( self, test_app: FastAPI, authenticated_client: AsyncClient, authenticated_user: User, ) -> None: """Test stopping VLC instances with partial success.""" # Set up mock mock_vlc_service = AsyncMock() mock_result = { "success": True, "processes_found": 3, "processes_killed": 2, "processes_remaining": 1, "message": "Killed 2 VLC processes", } mock_vlc_service.stop_all_vlc_instances.return_value = mock_result # Override dependency test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service try: response = await authenticated_client.post("/api/v1/sounds/vlc/stop-all") assert response.status_code == 200 data = response.json() assert data["success"] is True assert data["processes_found"] == 3 assert data["processes_killed"] == 2 assert data["processes_remaining"] == 1 finally: # Clean up dependency override test_app.dependency_overrides.pop(get_vlc_player, None) @pytest.mark.asyncio async def test_stop_all_vlc_instances_failure( self, test_app: FastAPI, authenticated_client: AsyncClient, authenticated_user: User, ) -> None: """Test stopping VLC instances when service fails.""" # Set up mock mock_vlc_service = AsyncMock() mock_result = { "success": False, "processes_found": 0, "processes_killed": 0, "error": "Command failed", "message": "Failed to stop VLC processes", } mock_vlc_service.stop_all_vlc_instances.return_value = mock_result # Override dependency test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service try: response = await authenticated_client.post("/api/v1/sounds/vlc/stop-all") assert response.status_code == 200 data = response.json() assert data["success"] is False assert data["error"] == "Command failed" assert data["message"] == "Failed to stop VLC processes" finally: # Clean up dependency override test_app.dependency_overrides.pop(get_vlc_player, None) @pytest.mark.asyncio async def test_stop_all_vlc_instances_service_exception( self, test_app: FastAPI, authenticated_client: AsyncClient, authenticated_user: User, ) -> None: """Test stopping VLC instances when service raises an exception.""" # Set up mock to raise an exception mock_vlc_service = AsyncMock() mock_vlc_service.stop_all_vlc_instances.side_effect = Exception("Service error") # Override dependency test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service try: response = await authenticated_client.post("/api/v1/sounds/vlc/stop-all") assert response.status_code == 500 data = response.json() assert "Failed to stop VLC instances" in data["detail"] finally: # Clean up dependency override test_app.dependency_overrides.pop(get_vlc_player, None) @pytest.mark.asyncio async def test_stop_all_vlc_instances_unauthenticated( self, client: AsyncClient, ) -> None: """Test stopping VLC instances without authentication.""" response = await client.post("/api/v1/sounds/vlc/stop-all") assert response.status_code == 401 @pytest.mark.asyncio async def test_vlc_endpoints_with_admin_user( self, test_app: FastAPI, authenticated_admin_client: AsyncClient, admin_user: User, ) -> None: """Test VLC endpoints work with admin user.""" # Set up mocks mock_vlc_service = AsyncMock() mock_repo = AsyncMock() mock_credit_service = AsyncMock() # Set up test data mock_sound = Sound( id=1, type="SDB", name="Admin Test Sound", filename="admin_test.mp3", duration=3000, size=512, hash="admin_hash", ) # Configure mocks mock_repo.get_by_id.return_value = mock_sound mock_credit_service.validate_and_reserve_credits.return_value = None mock_credit_service.deduct_credits.return_value = None mock_vlc_service.play_sound.return_value = True # Override dependencies test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service test_app.dependency_overrides[get_sound_repository] = lambda: mock_repo test_app.dependency_overrides[get_credit_service] = lambda: mock_credit_service try: response = await authenticated_admin_client.post("/api/v1/sounds/vlc/play/1") assert response.status_code == 200 data = response.json() assert data["sound_name"] == "Admin Test Sound" finally: # Clean up dependency overrides (except get_db which is needed for other tests) test_app.dependency_overrides.pop(get_vlc_player, None) test_app.dependency_overrides.pop(get_sound_repository, None) test_app.dependency_overrides.pop(get_credit_service, None) # Test stop-all endpoint with admin mock_vlc_service_2 = AsyncMock() mock_result = { "success": True, "processes_found": 1, "processes_killed": 1, "processes_remaining": 0, "message": "Killed 1 VLC processes", } mock_vlc_service_2.stop_all_vlc_instances.return_value = mock_result # Override dependency for stop-all test test_app.dependency_overrides[get_vlc_player] = lambda: mock_vlc_service_2 try: response = await authenticated_admin_client.post("/api/v1/sounds/vlc/stop-all") assert response.status_code == 200 data = response.json() assert data["success"] is True assert data["processes_killed"] == 1 finally: # Clean up dependency override test_app.dependency_overrides.pop(get_vlc_player, None)