486 lines
12 KiB
Rust
486 lines
12 KiB
Rust
#[cfg(test)]
|
|
mod tests {
|
|
use clap::{CommandFactory, Parser, error::ErrorKind};
|
|
|
|
use crate::{
|
|
Cli,
|
|
cli::{
|
|
Command,
|
|
radarr::{RadarrCommand, add_command_handler::RadarrAddCommand},
|
|
},
|
|
models::radarr_models::{MinimumAvailability, MovieMonitor},
|
|
};
|
|
use pretty_assertions::assert_eq;
|
|
|
|
#[test]
|
|
fn test_radarr_add_command_from() {
|
|
let command = RadarrAddCommand::Tag {
|
|
name: String::new(),
|
|
};
|
|
|
|
let result = Command::from(command.clone());
|
|
|
|
assert_eq!(result, Command::Radarr(RadarrCommand::Add(command)));
|
|
}
|
|
|
|
mod cli {
|
|
use super::*;
|
|
use pretty_assertions::assert_eq;
|
|
use rstest::rstest;
|
|
|
|
#[test]
|
|
fn test_add_movie_requires_arguments() {
|
|
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "add", "movie"]);
|
|
|
|
assert_err!(&result);
|
|
assert_eq!(
|
|
result.unwrap_err().kind(),
|
|
ErrorKind::MissingRequiredArgument
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_movie_requires_root_folder_path() {
|
|
let result = Cli::command().try_get_matches_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--tmdb-id",
|
|
"1",
|
|
"--quality-profile-id",
|
|
"1",
|
|
]);
|
|
|
|
assert_err!(&result);
|
|
assert_eq!(
|
|
result.unwrap_err().kind(),
|
|
ErrorKind::MissingRequiredArgument
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_movie_requires_quality_profile_id() {
|
|
let result = Cli::command().try_get_matches_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--tmdb-id",
|
|
"1",
|
|
"--root-folder-path",
|
|
"/test",
|
|
]);
|
|
|
|
assert_err!(&result);
|
|
assert_eq!(
|
|
result.unwrap_err().kind(),
|
|
ErrorKind::MissingRequiredArgument
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_movie_requires_tmdb_id() {
|
|
let result = Cli::command().try_get_matches_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--root-folder-path",
|
|
"/test",
|
|
"--quality-profile-id",
|
|
"1",
|
|
]);
|
|
|
|
assert_err!(&result);
|
|
assert_eq!(
|
|
result.unwrap_err().kind(),
|
|
ErrorKind::MissingRequiredArgument
|
|
);
|
|
}
|
|
|
|
#[rstest]
|
|
fn test_add_movie_assert_argument_flags_require_args(
|
|
#[values("--minimum-availability", "--tag", "--monitor")] flag: &str,
|
|
) {
|
|
let result = Cli::command().try_get_matches_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--root-folder-path",
|
|
"/test",
|
|
"--quality-profile-id",
|
|
"1",
|
|
"--tmdb-id",
|
|
"1",
|
|
flag,
|
|
]);
|
|
|
|
assert_err!(&result);
|
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_movie_all_arguments_satisfied() {
|
|
let result = Cli::command().try_get_matches_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--root-folder-path",
|
|
"/test",
|
|
"--quality-profile-id",
|
|
"1",
|
|
"--tmdb-id",
|
|
"1",
|
|
]);
|
|
|
|
assert_ok!(&result);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_movie_minimum_availability_validation() {
|
|
let result = Cli::command().try_get_matches_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--root-folder-path",
|
|
"/test",
|
|
"--quality-profile-id",
|
|
"1",
|
|
"--tmdb-id",
|
|
"1",
|
|
"--minimum-availability",
|
|
"test",
|
|
]);
|
|
|
|
assert_err!(&result);
|
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_movie_monitor_validation() {
|
|
let result = Cli::command().try_get_matches_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--root-folder-path",
|
|
"/test",
|
|
"--quality-profile-id",
|
|
"1",
|
|
"--tmdb-id",
|
|
"1",
|
|
"--monitor",
|
|
"test",
|
|
]);
|
|
|
|
assert_err!(&result);
|
|
assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_movie_defaults() {
|
|
let expected_args = RadarrAddCommand::Movie {
|
|
tmdb_id: 1,
|
|
root_folder_path: "/test".to_owned(),
|
|
quality_profile_id: 1,
|
|
minimum_availability: MinimumAvailability::default(),
|
|
disable_monitoring: false,
|
|
tag: vec![],
|
|
monitor: MovieMonitor::default(),
|
|
no_search_for_movie: false,
|
|
};
|
|
|
|
let result = Cli::try_parse_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--root-folder-path",
|
|
"/test",
|
|
"--quality-profile-id",
|
|
"1",
|
|
"--tmdb-id",
|
|
"1",
|
|
]);
|
|
|
|
assert_ok!(&result);
|
|
let Some(Command::Radarr(RadarrCommand::Add(add_command))) = result.unwrap().command else {
|
|
panic!("Unexpected command type")
|
|
};
|
|
assert_eq!(add_command, expected_args);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_movie_tags_is_repeatable() {
|
|
let expected_args = RadarrAddCommand::Movie {
|
|
tmdb_id: 1,
|
|
root_folder_path: "/test".to_owned(),
|
|
quality_profile_id: 1,
|
|
minimum_availability: MinimumAvailability::default(),
|
|
disable_monitoring: false,
|
|
tag: vec![1, 2],
|
|
monitor: MovieMonitor::default(),
|
|
no_search_for_movie: false,
|
|
};
|
|
|
|
let result = Cli::try_parse_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--root-folder-path",
|
|
"/test",
|
|
"--quality-profile-id",
|
|
"1",
|
|
"--tmdb-id",
|
|
"1",
|
|
"--tag",
|
|
"1",
|
|
"--tag",
|
|
"2",
|
|
]);
|
|
|
|
assert_ok!(&result);
|
|
let Some(Command::Radarr(RadarrCommand::Add(add_command))) = result.unwrap().command else {
|
|
panic!("Unexpected command type")
|
|
};
|
|
assert_eq!(add_command, expected_args);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_movie_all_args_defined() {
|
|
let expected_args = RadarrAddCommand::Movie {
|
|
tmdb_id: 1,
|
|
root_folder_path: "/test".to_owned(),
|
|
quality_profile_id: 1,
|
|
minimum_availability: MinimumAvailability::Released,
|
|
disable_monitoring: true,
|
|
tag: vec![1, 2],
|
|
monitor: MovieMonitor::MovieAndCollection,
|
|
no_search_for_movie: true,
|
|
};
|
|
|
|
let result = Cli::try_parse_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"movie",
|
|
"--root-folder-path",
|
|
"/test",
|
|
"--quality-profile-id",
|
|
"1",
|
|
"--minimum-availability",
|
|
"released",
|
|
"--disable-monitoring",
|
|
"--tmdb-id",
|
|
"1",
|
|
"--tag",
|
|
"1",
|
|
"--tag",
|
|
"2",
|
|
"--monitor",
|
|
"movie-and-collection",
|
|
"--no-search-for-movie",
|
|
]);
|
|
|
|
assert_ok!(&result);
|
|
let Some(Command::Radarr(RadarrCommand::Add(add_command))) = result.unwrap().command else {
|
|
panic!("Unexpected command type")
|
|
};
|
|
assert_eq!(add_command, expected_args);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_root_folder_requires_arguments() {
|
|
let result =
|
|
Cli::command().try_get_matches_from(["managarr", "radarr", "add", "root-folder"]);
|
|
|
|
assert_err!(&result);
|
|
assert_eq!(
|
|
result.unwrap_err().kind(),
|
|
ErrorKind::MissingRequiredArgument
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_root_folder_success() {
|
|
let expected_args = RadarrAddCommand::RootFolder {
|
|
root_folder_path: "/nfs/test".to_owned(),
|
|
};
|
|
|
|
let result = Cli::try_parse_from([
|
|
"managarr",
|
|
"radarr",
|
|
"add",
|
|
"root-folder",
|
|
"--root-folder-path",
|
|
"/nfs/test",
|
|
]);
|
|
|
|
assert_ok!(&result);
|
|
|
|
let Some(Command::Radarr(RadarrCommand::Add(add_command))) = result.unwrap().command else {
|
|
panic!("Unexpected command type")
|
|
};
|
|
assert_eq!(add_command, expected_args);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_tag_requires_arguments() {
|
|
let result = Cli::command().try_get_matches_from(["managarr", "radarr", "add", "tag"]);
|
|
|
|
assert_err!(&result);
|
|
assert_eq!(
|
|
result.unwrap_err().kind(),
|
|
ErrorKind::MissingRequiredArgument
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn test_add_tag_success() {
|
|
let expected_args = RadarrAddCommand::Tag {
|
|
name: "test".to_owned(),
|
|
};
|
|
|
|
let result = Cli::try_parse_from(["managarr", "radarr", "add", "tag", "--name", "test"]);
|
|
|
|
assert_ok!(&result);
|
|
|
|
let Some(Command::Radarr(RadarrCommand::Add(add_command))) = result.unwrap().command else {
|
|
panic!("Unexpected command type")
|
|
};
|
|
assert_eq!(add_command, expected_args);
|
|
}
|
|
}
|
|
|
|
mod handler {
|
|
use std::sync::Arc;
|
|
|
|
use crate::{
|
|
app::App,
|
|
cli::{CliCommandHandler, radarr::add_command_handler::RadarrAddCommandHandler},
|
|
models::{
|
|
Serdeable,
|
|
radarr_models::{AddMovieBody, AddMovieOptions, RadarrSerdeable},
|
|
},
|
|
network::{MockNetworkTrait, NetworkEvent, radarr_network::RadarrEvent},
|
|
};
|
|
|
|
use super::*;
|
|
use mockall::predicate::eq;
|
|
|
|
use crate::models::servarr_models::AddRootFolderBody;
|
|
use serde_json::json;
|
|
use tokio::sync::Mutex;
|
|
|
|
#[tokio::test]
|
|
async fn test_handle_add_movie_command() {
|
|
let expected_add_movie_body = AddMovieBody {
|
|
tmdb_id: 1,
|
|
title: String::new(),
|
|
root_folder_path: "/test".to_owned(),
|
|
quality_profile_id: 1,
|
|
minimum_availability: MinimumAvailability::Released,
|
|
monitored: false,
|
|
tags: vec![1, 2],
|
|
tag_input_string: None,
|
|
add_options: AddMovieOptions {
|
|
monitor: MovieMonitor::MovieAndCollection,
|
|
search_for_movie: false,
|
|
},
|
|
};
|
|
let mut mock_network = MockNetworkTrait::new();
|
|
mock_network
|
|
.expect_handle_network_event()
|
|
.with(eq::<NetworkEvent>(
|
|
RadarrEvent::AddMovie(expected_add_movie_body).into(),
|
|
))
|
|
.times(1)
|
|
.returning(|_| {
|
|
Ok(Serdeable::Radarr(RadarrSerdeable::Value(
|
|
json!({"testResponse": "response"}),
|
|
)))
|
|
});
|
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
|
let add_movie_command = RadarrAddCommand::Movie {
|
|
tmdb_id: 1,
|
|
root_folder_path: "/test".to_owned(),
|
|
quality_profile_id: 1,
|
|
minimum_availability: MinimumAvailability::Released,
|
|
disable_monitoring: true,
|
|
tag: vec![1, 2],
|
|
monitor: MovieMonitor::MovieAndCollection,
|
|
no_search_for_movie: true,
|
|
};
|
|
|
|
let result = RadarrAddCommandHandler::with(&app_arc, add_movie_command, &mut mock_network)
|
|
.handle()
|
|
.await;
|
|
|
|
assert_ok!(&result);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_handle_add_root_folder_command() {
|
|
let expected_root_folder_path = "/nfs/test".to_owned();
|
|
let expected_add_root_folder_body = AddRootFolderBody {
|
|
path: expected_root_folder_path.clone(),
|
|
};
|
|
let mut mock_network = MockNetworkTrait::new();
|
|
mock_network
|
|
.expect_handle_network_event()
|
|
.with(eq::<NetworkEvent>(
|
|
RadarrEvent::AddRootFolder(expected_add_root_folder_body).into(),
|
|
))
|
|
.times(1)
|
|
.returning(|_| {
|
|
Ok(Serdeable::Radarr(RadarrSerdeable::Value(
|
|
json!({"testResponse": "response"}),
|
|
)))
|
|
});
|
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
|
let add_root_folder_command = RadarrAddCommand::RootFolder {
|
|
root_folder_path: expected_root_folder_path,
|
|
};
|
|
|
|
let result =
|
|
RadarrAddCommandHandler::with(&app_arc, add_root_folder_command, &mut mock_network)
|
|
.handle()
|
|
.await;
|
|
|
|
assert_ok!(&result);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_handle_add_tag_command() {
|
|
let expected_tag_name = "test".to_owned();
|
|
let mut mock_network = MockNetworkTrait::new();
|
|
mock_network
|
|
.expect_handle_network_event()
|
|
.with(eq::<NetworkEvent>(
|
|
RadarrEvent::AddTag(expected_tag_name.clone()).into(),
|
|
))
|
|
.times(1)
|
|
.returning(|_| {
|
|
Ok(Serdeable::Radarr(RadarrSerdeable::Value(
|
|
json!({"testResponse": "response"}),
|
|
)))
|
|
});
|
|
let app_arc = Arc::new(Mutex::new(App::test_default()));
|
|
let add_tag_command = RadarrAddCommand::Tag {
|
|
name: expected_tag_name,
|
|
};
|
|
|
|
let result = RadarrAddCommandHandler::with(&app_arc, add_tag_command, &mut mock_network)
|
|
.handle()
|
|
.await;
|
|
|
|
assert_ok!(&result);
|
|
}
|
|
}
|
|
}
|