diff --git a/src/cli/sonarr/list_command_handler.rs b/src/cli/sonarr/list_command_handler.rs index 567d2db..697292b 100644 --- a/src/cli/sonarr/list_command_handler.rs +++ b/src/cli/sonarr/list_command_handler.rs @@ -31,6 +31,15 @@ pub enum SonarrListCommand { )] series_id: i64, }, + #[command(about = "Fetch all history events for the episode with the given ID")] + EpisodeHistory { + #[arg( + long, + help = "The Sonarr ID of the episode whose history you wish to fetch", + required = true + )] + episode_id: i64, + }, #[command(about = "Fetch all Sonarr history events")] History { #[arg(long, help = "How many history events to fetch", default_value_t = 500)] @@ -113,6 +122,13 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrListCommand> for SonarrListCommandH .await?; serde_json::to_string_pretty(&resp)? } + SonarrListCommand::EpisodeHistory { episode_id } => { + let resp = self + .network + .handle_network_event((SonarrEvent::GetEpisodeHistory(Some(episode_id))).into()) + .await?; + serde_json::to_string_pretty(&resp)? + } SonarrListCommand::History { events: items } => { let resp = self .network diff --git a/src/cli/sonarr/list_command_handler_tests.rs b/src/cli/sonarr/list_command_handler_tests.rs index 6d7d9c1..e0d2e8a 100644 --- a/src/cli/sonarr/list_command_handler_tests.rs +++ b/src/cli/sonarr/list_command_handler_tests.rs @@ -50,6 +50,39 @@ mod tests { ); } + #[test] + fn test_list_episode_history_requires_series_id() { + let result = + Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "episode-history"]); + + assert!(result.is_err()); + assert_eq!( + result.unwrap_err().kind(), + ErrorKind::MissingRequiredArgument + ); + } + + #[test] + fn test_list_episode_history_success() { + let expected_args = SonarrListCommand::EpisodeHistory { episode_id: 1 }; + let result = Cli::try_parse_from([ + "managarr", + "sonarr", + "list", + "episode-history", + "--episode-id", + "1", + ]); + + assert!(result.is_ok()); + + if let Some(Command::Sonarr(SonarrCommand::List(episode_history_command))) = + result.unwrap().command + { + assert_eq!(episode_history_command, expected_args); + } + } + #[test] fn test_list_history_events_flag_requires_arguments() { let result = @@ -296,5 +329,31 @@ mod tests { assert!(result.is_ok()); } + + #[tokio::test] + async fn test_handle_list_episode_history_command() { + let expected_episode_id = 1; + let mut mock_network = MockNetworkTrait::new(); + mock_network + .expect_handle_network_event() + .with(eq::( + SonarrEvent::GetEpisodeHistory(Some(expected_episode_id)).into(), + )) + .times(1) + .returning(|_| { + Ok(Serdeable::Sonarr(SonarrSerdeable::Value( + json!({"testResponse": "response"}), + ))) + }); + let app_arc = Arc::new(Mutex::new(App::default())); + let list_episode_history_command = SonarrListCommand::EpisodeHistory { episode_id: 1 }; + + let result = + SonarrListCommandHandler::with(&app_arc, list_episode_history_command, &mut mock_network) + .handle() + .await; + + assert!(result.is_ok()); + } } }