Add new API endpoints for media retrieval by country and enhance configuration
Some checks failed
Test Suite / test (push) Has been cancelled

- Introduced `/api/tmdb` and `/api/collection/missing-locations` endpoints to the backend for improved media management.
- Added a new `get_media_by_country` function in the collection API to fetch media items based on country codes.
- Updated configuration to allow overriding *arr base URLs via environment variables for better flexibility.
- Enhanced frontend with a new `MissingLocations` component and integrated it into the routing structure.
- Improved the `CollectionMap` component to handle country selection and display media items accordingly.
- Added testing dependencies in `requirements.txt` and updated frontend configuration for testing support.
This commit is contained in:
Danilo Reyes
2025-12-28 22:35:06 -06:00
parent 4caba81599
commit 2b1a92fb49
32 changed files with 2733 additions and 76 deletions

204
backend/tests/test_sync.py Normal file
View File

@@ -0,0 +1,204 @@
"""Tests for sync service"""
import pytest
from unittest.mock import AsyncMock, patch
from app.services.sync import (
extract_country_from_radarr,
extract_country_from_sonarr,
extract_country_from_lidarr,
upsert_media_item,
)
def test_extract_country_from_radarr_with_production_countries():
"""Test extracting country from Radarr movie with productionCountries"""
movie = {
"id": 1,
"title": "Test Movie",
"productionCountries": [{"iso_3166_1": "US"}]
}
country = extract_country_from_radarr(movie)
assert country == "US"
def test_extract_country_from_radarr_with_metadata():
"""Test extracting country from Radarr movie with movieMetadata"""
movie = {
"id": 1,
"title": "Test Movie",
"movieMetadata": {
"productionCountries": [{"iso_3166_1": "GB"}]
}
}
country = extract_country_from_radarr(movie)
assert country == "GB"
def test_extract_country_from_radarr_no_country():
"""Test extracting country from Radarr movie with no country"""
movie = {
"id": 1,
"title": "Test Movie"
}
country = extract_country_from_radarr(movie)
assert country is None
def test_extract_country_from_sonarr_with_metadata():
"""Test extracting country from Sonarr series with seriesMetadata"""
series = {
"id": 1,
"title": "Test Show",
"seriesMetadata": {
"originCountry": ["US"]
}
}
country = extract_country_from_sonarr(series)
assert country == "US"
def test_extract_country_from_sonarr_string_country():
"""Test extracting country from Sonarr series with string country"""
series = {
"id": 1,
"title": "Test Show",
"seriesMetadata": {
"originCountry": "US"
}
}
country = extract_country_from_sonarr(series)
assert country == "US"
def test_extract_country_from_lidarr_with_country():
"""Test extracting country from Lidarr artist with country field"""
artist = {
"id": 1,
"artistName": "Test Artist",
"country": "US"
}
country = extract_country_from_lidarr(artist)
assert country == "US"
def test_extract_country_from_lidarr_no_country():
"""Test extracting country from Lidarr artist with no country"""
artist = {
"id": 1,
"artistName": "Test Artist"
}
country = extract_country_from_lidarr(artist)
assert country is None
@pytest.mark.asyncio
async def test_upsert_media_item_new(test_db_pool):
"""Test upserting a new media item"""
# Temporarily replace the global pool
import app.core.database
original_pool = app.core.database.pool
app.core.database.pool = test_db_pool
try:
media_id = await upsert_media_item(
source_kind="radarr",
source_item_id=1,
title="Test Movie",
year=2020,
media_type="movie",
arr_raw={"id": 1, "title": "Test Movie", "productionCountries": [{"iso_3166_1": "US"}]}
)
assert media_id is not None
# Verify it was inserted
async with test_db_pool.connection() as conn:
async with conn.cursor() as cur:
await cur.execute("""
SELECT title, year, media_type FROM moviemap.media_item
WHERE id = %s
""", (media_id,))
result = await cur.fetchone()
assert result is not None
assert result[0] == "Test Movie"
assert result[1] == 2020
assert result[2] == "movie"
finally:
app.core.database.pool = original_pool
@pytest.mark.asyncio
async def test_upsert_media_item_with_country(test_db_pool):
"""Test upserting a media item with country association"""
# Temporarily replace the global pool
import app.core.database
original_pool = app.core.database.pool
app.core.database.pool = test_db_pool
try:
media_id = await upsert_media_item(
source_kind="radarr",
source_item_id=1,
title="Test Movie",
year=2020,
media_type="movie",
arr_raw={"id": 1, "title": "Test Movie", "productionCountries": [{"iso_3166_1": "US"}]}
)
# Verify country was associated
async with test_db_pool.connection() as conn:
async with conn.cursor() as cur:
await cur.execute("""
SELECT country_code FROM moviemap.media_country
WHERE media_item_id = %s
""", (media_id,))
result = await cur.fetchone()
assert result is not None
assert result[0] == "US"
finally:
app.core.database.pool = original_pool
@pytest.mark.asyncio
async def test_upsert_media_item_update(test_db_pool):
"""Test updating an existing media item"""
# Temporarily replace the global pool
import app.core.database
original_pool = app.core.database.pool
app.core.database.pool = test_db_pool
try:
# Insert first
media_id = await upsert_media_item(
source_kind="radarr",
source_item_id=1,
title="Test Movie",
year=2020,
media_type="movie",
arr_raw={"id": 1, "title": "Test Movie"}
)
# Update with new title
updated_id = await upsert_media_item(
source_kind="radarr",
source_item_id=1,
title="Updated Movie",
year=2021,
media_type="movie",
arr_raw={"id": 1, "title": "Updated Movie"}
)
assert updated_id == media_id # Same ID
# Verify it was updated
async with test_db_pool.connection() as conn:
async with conn.cursor() as cur:
await cur.execute("""
SELECT title, year FROM moviemap.media_item
WHERE id = %s
""", (media_id,))
result = await cur.fetchone()
assert result[0] == "Updated Movie"
assert result[1] == 2021
finally:
app.core.database.pool = original_pool