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

This commit is contained in:
2024-12-17 23:22:56 -07:00
parent 89d106c03e
commit 38c0ad29dd
6 changed files with 453 additions and 400 deletions
+1 -1
View File
@@ -319,7 +319,7 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrEditCommand> for SonarrEditCommandH
self self
.network .network
.handle_network_event(SonarrEvent::EditIndexer(Some(edit_indexer_params)).into()) .handle_network_event(SonarrEvent::EditIndexer(edit_indexer_params).into())
.await?; .await?;
"Indexer updated".to_owned() "Indexer updated".to_owned()
} }
+1 -1
View File
@@ -697,7 +697,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
SonarrEvent::EditIndexer(Some(expected_edit_indexer_params)).into(), SonarrEvent::EditIndexer(expected_edit_indexer_params).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
@@ -2,7 +2,9 @@ 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::servarr_data::modals::EditIndexerModal;
use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, EDIT_INDEXER_BLOCKS}; use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, EDIT_INDEXER_BLOCKS};
use crate::models::servarr_models::EditIndexerParams;
use crate::network::sonarr_network::SonarrEvent; use crate::network::sonarr_network::SonarrEvent;
use crate::{handle_prompt_left_right_keys, handle_text_box_keys, handle_text_box_left_right_keys}; use crate::{handle_prompt_left_right_keys, handle_text_box_keys, handle_text_box_left_right_keys};
@@ -17,6 +19,60 @@ pub(super) struct EditIndexerHandler<'a, 'b> {
_context: Option<ActiveSonarrBlock>, _context: Option<ActiveSonarrBlock>,
} }
impl<'a, 'b> EditIndexerHandler<'a, 'b> {
fn build_edit_indexer_params(&mut self) -> EditIndexerParams {
let indexer_id = self.app.data.sonarr_data.indexers.current_selection().id;
let tags = self
.app
.data
.sonarr_data
.edit_indexer_modal
.as_ref()
.unwrap()
.tags
.text
.clone();
let params = {
let EditIndexerModal {
name,
enable_rss,
enable_automatic_search,
enable_interactive_search,
url,
api_key,
seed_ratio,
priority,
..
} = self
.app
.data
.sonarr_data
.edit_indexer_modal
.as_ref()
.unwrap();
EditIndexerParams {
indexer_id,
name: Some(name.text.clone()),
enable_rss: Some(enable_rss.unwrap_or_default()),
enable_automatic_search: Some(enable_automatic_search.unwrap_or_default()),
enable_interactive_search: Some(enable_interactive_search.unwrap_or_default()),
url: Some(url.text.clone()),
api_key: Some(api_key.text.clone()),
seed_ratio: Some(seed_ratio.text.clone()),
tags: None,
tag_input_string: Some(tags),
priority: Some(*priority),
clear_tags: false,
}
};
self.app.data.sonarr_data.edit_indexer_modal = None;
params
}
}
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for EditIndexerHandler<'a, 'b> { impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for EditIndexerHandler<'a, 'b> {
fn accepts(active_block: ActiveSonarrBlock) -> bool { fn accepts(active_block: ActiveSonarrBlock) -> bool {
EDIT_INDEXER_BLOCKS.contains(&active_block) EDIT_INDEXER_BLOCKS.contains(&active_block)
@@ -297,12 +353,12 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for EditIndexerHandler<'
let selected_block = self.app.data.sonarr_data.selected_block.get_active_block(); let selected_block = self.app.data.sonarr_data.selected_block.get_active_block();
match selected_block { match selected_block {
ActiveSonarrBlock::EditIndexerConfirmPrompt => { ActiveSonarrBlock::EditIndexerConfirmPrompt => {
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 =
sonarr_data.prompt_confirm_action = Some(SonarrEvent::EditIndexer(None)); Some(SonarrEvent::EditIndexer(self.build_edit_indexer_params()));
self.app.should_refresh = true; self.app.should_refresh = true;
} else { } else {
sonarr_data.edit_indexer_modal = None; self.app.data.sonarr_data.edit_indexer_modal = None;
} }
self.app.pop_navigation_stack(); self.app.pop_navigation_stack();
@@ -464,7 +520,8 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for EditIndexerHandler<'
&& 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 = Some(SonarrEvent::EditIndexer(None)); self.app.data.sonarr_data.prompt_confirm_action =
Some(SonarrEvent::EditIndexer(self.build_edit_indexer_params()));
self.app.should_refresh = true; self.app.should_refresh = true;
self.app.pop_navigation_stack(); self.app.pop_navigation_stack();
@@ -4,9 +4,12 @@ mod tests {
use crate::app::App; use crate::app::App;
use crate::event::Key; use crate::event::Key;
use crate::handlers::sonarr_handlers::indexers::edit_indexer_handler::EditIndexerHandler; use crate::handlers::sonarr_handlers::indexers::edit_indexer_handler::EditIndexerHandler;
use crate::handlers::sonarr_handlers::sonarr_handler_test_utils::utils::indexer;
use crate::handlers::KeyEventHandler; use crate::handlers::KeyEventHandler;
use crate::models::servarr_data::modals::EditIndexerModal; use crate::models::servarr_data::modals::EditIndexerModal;
use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, EDIT_INDEXER_BLOCKS}; use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, EDIT_INDEXER_BLOCKS};
use crate::models::servarr_models::EditIndexerParams;
use pretty_assertions::assert_eq;
use strum::IntoEnumIterator; use strum::IntoEnumIterator;
mod test_handle_scroll_up_and_down { mod test_handle_scroll_up_and_down {
@@ -896,7 +899,32 @@ mod tests {
.sonarr_data .sonarr_data
.selected_block .selected_block
.set_index(0, EDIT_INDEXER_TORRENT_SELECTION_BLOCKS.len() - 1); .set_index(0, EDIT_INDEXER_TORRENT_SELECTION_BLOCKS.len() - 1);
app.data.sonarr_data.edit_indexer_modal = Some(EditIndexerModal::default()); let edit_indexer_modal = EditIndexerModal {
name: "Test Update".into(),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: "https://localhost:9696/1/".into(),
api_key: "test1234".into(),
seed_ratio: "1.3".into(),
tags: "usenet, testing".into(),
priority: 0,
};
app.data.sonarr_data.edit_indexer_modal = Some(edit_indexer_modal);
app.data.sonarr_data.indexers.set_items(vec![indexer()]);
let expected_edit_indexer_params = EditIndexerParams {
indexer_id: 1,
name: Some("Test Update".to_owned()),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: Some("https://localhost:9696/1/".to_owned()),
api_key: Some("test1234".to_owned()),
seed_ratio: Some("1.3".to_owned()),
tag_input_string: Some("usenet, testing".to_owned()),
priority: Some(0),
..EditIndexerParams::default()
};
app.data.sonarr_data.prompt_confirm = true; app.data.sonarr_data.prompt_confirm = true;
EditIndexerHandler::with( EditIndexerHandler::with(
@@ -908,11 +936,11 @@ mod tests {
.handle(); .handle();
assert_eq!(app.get_current_route(), ActiveSonarrBlock::Indexers.into()); assert_eq!(app.get_current_route(), ActiveSonarrBlock::Indexers.into());
assert!(app.data.sonarr_data.edit_indexer_modal.is_some()); assert!(app.data.sonarr_data.edit_indexer_modal.is_none());
assert!(app.should_refresh); assert!(app.should_refresh);
assert_eq!( assert_eq!(
app.data.sonarr_data.prompt_confirm_action, app.data.sonarr_data.prompt_confirm_action,
Some(SonarrEvent::EditIndexer(None)) Some(SonarrEvent::EditIndexer(expected_edit_indexer_params))
); );
} }
@@ -1408,7 +1436,7 @@ mod tests {
use crate::models::servarr_data::sonarr::sonarr_data::EDIT_INDEXER_TORRENT_SELECTION_BLOCKS; use crate::models::servarr_data::sonarr::sonarr_data::EDIT_INDEXER_TORRENT_SELECTION_BLOCKS;
use crate::models::BlockSelectionState; use crate::models::BlockSelectionState;
use crate::network::sonarr_network::SonarrEvent; use crate::network::sonarr_network::SonarrEvent;
use pretty_assertions::assert_str_eq; use pretty_assertions::{assert_eq, assert_str_eq};
use super::*; use super::*;
@@ -1709,7 +1737,32 @@ mod tests {
.sonarr_data .sonarr_data
.selected_block .selected_block
.set_index(0, EDIT_INDEXER_TORRENT_SELECTION_BLOCKS.len() - 1); .set_index(0, EDIT_INDEXER_TORRENT_SELECTION_BLOCKS.len() - 1);
app.data.sonarr_data.edit_indexer_modal = Some(EditIndexerModal::default()); let edit_indexer_modal = EditIndexerModal {
name: "Test Update".into(),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: "https://localhost:9696/1/".into(),
api_key: "test1234".into(),
seed_ratio: "1.3".into(),
tags: "usenet, testing".into(),
priority: 0,
};
app.data.sonarr_data.edit_indexer_modal = Some(edit_indexer_modal);
app.data.sonarr_data.indexers.set_items(vec![indexer()]);
let expected_edit_indexer_params = EditIndexerParams {
indexer_id: 1,
name: Some("Test Update".to_owned()),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: Some("https://localhost:9696/1/".to_owned()),
api_key: Some("test1234".to_owned()),
seed_ratio: Some("1.3".to_owned()),
tag_input_string: Some("usenet, testing".to_owned()),
priority: Some(0),
..EditIndexerParams::default()
};
EditIndexerHandler::with( EditIndexerHandler::with(
DEFAULT_KEYBINDINGS.confirm.key, DEFAULT_KEYBINDINGS.confirm.key,
@@ -1720,11 +1773,11 @@ mod tests {
.handle(); .handle();
assert_eq!(app.get_current_route(), ActiveSonarrBlock::Indexers.into()); assert_eq!(app.get_current_route(), ActiveSonarrBlock::Indexers.into());
assert!(app.data.sonarr_data.edit_indexer_modal.is_some()); assert!(app.data.sonarr_data.edit_indexer_modal.is_none());
assert!(app.should_refresh); assert!(app.should_refresh);
assert_eq!( assert_eq!(
app.data.sonarr_data.prompt_confirm_action, app.data.sonarr_data.prompt_confirm_action,
Some(SonarrEvent::EditIndexer(None)) Some(SonarrEvent::EditIndexer(expected_edit_indexer_params))
); );
} }
} }
@@ -1740,6 +1793,48 @@ mod tests {
}) })
} }
#[test]
fn test_build_edit_indexer_params() {
let mut app = App::default();
let edit_indexer_modal = EditIndexerModal {
name: "Test Update".into(),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: "https://localhost:9696/1/".into(),
api_key: "test1234".into(),
seed_ratio: "1.3".into(),
tags: "usenet, testing".into(),
priority: 0,
};
app.data.sonarr_data.edit_indexer_modal = Some(edit_indexer_modal);
app.data.sonarr_data.indexers.set_items(vec![indexer()]);
let expected_edit_indexer_params = EditIndexerParams {
indexer_id: 1,
name: Some("Test Update".to_owned()),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: Some("https://localhost:9696/1/".to_owned()),
api_key: Some("test1234".to_owned()),
seed_ratio: Some("1.3".to_owned()),
tag_input_string: Some("usenet, testing".to_owned()),
priority: Some(0),
..EditIndexerParams::default()
};
let params = EditIndexerHandler::with(
DEFAULT_KEYBINDINGS.confirm.key,
&mut app,
ActiveSonarrBlock::EditIndexerPrompt,
None,
)
.build_edit_indexer_params();
assert_eq!(params, expected_edit_indexer_params);
assert!(app.data.sonarr_data.edit_indexer_modal.is_none());
}
#[test] #[test]
fn test_edit_indexer_handler_is_not_ready_when_loading() { fn test_edit_indexer_handler_is_not_ready_when_loading() {
let mut app = App::default(); let mut app = App::default();
+22 -74
View File
@@ -10,7 +10,7 @@ use crate::{
models::{ models::{
radarr_models::IndexerTestResult, radarr_models::IndexerTestResult,
servarr_data::{ servarr_data::{
modals::{EditIndexerModal, IndexerTestResultModalItem}, modals::IndexerTestResultModalItem,
sonarr::{ sonarr::{
modals::{EditSeriesModal, EpisodeDetailsModal, SeasonDetailsModal}, modals::{EditSeriesModal, EpisodeDetailsModal, SeasonDetailsModal},
sonarr_data::ActiveSonarrBlock, sonarr_data::ActiveSonarrBlock,
@@ -51,7 +51,7 @@ pub enum SonarrEvent {
DeleteTag(i64), DeleteTag(i64),
DownloadRelease(SonarrReleaseDownloadBody), DownloadRelease(SonarrReleaseDownloadBody),
EditAllIndexerSettings(IndexerSettings), EditAllIndexerSettings(IndexerSettings),
EditIndexer(Option<EditIndexerParams>), EditIndexer(EditIndexerParams),
EditSeries(Option<EditSeriesParams>), EditSeries(Option<EditSeriesParams>),
GetAllIndexerSettings, GetAllIndexerSettings,
GetBlocklist, GetBlocklist,
@@ -638,25 +638,18 @@ impl<'a, 'b> Network<'a, 'b> {
async fn edit_sonarr_indexer( async fn edit_sonarr_indexer(
&mut self, &mut self,
edit_indexer_params: Option<EditIndexerParams>, mut edit_indexer_params: EditIndexerParams,
) -> Result<()> { ) -> Result<()> {
if let Some(tag_input_string) = edit_indexer_params.tag_input_string.as_ref() {
let tag_ids_vec = self
.extract_and_add_sonarr_tag_ids_vec(tag_input_string.clone())
.await;
edit_indexer_params.tags = Some(tag_ids_vec);
}
let detail_event = SonarrEvent::GetIndexers; let detail_event = SonarrEvent::GetIndexers;
let event = SonarrEvent::EditIndexer(None); let event = SonarrEvent::EditIndexer(edit_indexer_params.clone());
let id = if let Some(ref params) = edit_indexer_params { let id = edit_indexer_params.indexer_id;
params.indexer_id
} else {
self
.app
.lock()
.await
.data
.sonarr_data
.indexers
.current_selection()
.id
};
info!("Updating Sonarr indexer with ID: {id}"); info!("Updating Sonarr indexer with ID: {id}");
info!("Fetching indexer details for indexer with ID: {id}"); info!("Fetching indexer details for indexer with ID: {id}");
let request_props = self let request_props = self
@@ -691,7 +684,7 @@ impl<'a, 'b> Network<'a, 'b> {
seed_ratio, seed_ratio,
tags, tags,
priority, priority,
) = if let Some(params) = edit_indexer_params { ) = {
let priority = detailed_indexer_body["priority"] let priority = detailed_indexer_body["priority"]
.as_i64() .as_i64()
.expect("Unable to deserialize 'priority'"); .expect("Unable to deserialize 'priority'");
@@ -700,28 +693,28 @@ impl<'a, 'b> Network<'a, 'b> {
.unwrap() .unwrap()
.iter() .iter()
.find(|field| field["name"] == "seedCriteria.seedRatio"); .find(|field| field["name"] == "seedCriteria.seedRatio");
let name = params.name.unwrap_or( let name = edit_indexer_params.name.unwrap_or(
detailed_indexer_body["name"] detailed_indexer_body["name"]
.as_str() .as_str()
.expect("Unable to deserialize 'name'") .expect("Unable to deserialize 'name'")
.to_owned(), .to_owned(),
); );
let enable_rss = params.enable_rss.unwrap_or( let enable_rss = edit_indexer_params.enable_rss.unwrap_or(
detailed_indexer_body["enableRss"] detailed_indexer_body["enableRss"]
.as_bool() .as_bool()
.expect("Unable to deserialize 'enableRss'"), .expect("Unable to deserialize 'enableRss'"),
); );
let enable_automatic_search = params.enable_automatic_search.unwrap_or( let enable_automatic_search = edit_indexer_params.enable_automatic_search.unwrap_or(
detailed_indexer_body["enableAutomaticSearch"] detailed_indexer_body["enableAutomaticSearch"]
.as_bool() .as_bool()
.expect("Unable to deserialize 'enableAutomaticSearch"), .expect("Unable to deserialize 'enableAutomaticSearch"),
); );
let enable_interactive_search = params.enable_interactive_search.unwrap_or( let enable_interactive_search = edit_indexer_params.enable_interactive_search.unwrap_or(
detailed_indexer_body["enableInteractiveSearch"] detailed_indexer_body["enableInteractiveSearch"]
.as_bool() .as_bool()
.expect("Unable to deserialize 'enableInteractiveSearch'"), .expect("Unable to deserialize 'enableInteractiveSearch'"),
); );
let url = params.url.unwrap_or( let url = edit_indexer_params.url.unwrap_or(
detailed_indexer_body["fields"] detailed_indexer_body["fields"]
.as_array() .as_array()
.expect("Unable to deserialize 'fields'") .expect("Unable to deserialize 'fields'")
@@ -734,7 +727,7 @@ impl<'a, 'b> Network<'a, 'b> {
.expect("Unable to deserialize 'baseUrl value'") .expect("Unable to deserialize 'baseUrl value'")
.to_owned(), .to_owned(),
); );
let api_key = params.api_key.unwrap_or( let api_key = edit_indexer_params.api_key.unwrap_or(
detailed_indexer_body["fields"] detailed_indexer_body["fields"]
.as_array() .as_array()
.expect("Unable to deserialize 'fields'") .expect("Unable to deserialize 'fields'")
@@ -747,7 +740,7 @@ impl<'a, 'b> Network<'a, 'b> {
.expect("Unable to deserialize 'apiKey value'") .expect("Unable to deserialize 'apiKey value'")
.to_owned(), .to_owned(),
); );
let seed_ratio = params.seed_ratio.unwrap_or_else(|| { let seed_ratio = edit_indexer_params.seed_ratio.unwrap_or_else(|| {
if let Some(seed_ratio_field) = seed_ratio_field_option { if let Some(seed_ratio_field) = seed_ratio_field_option {
return seed_ratio_field return seed_ratio_field
.get("value") .get("value")
@@ -759,10 +752,10 @@ impl<'a, 'b> Network<'a, 'b> {
String::new() String::new()
}); });
let tags = if params.clear_tags { let tags = if edit_indexer_params.clear_tags {
vec![] vec![]
} else { } else {
params.tags.unwrap_or( edit_indexer_params.tags.unwrap_or(
detailed_indexer_body["tags"] detailed_indexer_body["tags"]
.as_array() .as_array()
.expect("Unable to deserialize 'tags'") .expect("Unable to deserialize 'tags'")
@@ -771,7 +764,7 @@ impl<'a, 'b> Network<'a, 'b> {
.collect(), .collect(),
) )
}; };
let priority = params.priority.unwrap_or(priority); let priority = edit_indexer_params.priority.unwrap_or(priority);
( (
name, name,
@@ -784,51 +777,6 @@ impl<'a, 'b> Network<'a, 'b> {
tags, tags,
priority, priority,
) )
} else {
let tags = self
.app
.lock()
.await
.data
.sonarr_data
.edit_indexer_modal
.as_ref()
.unwrap()
.tags
.text
.clone();
let tag_ids_vec = self.extract_and_add_sonarr_tag_ids_vec(tags).await;
let mut app = self.app.lock().await;
let params = {
let EditIndexerModal {
name,
enable_rss,
enable_automatic_search,
enable_interactive_search,
url,
api_key,
seed_ratio,
priority,
..
} = app.data.sonarr_data.edit_indexer_modal.as_ref().unwrap();
(
name.text.clone(),
enable_rss.unwrap_or_default(),
enable_automatic_search.unwrap_or_default(),
enable_interactive_search.unwrap_or_default(),
url.text.clone(),
api_key.text.clone(),
seed_ratio.text.clone(),
tag_ids_vec,
*priority,
)
};
app.data.sonarr_data.edit_indexer_modal = None;
params
}; };
*detailed_indexer_body.get_mut("name").unwrap() = json!(name); *detailed_indexer_body.get_mut("name").unwrap() = json!(name);
+265 -312
View File
@@ -22,7 +22,7 @@ mod test {
use crate::app::{App, ServarrConfig}; use crate::app::{App, ServarrConfig};
use crate::models::radarr_models::IndexerTestResult; use crate::models::radarr_models::IndexerTestResult;
use crate::models::servarr_data::modals::{EditIndexerModal, IndexerTestResultModalItem}; use crate::models::servarr_data::modals::IndexerTestResultModalItem;
use crate::models::servarr_data::sonarr::modals::{ use crate::models::servarr_data::sonarr::modals::{
AddSeriesModal, EditSeriesModal, EpisodeDetailsModal, SeasonDetailsModal, AddSeriesModal, EditSeriesModal, EpisodeDetailsModal, SeasonDetailsModal,
}; };
@@ -212,7 +212,7 @@ mod test {
#[values( #[values(
SonarrEvent::GetIndexers, SonarrEvent::GetIndexers,
SonarrEvent::DeleteIndexer(0), SonarrEvent::DeleteIndexer(0),
SonarrEvent::EditIndexer(None) SonarrEvent::EditIndexer(EditIndexerParams::default())
)] )]
event: SonarrEvent, event: SonarrEvent,
) { ) {
@@ -776,6 +776,19 @@ mod test {
#[tokio::test] #[tokio::test]
async fn test_handle_edit_sonarr_indexer_event() { async fn test_handle_edit_sonarr_indexer_event() {
let expected_edit_indexer_params = EditIndexerParams {
indexer_id: 1,
name: Some("Test Update".to_owned()),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: Some("https://localhost:9696/1/".to_owned()),
api_key: Some("test1234".to_owned()),
seed_ratio: Some("1.3".to_owned()),
tag_input_string: Some("usenet, testing".to_owned()),
priority: Some(0),
..EditIndexerParams::default()
};
let indexer_details_json = json!({ let indexer_details_json = json!({
"enableRss": true, "enableRss": true,
"enableAutomaticSearch": true, "enableAutomaticSearch": true,
@@ -822,7 +835,6 @@ mod test {
"tags": [1, 2], "tags": [1, 2],
"id": 1 "id": 1
}); });
let (async_details_server, app_arc, mut server) = mock_servarr_api( let (async_details_server, app_arc, mut server) = mock_servarr_api(
RequestMethod::Get, RequestMethod::Get,
None, None,
@@ -838,7 +850,7 @@ mod test {
"PUT", "PUT",
format!( format!(
"/api/v3{}/1?forceSave=true", "/api/v3{}/1?forceSave=true",
SonarrEvent::EditIndexer(None).resource() SonarrEvent::EditIndexer(expected_edit_indexer_params.clone()).resource()
) )
.as_str(), .as_str(),
) )
@@ -847,315 +859,23 @@ mod test {
.match_body(Matcher::Json(expected_indexer_edit_body_json)) .match_body(Matcher::Json(expected_indexer_edit_body_json))
.create_async() .create_async()
.await; .await;
{ app_arc.lock().await.data.sonarr_data.tags_map =
let mut app = app_arc.lock().await; BiMap::from_iter([(1, "usenet".to_owned()), (2, "testing".to_owned())]);
app.data.sonarr_data.tags_map =
BiMap::from_iter([(1, "usenet".to_owned()), (2, "testing".to_owned())]);
let edit_indexer_modal = EditIndexerModal {
name: "Test Update".into(),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: "https://localhost:9696/1/".into(),
api_key: "test1234".into(),
seed_ratio: "1.3".into(),
tags: "usenet, testing".into(),
priority: 0,
};
app.data.sonarr_data.edit_indexer_modal = Some(edit_indexer_modal);
app.data.sonarr_data.indexers.set_items(vec![indexer()]);
}
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::EditIndexer(None)) .handle_sonarr_event(SonarrEvent::EditIndexer(expected_edit_indexer_params))
.await .await
.is_ok()); .is_ok());
async_details_server.assert_async().await; async_details_server.assert_async().await;
async_edit_server.assert_async().await; async_edit_server.assert_async().await;
let app = app_arc.lock().await;
assert!(app.data.sonarr_data.edit_indexer_modal.is_none());
} }
#[tokio::test] #[tokio::test]
async fn test_handle_edit_sonarr_indexer_event_does_not_add_seed_ratio_when_seed_ratio_field_is_none_in_details( async fn test_handle_edit_sonarr_indexer_event_does_not_overwrite_tags_vec_if_tag_input_string_is_none(
) { ) {
let indexer_details_json = json!({ let expected_edit_indexer_params = EditIndexerParams {
"enableRss": true,
"enableAutomaticSearch": true,
"enableInteractiveSearch": true,
"name": "Test Indexer",
"priority": 1,
"fields": [
{
"name": "baseUrl",
"value": "https://test.com",
},
{
"name": "apiKey",
"value": "",
},
],
"tags": [1],
"id": 1
});
let expected_indexer_edit_body_json = json!({
"enableRss": false,
"enableAutomaticSearch": false,
"enableInteractiveSearch": false,
"name": "Test Update",
"priority": 0,
"fields": [
{
"name": "baseUrl",
"value": "https://localhost:9696/1/",
},
{
"name": "apiKey",
"value": "test1234",
},
],
"tags": [1, 2],
"id": 1
});
let (async_details_server, app_arc, mut server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(indexer_details_json),
None,
SonarrEvent::GetIndexers,
Some("/1"),
None,
)
.await;
let async_edit_server = server
.mock(
"PUT",
format!(
"/api/v3{}/1?forceSave=true",
SonarrEvent::EditIndexer(None).resource()
)
.as_str(),
)
.with_status(202)
.match_header("X-Api-Key", "test1234")
.match_body(Matcher::Json(expected_indexer_edit_body_json))
.create_async()
.await;
{
let mut app = app_arc.lock().await;
app.data.sonarr_data.tags_map =
BiMap::from_iter([(1, "usenet".to_owned()), (2, "testing".to_owned())]);
let edit_indexer_modal = EditIndexerModal {
name: "Test Update".into(),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: "https://localhost:9696/1/".into(),
api_key: "test1234".into(),
seed_ratio: "1.3".into(),
tags: "usenet, testing".into(),
priority: 0,
};
app.data.sonarr_data.edit_indexer_modal = Some(edit_indexer_modal);
let mut indexer = indexer();
indexer.fields = Some(
indexer
.fields
.unwrap()
.into_iter()
.filter(|field| field.name != Some("seedCriteria.seedRatio".to_string()))
.collect(),
);
app.data.sonarr_data.indexers.set_items(vec![indexer]);
}
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_sonarr_event(SonarrEvent::EditIndexer(None))
.await
.is_ok());
async_details_server.assert_async().await;
async_edit_server.assert_async().await;
let app = app_arc.lock().await;
assert!(app.data.sonarr_data.edit_indexer_modal.is_none());
}
#[tokio::test]
async fn test_handle_edit_sonarr_indexer_event_populates_the_seed_ratio_value_when_seed_ratio_field_is_present_in_details(
) {
let indexer_details_json = json!({
"enableRss": true,
"enableAutomaticSearch": true,
"enableInteractiveSearch": true,
"name": "Test Indexer",
"priority": 1,
"fields": [
{
"name": "baseUrl",
"value": "https://test.com",
},
{
"name": "apiKey",
"value": "",
},
{
"name": "seedCriteria.seedRatio",
},
],
"tags": [1],
"id": 1
});
let expected_indexer_edit_body_json = json!({
"enableRss": false,
"enableAutomaticSearch": false,
"enableInteractiveSearch": false,
"name": "Test Update",
"priority": 0,
"fields": [
{
"name": "baseUrl",
"value": "https://localhost:9696/1/",
},
{
"name": "apiKey",
"value": "test1234",
},
{
"name": "seedCriteria.seedRatio",
"value": "1.3",
},
],
"tags": [1, 2],
"id": 1
});
let (async_details_server, app_arc, mut server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(indexer_details_json),
None,
SonarrEvent::GetIndexers,
Some("/1"),
None,
)
.await;
let async_edit_server = server
.mock(
"PUT",
format!(
"/api/v3{}/1?forceSave=true",
SonarrEvent::EditIndexer(None).resource()
)
.as_str(),
)
.with_status(202)
.match_header("X-Api-Key", "test1234")
.match_body(Matcher::Json(expected_indexer_edit_body_json))
.create_async()
.await;
{
let mut app = app_arc.lock().await;
app.data.sonarr_data.tags_map =
BiMap::from_iter([(1, "usenet".to_owned()), (2, "testing".to_owned())]);
let edit_indexer_modal = EditIndexerModal {
name: "Test Update".into(),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: "https://localhost:9696/1/".into(),
api_key: "test1234".into(),
seed_ratio: "1.3".into(),
tags: "usenet, testing".into(),
priority: 0,
};
app.data.sonarr_data.edit_indexer_modal = Some(edit_indexer_modal);
let mut indexer = indexer();
indexer.fields = Some(
indexer
.fields
.unwrap()
.into_iter()
.map(|mut field| {
if field.name == Some("seedCriteria.seedRatio".to_string()) {
field.value = None;
field
} else {
field
}
})
.collect(),
);
app.data.sonarr_data.indexers.set_items(vec![indexer]);
}
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_sonarr_event(SonarrEvent::EditIndexer(None))
.await
.is_ok());
async_details_server.assert_async().await;
async_edit_server.assert_async().await;
let app = app_arc.lock().await;
assert!(app.data.sonarr_data.edit_indexer_modal.is_none());
}
#[tokio::test]
async fn test_handle_edit_sonarr_indexer_event_uses_provided_parameters() {
let indexer_details_json = json!({
"enableRss": true,
"enableAutomaticSearch": true,
"enableInteractiveSearch": true,
"name": "Test Indexer",
"priority": 1,
"fields": [
{
"name": "baseUrl",
"value": "https://test.com",
},
{
"name": "apiKey",
"value": "",
},
{
"name": "seedCriteria.seedRatio",
"value": "1.2",
},
],
"tags": [1],
"id": 1
});
let expected_indexer_edit_body_json = json!({
"enableRss": false,
"enableAutomaticSearch": false,
"enableInteractiveSearch": false,
"name": "Test Update",
"priority": 25,
"fields": [
{
"name": "baseUrl",
"value": "https://localhost:9696/1/",
},
{
"name": "apiKey",
"value": "test1234",
},
{
"name": "seedCriteria.seedRatio",
"value": "1.3",
},
],
"tags": [1, 2],
"id": 1
});
let edit_indexer_params = EditIndexerParams {
indexer_id: 1, indexer_id: 1,
name: Some("Test Update".to_owned()), name: Some("Test Update".to_owned()),
enable_rss: Some(false), enable_rss: Some(false),
@@ -1165,9 +885,144 @@ mod test {
api_key: Some("test1234".to_owned()), api_key: Some("test1234".to_owned()),
seed_ratio: Some("1.3".to_owned()), seed_ratio: Some("1.3".to_owned()),
tags: Some(vec![1, 2]), tags: Some(vec![1, 2]),
priority: Some(25), priority: Some(0),
..EditIndexerParams::default() ..EditIndexerParams::default()
}; };
let indexer_details_json = json!({
"enableRss": true,
"enableAutomaticSearch": true,
"enableInteractiveSearch": true,
"name": "Test Indexer",
"priority": 1,
"fields": [
{
"name": "baseUrl",
"value": "https://test.com",
},
{
"name": "apiKey",
"value": "",
},
{
"name": "seedCriteria.seedRatio",
"value": "1.2",
},
],
"tags": [1],
"id": 1
});
let expected_indexer_edit_body_json = json!({
"enableRss": false,
"enableAutomaticSearch": false,
"enableInteractiveSearch": false,
"name": "Test Update",
"priority": 0,
"fields": [
{
"name": "baseUrl",
"value": "https://localhost:9696/1/",
},
{
"name": "apiKey",
"value": "test1234",
},
{
"name": "seedCriteria.seedRatio",
"value": "1.3",
},
],
"tags": [1, 2],
"id": 1
});
let (async_details_server, app_arc, mut server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(indexer_details_json),
None,
SonarrEvent::GetIndexers,
Some("/1"),
None,
)
.await;
let async_edit_server = server
.mock(
"PUT",
format!(
"/api/v3{}/1?forceSave=true",
SonarrEvent::EditIndexer(expected_edit_indexer_params.clone()).resource()
)
.as_str(),
)
.with_status(202)
.match_header("X-Api-Key", "test1234")
.match_body(Matcher::Json(expected_indexer_edit_body_json))
.create_async()
.await;
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_sonarr_event(SonarrEvent::EditIndexer(expected_edit_indexer_params))
.await
.is_ok());
async_details_server.assert_async().await;
async_edit_server.assert_async().await;
}
#[tokio::test]
async fn test_handle_edit_sonarr_indexer_event_does_not_add_seed_ratio_when_seed_ratio_field_is_none_in_details(
) {
let expected_edit_indexer_params = EditIndexerParams {
indexer_id: 1,
name: Some("Test Update".to_owned()),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: Some("https://localhost:9696/1/".to_owned()),
api_key: Some("test1234".to_owned()),
seed_ratio: Some("1.3".to_owned()),
tag_input_string: Some("usenet, testing".to_owned()),
priority: Some(0),
..EditIndexerParams::default()
};
let indexer_details_json = json!({
"enableRss": true,
"enableAutomaticSearch": true,
"enableInteractiveSearch": true,
"name": "Test Indexer",
"priority": 1,
"fields": [
{
"name": "baseUrl",
"value": "https://test.com",
},
{
"name": "apiKey",
"value": "",
},
],
"tags": [1],
"id": 1
});
let expected_indexer_edit_body_json = json!({
"enableRss": false,
"enableAutomaticSearch": false,
"enableInteractiveSearch": false,
"name": "Test Update",
"priority": 0,
"fields": [
{
"name": "baseUrl",
"value": "https://localhost:9696/1/",
},
{
"name": "apiKey",
"value": "test1234",
},
],
"tags": [1, 2],
"id": 1
});
let (async_details_server, app_arc, mut server) = mock_servarr_api( let (async_details_server, app_arc, mut server) = mock_servarr_api(
RequestMethod::Get, RequestMethod::Get,
@@ -1184,7 +1039,7 @@ mod test {
"PUT", "PUT",
format!( format!(
"/api/v3{}/1?forceSave=true", "/api/v3{}/1?forceSave=true",
SonarrEvent::EditIndexer(None).resource() SonarrEvent::EditIndexer(expected_edit_indexer_params.clone()).resource()
) )
.as_str(), .as_str(),
) )
@@ -1193,10 +1048,12 @@ mod test {
.match_body(Matcher::Json(expected_indexer_edit_body_json)) .match_body(Matcher::Json(expected_indexer_edit_body_json))
.create_async() .create_async()
.await; .await;
app_arc.lock().await.data.sonarr_data.tags_map =
BiMap::from_iter([(1, "usenet".to_owned()), (2, "testing".to_owned())]);
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::EditIndexer(Some(edit_indexer_params))) .handle_sonarr_event(SonarrEvent::EditIndexer(expected_edit_indexer_params))
.await .await
.is_ok()); .is_ok());
@@ -1205,8 +1062,106 @@ mod test {
} }
#[tokio::test] #[tokio::test]
async fn test_handle_edit_sonarr_indexer_event_uses_provided_parameters_defaults_to_previous_values( async fn test_handle_edit_sonarr_indexer_event_populates_the_seed_ratio_value_when_seed_ratio_field_is_present_in_details(
) { ) {
let expected_edit_indexer_params = EditIndexerParams {
indexer_id: 1,
name: Some("Test Update".to_owned()),
enable_rss: Some(false),
enable_automatic_search: Some(false),
enable_interactive_search: Some(false),
url: Some("https://localhost:9696/1/".to_owned()),
api_key: Some("test1234".to_owned()),
seed_ratio: Some("1.3".to_owned()),
tag_input_string: Some("usenet, testing".to_owned()),
priority: Some(0),
..EditIndexerParams::default()
};
let indexer_details_json = json!({
"enableRss": true,
"enableAutomaticSearch": true,
"enableInteractiveSearch": true,
"name": "Test Indexer",
"priority": 1,
"fields": [
{
"name": "baseUrl",
"value": "https://test.com",
},
{
"name": "apiKey",
"value": "",
},
{
"name": "seedCriteria.seedRatio",
},
],
"tags": [1],
"id": 1
});
let expected_indexer_edit_body_json = json!({
"enableRss": false,
"enableAutomaticSearch": false,
"enableInteractiveSearch": false,
"name": "Test Update",
"priority": 0,
"fields": [
{
"name": "baseUrl",
"value": "https://localhost:9696/1/",
},
{
"name": "apiKey",
"value": "test1234",
},
{
"name": "seedCriteria.seedRatio",
"value": "1.3",
},
],
"tags": [1, 2],
"id": 1
});
let (async_details_server, app_arc, mut server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(indexer_details_json),
None,
SonarrEvent::GetIndexers,
Some("/1"),
None,
)
.await;
let async_edit_server = server
.mock(
"PUT",
format!(
"/api/v3{}/1?forceSave=true",
SonarrEvent::EditIndexer(expected_edit_indexer_params.clone()).resource()
)
.as_str(),
)
.with_status(202)
.match_header("X-Api-Key", "test1234")
.match_body(Matcher::Json(expected_indexer_edit_body_json))
.create_async()
.await;
app_arc.lock().await.data.sonarr_data.tags_map =
BiMap::from_iter([(1, "usenet".to_owned()), (2, "testing".to_owned())]);
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_sonarr_event(SonarrEvent::EditIndexer(expected_edit_indexer_params))
.await
.is_ok());
async_details_server.assert_async().await;
async_edit_server.assert_async().await;
}
#[tokio::test]
async fn test_handle_edit_sonarr_indexer_event_defaults_to_previous_values() {
let indexer_details_json = json!({ let indexer_details_json = json!({
"enableRss": true, "enableRss": true,
"enableAutomaticSearch": true, "enableAutomaticSearch": true,
@@ -1234,7 +1189,6 @@ mod test {
indexer_id: 1, indexer_id: 1,
..EditIndexerParams::default() ..EditIndexerParams::default()
}; };
let (async_details_server, app_arc, mut server) = mock_servarr_api( let (async_details_server, app_arc, mut server) = mock_servarr_api(
RequestMethod::Get, RequestMethod::Get,
None, None,
@@ -1250,7 +1204,7 @@ mod test {
"PUT", "PUT",
format!( format!(
"/api/v3{}/1?forceSave=true", "/api/v3{}/1?forceSave=true",
SonarrEvent::EditIndexer(None).resource() SonarrEvent::EditIndexer(edit_indexer_params.clone()).resource()
) )
.as_str(), .as_str(),
) )
@@ -1262,7 +1216,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::EditIndexer(Some(edit_indexer_params))) .handle_sonarr_event(SonarrEvent::EditIndexer(edit_indexer_params))
.await .await
.is_ok()); .is_ok());
@@ -1271,8 +1225,7 @@ mod test {
} }
#[tokio::test] #[tokio::test]
async fn test_handle_edit_sonarr_indexer_event_uses_provided_parameters_clears_tags_when_clear_tags_is_true( async fn test_handle_edit_sonarr_indexer_event_clears_tags_when_clear_tags_is_true() {
) {
let indexer_details_json = json!({ let indexer_details_json = json!({
"enableRss": true, "enableRss": true,
"enableAutomaticSearch": true, "enableAutomaticSearch": true,
@@ -1340,7 +1293,7 @@ mod test {
"PUT", "PUT",
format!( format!(
"/api/v3{}/1?forceSave=true", "/api/v3{}/1?forceSave=true",
SonarrEvent::EditIndexer(None).resource() SonarrEvent::EditIndexer(edit_indexer_params.clone()).resource()
) )
.as_str(), .as_str(),
) )
@@ -1352,7 +1305,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::EditIndexer(Some(edit_indexer_params))) .handle_sonarr_event(SonarrEvent::EditIndexer(edit_indexer_params))
.await .await
.is_ok()); .is_ok());