fix(sonarr): Pass the episode ID alongside all ToggleEpisodeMonitoring events when publishing to the networking channel
This commit is contained in:
@@ -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)?
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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();
|
||||||
|
|||||||
@@ -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,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
@@ -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());
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user