fix(sonarr): Pass the episode ID alongside all ManualEpisodeSearch events when publishing to the networking channel

This commit is contained in:
2024-12-18 00:12:18 -07:00
parent 924f8d5eff
commit fcb87a6779
7 changed files with 22 additions and 121 deletions
+5 -7
View File
@@ -69,7 +69,7 @@ pub enum SonarrEvent {
GetQualityProfiles,
GetQueuedEvents,
GetRootFolders,
GetEpisodeReleases(Option<i64>),
GetEpisodeReleases(i64),
GetSeasonHistory(Option<(i64, i64)>),
GetSeasonReleases(Option<(i64, i64)>),
GetSecurityConfig,
@@ -1613,11 +1613,9 @@ impl<'a, 'b> Network<'a, 'b> {
.await
}
async fn get_episode_releases(&mut self, episode_id: Option<i64>) -> Result<Vec<SonarrRelease>> {
let event = SonarrEvent::GetEpisodeReleases(None);
let id = self.extract_episode_id(episode_id).await;
info!("Fetching releases for episode with ID: {id}");
async fn get_episode_releases(&mut self, episode_id: i64) -> Result<Vec<SonarrRelease>> {
let event = SonarrEvent::GetEpisodeReleases(episode_id);
info!("Fetching releases for episode with ID: {episode_id}");
let request_props = self
.request_props_from(
@@ -1625,7 +1623,7 @@ impl<'a, 'b> Network<'a, 'b> {
RequestMethod::Get,
None::<()>,
None,
Some(format!("episodeId={id}")),
Some(format!("episodeId={episode_id}")),
)
.await;
+5 -108
View File
@@ -258,7 +258,7 @@ mod test {
fn test_resource_release(
#[values(
SonarrEvent::GetSeasonReleases(None),
SonarrEvent::GetEpisodeReleases(None)
SonarrEvent::GetEpisodeReleases(0)
)]
event: SonarrEvent,
) {
@@ -3360,7 +3360,7 @@ mod test {
None,
Some(release_json),
None,
SonarrEvent::GetEpisodeReleases(None),
SonarrEvent::GetEpisodeReleases(1),
None,
Some("episodeId=1"),
)
@@ -3380,7 +3380,7 @@ mod test {
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Releases(releases_vec) = network
.handle_sonarr_event(SonarrEvent::GetEpisodeReleases(None))
.handle_sonarr_event(SonarrEvent::GetEpisodeReleases(1))
.await
.unwrap()
{
@@ -3427,7 +3427,7 @@ mod test {
None,
Some(release_json),
None,
SonarrEvent::GetEpisodeReleases(None),
SonarrEvent::GetEpisodeReleases(1),
None,
Some("episodeId=1"),
)
@@ -3438,110 +3438,7 @@ mod test {
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Releases(releases_vec) = network
.handle_sonarr_event(SonarrEvent::GetEpisodeReleases(None))
.await
.unwrap()
{
async_server.assert_async().await;
assert_eq!(
app_arc
.lock()
.await
.data
.sonarr_data
.season_details_modal
.as_ref()
.unwrap()
.episode_details_modal
.as_ref()
.unwrap()
.episode_releases
.items,
vec![release()]
);
assert_eq!(releases_vec, vec![release()]);
}
}
#[tokio::test]
#[should_panic(expected = "Season details have not been loaded")]
async fn test_handle_get_episode_releases_event_empty_season_details_modal_panics() {
let release_json = json!([{
"guid": "1234",
"protocol": "torrent",
"age": 1,
"title": "Test Release",
"indexer": "kickass torrents",
"indexerId": 2,
"size": 1234,
"rejected": true,
"rejections": [ "Unknown quality profile", "Release is already mapped" ],
"seeders": 2,
"leechers": 1,
"languages": [ { "id": 1, "name": "English" } ],
"quality": { "quality": { "name": "Bluray-1080p" }}
}]);
let (_async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(release_json),
None,
SonarrEvent::GetEpisodeReleases(None),
None,
Some("episodeId=1"),
)
.await;
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
network
.handle_sonarr_event(SonarrEvent::GetEpisodeReleases(None))
.await
.unwrap();
}
#[tokio::test]
async fn test_handle_get_episode_releases_event_uses_provided_series_id() {
let release_json = json!([{
"guid": "1234",
"protocol": "torrent",
"age": 1,
"title": "Test Release",
"indexer": "kickass torrents",
"indexerId": 2,
"size": 1234,
"rejected": true,
"rejections": [ "Unknown quality profile", "Release is already mapped" ],
"seeders": 2,
"leechers": 1,
"languages": [ { "id": 1, "name": "English" } ],
"quality": { "quality": { "name": "Bluray-1080p" }}
}]);
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(release_json),
None,
SonarrEvent::GetEpisodeReleases(None),
None,
Some("episodeId=2"),
)
.await;
let mut season_details_modal = SeasonDetailsModal::default();
season_details_modal.episodes.set_items(vec![episode()]);
app_arc.lock().await.data.sonarr_data.season_details_modal = Some(season_details_modal);
app_arc
.lock()
.await
.data
.sonarr_data
.season_details_modal
.as_mut()
.unwrap()
.episode_details_modal = Some(EpisodeDetailsModal::default());
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Releases(releases_vec) = network
.handle_sonarr_event(SonarrEvent::GetEpisodeReleases(Some(2)))
.handle_sonarr_event(SonarrEvent::GetEpisodeReleases(1))
.await
.unwrap()
{