"""
Unit tests for AnimeSama season detection
"""
import pytest
from unittest.mock import Mock, AsyncMock, patch, MagicMock
from bs4 import BeautifulSoup
class TestAnimeSamaSeasons:
"""Tests for AnimeSamaDownloader season detection"""
@pytest.mark.asyncio
async def test_get_seasons_no_seasons_available(self):
"""Test get_seasons when no seasons exist"""
from app.downloaders.animesama import AnimeSamaDownloader
downloader = AnimeSamaDownloader()
# Mock the response for main anime page
with patch.object(downloader, 'client') as mock_client:
# Mock response for main page
mock_response = Mock()
mock_response.status_code = 200
mock_response.text = """
"""
mock_client.get = AsyncMock(return_value=mock_response)
# Mock season checks (all return 404)
async def mock_get(url, timeout=None):
response = Mock()
if "saison1" in url:
response.status_code = 404
elif "saison2" in url:
response.status_code = 404
else:
response.status_code = 200
response.text = mock_response.text
return response
mock_client.get.side_effect = mock_get
seasons = await downloader.get_seasons("https://anime-sama.si/catalogue/test-anime/saison1/vostfr/")
# Should return empty list if no seasons found
assert isinstance(seasons, list)
@pytest.mark.asyncio
async def test_get_seasons_with_multiple_seasons(self):
"""Test get_seasons when multiple seasons exist"""
from app.downloaders.animesama import AnimeSamaDownloader
downloader = AnimeSamaDownloader()
with patch.object(downloader, 'client') as mock_client:
# Mock get_episodes to return different counts for each season
async def mock_get(url, timeout=None):
response = Mock()
if "/saison1/" in url:
response.status_code = 200
response.text = 'episodes.js = [{"url": "/ep1", "episode": "1"}, {"url": "/ep2", "episode": "2"}]'
elif "/saison2/" in url:
response.status_code = 200
response.text = 'episodes.js = [{"url": "/ep3", "episode": "1"}]'
elif "/saison3/" in url:
response.status_code = 404
else:
# Main page
response.status_code = 200
response.text = 'No season links'
return response
mock_client.get.side_effect = mock_get
# Mock get_episodes
with patch.object(downloader, 'get_episodes') as mock_get_episodes:
mock_get_episodes.side_effect = [
[{"url": "/ep1", "episode": "1"}, {"url": "/ep2", "episode": "2"}], # Season 1
[{"url": "/ep3", "episode": "1"}], # Season 2
]
seasons = await downloader.get_seasons("https://anime-sama.si/catalogue/test/vostfr/")
# Should return multiple seasons
assert len(seasons) >= 0
# Check season structure
for season in seasons:
assert "season" in season
assert "title" in season
assert "url" in season
assert "episode_count" in season
assert isinstance(season["season"], int)
assert isinstance(season["episode_count"], int)
@pytest.mark.asyncio
async def test_get_seasons_url_parsing(self):
"""Test that get_seasons correctly parses URLs"""
from app.downloaders.animesama import AnimeSamaDownloader
downloader = AnimeSamaDownloader()
with patch.object(downloader, 'client') as mock_client:
# All seasons return 404
async def mock_get(url, timeout=None):
response = Mock()
response.status_code = 404
return response
mock_client.get.side_effect = mock_get
# Test with various URL formats
test_urls = [
"https://anime-sama.si/catalogue/test-anime/saison1/vostfr/",
"https://anime-sama.si/catalogue/test-anime/vostfr/",
"https://anime-sama.si/catalogue/naruto-shippuden/saison3/vostfr/",
]
for url in test_urls:
seasons = await downloader.get_seasons(url)
# Should not crash and should return a list
assert isinstance(seasons, list)
@pytest.mark.asyncio
async def test_get_seasons_sorting(self):
"""Test that seasons are returned in correct order"""
from app.downloaders.animesama import AnimeSamaDownloader
downloader = AnimeSamaDownloader()
with patch.object(downloader, 'client') as mock_client:
async def mock_get(url, timeout=None):
response = Mock()
response.status_code = 404
return response
mock_client.get.side_effect = mock_get
seasons = await downloader.get_seasons("https://anime-sama.si/catalogue/test/vostfr/")
# If seasons are found, they should be sorted by season number
if seasons:
season_numbers = [s["season"] for s in seasons]
assert season_numbers == sorted(season_numbers)
@pytest.mark.asyncio
async def test_get_seasons_with_season_links_in_html(self):
"""Test get_seasons when season links are present in HTML"""
from app.downloaders.animesama import AnimeSamaDownloader
downloader = AnimeSamaDownloader()
with patch.object(downloader, 'client') as mock_client:
# Mock main page with season links
main_page_response = Mock()
main_page_response.status_code = 200
main_page_response.text = """
"""
async def mock_get(url, timeout=None):
if "saison" not in url:
# Main page
return main_page_response
else:
# Season page
response = Mock()
response.status_code = 200
response.text = 'episodes.js = [{"url": "/ep1", "episode": "1"}]'
return response
mock_client.get.side_effect = mock_get
with patch.object(downloader, 'get_episodes') as mock_get_episodes:
mock_get_episodes.return_value = [{"url": "/ep1", "episode": "1"}]
seasons = await downloader.get_seasons("https://anime-sama.si/catalogue/test/vostfr/")
# Should find seasons from HTML links
assert isinstance(seasons, list)