28 lines
2.4 KiB
Python
28 lines
2.4 KiB
Python
import unittest
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock, patch
|
|
from backend.app.routers import requests
|
|
from backend.app.models import Snapshot, RequestType
|
|
|
|
class RecheckMonitoringTests(unittest.IsolatedAsyncioTestCase):
|
|
async def test_series_restores_only_requested_seasons_and_episodes(self):
|
|
original={'id':10,'monitored':False,'qualityProfileId':7,'seasons':[{'seasonNumber':1,'monitored':False},{'seasonNumber':2,'monitored':False}]}
|
|
restored={**original,'monitored':True,'seasons':[{'seasonNumber':1,'monitored':True},{'seasonNumber':2,'monitored':False}]}
|
|
client=SimpleNamespace(get_series=AsyncMock(side_effect=[original,restored]),update_series=AsyncMock(),get_episodes=AsyncMock(side_effect=[[{'id':1,'seasonNumber':1,'monitored':False},{'id':2,'seasonNumber':2,'monitored':False}],[{'id':1,'seasonNumber':1,'monitored':True},{'id':2,'seasonNumber':2,'monitored':False}]]),monitor_episodes=AsyncMock())
|
|
snapshot=Snapshot(request_id='42',title='Test',request_type=RequestType.tv,raw={'arr':{'item':{'id':10}}})
|
|
runtime=SimpleNamespace(sonarr_base_url='http://sonarr',sonarr_api_key='test')
|
|
with patch.object(requests,'get_runtime_settings',return_value=runtime),patch.object(requests,'SonarrClient',return_value=client):
|
|
self.assertTrue(await requests._restore_request_monitoring(snapshot,{'status':2,'seasons':[{'seasonNumber':1}]}))
|
|
client.update_series.assert_awaited_once_with(restored)
|
|
client.monitor_episodes.assert_awaited_once_with([1],True)
|
|
|
|
async def test_movie_monitoring_preserves_profile_and_pending_is_noop(self):
|
|
movie={'id':10,'monitored':False,'qualityProfileId':7}
|
|
client=SimpleNamespace(get_movie=AsyncMock(side_effect=[movie,{**movie,'monitored':True}]),update_movie=AsyncMock())
|
|
snapshot=Snapshot(request_id='42',title='Test',request_type=RequestType.movie,raw={'arr':{'item':{'id':10}}})
|
|
runtime=SimpleNamespace(radarr_base_url='http://radarr',radarr_api_key='test')
|
|
with patch.object(requests,'get_runtime_settings',return_value=runtime),patch.object(requests,'RadarrClient',return_value=client):
|
|
self.assertFalse(await requests._restore_request_monitoring(snapshot,{'status':1}))
|
|
self.assertTrue(await requests._restore_request_monitoring(snapshot,{'status':2}))
|
|
client.update_movie.assert_awaited_once_with({**movie,'monitored':True})
|