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

This commit is contained in:
2024-12-18 01:22:28 -07:00
parent 18a8b81631
commit c16ecfb188
6 changed files with 55 additions and 64 deletions
+1 -1
View File
@@ -274,7 +274,7 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrCommand> for SonarrCliHandler<'a, '
SonarrCommand::ToggleEpisodeMonitoring { episode_id } => { SonarrCommand::ToggleEpisodeMonitoring { episode_id } => {
let resp = self let resp = self
.network .network
.handle_network_event(SonarrEvent::ToggleEpisodeMonitoring(Some(episode_id)).into()) .handle_network_event(SonarrEvent::ToggleEpisodeMonitoring(episode_id).into())
.await?; .await?;
serde_json::to_string_pretty(&resp)? serde_json::to_string_pretty(&resp)?
} }
+1 -1
View File
@@ -698,7 +698,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
SonarrEvent::ToggleEpisodeMonitoring(Some(expected_episode_id)).into(), SonarrEvent::ToggleEpisodeMonitoring(expected_episode_id).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
@@ -78,6 +78,19 @@ impl<'a, 'b> SeasonDetailsHandler<'a, 'b> {
.current_selection() .current_selection()
.episode_file_id .episode_file_id
} }
fn extract_episode_id(&self) -> i64 {
self
.app
.data
.sonarr_data
.season_details_modal
.as_ref()
.expect("Season details have not been loaded")
.episodes
.current_selection()
.id
}
} }
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for SeasonDetailsHandler<'a, 'b> { impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for SeasonDetailsHandler<'a, 'b> {
@@ -354,8 +367,9 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for SeasonDetailsHandler
match self.active_sonarr_block { match self.active_sonarr_block {
ActiveSonarrBlock::SeasonDetails if self.key == DEFAULT_KEYBINDINGS.toggle_monitoring.key => { ActiveSonarrBlock::SeasonDetails if self.key == DEFAULT_KEYBINDINGS.toggle_monitoring.key => {
self.app.data.sonarr_data.prompt_confirm = true; self.app.data.sonarr_data.prompt_confirm = true;
self.app.data.sonarr_data.prompt_confirm_action = self.app.data.sonarr_data.prompt_confirm_action = Some(
Some(SonarrEvent::ToggleEpisodeMonitoring(None)); SonarrEvent::ToggleEpisodeMonitoring(self.extract_episode_id()),
);
self self
.app .app
@@ -5,6 +5,7 @@ mod tests {
use crate::handlers::sonarr_handlers::library::season_details_handler::{ use crate::handlers::sonarr_handlers::library::season_details_handler::{
releases_sorting_options, SeasonDetailsHandler, releases_sorting_options, SeasonDetailsHandler,
}; };
use crate::handlers::sonarr_handlers::sonarr_handler_test_utils::utils::episode;
use crate::handlers::KeyEventHandler; use crate::handlers::KeyEventHandler;
use crate::models::servarr_data::sonarr::modals::SeasonDetailsModal; use crate::models::servarr_data::sonarr::modals::SeasonDetailsModal;
use crate::models::servarr_data::sonarr::sonarr_data::sonarr_test_utils::utils::create_test_sonarr_data; use crate::models::servarr_data::sonarr::sonarr_data::sonarr_test_utils::utils::create_test_sonarr_data;
@@ -550,6 +551,14 @@ mod tests {
fn test_toggle_monitoring_key() { fn test_toggle_monitoring_key() {
let mut app = App::default(); let mut app = App::default();
app.data.sonarr_data = create_test_sonarr_data(); app.data.sonarr_data = create_test_sonarr_data();
app
.data
.sonarr_data
.season_details_modal
.as_mut()
.unwrap()
.episodes
.set_items(vec![episode()]);
app.push_navigation_stack(ActiveSonarrBlock::SeasonDetails.into()); app.push_navigation_stack(ActiveSonarrBlock::SeasonDetails.into());
app.is_routing = false; app.is_routing = false;
@@ -569,7 +578,7 @@ mod tests {
assert!(app.is_routing); assert!(app.is_routing);
assert_eq!( assert_eq!(
app.data.sonarr_data.prompt_confirm_action, app.data.sonarr_data.prompt_confirm_action,
Some(SonarrEvent::ToggleEpisodeMonitoring(None)) Some(SonarrEvent::ToggleEpisodeMonitoring(1))
); );
} }
@@ -813,6 +822,24 @@ mod tests {
.extract_episode_file_id(); .extract_episode_file_id();
} }
#[test]
fn test_extract_episode_id() {
let mut app = App::default();
let mut season_details_modal = SeasonDetailsModal::default();
season_details_modal.episodes.set_items(vec![episode()]);
app.data.sonarr_data.season_details_modal = Some(season_details_modal);
let episode_id = SeasonDetailsHandler::with(
DEFAULT_KEYBINDINGS.esc.key,
&mut app,
ActiveSonarrBlock::SeasonDetails,
None,
)
.extract_episode_id();
assert_eq!(episode_id, 1);
}
#[test] #[test]
fn test_season_details_handler_is_not_ready_when_loading() { fn test_season_details_handler_is_not_ready_when_loading() {
let mut app = App::default(); let mut app = App::default();
+6 -17
View File
@@ -87,7 +87,7 @@ pub enum SonarrEvent {
TestIndexer(i64), TestIndexer(i64),
TestAllIndexers, TestAllIndexers,
ToggleSeasonMonitoring((i64, i64)), ToggleSeasonMonitoring((i64, i64)),
ToggleEpisodeMonitoring(Option<i64>), ToggleEpisodeMonitoring(i64),
TriggerAutomaticEpisodeSearch(Option<i64>), TriggerAutomaticEpisodeSearch(Option<i64>),
TriggerAutomaticSeasonSearch(Option<(i64, i64)>), TriggerAutomaticSeasonSearch(Option<(i64, i64)>),
TriggerAutomaticSeriesSearch(Option<i64>), TriggerAutomaticSeriesSearch(Option<i64>),
@@ -2131,11 +2131,11 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
} }
async fn toggle_sonarr_episode_monitoring(&mut self, episode_id: Option<i64>) -> Result<()> { async fn toggle_sonarr_episode_monitoring(&mut self, episode_id: i64) -> Result<()> {
let event = SonarrEvent::ToggleEpisodeMonitoring(episode_id); let event = SonarrEvent::ToggleEpisodeMonitoring(episode_id);
let detail_event = SonarrEvent::GetEpisodeDetails(0); let detail_event = SonarrEvent::GetEpisodeDetails(0);
let (id, monitored) = if let Some(episode_id) = episode_id { let monitored = {
info!("Fetching episode details for episode id: {episode_id}"); info!("Fetching episode details for episode id: {episode_id}");
let request_props = self let request_props = self
.request_props_from( .request_props_from(
@@ -2159,24 +2159,13 @@ impl<'a, 'b> Network<'a, 'b> {
}) })
.await?; .await?;
(episode_id, monitored) monitored
} else {
let app = self.app.lock().await;
let current_selection = app
.data
.sonarr_data
.season_details_modal
.as_ref()
.unwrap()
.episodes
.current_selection();
(current_selection.id, current_selection.monitored)
}; };
info!("Toggling monitoring for episode id: {id}"); info!("Toggling monitoring for episode id: {episode_id}");
let body = MonitorEpisodeBody { let body = MonitorEpisodeBody {
episode_ids: vec![id], episode_ids: vec![episode_id],
monitored: !monitored, monitored: !monitored,
}; };
+3 -42
View File
@@ -289,7 +289,7 @@ mod test {
#[case(SonarrEvent::SearchNewSeries(String::new()), "/series/lookup")] #[case(SonarrEvent::SearchNewSeries(String::new()), "/series/lookup")]
#[case(SonarrEvent::TestIndexer(0), "/indexer/test")] #[case(SonarrEvent::TestIndexer(0), "/indexer/test")]
#[case(SonarrEvent::TestAllIndexers, "/indexer/testall")] #[case(SonarrEvent::TestAllIndexers, "/indexer/testall")]
#[case(SonarrEvent::ToggleEpisodeMonitoring(None), "/episode/monitor")] #[case(SonarrEvent::ToggleEpisodeMonitoring(0), "/episode/monitor")]
fn test_resource(#[case] event: SonarrEvent, #[case] expected_uri: String) { fn test_resource(#[case] event: SonarrEvent, #[case] expected_uri: String) {
assert_str_eq!(event.resource(), expected_uri); assert_str_eq!(event.resource(), expected_uri);
} }
@@ -5036,39 +5036,6 @@ mod test {
#[tokio::test] #[tokio::test]
async fn test_handle_toggle_episode_monitoring_event() { async fn test_handle_toggle_episode_monitoring_event() {
let expected_body = MonitorEpisodeBody {
episode_ids: vec![1],
monitored: false,
};
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Put,
Some(json!(expected_body)),
None,
None,
SonarrEvent::ToggleEpisodeMonitoring(None),
None,
None,
)
.await;
{
let mut app = app_arc.lock().await;
let mut season_details_modal = SeasonDetailsModal::default();
season_details_modal.episodes.set_items(vec![episode()]);
app.data.sonarr_data.season_details_modal = Some(season_details_modal);
}
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_sonarr_event(SonarrEvent::ToggleEpisodeMonitoring(None))
.await
.is_ok());
async_server.assert_async().await;
}
#[tokio::test]
async fn test_handle_toggle_episode_monitoring_event_uses_provided_episode_id() {
let expected_body = MonitorEpisodeBody { let expected_body = MonitorEpisodeBody {
episode_ids: vec![2], episode_ids: vec![2],
monitored: false, monitored: false,
@@ -5090,7 +5057,7 @@ mod test {
"PUT", "PUT",
format!( format!(
"/api/v3{}", "/api/v3{}",
SonarrEvent::ToggleEpisodeMonitoring(None).resource() SonarrEvent::ToggleEpisodeMonitoring(2).resource()
) )
.as_str(), .as_str(),
) )
@@ -5099,16 +5066,10 @@ mod test {
.match_body(Matcher::Json(json!(expected_body))) .match_body(Matcher::Json(json!(expected_body)))
.create_async() .create_async()
.await; .await;
{
let mut app = app_arc.lock().await;
let mut season_details_modal = SeasonDetailsModal::default();
season_details_modal.episodes.set_items(vec![episode()]);
app.data.sonarr_data.season_details_modal = Some(season_details_modal);
}
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new()); let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network assert!(network
.handle_sonarr_event(SonarrEvent::ToggleEpisodeMonitoring(Some(2))) .handle_sonarr_event(SonarrEvent::ToggleEpisodeMonitoring(2))
.await .await
.is_ok()); .is_ok());