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

This commit is contained in:
2024-12-17 23:59:49 -07:00
parent 2ecc591966
commit 5f94dbcabe
6 changed files with 15 additions and 67 deletions
+3 -1
View File
@@ -46,7 +46,9 @@ impl<'a> App<'a> {
.dispatch_network_event(SonarrEvent::GetEpisodes(self.extract_series_id().await).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(self.extract_series_id().await).into(),
)
.await; .await;
self self
.dispatch_network_event(SonarrEvent::GetDownloads.into()) .dispatch_network_event(SonarrEvent::GetDownloads.into())
+1 -1
View File
@@ -97,7 +97,7 @@ mod tests {
); );
assert_eq!( assert_eq!(
sync_network_rx.recv().await.unwrap(), sync_network_rx.recv().await.unwrap(),
SonarrEvent::GetEpisodeFiles(None).into() SonarrEvent::GetEpisodeFiles(1).into()
); );
assert_eq!( assert_eq!(
sync_network_rx.recv().await.unwrap(), sync_network_rx.recv().await.unwrap(),
+1 -1
View File
@@ -170,7 +170,7 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrListCommand> for SonarrListCommandH
SonarrListCommand::EpisodeFiles { series_id } => { SonarrListCommand::EpisodeFiles { series_id } => {
let resp = self let resp = self
.network .network
.handle_network_event(SonarrEvent::GetEpisodeFiles(Some(series_id)).into()) .handle_network_event(SonarrEvent::GetEpisodeFiles(series_id).into())
.await?; .await?;
serde_json::to_string_pretty(&resp)? serde_json::to_string_pretty(&resp)?
} }
+1 -1
View File
@@ -355,7 +355,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
SonarrEvent::GetEpisodeFiles(Some(expected_series_id)).into(), SonarrEvent::GetEpisodeFiles(expected_series_id).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
+4 -5
View File
@@ -61,7 +61,7 @@ pub enum SonarrEvent {
GetIndexers, GetIndexers,
GetEpisodeDetails(i64), GetEpisodeDetails(i64),
GetEpisodes(i64), GetEpisodes(i64),
GetEpisodeFiles(Option<i64>), GetEpisodeFiles(i64),
GetEpisodeHistory(Option<i64>), GetEpisodeHistory(Option<i64>),
GetLanguageProfiles, GetLanguageProfiles,
GetLogs(Option<u64>), GetLogs(Option<u64>),
@@ -1206,10 +1206,9 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
} }
async fn get_episode_files(&mut self, series_id: Option<i64>) -> Result<Vec<EpisodeFile>> { async fn get_episode_files(&mut self, series_id: i64) -> Result<Vec<EpisodeFile>> {
let event = SonarrEvent::GetEpisodeFiles(series_id); let event = SonarrEvent::GetEpisodeFiles(series_id);
let (id, series_id_param) = self.extract_series_id(series_id).await; info!("Fetching episodes files for Sonarr series with ID: {series_id}");
info!("Fetching episodes files for Sonarr series with ID: {id}");
let request_props = self let request_props = self
.request_props_from( .request_props_from(
@@ -1217,7 +1216,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;
+5 -58
View File
@@ -267,7 +267,7 @@ mod test {
#[rstest] #[rstest]
fn test_resource_episode_file( fn test_resource_episode_file(
#[values(SonarrEvent::GetEpisodeFiles(None), SonarrEvent::DeleteEpisodeFile(0))] #[values(SonarrEvent::GetEpisodeFiles(0), SonarrEvent::DeleteEpisodeFile(0))]
event: SonarrEvent, event: SonarrEvent,
) { ) {
assert_str_eq!(event.resource(), "/episodefile"); assert_str_eq!(event.resource(), "/episodefile");
@@ -2163,7 +2163,7 @@ mod test {
None, None,
Some(json!([episode_file()])), Some(json!([episode_file()])),
None, None,
SonarrEvent::GetEpisodeFiles(None), SonarrEvent::GetEpisodeFiles(1),
None, None,
Some("seriesId=1"), Some("seriesId=1"),
) )
@@ -2183,7 +2183,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::EpisodeFiles(episode_files) = network if let SonarrSerdeable::EpisodeFiles(episode_files) = network
.handle_sonarr_event(SonarrEvent::GetEpisodeFiles(None)) .handle_sonarr_event(SonarrEvent::GetEpisodeFiles(1))
.await .await
.unwrap() .unwrap()
{ {
@@ -2212,7 +2212,7 @@ mod test {
None, None,
Some(json!([episode_file()])), Some(json!([episode_file()])),
None, None,
SonarrEvent::GetEpisodeFiles(None), SonarrEvent::GetEpisodeFiles(1),
None, None,
Some("seriesId=1"), Some("seriesId=1"),
) )
@@ -2230,7 +2230,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::EpisodeFiles(episode_files) = network if let SonarrSerdeable::EpisodeFiles(episode_files) = network
.handle_sonarr_event(SonarrEvent::GetEpisodeFiles(None)) .handle_sonarr_event(SonarrEvent::GetEpisodeFiles(1))
.await .await
.unwrap() .unwrap()
{ {
@@ -2259,59 +2259,6 @@ mod test {
} }
} }
#[tokio::test]
async fn test_handle_get_episode_files_event_uses_provided_series_id() {
let episode_file = EpisodeFile {
id: 2,
..episode_file()
};
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(json!([episode_file.clone()])),
None,
SonarrEvent::GetEpisodeFiles(Some(2)),
None,
Some("seriesId=2"),
)
.await;
app_arc.lock().await.data.sonarr_data.season_details_modal =
Some(SeasonDetailsModal::default());
app_arc
.lock()
.await
.data
.sonarr_data
.series
.set_items(vec![Series {
id: 1,
..Series::default()
}]);
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::EpisodeFiles(episode_files) = network
.handle_sonarr_event(SonarrEvent::GetEpisodeFiles(Some(2)))
.await
.unwrap()
{
async_server.assert_async().await;
assert_eq!(
app_arc
.lock()
.await
.data
.sonarr_data
.season_details_modal
.as_ref()
.unwrap()
.episode_files
.items,
vec![episode_file.clone()]
);
assert_eq!(episode_files, vec![episode_file]);
}
}
#[tokio::test] #[tokio::test]
async fn test_handle_get_sonarr_host_config_event() { async fn test_handle_get_sonarr_host_config_event() {
let host_config_response = json!({ let host_config_response = json!({