diff --git a/src/cli/sonarr/list_command_handler.rs b/src/cli/sonarr/list_command_handler.rs index 7bc46ff..1603308 100644 --- a/src/cli/sonarr/list_command_handler.rs +++ b/src/cli/sonarr/list_command_handler.rs @@ -33,6 +33,15 @@ pub enum SonarrListCommand { )] series_id: i64, }, + #[command(about = "List the episode files for the series with the given ID")] + EpisodeFiles { + #[arg( + long, + help = "The Sonarr ID of the series whose episode files you wish to fetch", + required = true + )] + series_id: i64, + }, #[command(about = "Fetch all history events for the episode with the given ID")] EpisodeHistory { #[arg( @@ -158,6 +167,13 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrListCommand> for SonarrListCommandH .await?; serde_json::to_string_pretty(&resp)? } + SonarrListCommand::EpisodeFiles { series_id } => { + let resp = self + .network + .handle_network_event(SonarrEvent::GetEpisodeFiles(Some(series_id)).into()) + .await?; + serde_json::to_string_pretty(&resp)? + } SonarrListCommand::EpisodeHistory { episode_id } => { 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 54cc3b0..c333193 100644 --- a/src/cli/sonarr/list_command_handler_tests.rs +++ b/src/cli/sonarr/list_command_handler_tests.rs @@ -57,6 +57,18 @@ mod tests { ); } + #[test] + fn test_list_episode_files_requires_series_id() { + let result = + Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "episode-files"]); + + assert!(result.is_err()); + assert_eq!( + result.unwrap_err().kind(), + ErrorKind::MissingRequiredArgument + ); + } + #[test] fn test_list_episode_history_requires_series_id() { let result = @@ -149,6 +161,27 @@ mod tests { } } + #[test] + fn test_list_episode_files_success() { + let expected_args = SonarrListCommand::EpisodeFiles { series_id: 1 }; + let result = Cli::try_parse_from([ + "managarr", + "sonarr", + "list", + "episode-files", + "--series-id", + "1", + ]); + + assert!(result.is_ok()); + + if let Some(Command::Sonarr(SonarrCommand::List(episode_files_command))) = + result.unwrap().command + { + assert_eq!(episode_files_command, expected_args); + } + } + #[test] fn test_season_history_requires_series_id() { let result = Cli::command().try_get_matches_from([ @@ -315,6 +348,32 @@ mod tests { assert!(result.is_ok()); } + #[tokio::test] + async fn test_handle_list_episode_files_command() { + let expected_series_id = 1; + let mut mock_network = MockNetworkTrait::new(); + mock_network + .expect_handle_network_event() + .with(eq::( + SonarrEvent::GetEpisodeFiles(Some(expected_series_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_files_command = SonarrListCommand::EpisodeFiles { series_id: 1 }; + + let result = + SonarrListCommandHandler::with(&app_arc, list_episode_files_command, &mut mock_network) + .handle() + .await; + + assert!(result.is_ok()); + } + #[tokio::test] async fn test_handle_list_history_command() { let expected_events = 1000;