fix(sonarr): Construct and pass edit all indexer settings alongside all EditAllIndexerSettings events when publishing to the networking channel

This commit is contained in:
2024-12-17 23:05:29 -07:00
parent 3e36bcf307
commit 89d106c03e
7 changed files with 66 additions and 90 deletions
+1 -1
View File
@@ -274,7 +274,7 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrEditCommand> for SonarrEditCommandH
}; };
self self
.network .network
.handle_network_event(SonarrEvent::EditAllIndexerSettings(Some(params)).into()) .handle_network_event(SonarrEvent::EditAllIndexerSettings(params).into())
.await?; .await?;
"All indexer settings updated".to_owned() "All indexer settings updated".to_owned()
} else { } else {
+1 -1
View File
@@ -650,7 +650,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
SonarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(), SonarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
+1 -1
View File
@@ -434,7 +434,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
SonarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(), SonarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
@@ -6,6 +6,7 @@ use crate::handlers::{handle_prompt_toggle, KeyEventHandler};
use crate::models::servarr_data::sonarr::sonarr_data::{ use crate::models::servarr_data::sonarr::sonarr_data::{
ActiveSonarrBlock, INDEXER_SETTINGS_BLOCKS, ActiveSonarrBlock, INDEXER_SETTINGS_BLOCKS,
}; };
use crate::models::sonarr_models::IndexerSettings;
use crate::network::sonarr_network::SonarrEvent; use crate::network::sonarr_network::SonarrEvent;
#[cfg(test)] #[cfg(test)]
@@ -19,6 +20,23 @@ pub(super) struct IndexerSettingsHandler<'a, 'b> {
_context: Option<ActiveSonarrBlock>, _context: Option<ActiveSonarrBlock>,
} }
impl<'a, 'b> IndexerSettingsHandler<'a, 'b> {
fn build_edit_indexer_settings_params(&mut self) -> IndexerSettings {
let indexer_settings = self
.app
.data
.sonarr_data
.indexer_settings
.as_ref()
.unwrap()
.clone();
self.app.data.sonarr_data.indexer_settings = None;
indexer_settings
}
}
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for IndexerSettingsHandler<'a, 'b> { impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for IndexerSettingsHandler<'a, 'b> {
fn accepts(active_block: ActiveSonarrBlock) -> bool { fn accepts(active_block: ActiveSonarrBlock) -> bool {
INDEXER_SETTINGS_BLOCKS.contains(&active_block) INDEXER_SETTINGS_BLOCKS.contains(&active_block)
@@ -119,12 +137,13 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for IndexerSettingsHandl
ActiveSonarrBlock::AllIndexerSettingsPrompt => { ActiveSonarrBlock::AllIndexerSettingsPrompt => {
match self.app.data.sonarr_data.selected_block.get_active_block() { match self.app.data.sonarr_data.selected_block.get_active_block() {
ActiveSonarrBlock::IndexerSettingsConfirmPrompt => { ActiveSonarrBlock::IndexerSettingsConfirmPrompt => {
let sonarr_data = &mut self.app.data.sonarr_data; if self.app.data.sonarr_data.prompt_confirm {
if sonarr_data.prompt_confirm { self.app.data.sonarr_data.prompt_confirm_action = Some(
sonarr_data.prompt_confirm_action = Some(SonarrEvent::EditAllIndexerSettings(None)); SonarrEvent::EditAllIndexerSettings(self.build_edit_indexer_settings_params()),
);
self.app.should_refresh = true; self.app.should_refresh = true;
} else { } else {
sonarr_data.indexer_settings = None; self.app.data.sonarr_data.indexer_settings = None;
} }
self.app.pop_navigation_stack(); self.app.pop_navigation_stack();
@@ -172,8 +191,9 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for IndexerSettingsHandl
&& self.key == DEFAULT_KEYBINDINGS.confirm.key && self.key == DEFAULT_KEYBINDINGS.confirm.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(SonarrEvent::EditAllIndexerSettings(
Some(SonarrEvent::EditAllIndexerSettings(None)); self.build_edit_indexer_settings_params(),
));
self.app.should_refresh = true; self.app.should_refresh = true;
self.app.pop_navigation_stack(); self.app.pop_navigation_stack();
@@ -1,11 +1,13 @@
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use pretty_assertions::assert_eq;
use strum::IntoEnumIterator; use strum::IntoEnumIterator;
use crate::app::key_binding::DEFAULT_KEYBINDINGS; use crate::app::key_binding::DEFAULT_KEYBINDINGS;
use crate::app::App; use crate::app::App;
use crate::event::Key; use crate::event::Key;
use crate::handlers::sonarr_handlers::indexers::edit_indexer_settings_handler::IndexerSettingsHandler; use crate::handlers::sonarr_handlers::indexers::edit_indexer_settings_handler::IndexerSettingsHandler;
use crate::handlers::sonarr_handlers::sonarr_handler_test_utils::utils::indexer_settings;
use crate::handlers::KeyEventHandler; use crate::handlers::KeyEventHandler;
use crate::models::servarr_data::sonarr::sonarr_data::{ use crate::models::servarr_data::sonarr::sonarr_data::{
ActiveSonarrBlock, INDEXER_SETTINGS_BLOCKS, ActiveSonarrBlock, INDEXER_SETTINGS_BLOCKS,
@@ -285,7 +287,7 @@ mod tests {
.sonarr_data .sonarr_data
.selected_block .selected_block
.set_index(0, INDEXER_SETTINGS_SELECTION_BLOCKS.len() - 1); .set_index(0, INDEXER_SETTINGS_SELECTION_BLOCKS.len() - 1);
app.data.sonarr_data.indexer_settings = Some(IndexerSettings::default()); app.data.sonarr_data.indexer_settings = Some(indexer_settings());
app.data.sonarr_data.prompt_confirm = true; app.data.sonarr_data.prompt_confirm = true;
IndexerSettingsHandler::with( IndexerSettingsHandler::with(
@@ -299,9 +301,9 @@ mod tests {
assert_eq!(app.get_current_route(), ActiveSonarrBlock::Indexers.into()); assert_eq!(app.get_current_route(), ActiveSonarrBlock::Indexers.into());
assert_eq!( assert_eq!(
app.data.sonarr_data.prompt_confirm_action, app.data.sonarr_data.prompt_confirm_action,
Some(SonarrEvent::EditAllIndexerSettings(None)) Some(SonarrEvent::EditAllIndexerSettings(indexer_settings()))
); );
assert!(app.data.sonarr_data.indexer_settings.is_some()); assert!(app.data.sonarr_data.indexer_settings.is_none());
assert!(app.should_refresh); assert!(app.should_refresh);
} }
@@ -468,11 +470,11 @@ mod tests {
mod test_handle_key_char { mod test_handle_key_char {
use crate::{ use crate::{
models::{ models::{
servarr_data::sonarr::sonarr_data::INDEXER_SETTINGS_SELECTION_BLOCKS, servarr_data::sonarr::sonarr_data::INDEXER_SETTINGS_SELECTION_BLOCKS, BlockSelectionState,
sonarr_models::IndexerSettings, BlockSelectionState,
}, },
network::sonarr_network::SonarrEvent, network::sonarr_network::SonarrEvent,
}; };
use pretty_assertions::assert_eq;
use super::*; use super::*;
@@ -488,7 +490,7 @@ mod tests {
.sonarr_data .sonarr_data
.selected_block .selected_block
.set_index(0, INDEXER_SETTINGS_SELECTION_BLOCKS.len() - 1); .set_index(0, INDEXER_SETTINGS_SELECTION_BLOCKS.len() - 1);
app.data.sonarr_data.indexer_settings = Some(IndexerSettings::default()); app.data.sonarr_data.indexer_settings = Some(indexer_settings());
IndexerSettingsHandler::with( IndexerSettingsHandler::with(
DEFAULT_KEYBINDINGS.confirm.key, DEFAULT_KEYBINDINGS.confirm.key,
@@ -501,9 +503,9 @@ mod tests {
assert_eq!(app.get_current_route(), ActiveSonarrBlock::Indexers.into()); assert_eq!(app.get_current_route(), ActiveSonarrBlock::Indexers.into());
assert_eq!( assert_eq!(
app.data.sonarr_data.prompt_confirm_action, app.data.sonarr_data.prompt_confirm_action,
Some(SonarrEvent::EditAllIndexerSettings(None)) Some(SonarrEvent::EditAllIndexerSettings(indexer_settings()))
); );
assert!(app.data.sonarr_data.indexer_settings.is_some()); assert!(app.data.sonarr_data.indexer_settings.is_none());
assert!(app.should_refresh); assert!(app.should_refresh);
} }
} }
@@ -519,6 +521,23 @@ mod tests {
}) })
} }
#[test]
fn test_build_edit_indexer_settings_params() {
let mut app = App::default();
app.data.sonarr_data.indexer_settings = Some(indexer_settings());
let actual_indexer_settings = IndexerSettingsHandler::with(
DEFAULT_KEYBINDINGS.esc.key,
&mut app,
ActiveSonarrBlock::AllIndexerSettingsPrompt,
None,
)
.build_edit_indexer_settings_params();
assert_eq!(actual_indexer_settings, indexer_settings());
assert!(app.data.sonarr_data.indexer_settings.is_none());
}
#[test] #[test]
fn test_edit_indexer_settings_handler_not_ready_when_loading() { fn test_edit_indexer_settings_handler_not_ready_when_loading() {
let mut app = App::default(); let mut app = App::default();
+7 -30
View File
@@ -50,7 +50,7 @@ pub enum SonarrEvent {
DeleteSeries(DeleteSeriesParams), DeleteSeries(DeleteSeriesParams),
DeleteTag(i64), DeleteTag(i64),
DownloadRelease(SonarrReleaseDownloadBody), DownloadRelease(SonarrReleaseDownloadBody),
EditAllIndexerSettings(Option<IndexerSettings>), EditAllIndexerSettings(IndexerSettings),
EditIndexer(Option<EditIndexerParams>), EditIndexer(Option<EditIndexerParams>),
EditSeries(Option<EditSeriesParams>), EditSeries(Option<EditSeriesParams>),
GetAllIndexerSettings, GetAllIndexerSettings,
@@ -622,41 +622,18 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
} }
async fn edit_all_sonarr_indexer_settings( async fn edit_all_sonarr_indexer_settings(&mut self, params: IndexerSettings) -> Result<Value> {
&mut self,
params: Option<IndexerSettings>,
) -> Result<Value> {
info!("Updating Sonarr indexer settings"); info!("Updating Sonarr indexer settings");
let event = SonarrEvent::EditAllIndexerSettings(None); let event = SonarrEvent::EditAllIndexerSettings(params.clone());
debug!("Indexer settings body: {params:?}");
let body = if let Some(indexer_settings) = params {
indexer_settings
} else {
self
.app
.lock()
.await
.data
.sonarr_data
.indexer_settings
.as_ref()
.unwrap()
.clone()
};
debug!("Indexer settings body: {body:?}");
let request_props = self let request_props = self
.request_props_from(event, RequestMethod::Put, Some(body), None, None) .request_props_from(event, RequestMethod::Put, Some(params), None, None)
.await; .await;
let resp = self self
.handle_request::<IndexerSettings, Value>(request_props, |_, _| {}) .handle_request::<IndexerSettings, Value>(request_props, |_, _| {})
.await; .await
self.app.lock().await.data.sonarr_data.indexer_settings = None;
resp
} }
async fn edit_sonarr_indexer( async fn edit_sonarr_indexer(
+3 -43
View File
@@ -141,7 +141,7 @@ mod test {
fn test_resource_all_indexer_settings( fn test_resource_all_indexer_settings(
#[values( #[values(
SonarrEvent::GetAllIndexerSettings, SonarrEvent::GetAllIndexerSettings,
SonarrEvent::EditAllIndexerSettings(None) SonarrEvent::EditAllIndexerSettings(IndexerSettings::default())
)] )]
event: SonarrEvent, event: SonarrEvent,
) { ) {
@@ -758,45 +758,7 @@ mod test {
Some(indexer_settings_json), Some(indexer_settings_json),
None, None,
None, None,
SonarrEvent::EditAllIndexerSettings(None), SonarrEvent::EditAllIndexerSettings(indexer_settings()),
None,
None,
)
.await;
app_arc.lock().await.data.sonarr_data.indexer_settings = Some(indexer_settings());
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_sonarr_event(SonarrEvent::EditAllIndexerSettings(None))
.await
.is_ok());
async_server.assert_async().await;
assert!(app_arc
.lock()
.await
.data
.sonarr_data
.indexer_settings
.is_none());
}
#[tokio::test]
async fn test_handle_edit_all_indexer_settings_event_uses_provided_settings() {
let indexer_settings_json = json!({
"id": 1,
"minimumAge": 1,
"maximumSize": 12345,
"retention": 1,
"rssSyncInterval": 60
});
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Put,
Some(indexer_settings_json),
None,
None,
SonarrEvent::EditAllIndexerSettings(None),
None, None,
None, None,
) )
@@ -805,9 +767,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());
assert!(network assert!(network
.handle_sonarr_event(SonarrEvent::EditAllIndexerSettings( .handle_sonarr_event(SonarrEvent::EditAllIndexerSettings(indexer_settings()))
Some(indexer_settings())
))
.await .await
.is_ok()); .is_ok());