feat(cli): Support for starting a Sonarr task

This commit is contained in:
2024-11-22 17:01:00 -07:00
parent 33db3efacf
commit aa13735533
2 changed files with 86 additions and 1 deletions
+18
View File
@@ -10,6 +10,7 @@ use tokio::sync::Mutex;
use crate::{
app::App,
models::sonarr_models::SonarrTaskName,
network::{sonarr_network::SonarrEvent, NetworkTrait},
};
@@ -79,6 +80,16 @@ pub enum SonarrCommand {
#[arg(long, help = "The season number to search for", required = true)]
season_number: i64,
},
#[command(about = "Start the specified Sonarr task")]
StartTask {
#[arg(
long,
help = "The name of the task to trigger",
value_enum,
required = true
)]
task_name: SonarrTaskName,
},
}
impl From<SonarrCommand> for Command {
@@ -167,6 +178,13 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrCommand> for SonarrCliHandler<'a, '
.await?;
serde_json::to_string_pretty(&resp)?
}
SonarrCommand::StartTask { task_name } => {
let resp = self
.network
.handle_network_event(SonarrEvent::StartTask(Some(task_name)).into())
.await?;
serde_json::to_string_pretty(&resp)?
}
};
Ok(result)
+68 -1
View File
@@ -126,6 +126,44 @@ mod tests {
assert!(result.is_ok());
}
#[rstest]
fn test_start_task_requires_task_name() {
let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "start-task"]);
assert!(result.is_err());
assert_eq!(
result.unwrap_err().kind(),
ErrorKind::MissingRequiredArgument
);
}
#[rstest]
fn test_start_task_task_name_validation() {
let result = Cli::command().try_get_matches_from([
"managarr",
"sonarr",
"start-task",
"--task-name",
"test",
]);
assert!(result.is_err());
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
}
#[test]
fn test_start_task_requirements_satisfied() {
let result = Cli::command().try_get_matches_from([
"managarr",
"sonarr",
"start-task",
"--task-name",
"application-update-check",
]);
assert!(result.is_ok());
}
}
mod handler {
@@ -146,7 +184,9 @@ mod tests {
CliCommandHandler,
},
models::{
sonarr_models::{BlocklistItem, BlocklistResponse, Series, SonarrSerdeable},
sonarr_models::{
BlocklistItem, BlocklistResponse, Series, SonarrSerdeable, SonarrTaskName,
},
Serdeable,
},
network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
@@ -370,5 +410,32 @@ mod tests {
assert!(result.is_ok());
}
#[tokio::test]
async fn test_start_task_command() {
let expected_task_name = SonarrTaskName::ApplicationUpdateCheck;
let mut mock_network = MockNetworkTrait::new();
mock_network
.expect_handle_network_event()
.with(eq::<NetworkEvent>(
SonarrEvent::StartTask(Some(expected_task_name)).into(),
))
.times(1)
.returning(|_| {
Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
json!({"testResponse": "response"}),
)))
});
let app_arc = Arc::new(Mutex::new(App::default()));
let start_task_command = SonarrCommand::StartTask {
task_name: SonarrTaskName::ApplicationUpdateCheck,
};
let result = SonarrCliHandler::with(&app_arc, start_task_command, &mut mock_network)
.handle()
.await;
assert!(result.is_ok());
}
}
}