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}))