#[cfg(test)] mod tests { use pretty_assertions::assert_str_eq; use strum::IntoEnumIterator; use crate::app::key_binding::DEFAULT_KEYBINDINGS; use crate::app::App; use crate::event::Key; use crate::handlers::radarr_handlers::collections::edit_collection_handler::EditCollectionHandler; use crate::handlers::KeyEventHandler; use crate::models::radarr_models::MinimumAvailability; use crate::models::servarr_data::radarr_data::{ActiveRadarrBlock, EDIT_COLLECTION_BLOCKS}; mod test_handle_scroll_up_and_down { use pretty_assertions::assert_eq; use rstest::rstest; use strum::IntoEnumIterator; use crate::models::servarr_data::radarr_data::EDIT_COLLECTION_SELECTION_BLOCKS; use crate::models::BlockSelectionState; use crate::{test_enum_scroll, test_iterable_scroll}; use super::*; test_enum_scroll!( test_edit_collection_select_minimum_availability_scroll, EditCollectionHandler, MinimumAvailability, minimum_availability_list, ActiveRadarrBlock::EditCollectionSelectMinimumAvailability, None ); test_iterable_scroll!( test_edit_collection_select_quality_profile_scroll, EditCollectionHandler, quality_profile_list, ActiveRadarrBlock::EditCollectionSelectQualityProfile, None ); #[rstest] fn test_edit_collection_prompt_scroll(#[values(Key::Up, Key::Down)] key: Key) { let mut app = App::default(); app.data.radarr_data.selected_block = BlockSelectionState::new(&EDIT_COLLECTION_SELECTION_BLOCKS); app.data.radarr_data.selected_block.next(); EditCollectionHandler::with( &key, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &None, ) .handle(); if key == Key::Up { assert_eq!( app.data.radarr_data.selected_block.get_active_block(), &ActiveRadarrBlock::EditCollectionToggleMonitored ); } else { assert_eq!( app.data.radarr_data.selected_block.get_active_block(), &ActiveRadarrBlock::EditCollectionSelectQualityProfile ); } } } mod test_handle_home_end { use strum::IntoEnumIterator; use crate::{test_enum_home_and_end, test_iterable_home_and_end, test_text_box_home_end_keys}; use super::*; test_enum_home_and_end!( test_edit_collection_select_minimum_availability_home_end, EditCollectionHandler, MinimumAvailability, minimum_availability_list, ActiveRadarrBlock::EditCollectionSelectMinimumAvailability, None ); test_iterable_home_and_end!( test_edit_collection_select_quality_profile_scroll, EditCollectionHandler, quality_profile_list, ActiveRadarrBlock::EditCollectionSelectQualityProfile, None ); #[test] fn test_edit_collection_root_folder_path_input_home_end_keys() { test_text_box_home_end_keys!( EditCollectionHandler, ActiveRadarrBlock::EditCollectionRootFolderPathInput, edit_path ); } } mod test_handle_left_right_action { use rstest::rstest; use crate::test_text_box_left_right_keys; use super::*; #[rstest] fn test_left_right_prompt_toggle(#[values(Key::Left, Key::Right)] key: Key) { let mut app = App::default(); EditCollectionHandler::with( &key, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &None, ) .handle(); assert!(app.data.radarr_data.prompt_confirm); EditCollectionHandler::with( &key, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &None, ) .handle(); assert!(!app.data.radarr_data.prompt_confirm); } #[test] fn test_edit_collection_root_folder_path_input_left_right_keys() { test_text_box_left_right_keys!( EditCollectionHandler, ActiveRadarrBlock::EditCollectionRootFolderPathInput, edit_path ); } } mod test_handle_submit { use pretty_assertions::assert_eq; use rstest::rstest; use crate::models::servarr_data::radarr_data::EDIT_COLLECTION_SELECTION_BLOCKS; use crate::models::{BlockSelectionState, Route}; use crate::network::radarr_network::RadarrEvent; use super::*; const SUBMIT_KEY: Key = DEFAULT_KEYBINDINGS.submit.key; #[test] fn test_edit_collection_root_folder_path_input_submit() { let mut app = App::default(); app.should_ignore_quit_key = true; app.data.radarr_data.edit_path = "Test Path".to_owned().into(); app.push_navigation_stack(ActiveRadarrBlock::EditCollectionPrompt.into()); app.push_navigation_stack(ActiveRadarrBlock::EditCollectionRootFolderPathInput.into()); EditCollectionHandler::with( &SUBMIT_KEY, &mut app, &ActiveRadarrBlock::EditCollectionRootFolderPathInput, &None, ) .handle(); assert!(!app.should_ignore_quit_key); assert!(!app.data.radarr_data.edit_path.text.is_empty()); assert_eq!( app.get_current_route(), &ActiveRadarrBlock::EditCollectionPrompt.into() ); } #[test] fn test_edit_collection_prompt_prompt_decline_submit() { let mut app = App::default(); app.push_navigation_stack(ActiveRadarrBlock::Collections.into()); app.push_navigation_stack(ActiveRadarrBlock::EditCollectionPrompt.into()); app.data.radarr_data.selected_block = BlockSelectionState::new(&EDIT_COLLECTION_SELECTION_BLOCKS); app .data .radarr_data .selected_block .set_index(EDIT_COLLECTION_SELECTION_BLOCKS.len() - 1); EditCollectionHandler::with( &SUBMIT_KEY, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &None, ) .handle(); assert_eq!( app.get_current_route(), &ActiveRadarrBlock::Collections.into() ); assert_eq!(app.data.radarr_data.prompt_confirm_action, None); } #[test] fn test_edit_collection_confirm_prompt_prompt_confirmation_submit() { let mut app = App::default(); app.push_navigation_stack(ActiveRadarrBlock::Collections.into()); app.push_navigation_stack(ActiveRadarrBlock::EditCollectionPrompt.into()); app.data.radarr_data.prompt_confirm = true; app.data.radarr_data.selected_block = BlockSelectionState::new(&EDIT_COLLECTION_SELECTION_BLOCKS); app .data .radarr_data .selected_block .set_index(EDIT_COLLECTION_SELECTION_BLOCKS.len() - 1); EditCollectionHandler::with( &SUBMIT_KEY, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &None, ) .handle(); assert_eq!( app.get_current_route(), &ActiveRadarrBlock::Collections.into() ); assert_eq!( app.data.radarr_data.prompt_confirm_action, Some(RadarrEvent::EditCollection) ); assert!(app.should_refresh); } #[test] fn test_edit_collection_toggle_monitored_submit() { let current_route = Route::from(( ActiveRadarrBlock::EditCollectionPrompt, Some(ActiveRadarrBlock::Collections), )); let mut app = App::default(); app.data.radarr_data.selected_block = BlockSelectionState::new(&EDIT_COLLECTION_SELECTION_BLOCKS); app.push_navigation_stack(current_route); EditCollectionHandler::with( &SUBMIT_KEY, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &Some(ActiveRadarrBlock::Collections), ) .handle(); assert_eq!(app.get_current_route(), ¤t_route); assert_eq!(app.data.radarr_data.edit_monitored, Some(true)); EditCollectionHandler::with( &SUBMIT_KEY, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &Some(ActiveRadarrBlock::Collections), ) .handle(); assert_eq!(app.get_current_route(), ¤t_route); assert_eq!(app.data.radarr_data.edit_monitored, Some(false)); } #[test] fn test_edit_collection_toggle_search_on_add_submit() { let current_route = Route::from(( ActiveRadarrBlock::EditCollectionPrompt, Some(ActiveRadarrBlock::Collections), )); let mut app = App::default(); app.data.radarr_data.selected_block = BlockSelectionState::new(&EDIT_COLLECTION_SELECTION_BLOCKS); app .data .radarr_data .selected_block .set_index(EDIT_COLLECTION_SELECTION_BLOCKS.len() - 2); app.push_navigation_stack(current_route); EditCollectionHandler::with( &SUBMIT_KEY, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &Some(ActiveRadarrBlock::Collections), ) .handle(); assert_eq!(app.get_current_route(), ¤t_route); assert_eq!(app.data.radarr_data.edit_search_on_add, Some(true)); EditCollectionHandler::with( &SUBMIT_KEY, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &Some(ActiveRadarrBlock::Collections), ) .handle(); assert_eq!(app.get_current_route(), ¤t_route); assert_eq!(app.data.radarr_data.edit_search_on_add, Some(false)); } #[rstest] #[case(ActiveRadarrBlock::EditCollectionSelectMinimumAvailability, 1)] #[case(ActiveRadarrBlock::EditCollectionSelectQualityProfile, 2)] #[case(ActiveRadarrBlock::EditCollectionRootFolderPathInput, 3)] fn test_edit_collection_prompt_selected_block_submit( #[case] selected_block: ActiveRadarrBlock, #[case] index: usize, ) { let mut app = App::default(); app.push_navigation_stack( ( ActiveRadarrBlock::EditCollectionPrompt, Some(ActiveRadarrBlock::Collections), ) .into(), ); app.data.radarr_data.selected_block = BlockSelectionState::new(&EDIT_COLLECTION_SELECTION_BLOCKS); app.data.radarr_data.selected_block.set_index(index); EditCollectionHandler::with( &SUBMIT_KEY, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &Some(ActiveRadarrBlock::Collections), ) .handle(); assert_eq!( app.get_current_route(), &(selected_block, Some(ActiveRadarrBlock::Collections)).into() ); assert_eq!(app.data.radarr_data.prompt_confirm_action, None); if selected_block == ActiveRadarrBlock::EditCollectionRootFolderPathInput { assert!(app.should_ignore_quit_key); } } #[rstest] fn test_edit_collection_prompt_selecting_preferences_blocks_submit( #[values( ActiveRadarrBlock::EditCollectionSelectMinimumAvailability, ActiveRadarrBlock::EditCollectionSelectQualityProfile, ActiveRadarrBlock::EditCollectionRootFolderPathInput )] active_radarr_block: ActiveRadarrBlock, ) { let mut app = App::default(); app.push_navigation_stack(ActiveRadarrBlock::EditCollectionPrompt.into()); app.push_navigation_stack(active_radarr_block.into()); EditCollectionHandler::with( &SUBMIT_KEY, &mut app, &active_radarr_block, &Some(ActiveRadarrBlock::Collections), ) .handle(); assert_eq!( app.get_current_route(), &ActiveRadarrBlock::EditCollectionPrompt.into() ); if active_radarr_block == ActiveRadarrBlock::EditCollectionRootFolderPathInput { assert!(!app.should_ignore_quit_key); } } } mod test_handle_esc { use pretty_assertions::assert_eq; use rstest::rstest; use crate::models::servarr_data::radarr_data::radarr_test_utils::utils::create_test_radarr_data; use crate::{assert_edit_media_reset, assert_preferences_selections_reset}; use super::*; const ESC_KEY: Key = DEFAULT_KEYBINDINGS.esc.key; #[test] fn test_edit_collection_root_folder_path_input_esc() { let mut app = App::default(); app.data.radarr_data = create_test_radarr_data(); app.should_ignore_quit_key = true; app.push_navigation_stack(ActiveRadarrBlock::EditCollectionPrompt.into()); app.push_navigation_stack(ActiveRadarrBlock::EditCollectionRootFolderPathInput.into()); EditCollectionHandler::with( &ESC_KEY, &mut app, &ActiveRadarrBlock::EditCollectionRootFolderPathInput, &None, ) .handle(); assert!(!app.should_ignore_quit_key); assert_eq!( app.get_current_route(), &ActiveRadarrBlock::EditCollectionPrompt.into() ); } #[test] fn test_edit_collection_prompt_esc() { let mut app = App::default(); app.push_navigation_stack(ActiveRadarrBlock::Collections.into()); app.push_navigation_stack(ActiveRadarrBlock::EditCollectionPrompt.into()); app.data.radarr_data = create_test_radarr_data(); EditCollectionHandler::with( &ESC_KEY, &mut app, &ActiveRadarrBlock::EditCollectionPrompt, &None, ) .handle(); assert_eq!( app.get_current_route(), &ActiveRadarrBlock::Collections.into() ); let radarr_data = &app.data.radarr_data; assert_preferences_selections_reset!(radarr_data); assert_edit_media_reset!(radarr_data); assert!(!radarr_data.prompt_confirm); } #[rstest] fn test_edit_collection_esc( #[values( ActiveRadarrBlock::EditCollectionSelectMinimumAvailability, ActiveRadarrBlock::EditCollectionSelectQualityProfile )] active_radarr_block: ActiveRadarrBlock, ) { let mut app = App::default(); app.data.radarr_data = create_test_radarr_data(); app.push_navigation_stack(ActiveRadarrBlock::Collections.into()); app.push_navigation_stack(active_radarr_block.into()); EditCollectionHandler::with(&ESC_KEY, &mut app, &active_radarr_block, &None).handle(); assert_eq!( app.get_current_route(), &ActiveRadarrBlock::Collections.into() ); } } mod test_handle_key_char { use super::*; #[test] fn test_edit_collection_root_folder_path_input_backspace() { let mut app = App::default(); app.data.radarr_data.edit_path = "Test".to_owned().into(); EditCollectionHandler::with( &DEFAULT_KEYBINDINGS.backspace.key, &mut app, &ActiveRadarrBlock::EditCollectionRootFolderPathInput, &None, ) .handle(); assert_str_eq!(app.data.radarr_data.edit_path.text, "Tes"); } #[test] fn test_edit_collection_root_folder_path_input_char_key() { let mut app = App::default(); EditCollectionHandler::with( &Key::Char('h'), &mut app, &ActiveRadarrBlock::EditCollectionRootFolderPathInput, &None, ) .handle(); assert_str_eq!(app.data.radarr_data.edit_path.text, "h"); } } #[test] fn test_edit_collection_handler_accepts() { ActiveRadarrBlock::iter().for_each(|active_radarr_block| { if EDIT_COLLECTION_BLOCKS.contains(&active_radarr_block) { assert!(EditCollectionHandler::accepts(&active_radarr_block)); } else { assert!(!EditCollectionHandler::accepts(&active_radarr_block)); } }); } }