fix(sonarr): Pass series ID alognside all GetEpisodes events when publishing to the networking channel

This commit is contained in:
2024-12-17 23:57:13 -07:00
parent 30ba1f3317
commit 2ecc591966
6 changed files with 36 additions and 85 deletions
+5 -1
View File
@@ -43,7 +43,7 @@ impl<'a> App<'a> {
} }
ActiveSonarrBlock::SeasonDetails => { ActiveSonarrBlock::SeasonDetails => {
self self
.dispatch_network_event(SonarrEvent::GetEpisodes(None).into()) .dispatch_network_event(SonarrEvent::GetEpisodes(self.extract_series_id().await).into())
.await; .await;
self self
.dispatch_network_event(SonarrEvent::GetEpisodeFiles(None).into()) .dispatch_network_event(SonarrEvent::GetEpisodeFiles(None).into())
@@ -256,4 +256,8 @@ impl<'a> App<'a> {
.current_selection() .current_selection()
.id .id
} }
async fn extract_series_id(&self) -> i64 {
self.data.sonarr_data.series.current_selection().id
}
} }
+16 -1
View File
@@ -81,6 +81,10 @@ mod tests {
#[tokio::test] #[tokio::test]
async fn test_dispatch_by_season_details_block() { async fn test_dispatch_by_season_details_block() {
let (mut app, mut sync_network_rx) = construct_app_unit(); let (mut app, mut sync_network_rx) = construct_app_unit();
app.data.sonarr_data.series.set_items(vec![Series {
id: 1,
..Series::default()
}]);
app app
.dispatch_by_sonarr_block(&ActiveSonarrBlock::SeasonDetails) .dispatch_by_sonarr_block(&ActiveSonarrBlock::SeasonDetails)
@@ -89,7 +93,7 @@ mod tests {
assert!(app.is_loading); assert!(app.is_loading);
assert_eq!( assert_eq!(
sync_network_rx.recv().await.unwrap(), sync_network_rx.recv().await.unwrap(),
SonarrEvent::GetEpisodes(None).into() SonarrEvent::GetEpisodes(1).into()
); );
assert_eq!( assert_eq!(
sync_network_rx.recv().await.unwrap(), sync_network_rx.recv().await.unwrap(),
@@ -746,6 +750,17 @@ mod tests {
assert_eq!(app.extract_episode_id().await, 0); assert_eq!(app.extract_episode_id().await, 0);
} }
#[tokio::test]
async fn test_extract_series_id() {
let mut app = App::default();
app.data.sonarr_data.series.set_items(vec![Series {
id: 1,
..Series::default()
}]);
assert_eq!(app.extract_series_id().await, 1);
}
fn construct_app_unit<'a>() -> (App<'a>, mpsc::Receiver<NetworkEvent>) { fn construct_app_unit<'a>() -> (App<'a>, mpsc::Receiver<NetworkEvent>) {
let (sync_network_tx, sync_network_rx) = mpsc::channel::<NetworkEvent>(500); let (sync_network_tx, sync_network_rx) = mpsc::channel::<NetworkEvent>(500);
let mut app = App { let mut app = App {
+1 -1
View File
@@ -163,7 +163,7 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrListCommand> for SonarrListCommandH
SonarrListCommand::Episodes { series_id } => { SonarrListCommand::Episodes { series_id } => {
let resp = self let resp = self
.network .network
.handle_network_event(SonarrEvent::GetEpisodes(Some(series_id)).into()) .handle_network_event(SonarrEvent::GetEpisodes(series_id).into())
.await?; .await?;
serde_json::to_string_pretty(&resp)? serde_json::to_string_pretty(&resp)?
} }
+1 -1
View File
@@ -329,7 +329,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
SonarrEvent::GetEpisodes(Some(expected_series_id)).into(), SonarrEvent::GetEpisodes(expected_series_id).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
+4 -5
View File
@@ -60,7 +60,7 @@ pub enum SonarrEvent {
GetHostConfig, GetHostConfig,
GetIndexers, GetIndexers,
GetEpisodeDetails(i64), GetEpisodeDetails(i64),
GetEpisodes(Option<i64>), GetEpisodes(i64),
GetEpisodeFiles(Option<i64>), GetEpisodeFiles(Option<i64>),
GetEpisodeHistory(Option<i64>), GetEpisodeHistory(Option<i64>),
GetLanguageProfiles, GetLanguageProfiles,
@@ -1144,10 +1144,9 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
} }
async fn get_episodes(&mut self, series_id: Option<i64>) -> Result<Vec<Episode>> { async fn get_episodes(&mut self, series_id: i64) -> Result<Vec<Episode>> {
let event = SonarrEvent::GetEpisodes(series_id); let event = SonarrEvent::GetEpisodes(series_id);
let (id, series_id_param) = self.extract_series_id(series_id).await; info!("Fetching episodes for Sonarr series with ID: {series_id}");
info!("Fetching episodes for Sonarr series with ID: {id}");
let request_props = self let request_props = self
.request_props_from( .request_props_from(
@@ -1155,7 +1154,7 @@ impl<'a, 'b> Network<'a, 'b> {
RequestMethod::Get, RequestMethod::Get,
None::<()>, None::<()>,
None, None,
Some(series_id_param), Some(format!("seriesId={series_id}")),
) )
.await; .await;
+9 -76
View File
@@ -149,7 +149,7 @@ mod test {
#[rstest] #[rstest]
fn test_resource_episode( fn test_resource_episode(
#[values(SonarrEvent::GetEpisodes(None), SonarrEvent::GetEpisodeDetails(0))] event: SonarrEvent, #[values(SonarrEvent::GetEpisodes(0), SonarrEvent::GetEpisodeDetails(0))] event: SonarrEvent,
) { ) {
assert_str_eq!(event.resource(), "/episode"); assert_str_eq!(event.resource(), "/episode");
} }
@@ -1834,7 +1834,7 @@ mod test {
None, None,
Some(json!([episode_1, episode_2, episode_3])), Some(json!([episode_1, episode_2, episode_3])),
None, None,
SonarrEvent::GetEpisodes(None), SonarrEvent::GetEpisodes(1),
None, None,
Some("seriesId=1"), Some("seriesId=1"),
) )
@@ -1876,7 +1876,7 @@ mod test {
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new()); let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Episodes(episodes) = network if let SonarrSerdeable::Episodes(episodes) = network
.handle_sonarr_event(SonarrEvent::GetEpisodes(None)) .handle_sonarr_event(SonarrEvent::GetEpisodes(1))
.await .await
.unwrap() .unwrap()
{ {
@@ -1941,7 +1941,7 @@ mod test {
None, None,
Some(json!([episode_1, episode_2, episode_3])), Some(json!([episode_1, episode_2, episode_3])),
None, None,
SonarrEvent::GetEpisodes(None), SonarrEvent::GetEpisodes(1),
None, None,
Some("seriesId=1"), Some("seriesId=1"),
) )
@@ -1962,7 +1962,7 @@ mod test {
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new()); let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Episodes(episodes) = network if let SonarrSerdeable::Episodes(episodes) = network
.handle_sonarr_event(SonarrEvent::GetEpisodes(None)) .handle_sonarr_event(SonarrEvent::GetEpisodes(1))
.await .await
.unwrap() .unwrap()
{ {
@@ -2003,7 +2003,7 @@ mod test {
None, None,
Some(json!([episode()])), Some(json!([episode()])),
None, None,
SonarrEvent::GetEpisodes(None), SonarrEvent::GetEpisodes(1),
None, None,
Some("seriesId=1"), Some("seriesId=1"),
) )
@@ -2021,7 +2021,7 @@ mod test {
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new()); let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Episodes(episodes) = network if let SonarrSerdeable::Episodes(episodes) = network
.handle_sonarr_event(SonarrEvent::GetEpisodes(None)) .handle_sonarr_event(SonarrEvent::GetEpisodes(1))
.await .await
.unwrap() .unwrap()
{ {
@@ -2091,7 +2091,7 @@ mod test {
None, None,
Some(episodes_json), Some(episodes_json),
None, None,
SonarrEvent::GetEpisodes(None), SonarrEvent::GetEpisodes(1),
None, None,
Some("seriesId=1"), Some("seriesId=1"),
) )
@@ -2125,7 +2125,7 @@ mod test {
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new()); let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Episodes(episodes) = network if let SonarrSerdeable::Episodes(episodes) = network
.handle_sonarr_event(SonarrEvent::GetEpisodes(None)) .handle_sonarr_event(SonarrEvent::GetEpisodes(1))
.await .await
.unwrap() .unwrap()
{ {
@@ -2575,73 +2575,6 @@ mod test {
} }
} }
#[tokio::test]
async fn test_handle_get_episodes_event_uses_provided_series_id() {
let episodes_json = json!([
{
"id": 2,
"seriesId": 2,
"tvdbId": 1234,
"episodeFileId": 2,
"seasonNumber": 2,
"episodeNumber": 2,
"title": "Something cool",
"airDateUtc": "2024-02-10T07:28:45Z",
"overview": "Okay so this one time at band camp...",
"hasFile": true,
"monitored": true
},
{
"id": 1,
"seriesId": 2,
"tvdbId": 1234,
"episodeFileId": 1,
"seasonNumber": 1,
"episodeNumber": 1,
"title": "Something cool",
"airDateUtc": "2024-02-10T07:28:45Z",
"overview": "Okay so this one time at band camp...",
"hasFile": true,
"monitored": true
}
]);
let episode_1 = Episode {
series_id: 2,
episode_file: None,
..episode()
};
let episode_2 = Episode {
id: 2,
episode_file_id: 2,
season_number: 2,
episode_number: 2,
series_id: 2,
episode_file: None,
..episode()
};
let expected_episodes = vec![episode_2.clone(), episode_1.clone()];
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(episodes_json),
None,
SonarrEvent::GetEpisodes(None),
None,
Some("seriesId=2"),
)
.await;
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Episodes(episodes) = network
.handle_sonarr_event(SonarrEvent::GetEpisodes(Some(2)))
.await
.unwrap()
{
async_server.assert_async().await;
assert_eq!(episodes, expected_episodes);
}
}
#[tokio::test] #[tokio::test]
async fn test_handle_get_episode_details_event() { async fn test_handle_get_episode_details_event() {
let response: Episode = serde_json::from_str(EPISODE_JSON).unwrap(); let response: Episode = serde_json::from_str(EPISODE_JSON).unwrap();