fix(radarr): Pass in the search query for the SearchNewMovie event when publishing to the networking channel
This commit is contained in:
@@ -103,7 +103,7 @@ impl<'a> App<'a> {
|
|||||||
}
|
}
|
||||||
ActiveRadarrBlock::AddMovieSearchResults => {
|
ActiveRadarrBlock::AddMovieSearchResults => {
|
||||||
self
|
self
|
||||||
.dispatch_network_event(RadarrEvent::SearchNewMovie(None).into())
|
.dispatch_network_event(RadarrEvent::SearchNewMovie(self.extract_movie_search_query().await).into())
|
||||||
.await;
|
.await;
|
||||||
}
|
}
|
||||||
ActiveRadarrBlock::MovieDetails | ActiveRadarrBlock::FileInfo => {
|
ActiveRadarrBlock::MovieDetails | ActiveRadarrBlock::FileInfo => {
|
||||||
@@ -219,7 +219,7 @@ impl<'a> App<'a> {
|
|||||||
.collection_movies
|
.collection_movies
|
||||||
.set_items(collection_movies);
|
.set_items(collection_movies);
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn extract_movie_id(&self) -> i64 {
|
async fn extract_movie_id(&self) -> i64 {
|
||||||
self
|
self
|
||||||
.data
|
.data
|
||||||
@@ -229,4 +229,8 @@ impl<'a> App<'a> {
|
|||||||
.clone()
|
.clone()
|
||||||
.id
|
.id
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn extract_movie_search_query(&self) -> String {
|
||||||
|
self.data.radarr_data.add_movie_search.as_ref().expect("Add movie search is empty").text.clone()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -305,6 +305,7 @@ mod tests {
|
|||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_dispatch_by_add_movie_search_results_block() {
|
async fn test_dispatch_by_add_movie_search_results_block() {
|
||||||
let (mut app, mut sync_network_rx) = construct_app_unit();
|
let (mut app, mut sync_network_rx) = construct_app_unit();
|
||||||
|
app.data.radarr_data.add_movie_search = Some("test".into());
|
||||||
|
|
||||||
app
|
app
|
||||||
.dispatch_by_radarr_block(&ActiveRadarrBlock::AddMovieSearchResults)
|
.dispatch_by_radarr_block(&ActiveRadarrBlock::AddMovieSearchResults)
|
||||||
@@ -313,7 +314,7 @@ mod tests {
|
|||||||
assert!(app.is_loading);
|
assert!(app.is_loading);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
sync_network_rx.recv().await.unwrap(),
|
sync_network_rx.recv().await.unwrap(),
|
||||||
RadarrEvent::SearchNewMovie(None).into()
|
RadarrEvent::SearchNewMovie("test".into()).into()
|
||||||
);
|
);
|
||||||
assert!(!app.data.radarr_data.prompt_confirm);
|
assert!(!app.data.radarr_data.prompt_confirm);
|
||||||
assert_eq!(app.tick_count, 0);
|
assert_eq!(app.tick_count, 0);
|
||||||
|
|||||||
@@ -224,7 +224,7 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, RadarrCommand> for RadarrCliHandler<'a, '
|
|||||||
RadarrCommand::SearchNewMovie { query } => {
|
RadarrCommand::SearchNewMovie { query } => {
|
||||||
let resp = self
|
let resp = self
|
||||||
.network
|
.network
|
||||||
.handle_network_event(RadarrEvent::SearchNewMovie(Some(query)).into())
|
.handle_network_event(RadarrEvent::SearchNewMovie(query).into())
|
||||||
.await?;
|
.await?;
|
||||||
serde_json::to_string_pretty(&resp)?
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -367,7 +367,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
RadarrEvent::SearchNewMovie(Some(expected_search_query)).into(),
|
RadarrEvent::SearchNewMovie(expected_search_query).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
|
|||||||
@@ -6,6 +6,7 @@ use crate::models::servarr_data::radarr::modals::AddMovieModal;
|
|||||||
use crate::models::servarr_data::radarr::radarr_data::{
|
use crate::models::servarr_data::radarr::radarr_data::{
|
||||||
ActiveRadarrBlock, ADD_MOVIE_BLOCKS, ADD_MOVIE_SELECTION_BLOCKS,
|
ActiveRadarrBlock, ADD_MOVIE_BLOCKS, ADD_MOVIE_SELECTION_BLOCKS,
|
||||||
};
|
};
|
||||||
|
use crate::models::stateful_table::StatefulTable;
|
||||||
use crate::models::{BlockSelectionState, Scrollable};
|
use crate::models::{BlockSelectionState, Scrollable};
|
||||||
use crate::network::radarr_network::RadarrEvent;
|
use crate::network::radarr_network::RadarrEvent;
|
||||||
use crate::{handle_table_events, handle_text_box_keys, handle_text_box_left_right_keys, App, Key};
|
use crate::{handle_table_events, handle_text_box_keys, handle_text_box_left_right_keys, App, Key};
|
||||||
@@ -31,7 +32,7 @@ impl<'a, 'b> AddMovieHandler<'a, 'b> {
|
|||||||
.radarr_data
|
.radarr_data
|
||||||
.add_searched_movies
|
.add_searched_movies
|
||||||
.as_mut()
|
.as_mut()
|
||||||
.unwrap(),
|
.unwrap_or(&mut StatefulTable::default()),
|
||||||
AddMovieSearchResult
|
AddMovieSearchResult
|
||||||
);
|
);
|
||||||
|
|
||||||
|
|||||||
@@ -1522,6 +1522,20 @@ mod tests {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_movie_search_no_panic_on_none_search_result() {
|
||||||
|
let mut app = App::default();
|
||||||
|
app.data.radarr_data.add_searched_movies = None;
|
||||||
|
|
||||||
|
AddMovieHandler::with(
|
||||||
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
|
&mut app,
|
||||||
|
ActiveRadarrBlock::AddMovieSearchResults,
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.handle();
|
||||||
|
}
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn test_build_add_movie_body(#[values(true, false)] movie_details_context: bool) {
|
fn test_build_add_movie_body(#[values(true, false)] movie_details_context: bool) {
|
||||||
|
|||||||
+105
-132
@@ -1,17 +1,16 @@
|
|||||||
use anyhow::{anyhow, Result};
|
use anyhow::Result;
|
||||||
use std::fmt::Debug;
|
use std::fmt::Debug;
|
||||||
|
|
||||||
use indoc::formatdoc;
|
use indoc::formatdoc;
|
||||||
use log::{debug, info, warn};
|
use log::{debug, info};
|
||||||
use serde_json::{json, Value};
|
use serde_json::{json, Value};
|
||||||
use urlencoding::encode;
|
use urlencoding::encode;
|
||||||
|
|
||||||
use crate::models::radarr_models::{
|
use crate::models::radarr_models::{
|
||||||
AddMovieBody, AddMovieSearchResult, BlocklistResponse, Collection,
|
AddMovieBody, AddMovieSearchResult, BlocklistResponse, Collection, Credit, CreditType,
|
||||||
Credit, CreditType, DeleteMovieParams, DownloadRecord, DownloadsResponse,
|
DeleteMovieParams, DownloadRecord, DownloadsResponse, EditCollectionParams, EditMovieParams,
|
||||||
EditCollectionParams, EditMovieParams, IndexerSettings, IndexerTestResult, Movie,
|
IndexerSettings, IndexerTestResult, Movie, MovieCommandBody, MovieHistoryItem, RadarrRelease,
|
||||||
MovieCommandBody, MovieHistoryItem, RadarrRelease, RadarrReleaseDownloadBody, RadarrSerdeable,
|
RadarrReleaseDownloadBody, RadarrSerdeable, RadarrTask, RadarrTaskName, SystemStatus,
|
||||||
RadarrTask, RadarrTaskName, SystemStatus,
|
|
||||||
};
|
};
|
||||||
use crate::models::servarr_data::modals::IndexerTestResultModalItem;
|
use crate::models::servarr_data::modals::IndexerTestResultModalItem;
|
||||||
use crate::models::servarr_data::radarr::modals::MovieDetailsModal;
|
use crate::models::servarr_data::radarr::modals::MovieDetailsModal;
|
||||||
@@ -70,7 +69,7 @@ pub enum RadarrEvent {
|
|||||||
GetTasks,
|
GetTasks,
|
||||||
GetUpdates,
|
GetUpdates,
|
||||||
HealthCheck,
|
HealthCheck,
|
||||||
SearchNewMovie(Option<String>),
|
SearchNewMovie(String),
|
||||||
StartTask(Option<RadarrTaskName>),
|
StartTask(Option<RadarrTaskName>),
|
||||||
TestIndexer(Option<i64>),
|
TestIndexer(Option<i64>),
|
||||||
TestAllIndexers,
|
TestAllIndexers,
|
||||||
@@ -301,14 +300,23 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn add_radarr_root_folder(&mut self, add_root_folder_body: AddRootFolderBody) -> Result<Value> {
|
async fn add_radarr_root_folder(
|
||||||
|
&mut self,
|
||||||
|
add_root_folder_body: AddRootFolderBody,
|
||||||
|
) -> Result<Value> {
|
||||||
info!("Adding new root folder to Radarr");
|
info!("Adding new root folder to Radarr");
|
||||||
let event = RadarrEvent::AddRootFolder(add_root_folder_body.clone());
|
let event = RadarrEvent::AddRootFolder(add_root_folder_body.clone());
|
||||||
|
|
||||||
debug!("Add root folder body: {add_root_folder_body:?}");
|
debug!("Add root folder body: {add_root_folder_body:?}");
|
||||||
|
|
||||||
let request_props = self
|
let request_props = self
|
||||||
.request_props_from(event, RequestMethod::Post, Some(add_root_folder_body), None, None)
|
.request_props_from(
|
||||||
|
event,
|
||||||
|
RequestMethod::Post,
|
||||||
|
Some(add_root_folder_body),
|
||||||
|
None,
|
||||||
|
None,
|
||||||
|
)
|
||||||
.await;
|
.await;
|
||||||
|
|
||||||
self
|
self
|
||||||
@@ -447,7 +455,11 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
|
|
||||||
async fn delete_movie(&mut self, delete_movie_params: DeleteMovieParams) -> Result<()> {
|
async fn delete_movie(&mut self, delete_movie_params: DeleteMovieParams) -> Result<()> {
|
||||||
let event = RadarrEvent::DeleteMovie(delete_movie_params.clone());
|
let event = RadarrEvent::DeleteMovie(delete_movie_params.clone());
|
||||||
let DeleteMovieParams { id, delete_movie_files, add_list_exclusion } = delete_movie_params;
|
let DeleteMovieParams {
|
||||||
|
id,
|
||||||
|
delete_movie_files,
|
||||||
|
add_list_exclusion,
|
||||||
|
} = delete_movie_params;
|
||||||
info!("Deleting Radarr movie with ID: {id} with deleteFiles={delete_movie_files} and addImportExclusion={add_list_exclusion}");
|
info!("Deleting Radarr movie with ID: {id} with deleteFiles={delete_movie_files} and addImportExclusion={add_list_exclusion}");
|
||||||
|
|
||||||
let request_props = self
|
let request_props = self
|
||||||
@@ -486,10 +498,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn download_radarr_release(
|
async fn download_radarr_release(&mut self, params: RadarrReleaseDownloadBody) -> Result<Value> {
|
||||||
&mut self,
|
|
||||||
params: RadarrReleaseDownloadBody,
|
|
||||||
) -> Result<Value> {
|
|
||||||
let event = RadarrEvent::DownloadRelease(params.clone());
|
let event = RadarrEvent::DownloadRelease(params.clone());
|
||||||
info!("Downloading Radarr release with params: {params:?}");
|
info!("Downloading Radarr release with params: {params:?}");
|
||||||
|
|
||||||
@@ -502,10 +511,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn edit_all_radarr_indexer_settings(
|
async fn edit_all_radarr_indexer_settings(&mut self, params: IndexerSettings) -> Result<Value> {
|
||||||
&mut self,
|
|
||||||
params: IndexerSettings,
|
|
||||||
) -> Result<Value> {
|
|
||||||
info!("Updating Radarr indexer settings");
|
info!("Updating Radarr indexer settings");
|
||||||
let event = RadarrEvent::EditAllIndexerSettings(params.clone());
|
let event = RadarrEvent::EditAllIndexerSettings(params.clone());
|
||||||
|
|
||||||
@@ -520,10 +526,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn edit_collection(
|
async fn edit_collection(&mut self, edit_collection_params: EditCollectionParams) -> Result<()> {
|
||||||
&mut self,
|
|
||||||
edit_collection_params: EditCollectionParams,
|
|
||||||
) -> Result<()> {
|
|
||||||
info!("Editing Radarr collection");
|
info!("Editing Radarr collection");
|
||||||
let detail_event = RadarrEvent::GetCollections;
|
let detail_event = RadarrEvent::GetCollections;
|
||||||
let event = RadarrEvent::EditCollection(edit_collection_params.clone());
|
let event = RadarrEvent::EditCollection(edit_collection_params.clone());
|
||||||
@@ -564,11 +567,13 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.expect("Unable to deserialize 'minimumAvailability'")
|
.expect("Unable to deserialize 'minimumAvailability'")
|
||||||
})
|
})
|
||||||
.to_string();
|
.to_string();
|
||||||
let quality_profile_id = edit_collection_params.quality_profile_id.unwrap_or_else(|| {
|
let quality_profile_id = edit_collection_params
|
||||||
detailed_collection_body["qualityProfileId"]
|
.quality_profile_id
|
||||||
.as_i64()
|
.unwrap_or_else(|| {
|
||||||
.expect("Unable to deserialize 'qualityProfileId'")
|
detailed_collection_body["qualityProfileId"]
|
||||||
});
|
.as_i64()
|
||||||
|
.expect("Unable to deserialize 'qualityProfileId'")
|
||||||
|
});
|
||||||
let root_folder_path = edit_collection_params.root_folder_path.unwrap_or_else(|| {
|
let root_folder_path = edit_collection_params.root_folder_path.unwrap_or_else(|| {
|
||||||
detailed_collection_body["rootFolderPath"]
|
detailed_collection_body["rootFolderPath"]
|
||||||
.as_str()
|
.as_str()
|
||||||
@@ -590,7 +595,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
)
|
)
|
||||||
};
|
};
|
||||||
|
|
||||||
* detailed_collection_body.get_mut("monitored").unwrap() = json!(monitored);
|
*detailed_collection_body.get_mut("monitored").unwrap() = json!(monitored);
|
||||||
*detailed_collection_body
|
*detailed_collection_body
|
||||||
.get_mut("minimumAvailability")
|
.get_mut("minimumAvailability")
|
||||||
.unwrap() = json!(minimum_availability);
|
.unwrap() = json!(minimum_availability);
|
||||||
@@ -631,7 +636,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
edit_indexer_params.tags = Some(tag_ids_vec);
|
edit_indexer_params.tags = Some(tag_ids_vec);
|
||||||
}
|
}
|
||||||
info!("Updating Radarr indexer with ID: {id}");
|
info!("Updating Radarr indexer with ID: {id}");
|
||||||
|
|
||||||
info!("Fetching indexer details for indexer with ID: {id}");
|
info!("Fetching indexer details for indexer with ID: {id}");
|
||||||
|
|
||||||
let request_props = self
|
let request_props = self
|
||||||
@@ -857,53 +862,52 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
info!("Constructing edit movie body");
|
info!("Constructing edit movie body");
|
||||||
|
|
||||||
let mut detailed_movie_body: Value = serde_json::from_str(&response)?;
|
let mut detailed_movie_body: Value = serde_json::from_str(&response)?;
|
||||||
let (monitored, minimum_availability, quality_profile_id, root_folder_path, tags) =
|
let (monitored, minimum_availability, quality_profile_id, root_folder_path, tags) = {
|
||||||
{
|
let monitored = edit_movie_params.monitored.unwrap_or(
|
||||||
let monitored = edit_movie_params.monitored.unwrap_or(
|
detailed_movie_body["monitored"]
|
||||||
detailed_movie_body["monitored"]
|
.as_bool()
|
||||||
.as_bool()
|
.expect("Unable to deserialize 'monitored'"),
|
||||||
.expect("Unable to deserialize 'monitored'"),
|
);
|
||||||
);
|
let minimum_availability = edit_movie_params
|
||||||
let minimum_availability = edit_movie_params
|
.minimum_availability
|
||||||
.minimum_availability
|
.unwrap_or_else(|| {
|
||||||
.unwrap_or_else(|| {
|
serde_json::from_value(detailed_movie_body["minimumAvailability"].clone())
|
||||||
serde_json::from_value(detailed_movie_body["minimumAvailability"].clone())
|
.expect("Unable to deserialize 'minimumAvailability'")
|
||||||
.expect("Unable to deserialize 'minimumAvailability'")
|
})
|
||||||
})
|
.to_string();
|
||||||
.to_string();
|
let quality_profile_id = edit_movie_params.quality_profile_id.unwrap_or_else(|| {
|
||||||
let quality_profile_id = edit_movie_params.quality_profile_id.unwrap_or_else(|| {
|
detailed_movie_body["qualityProfileId"]
|
||||||
detailed_movie_body["qualityProfileId"]
|
.as_i64()
|
||||||
.as_i64()
|
.expect("Unable to deserialize 'qualityProfileId'")
|
||||||
.expect("Unable to deserialize 'qualityProfileId'")
|
});
|
||||||
});
|
let root_folder_path = edit_movie_params.root_folder_path.unwrap_or_else(|| {
|
||||||
let root_folder_path = edit_movie_params.root_folder_path.unwrap_or_else(|| {
|
detailed_movie_body["path"]
|
||||||
detailed_movie_body["path"]
|
.as_str()
|
||||||
.as_str()
|
.expect("Unable to deserialize 'path'")
|
||||||
.expect("Unable to deserialize 'path'")
|
.to_owned()
|
||||||
.to_owned()
|
});
|
||||||
});
|
let tags = if edit_movie_params.clear_tags {
|
||||||
let tags = if edit_movie_params.clear_tags {
|
vec![]
|
||||||
vec![]
|
} else {
|
||||||
} else {
|
edit_movie_params.tags.unwrap_or(
|
||||||
edit_movie_params.tags.unwrap_or(
|
detailed_movie_body["tags"]
|
||||||
detailed_movie_body["tags"]
|
.as_array()
|
||||||
.as_array()
|
.expect("Unable to deserialize 'tags'")
|
||||||
.expect("Unable to deserialize 'tags'")
|
.iter()
|
||||||
.iter()
|
.map(|item| item.as_i64().expect("Unable to deserialize tag ID"))
|
||||||
.map(|item| item.as_i64().expect("Unable to deserialize tag ID"))
|
.collect(),
|
||||||
.collect(),
|
|
||||||
)
|
|
||||||
};
|
|
||||||
|
|
||||||
(
|
|
||||||
monitored,
|
|
||||||
minimum_availability,
|
|
||||||
quality_profile_id,
|
|
||||||
root_folder_path,
|
|
||||||
tags,
|
|
||||||
)
|
)
|
||||||
};
|
};
|
||||||
|
|
||||||
|
(
|
||||||
|
monitored,
|
||||||
|
minimum_availability,
|
||||||
|
quality_profile_id,
|
||||||
|
root_folder_path,
|
||||||
|
tags,
|
||||||
|
)
|
||||||
|
};
|
||||||
|
|
||||||
*detailed_movie_body.get_mut("monitored").unwrap() = json!(monitored);
|
*detailed_movie_body.get_mut("monitored").unwrap() = json!(monitored);
|
||||||
*detailed_movie_body.get_mut("minimumAvailability").unwrap() = json!(minimum_availability);
|
*detailed_movie_body.get_mut("minimumAvailability").unwrap() = json!(minimum_availability);
|
||||||
*detailed_movie_body.get_mut("qualityProfileId").unwrap() = json!(quality_profile_id);
|
*detailed_movie_body.get_mut("qualityProfileId").unwrap() = json!(quality_profile_id);
|
||||||
@@ -1121,10 +1125,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
info!("Fetching Radarr logs");
|
info!("Fetching Radarr logs");
|
||||||
let event = RadarrEvent::GetLogs(events);
|
let event = RadarrEvent::GetLogs(events);
|
||||||
|
|
||||||
let params = format!(
|
let params = format!("pageSize={}&sortDirection=descending&sortKey=time", events);
|
||||||
"pageSize={}&sortDirection=descending&sortKey=time",
|
|
||||||
events
|
|
||||||
);
|
|
||||||
let request_props = self
|
let request_props = self
|
||||||
.request_props_from(event, RequestMethod::Get, None::<()>, None, Some(params))
|
.request_props_from(event, RequestMethod::Get, None::<()>, None, Some(params))
|
||||||
.await;
|
.await;
|
||||||
@@ -1600,62 +1601,34 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn search_movie(&mut self, query: Option<String>) -> Result<Vec<AddMovieSearchResult>> {
|
async fn search_movie(&mut self, query: String) -> Result<Vec<AddMovieSearchResult>> {
|
||||||
info!("Searching for specific Radarr movie");
|
info!("Searching for specific Radarr movie");
|
||||||
let event = RadarrEvent::SearchNewMovie(None);
|
let event = RadarrEvent::SearchNewMovie(query.clone());
|
||||||
let search = if let Some(search_query) = query {
|
|
||||||
Ok(search_query.into())
|
|
||||||
} else {
|
|
||||||
self
|
|
||||||
.app
|
|
||||||
.lock()
|
|
||||||
.await
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.add_movie_search
|
|
||||||
.clone()
|
|
||||||
.ok_or(anyhow!("Encountered a race condition"))
|
|
||||||
};
|
|
||||||
|
|
||||||
match search {
|
let request_props = self
|
||||||
Ok(search_string) => {
|
.request_props_from(
|
||||||
let request_props = self
|
event,
|
||||||
.request_props_from(
|
RequestMethod::Get,
|
||||||
event,
|
None::<()>,
|
||||||
RequestMethod::Get,
|
None,
|
||||||
None::<()>,
|
Some(format!("term={}", encode(&query))),
|
||||||
None,
|
)
|
||||||
Some(format!("term={}", encode(&search_string.text))),
|
.await;
|
||||||
)
|
|
||||||
.await;
|
|
||||||
|
|
||||||
self
|
self
|
||||||
.handle_request::<(), Vec<AddMovieSearchResult>>(request_props, |movie_vec, mut app| {
|
.handle_request::<(), Vec<AddMovieSearchResult>>(request_props, |movie_vec, mut app| {
|
||||||
if movie_vec.is_empty() {
|
if movie_vec.is_empty() {
|
||||||
app.pop_and_push_navigation_stack(
|
app.pop_and_push_navigation_stack(ActiveRadarrBlock::AddMovieEmptySearchResults.into());
|
||||||
ActiveRadarrBlock::AddMovieEmptySearchResults.into(),
|
} else if let Some(add_searched_movies) = app.data.radarr_data.add_searched_movies.as_mut()
|
||||||
);
|
{
|
||||||
} else if let Some(add_searched_movies) =
|
add_searched_movies.set_items(movie_vec);
|
||||||
app.data.radarr_data.add_searched_movies.as_mut()
|
} else {
|
||||||
{
|
let mut add_searched_movies = StatefulTable::default();
|
||||||
add_searched_movies.set_items(movie_vec);
|
add_searched_movies.set_items(movie_vec);
|
||||||
} else {
|
app.data.radarr_data.add_searched_movies = Some(add_searched_movies);
|
||||||
let mut add_searched_movies = StatefulTable::default();
|
}
|
||||||
add_searched_movies.set_items(movie_vec);
|
})
|
||||||
app.data.radarr_data.add_searched_movies = Some(add_searched_movies);
|
.await
|
||||||
}
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
warn!(
|
|
||||||
"Encountered a race condition: {e}\n \
|
|
||||||
This is most likely caused by the user trying to navigate between modals rapidly. \
|
|
||||||
Ignoring search request."
|
|
||||||
);
|
|
||||||
Ok(Vec::default())
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn start_radarr_task(&mut self, task: Option<RadarrTaskName>) -> Result<Value> {
|
async fn start_radarr_task(&mut self, task: Option<RadarrTaskName>) -> Result<Value> {
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ mod test {
|
|||||||
|
|
||||||
use bimap::BiMap;
|
use bimap::BiMap;
|
||||||
use chrono::DateTime;
|
use chrono::DateTime;
|
||||||
use mockito::{Matcher, Server};
|
use mockito::Matcher;
|
||||||
use pretty_assertions::{assert_eq, assert_str_eq};
|
use pretty_assertions::{assert_eq, assert_str_eq};
|
||||||
use reqwest::Client;
|
use reqwest::Client;
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
@@ -13,7 +13,6 @@ mod test {
|
|||||||
use tokio_util::sync::CancellationToken;
|
use tokio_util::sync::CancellationToken;
|
||||||
|
|
||||||
use super::super::*;
|
use super::super::*;
|
||||||
use crate::app::ServarrConfig;
|
|
||||||
use crate::models::radarr_models::{
|
use crate::models::radarr_models::{
|
||||||
AddMovieOptions, BlocklistItem, BlocklistItemMovie, CollectionMovie, EditCollectionParams,
|
AddMovieOptions, BlocklistItem, BlocklistItemMovie, CollectionMovie, EditCollectionParams,
|
||||||
EditMovieParams, IndexerSettings, MediaInfo, MinimumAvailability, MovieCollection, MovieFile, Rating, RatingsList
|
EditMovieParams, IndexerSettings, MediaInfo, MinimumAvailability, MovieCollection, MovieFile, Rating, RatingsList
|
||||||
@@ -220,7 +219,7 @@ mod test {
|
|||||||
#[case(RadarrEvent::DeleteBlocklistItem(1), "/blocklist")]
|
#[case(RadarrEvent::DeleteBlocklistItem(1), "/blocklist")]
|
||||||
#[case(RadarrEvent::GetBlocklist, "/blocklist?page=1&pageSize=10000")]
|
#[case(RadarrEvent::GetBlocklist, "/blocklist?page=1&pageSize=10000")]
|
||||||
#[case(RadarrEvent::GetLogs(500), "/log")]
|
#[case(RadarrEvent::GetLogs(500), "/log")]
|
||||||
#[case(RadarrEvent::SearchNewMovie(None), "/movie/lookup")]
|
#[case(RadarrEvent::SearchNewMovie(String::new()), "/movie/lookup")]
|
||||||
#[case(RadarrEvent::GetMovieCredits(0), "/credit")]
|
#[case(RadarrEvent::GetMovieCredits(0), "/credit")]
|
||||||
#[case(RadarrEvent::GetMovieHistory(0), "/history/movie")]
|
#[case(RadarrEvent::GetMovieHistory(0), "/history/movie")]
|
||||||
#[case(RadarrEvent::GetDiskSpace, "/diskspace")]
|
#[case(RadarrEvent::GetDiskSpace, "/diskspace")]
|
||||||
@@ -616,16 +615,15 @@ mod test {
|
|||||||
None,
|
None,
|
||||||
Some(add_movie_search_result_json),
|
Some(add_movie_search_result_json),
|
||||||
None,
|
None,
|
||||||
RadarrEvent::SearchNewMovie(None),
|
RadarrEvent::SearchNewMovie("test term".into()),
|
||||||
None,
|
None,
|
||||||
Some("term=test%20term"),
|
Some("term=test%20term"),
|
||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
app_arc.lock().await.data.radarr_data.add_movie_search = Some("test term".into());
|
|
||||||
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
if let RadarrSerdeable::AddMovieSearchResults(add_movie_search_results) = network
|
if let RadarrSerdeable::AddMovieSearchResults(add_movie_search_results) = network
|
||||||
.handle_radarr_event(RadarrEvent::SearchNewMovie(None))
|
.handle_radarr_event(RadarrEvent::SearchNewMovie("test term".into()))
|
||||||
.await
|
.await
|
||||||
.unwrap()
|
.unwrap()
|
||||||
{
|
{
|
||||||
@@ -653,51 +651,6 @@ mod test {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn test_handle_search_new_movie_event_uses_provided_query() {
|
|
||||||
let add_movie_search_result_json = json!([{
|
|
||||||
"tmdbId": 1234,
|
|
||||||
"title": "Test",
|
|
||||||
"originalLanguage": { "id": 1, "name": "English" },
|
|
||||||
"status": "released",
|
|
||||||
"overview": "New movie blah blah blah",
|
|
||||||
"genres": ["cool", "family", "fun"],
|
|
||||||
"year": 2023,
|
|
||||||
"runtime": 120,
|
|
||||||
"ratings": {
|
|
||||||
"imdb": {
|
|
||||||
"value": 9.9
|
|
||||||
},
|
|
||||||
"tmdb": {
|
|
||||||
"value": 9.9
|
|
||||||
},
|
|
||||||
"rottenTomatoes": {
|
|
||||||
"value": 9.9
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}]);
|
|
||||||
let (async_server, app_arc, _server) = mock_servarr_api(
|
|
||||||
RequestMethod::Get,
|
|
||||||
None,
|
|
||||||
Some(add_movie_search_result_json),
|
|
||||||
None,
|
|
||||||
RadarrEvent::SearchNewMovie(None),
|
|
||||||
None,
|
|
||||||
Some("term=test%20term"),
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
|
||||||
|
|
||||||
if let RadarrSerdeable::AddMovieSearchResults(add_movie_search_results) = network
|
|
||||||
.handle_radarr_event(RadarrEvent::SearchNewMovie(Some("test term".into())))
|
|
||||||
.await
|
|
||||||
.unwrap()
|
|
||||||
{
|
|
||||||
async_server.assert_async().await;
|
|
||||||
assert_eq!(add_movie_search_results, vec![add_movie_search_result()]);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_handle_start_radarr_task_event() {
|
async fn test_handle_start_radarr_task_event() {
|
||||||
let response = json!({ "test": "test"});
|
let response = json!({ "test": "test"});
|
||||||
@@ -742,16 +695,15 @@ mod test {
|
|||||||
None,
|
None,
|
||||||
Some(json!([])),
|
Some(json!([])),
|
||||||
None,
|
None,
|
||||||
RadarrEvent::SearchNewMovie(None),
|
RadarrEvent::SearchNewMovie("test term".into()),
|
||||||
None,
|
None,
|
||||||
Some("term=test%20term"),
|
Some("term=test%20term"),
|
||||||
)
|
)
|
||||||
.await;
|
.await;
|
||||||
app_arc.lock().await.data.radarr_data.add_movie_search = Some("test term".into());
|
|
||||||
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
assert!(network
|
assert!(network
|
||||||
.handle_radarr_event(RadarrEvent::SearchNewMovie(None))
|
.handle_radarr_event(RadarrEvent::SearchNewMovie("test term".into()))
|
||||||
.await
|
.await
|
||||||
.is_ok());
|
.is_ok());
|
||||||
|
|
||||||
@@ -769,57 +721,6 @@ mod test {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn test_handle_search_new_movie_event_no_panic_on_race_condition() {
|
|
||||||
let resource = format!(
|
|
||||||
"{}?term=test%20term",
|
|
||||||
RadarrEvent::SearchNewMovie(None).resource()
|
|
||||||
);
|
|
||||||
let mut server = Server::new_async().await;
|
|
||||||
let mut async_server = server
|
|
||||||
.mock(
|
|
||||||
&RequestMethod::Get.to_string().to_uppercase(),
|
|
||||||
format!("/api/v3{resource}").as_str(),
|
|
||||||
)
|
|
||||||
.match_header("X-Api-Key", "test1234");
|
|
||||||
async_server = async_server.expect_at_most(0).create_async().await;
|
|
||||||
|
|
||||||
let host = Some(server.host_with_port().split(':').collect::<Vec<&str>>()[0].to_owned());
|
|
||||||
let port = Some(
|
|
||||||
server.host_with_port().split(':').collect::<Vec<&str>>()[1]
|
|
||||||
.parse()
|
|
||||||
.unwrap(),
|
|
||||||
);
|
|
||||||
let mut app = App::default();
|
|
||||||
let radarr_config = ServarrConfig {
|
|
||||||
host,
|
|
||||||
port,
|
|
||||||
api_token: "test1234".to_owned(),
|
|
||||||
..ServarrConfig::default()
|
|
||||||
};
|
|
||||||
app.config.radarr = Some(radarr_config);
|
|
||||||
let app_arc = Arc::new(Mutex::new(app));
|
|
||||||
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
|
||||||
|
|
||||||
assert!(network
|
|
||||||
.handle_radarr_event(RadarrEvent::SearchNewMovie(None))
|
|
||||||
.await
|
|
||||||
.is_ok());
|
|
||||||
|
|
||||||
async_server.assert_async().await;
|
|
||||||
assert!(app_arc
|
|
||||||
.lock()
|
|
||||||
.await
|
|
||||||
.data
|
|
||||||
.radarr_data
|
|
||||||
.add_searched_movies
|
|
||||||
.is_none());
|
|
||||||
assert_eq!(
|
|
||||||
app_arc.lock().await.get_current_route(),
|
|
||||||
ActiveRadarrBlock::Movies.into()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_handle_start_radarr_task_event_uses_provided_task_name() {
|
async fn test_handle_start_radarr_task_event_uses_provided_task_name() {
|
||||||
let response = json!({ "test": "test"});
|
let response = json!({ "test": "test"});
|
||||||
|
|||||||
Reference in New Issue
Block a user