Add new API endpoints for media retrieval by country and enhance configuration
Some checks failed
Test Suite / test (push) Has been cancelled
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:
204
backend/tests/test_sync.py
Normal file
204
backend/tests/test_sync.py
Normal 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
|
||||
|
||||
Reference in New Issue
Block a user