diff --git a/src/cli/sonarr/delete_command_handler.rs b/src/cli/sonarr/delete_command_handler.rs index 32a9d28..15d052f 100644 --- a/src/cli/sonarr/delete_command_handler.rs +++ b/src/cli/sonarr/delete_command_handler.rs @@ -7,6 +7,7 @@ use tokio::sync::Mutex; use crate::{ app::App, cli::{CliCommandHandler, Command}, + models::sonarr_models::DeleteSeriesParams, network::{sonarr_network::SonarrEvent, NetworkTrait}, }; @@ -42,6 +43,15 @@ pub enum SonarrDeleteCommand { #[arg(long, help = "The ID of the root folder to delete", required = true)] root_folder_id: i64, }, + #[command(about = "Delete a series from your Sonarr library")] + Series { + #[arg(long, help = "The ID of the series to delete", required = true)] + series_id: i64, + #[arg(long, help = "Delete the series files from disk as well")] + delete_files_from_disk: bool, + #[arg(long, help = "Add a list exclusion for this series")] + add_list_exclusion: bool, + }, #[command(about = "Delete the tag with the specified ID")] Tag { #[arg(long, help = "The ID of the tag to delete", required = true)] @@ -104,6 +114,22 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrDeleteCommand> for SonarrDeleteComm .await?; serde_json::to_string_pretty(&resp)? } + SonarrDeleteCommand::Series { + series_id, + delete_files_from_disk, + add_list_exclusion, + } => { + let delete_series_params = DeleteSeriesParams { + id: series_id, + delete_series_files: delete_files_from_disk, + add_list_exclusion, + }; + let resp = self + .network + .handle_network_event(SonarrEvent::DeleteSeries(Some(delete_series_params)).into()) + .await?; + serde_json::to_string_pretty(&resp)? + } SonarrDeleteCommand::Tag { tag_id } => { let resp = self .network diff --git a/src/cli/sonarr/delete_command_handler_tests.rs b/src/cli/sonarr/delete_command_handler_tests.rs index 4400a90..35d7d4d 100644 --- a/src/cli/sonarr/delete_command_handler_tests.rs +++ b/src/cli/sonarr/delete_command_handler_tests.rs @@ -158,6 +158,63 @@ mod tests { } } + #[test] + fn test_delete_series_requires_arguments() { + let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "delete", "series"]); + + assert!(result.is_err()); + assert_eq!( + result.unwrap_err().kind(), + ErrorKind::MissingRequiredArgument + ); + } + + #[test] + fn test_delete_series_defaults() { + let expected_args = SonarrDeleteCommand::Series { + series_id: 1, + delete_files_from_disk: false, + add_list_exclusion: false, + }; + + let result = + Cli::try_parse_from(["managarr", "sonarr", "delete", "series", "--series-id", "1"]); + + assert!(result.is_ok()); + + if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command + { + assert_eq!(delete_command, expected_args); + } + } + + #[test] + fn test_delete_series_all_args_defined() { + let expected_args = SonarrDeleteCommand::Series { + series_id: 1, + delete_files_from_disk: true, + add_list_exclusion: true, + }; + + let result = Cli::try_parse_from([ + "managarr", + "sonarr", + "delete", + "series", + "--series-id", + "1", + "--delete-files-from-disk", + "--add-list-exclusion", + ]); + + assert!(result.is_ok()); + + if let Some(Command::Sonarr(SonarrCommand::Delete(delete_command))) = result.unwrap().command + { + assert_eq!(delete_command, expected_args); + } + } + #[test] fn test_delete_tag_requires_arguments() { let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "delete", "tag"]); @@ -197,7 +254,10 @@ mod tests { sonarr::delete_command_handler::{SonarrDeleteCommand, SonarrDeleteCommandHandler}, CliCommandHandler, }, - models::{sonarr_models::SonarrSerdeable, Serdeable}, + models::{ + sonarr_models::{DeleteSeriesParams, SonarrSerdeable}, + Serdeable, + }, network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent}, }; @@ -310,6 +370,40 @@ mod tests { assert!(result.is_ok()); } + #[tokio::test] + async fn test_handle_delete_series_command() { + let expected_delete_series_params = DeleteSeriesParams { + id: 1, + delete_series_files: true, + add_list_exclusion: true, + }; + let mut mock_network = MockNetworkTrait::new(); + mock_network + .expect_handle_network_event() + .with(eq::( + SonarrEvent::DeleteSeries(Some(expected_delete_series_params)).into(), + )) + .times(1) + .returning(|_| { + Ok(Serdeable::Sonarr(SonarrSerdeable::Value( + json!({"testResponse": "response"}), + ))) + }); + let app_arc = Arc::new(Mutex::new(App::default())); + let delete_series_command = SonarrDeleteCommand::Series { + series_id: 1, + delete_files_from_disk: true, + add_list_exclusion: true, + }; + + let result = + SonarrDeleteCommandHandler::with(&app_arc, delete_series_command, &mut mock_network) + .handle() + .await; + + assert!(result.is_ok()); + } + #[tokio::test] async fn test_handle_delete_tag_command() { let expected_tag_id = 1;