fix(radarr): Build and pass the edit indexer settings body with the EditAllIndexerSettings event when publishing to the networking channel

This commit is contained in:
2024-12-17 16:10:11 -07:00
parent 4afde8b750
commit f8792ea012
7 changed files with 53 additions and 89 deletions
+1 -1
View File
@@ -390,7 +390,7 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrEditCommand> for RadarrEditCommandH
}; };
self self
.network .network
.handle_network_event(RadarrEvent::EditAllIndexerSettings(Some(params)).into()) .handle_network_event(RadarrEvent::EditAllIndexerSettings(params).into())
.await?; .await?;
"All indexer settings updated".to_owned() "All indexer settings updated".to_owned()
} else { } else {
+3 -3
View File
@@ -857,7 +857,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
RadarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(), RadarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
@@ -928,7 +928,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
RadarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(), RadarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
@@ -1000,7 +1000,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
RadarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(), RadarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
+1 -1
View File
@@ -584,7 +584,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
RadarrEvent::EditAllIndexerSettings(Some(expected_edit_all_indexer_settings)).into(), RadarrEvent::EditAllIndexerSettings(expected_edit_all_indexer_settings).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
@@ -2,6 +2,7 @@ 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::{handle_prompt_toggle, KeyEventHandler}; use crate::handlers::{handle_prompt_toggle, KeyEventHandler};
use crate::models::radarr_models::IndexerSettings;
use crate::models::servarr_data::radarr::radarr_data::{ use crate::models::servarr_data::radarr::radarr_data::{
ActiveRadarrBlock, INDEXER_SETTINGS_BLOCKS, ActiveRadarrBlock, INDEXER_SETTINGS_BLOCKS,
}; };
@@ -19,6 +20,14 @@ pub(super) struct IndexerSettingsHandler<'a, 'b> {
_context: Option<ActiveRadarrBlock>, _context: Option<ActiveRadarrBlock>,
} }
impl<'a, 'b> IndexerSettingsHandler<'a, 'b> {
fn build_edit_indexer_settings_body(&mut self) -> IndexerSettings {
let indexer_settings = self.app.data.radarr_data.indexer_settings.clone().unwrap();
self.app.data.radarr_data.indexer_settings = None;
indexer_settings
}
}
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for IndexerSettingsHandler<'a, 'b> { impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for IndexerSettingsHandler<'a, 'b> {
fn accepts(active_block: ActiveRadarrBlock) -> bool { fn accepts(active_block: ActiveRadarrBlock) -> bool {
INDEXER_SETTINGS_BLOCKS.contains(&active_block) INDEXER_SETTINGS_BLOCKS.contains(&active_block)
@@ -166,12 +175,11 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for IndexerSettingsHandl
ActiveRadarrBlock::AllIndexerSettingsPrompt => { ActiveRadarrBlock::AllIndexerSettingsPrompt => {
match self.app.data.radarr_data.selected_block.get_active_block() { match self.app.data.radarr_data.selected_block.get_active_block() {
ActiveRadarrBlock::IndexerSettingsConfirmPrompt => { ActiveRadarrBlock::IndexerSettingsConfirmPrompt => {
let radarr_data = &mut self.app.data.radarr_data; if self.app.data.radarr_data.prompt_confirm {
if radarr_data.prompt_confirm { self.app.data.radarr_data.prompt_confirm_action = Some(RadarrEvent::EditAllIndexerSettings(self.build_edit_indexer_settings_body()));
radarr_data.prompt_confirm_action = Some(RadarrEvent::EditAllIndexerSettings(None));
self.app.should_refresh = true; self.app.should_refresh = true;
} else { } else {
radarr_data.indexer_settings = None; self.app.data.radarr_data.indexer_settings = None;
} }
self.app.pop_navigation_stack(); self.app.pop_navigation_stack();
@@ -259,7 +267,7 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveRadarrBlock> for IndexerSettingsHandl
{ {
self.app.data.radarr_data.prompt_confirm = true; self.app.data.radarr_data.prompt_confirm = true;
self.app.data.radarr_data.prompt_confirm_action = self.app.data.radarr_data.prompt_confirm_action =
Some(RadarrEvent::EditAllIndexerSettings(None)); Some(RadarrEvent::EditAllIndexerSettings(self.build_edit_indexer_settings_body()));
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::radarr_handlers::indexers::edit_indexer_settings_handler::IndexerSettingsHandler; use crate::handlers::radarr_handlers::indexers::edit_indexer_settings_handler::IndexerSettingsHandler;
use crate::handlers::radarr_handlers::radarr_handler_test_utils::utils::indexer_settings;
use crate::handlers::KeyEventHandler; use crate::handlers::KeyEventHandler;
use crate::models::radarr_models::IndexerSettings; use crate::models::radarr_models::IndexerSettings;
use crate::models::servarr_data::radarr::radarr_data::{ use crate::models::servarr_data::radarr::radarr_data::{
@@ -472,7 +474,7 @@ mod tests {
.radarr_data .radarr_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.radarr_data.indexer_settings = Some(IndexerSettings::default()); app.data.radarr_data.indexer_settings = Some(indexer_settings());
app.data.radarr_data.prompt_confirm = true; app.data.radarr_data.prompt_confirm = true;
IndexerSettingsHandler::with( IndexerSettingsHandler::with(
@@ -486,9 +488,9 @@ mod tests {
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Indexers.into()); assert_eq!(app.get_current_route(), ActiveRadarrBlock::Indexers.into());
assert_eq!( assert_eq!(
app.data.radarr_data.prompt_confirm_action, app.data.radarr_data.prompt_confirm_action,
Some(RadarrEvent::EditAllIndexerSettings(None)) Some(RadarrEvent::EditAllIndexerSettings(indexer_settings()))
); );
assert!(app.data.radarr_data.indexer_settings.is_some()); assert!(app.data.radarr_data.indexer_settings.is_none());
assert!(app.should_refresh); assert!(app.should_refresh);
} }
@@ -858,7 +860,7 @@ mod tests {
} }
mod test_handle_key_char { mod test_handle_key_char {
use pretty_assertions::assert_str_eq; use pretty_assertions::{assert_eq, assert_str_eq};
use crate::{ use crate::{
models::{ models::{
@@ -937,7 +939,7 @@ mod tests {
.radarr_data .radarr_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.radarr_data.indexer_settings = Some(IndexerSettings::default()); app.data.radarr_data.indexer_settings = Some(indexer_settings());
IndexerSettingsHandler::with( IndexerSettingsHandler::with(
DEFAULT_KEYBINDINGS.confirm.key, DEFAULT_KEYBINDINGS.confirm.key,
@@ -950,9 +952,9 @@ mod tests {
assert_eq!(app.get_current_route(), ActiveRadarrBlock::Indexers.into()); assert_eq!(app.get_current_route(), ActiveRadarrBlock::Indexers.into());
assert_eq!( assert_eq!(
app.data.radarr_data.prompt_confirm_action, app.data.radarr_data.prompt_confirm_action,
Some(RadarrEvent::EditAllIndexerSettings(None)) Some(RadarrEvent::EditAllIndexerSettings(indexer_settings()))
); );
assert!(app.data.radarr_data.indexer_settings.is_some()); assert!(app.data.radarr_data.indexer_settings.is_none());
assert!(app.should_refresh); assert!(app.should_refresh);
} }
} }
@@ -968,6 +970,22 @@ mod tests {
}) })
} }
#[test]
fn test_build_edit_indexer_settings_body() {
let mut app = App::default();
app.data.radarr_data.indexer_settings = Some(indexer_settings());
let body = IndexerSettingsHandler::with(
DEFAULT_KEYBINDINGS.esc.key,
&mut app,
ActiveRadarrBlock::AllIndexerSettingsPrompt,
None,
).build_edit_indexer_settings_body();
assert_eq!(body, indexer_settings());
assert!(app.data.radarr_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();
+5 -22
View File
@@ -46,7 +46,7 @@ pub enum RadarrEvent {
DeleteRootFolder(i64), DeleteRootFolder(i64),
DeleteTag(i64), DeleteTag(i64),
DownloadRelease(RadarrReleaseDownloadBody), DownloadRelease(RadarrReleaseDownloadBody),
EditAllIndexerSettings(Option<IndexerSettings>), EditAllIndexerSettings(IndexerSettings),
EditCollection(Option<EditCollectionParams>), EditCollection(Option<EditCollectionParams>),
EditIndexer(Option<EditIndexerParams>), EditIndexer(Option<EditIndexerParams>),
EditMovie(Option<EditMovieParams>), EditMovie(Option<EditMovieParams>),
@@ -506,38 +506,21 @@ impl<'a, 'b> Network<'a, 'b> {
async fn edit_all_radarr_indexer_settings( async fn edit_all_radarr_indexer_settings(
&mut self, &mut self,
params: Option<IndexerSettings>, params: IndexerSettings,
) -> Result<Value> { ) -> Result<Value> {
info!("Updating Radarr indexer settings"); info!("Updating Radarr indexer settings");
let event = RadarrEvent::EditAllIndexerSettings(None); let event = RadarrEvent::EditAllIndexerSettings(params.clone());
let body = if let Some(indexer_settings) = params { debug!("Indexer settings body: {params:?}");
indexer_settings
} else {
self
.app
.lock()
.await
.data
.radarr_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 let resp = self
.handle_request::<IndexerSettings, Value>(request_props, |_, _| {}) .handle_request::<IndexerSettings, Value>(request_props, |_, _| {})
.await; .await;
self.app.lock().await.data.radarr_data.indexer_settings = None;
resp resp
} }
+5 -50
View File
@@ -15,8 +15,8 @@ mod test {
use crate::app::ServarrConfig; use crate::app::ServarrConfig;
use crate::models::radarr_models::{ use crate::models::radarr_models::{
AddMovieOptions, BlocklistItem, BlocklistItemMovie, CollectionMovie, MediaInfo, AddMovieOptions, BlocklistItem, BlocklistItemMovie, CollectionMovie, IndexerSettings,
MinimumAvailability, MovieCollection, MovieFile, Rating, RatingsList, MediaInfo, MinimumAvailability, MovieCollection, MovieFile, Rating, RatingsList
}; };
use crate::models::servarr_data::radarr::radarr_data::ActiveRadarrBlock; use crate::models::servarr_data::radarr::radarr_data::ActiveRadarrBlock;
use crate::models::servarr_models::{ use crate::models::servarr_models::{
@@ -146,7 +146,7 @@ mod test {
fn test_resource_all_indexer_settings( fn test_resource_all_indexer_settings(
#[values( #[values(
RadarrEvent::GetAllIndexerSettings, RadarrEvent::GetAllIndexerSettings,
RadarrEvent::EditAllIndexerSettings(None) RadarrEvent::EditAllIndexerSettings(IndexerSettings::default())
)] )]
event: RadarrEvent, event: RadarrEvent,
) { ) {
@@ -3341,60 +3341,15 @@ mod test {
Some(indexer_settings_json), Some(indexer_settings_json),
None, None,
None, None,
RadarrEvent::EditAllIndexerSettings(None), RadarrEvent::EditAllIndexerSettings(indexer_settings()),
None, None,
None, None,
) )
.await; .await;
app_arc.lock().await.data.radarr_data.indexer_settings = Some(indexer_settings());
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_radarr_event(RadarrEvent::EditAllIndexerSettings(None)) .handle_radarr_event(RadarrEvent::EditAllIndexerSettings(indexer_settings()))
.await
.is_ok());
async_server.assert_async().await;
assert!(app_arc
.lock()
.await
.data
.radarr_data
.indexer_settings
.is_none());
}
#[tokio::test]
async fn test_handle_edit_all_radarr_indexer_settings_event_uses_provided_settings() {
let indexer_settings_json = json!({
"minimumAge": 0,
"maximumSize": 0,
"retention": 0,
"rssSyncInterval": 60,
"preferIndexerFlags": false,
"availabilityDelay": 0,
"allowHardcodedSubs": true,
"whitelistedHardcodedSubs": "",
"id": 1
});
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Put,
Some(indexer_settings_json),
None,
None,
RadarrEvent::EditAllIndexerSettings(None),
None,
None,
)
.await;
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_radarr_event(RadarrEvent::EditAllIndexerSettings(
Some(indexer_settings())
))
.await .await
.is_ok()); .is_ok());