31 lines
2.2 KiB
Python
31 lines
2.2 KiB
Python
import unittest
|
|
from unittest.mock import patch
|
|
import httpx
|
|
from backend.app.clients.jellyfin import JellyfinClient
|
|
from backend.app.services.snapshot import jellyfin_item_matches_request
|
|
from backend.app.models import RequestType
|
|
|
|
class JellyfinMatchingTests(unittest.IsolatedAsyncioTestCase):
|
|
async def test_search_includes_punctuation_variant_and_provider_metadata(self):
|
|
calls=[]
|
|
def handle(request):
|
|
calls.append(request)
|
|
items=[{'Id':'animated','Name':'Avatar: The Last Airbender','ProductionYear':2005,'ProviderIds':{'Tmdb':'246'}}] if ':' in request.url.params['SearchTerm'] else [{'Id':'live','Name':'Avatar the Last Airbender','ProductionYear':2024,'ProviderIds':{'Tmdb':'82452'}}]
|
|
return httpx.Response(200,json={'Items':items})
|
|
original=httpx.AsyncClient
|
|
with patch('backend.app.clients.jellyfin.httpx.AsyncClient',side_effect=lambda **kw:original(transport=httpx.MockTransport(handle),**kw)):
|
|
result=await JellyfinClient('http://jellyfin','test').search_items('Avatar: The Last Airbender',['Series'])
|
|
self.assertEqual({i['Id'] for i in result['Items']},{'live','animated'})
|
|
self.assertTrue(all('ProviderIds' in r.url.params['Fields'] for r in calls))
|
|
matches=[i for i in result['Items'] if jellyfin_item_matches_request(i,title='Avatar: The Last Airbender',year=2024,request_type=RequestType.tv,request_payload={'tmdbId':82452})]
|
|
self.assertEqual([i['Id'] for i in matches],['live'])
|
|
|
|
def test_fallback_rejects_remakes_prefixes_and_conflicting_ids(self):
|
|
def match(item,payload=None):
|
|
return jellyfin_item_matches_request(item,title='Avatar: The Last Airbender',year=2024,request_type=RequestType.tv,request_payload=payload)
|
|
self.assertTrue(match({'Name':'Avatar the Last Airbender','ProductionYear':2024}))
|
|
self.assertFalse(match({'Name':'Avatar the Last Airbender','ProductionYear':2005}))
|
|
self.assertFalse(match({'Name':'Avatar','ProductionYear':2024}))
|
|
self.assertFalse(match({'Name':'Avatar the Last Airbender','ProductionYear':2024,'ProviderIds':{'Tmdb':'246'}},{'tmdbId':82452}))
|
|
self.assertTrue(match({'Name':'Localized title','ProviderIds':{'Tmdb':'82452'}},{'tmdbId':82452}))
|