#[cfg(test)] mod tests { mod cli { use clap::CommandFactory; use clap::error::ErrorKind; use pretty_assertions::assert_eq; use crate::Cli; #[test] fn test_readarr_command_requires_a_subcommand() { let result = Cli::command().try_get_matches_from(["managarr", "readarr"]); assert_err!(&result); } #[test] fn test_search_new_author_requires_query() { let result = Cli::command().try_get_matches_from(["managarr", "readarr", "search-new-author"]); assert_err!(&result); assert_eq!( result.unwrap_err().kind(), ErrorKind::MissingRequiredArgument ); } #[test] fn test_search_new_author_requirements_satisfied() { let result = Cli::command().try_get_matches_from([ "managarr", "readarr", "search-new-author", "--query", "test query", ]); assert_ok!(&result); } #[test] fn test_start_task_requires_task_name() { let result = Cli::command().try_get_matches_from(["managarr", "readarr", "start-task"]); assert_err!(&result); assert_eq!( result.unwrap_err().kind(), ErrorKind::MissingRequiredArgument ); } #[test] fn test_start_task_task_name_validation() { let result = Cli::command().try_get_matches_from([ "managarr", "readarr", "start-task", "--task-name", "test", ]); assert_err!(&result); assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue); } #[test] fn test_start_task_requirements_satisfied() { let result = Cli::command().try_get_matches_from([ "managarr", "readarr", "start-task", "--task-name", "check-health", ]); assert_ok!(&result); } #[test] fn test_toggle_author_monitoring_requires_author_id() { let result = Cli::command().try_get_matches_from(["managarr", "readarr", "toggle-author-monitoring"]); assert_err!(&result); assert_eq!( result.unwrap_err().kind(), ErrorKind::MissingRequiredArgument ); } #[test] fn test_toggle_author_monitoring_requirements_satisfied() { let result = Cli::command().try_get_matches_from([ "managarr", "readarr", "toggle-author-monitoring", "--author-id", "1", ]); assert_ok!(&result); } } mod handler { use std::sync::Arc; use mockall::predicate::eq; use serde_json::json; use tokio::sync::Mutex; use crate::models::readarr_models::{ReadarrSerdeable, ReadarrTaskName}; use crate::{ app::App, cli::{ CliCommandHandler, readarr::{ReadarrCliHandler, ReadarrCommand}, }, models::Serdeable, network::{MockNetworkTrait, NetworkEvent, readarr_network::ReadarrEvent}, }; #[tokio::test] async fn test_search_new_author_command() { let expected_query = "test author".to_owned(); let mut mock_network = MockNetworkTrait::new(); mock_network .expect_handle_network_event() .with(eq::( ReadarrEvent::SearchNewAuthor(expected_query.clone()).into(), )) .times(1) .returning(|_| { Ok(Serdeable::Readarr(ReadarrSerdeable::Value( json!({"testResponse": "response"}), ))) }); let app_arc = Arc::new(Mutex::new(App::test_default())); let search_new_author_command = ReadarrCommand::SearchNewAuthor { query: expected_query, }; let result = ReadarrCliHandler::with(&app_arc, search_new_author_command, &mut mock_network) .handle() .await; assert_ok!(&result); } #[tokio::test] async fn test_start_task_command() { let expected_task_name = ReadarrTaskName::CheckHealth; let mut mock_network = MockNetworkTrait::new(); mock_network .expect_handle_network_event() .with(eq::( ReadarrEvent::StartTask(expected_task_name).into(), )) .times(1) .returning(|_| { Ok(Serdeable::Readarr(ReadarrSerdeable::Value( json!({"testResponse": "response"}), ))) }); let app_arc = Arc::new(Mutex::new(App::test_default())); let start_task_command = ReadarrCommand::StartTask { task_name: ReadarrTaskName::CheckHealth, }; let result = ReadarrCliHandler::with(&app_arc, start_task_command, &mut mock_network) .handle() .await; assert_ok!(&result); } #[tokio::test] async fn test_toggle_author_monitoring_command() { let mut mock_network = MockNetworkTrait::new(); mock_network .expect_handle_network_event() .with(eq::( ReadarrEvent::ToggleAuthorMonitoring(1).into(), )) .times(1) .returning(|_| { Ok(Serdeable::Readarr(ReadarrSerdeable::Value( json!({"testResponse": "response"}), ))) }); let app_arc = Arc::new(Mutex::new(App::test_default())); let toggle_author_monitoring_command = ReadarrCommand::ToggleAuthorMonitoring { author_id: 1 }; let result = ReadarrCliHandler::with( &app_arc, toggle_author_monitoring_command, &mut mock_network, ) .handle() .await; assert_ok!(&result); } } }