fix(sonarr): Pass the search query directly to the networking channel when searching for a new series

This commit is contained in:
2024-12-18 00:49:36 -07:00
parent ec8d748991
commit 1193b8c848
6 changed files with 70 additions and 160 deletions
+14 -1
View File
@@ -165,7 +165,9 @@ impl<'a> App<'a> {
} }
ActiveSonarrBlock::AddSeriesSearchResults => { ActiveSonarrBlock::AddSeriesSearchResults => {
self self
.dispatch_network_event(SonarrEvent::SearchNewSeries(None).into()) .dispatch_network_event(
SonarrEvent::SearchNewSeries(self.extract_add_new_series_search_query().await).into(),
)
.await; .await;
} }
ActiveSonarrBlock::SystemUpdates => { ActiveSonarrBlock::SystemUpdates => {
@@ -287,4 +289,15 @@ impl<'a> App<'a> {
.season_number; .season_number;
(series_id, season_number) (series_id, season_number)
} }
async fn extract_add_new_series_search_query(&self) -> String {
self
.data
.sonarr_data
.add_series_search
.as_ref()
.expect("Add series search is empty")
.text
.clone()
}
} }
+21 -1
View File
@@ -532,6 +532,7 @@ mod tests {
#[tokio::test] #[tokio::test]
async fn test_dispatch_by_add_series_search_results_block() { async fn test_dispatch_by_add_series_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.sonarr_data.add_series_search = Some("test search".into());
app app
.dispatch_by_sonarr_block(&ActiveSonarrBlock::AddSeriesSearchResults) .dispatch_by_sonarr_block(&ActiveSonarrBlock::AddSeriesSearchResults)
@@ -540,7 +541,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(),
SonarrEvent::SearchNewSeries(None).into() SonarrEvent::SearchNewSeries("test search".into()).into()
); );
assert!(!app.data.sonarr_data.prompt_confirm); assert!(!app.data.sonarr_data.prompt_confirm);
assert_eq!(app.tick_count, 0); assert_eq!(app.tick_count, 0);
@@ -829,6 +830,25 @@ mod tests {
assert_eq!(app.extract_series_id_season_number_tuple().await, (1, 1)); assert_eq!(app.extract_series_id_season_number_tuple().await, (1, 1));
} }
#[tokio::test]
async fn test_extract_add_new_series_search_query() {
let mut app = App::default();
app.data.sonarr_data.add_series_search = Some("test search".into());
assert_str_eq!(
app.extract_add_new_series_search_query().await,
"test search"
);
}
#[tokio::test]
#[should_panic(expected = "Add series search is empty")]
async fn test_extract_add_new_series_search_query_panics_when_the_query_is_not_set() {
let app = App::default();
app.extract_add_new_series_search_query().await;
}
fn construct_app_unit<'a>() -> (App<'a>, mpsc::Receiver<NetworkEvent>) { fn construct_app_unit<'a>() -> (App<'a>, mpsc::Receiver<NetworkEvent>) {
let (sync_network_tx, sync_network_rx) = mpsc::channel::<NetworkEvent>(500); let (sync_network_tx, sync_network_rx) = mpsc::channel::<NetworkEvent>(500);
let mut app = App { let mut app = App {
+1 -1
View File
@@ -245,7 +245,7 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrCommand> for SonarrCliHandler<'a, '
SonarrCommand::SearchNewSeries { query } => { SonarrCommand::SearchNewSeries { query } => {
let resp = self let resp = self
.network .network
.handle_network_event(SonarrEvent::SearchNewSeries(Some(query)).into()) .handle_network_event(SonarrEvent::SearchNewSeries(query).into())
.await?; .await?;
serde_json::to_string_pretty(&resp)? serde_json::to_string_pretty(&resp)?
} }
+1 -1
View File
@@ -597,7 +597,7 @@ mod tests {
mock_network mock_network
.expect_handle_network_event() .expect_handle_network_event()
.with(eq::<NetworkEvent>( .with(eq::<NetworkEvent>(
SonarrEvent::SearchNewSeries(Some(expected_search_query)).into(), SonarrEvent::SearchNewSeries(expected_search_query).into(),
)) ))
.times(1) .times(1)
.returning(|_| { .returning(|_| {
+6 -37
View File
@@ -82,7 +82,7 @@ pub enum SonarrEvent {
HealthCheck, HealthCheck,
ListSeries, ListSeries,
MarkHistoryItemAsFailed(i64), MarkHistoryItemAsFailed(i64),
SearchNewSeries(Option<String>), SearchNewSeries(String),
StartTask(Option<SonarrTaskName>), StartTask(Option<SonarrTaskName>),
TestIndexer(Option<i64>), TestIndexer(Option<i64>),
TestAllIndexers, TestAllIndexers,
@@ -1995,46 +1995,25 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
} }
async fn search_sonarr_series( async fn search_sonarr_series(&mut self, query: String) -> Result<Vec<AddSeriesSearchResult>> {
&mut self,
query: Option<String>,
) -> Result<Vec<AddSeriesSearchResult>> {
info!("Searching for specific Sonarr series"); info!("Searching for specific Sonarr series");
let event = SonarrEvent::SearchNewSeries(None); let event = SonarrEvent::SearchNewSeries(query.clone());
let search = if let Some(search_query) = query {
Ok(search_query.into())
} else {
self
.app
.lock()
.await
.data
.sonarr_data
.add_series_search
.clone()
.ok_or(anyhow!("Encountered a race condition"))
};
match search {
Ok(search_string) => {
let request_props = self let request_props = self
.request_props_from( .request_props_from(
event, event,
RequestMethod::Get, RequestMethod::Get,
None::<()>, None::<()>,
None, None,
Some(format!("term={}", encode(&search_string.text))), Some(format!("term={}", encode(&query))),
) )
.await; .await;
self self
.handle_request::<(), Vec<AddSeriesSearchResult>>(request_props, |series_vec, mut app| { .handle_request::<(), Vec<AddSeriesSearchResult>>(request_props, |series_vec, mut app| {
if series_vec.is_empty() { if series_vec.is_empty() {
app.pop_and_push_navigation_stack( app.pop_and_push_navigation_stack(ActiveSonarrBlock::AddSeriesEmptySearchResults.into());
ActiveSonarrBlock::AddSeriesEmptySearchResults.into(), } else if let Some(add_searched_seriess) = app.data.sonarr_data.add_searched_series.as_mut()
);
} else if let Some(add_searched_seriess) =
app.data.sonarr_data.add_searched_series.as_mut()
{ {
add_searched_seriess.set_items(series_vec); add_searched_seriess.set_items(series_vec);
} else { } else {
@@ -2045,16 +2024,6 @@ impl<'a, 'b> Network<'a, 'b> {
}) })
.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_sonarr_task(&mut self, task: Option<SonarrTaskName>) -> Result<Value> { async fn start_sonarr_task(&mut self, task: Option<SonarrTaskName>) -> Result<Value> {
let event = SonarrEvent::StartTask(None); let event = SonarrEvent::StartTask(None);
+7 -99
View File
@@ -5,7 +5,7 @@ mod test {
use bimap::BiMap; use bimap::BiMap;
use chrono::DateTime; use chrono::DateTime;
use indoc::formatdoc; use indoc::formatdoc;
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;
@@ -19,7 +19,7 @@ mod test {
DownloadStatus, EditSeriesParams, IndexerSettings, MonitorEpisodeBody, SonarrHistoryEventType, DownloadStatus, EditSeriesParams, IndexerSettings, MonitorEpisodeBody, SonarrHistoryEventType,
}; };
use crate::app::{App, ServarrConfig}; use crate::app::App;
use crate::models::radarr_models::IndexerTestResult; use crate::models::radarr_models::IndexerTestResult;
use crate::models::servarr_data::modals::IndexerTestResultModalItem; use crate::models::servarr_data::modals::IndexerTestResultModalItem;
use crate::models::servarr_data::sonarr::modals::{ use crate::models::servarr_data::sonarr::modals::{
@@ -286,7 +286,7 @@ mod test {
#[case(SonarrEvent::GetTasks, "/system/task")] #[case(SonarrEvent::GetTasks, "/system/task")]
#[case(SonarrEvent::GetUpdates, "/update")] #[case(SonarrEvent::GetUpdates, "/update")]
#[case(SonarrEvent::MarkHistoryItemAsFailed(0), "/history/failed")] #[case(SonarrEvent::MarkHistoryItemAsFailed(0), "/history/failed")]
#[case(SonarrEvent::SearchNewSeries(None), "/series/lookup")] #[case(SonarrEvent::SearchNewSeries(String::new()), "/series/lookup")]
#[case(SonarrEvent::TestIndexer(None), "/indexer/test")] #[case(SonarrEvent::TestIndexer(None), "/indexer/test")]
#[case(SonarrEvent::TestAllIndexers, "/indexer/testall")] #[case(SonarrEvent::TestAllIndexers, "/indexer/testall")]
#[case(SonarrEvent::ToggleEpisodeMonitoring(None), "/episode/monitor")] #[case(SonarrEvent::ToggleEpisodeMonitoring(None), "/episode/monitor")]
@@ -4682,7 +4682,7 @@ mod test {
None, None,
Some(add_series_search_result_json), Some(add_series_search_result_json),
None, None,
SonarrEvent::SearchNewSeries(None), SonarrEvent::SearchNewSeries("test term".into()),
None, None,
Some("term=test%20term"), Some("term=test%20term"),
) )
@@ -4691,7 +4691,7 @@ mod test {
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new()); let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::AddSeriesSearchResults(add_series_search_results) = network if let SonarrSerdeable::AddSeriesSearchResults(add_series_search_results) = network
.handle_sonarr_event(SonarrEvent::SearchNewSeries(None)) .handle_sonarr_event(SonarrEvent::SearchNewSeries("test term".into()))
.await .await
.unwrap() .unwrap()
{ {
@@ -4719,43 +4719,6 @@ mod test {
} }
} }
#[tokio::test]
async fn test_handle_search_new_series_event_uses_provided_query() {
let add_series_search_result_json = json!([{
"tvdbId": 1234,
"title": "Test",
"status": "continuing",
"ended": false,
"overview": "New series blah blah blah",
"genres": ["cool", "family", "fun"],
"year": 2023,
"network": "Prime Video",
"runtime": 60,
"ratings": { "votes": 406744, "value": 8.4 },
"statistics": { "seasonCount": 3 }
}]);
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(add_series_search_result_json),
None,
SonarrEvent::SearchNewSeries(None),
None,
Some("term=test%20term"),
)
.await;
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::AddSeriesSearchResults(add_series_search_results) = network
.handle_sonarr_event(SonarrEvent::SearchNewSeries(Some("test term".into())))
.await
.unwrap()
{
async_server.assert_async().await;
assert_eq!(add_series_search_results, vec![add_series_search_result()]);
}
}
#[tokio::test] #[tokio::test]
async fn test_handle_search_new_series_event_no_results() { async fn test_handle_search_new_series_event_no_results() {
let (async_server, app_arc, _server) = mock_servarr_api( let (async_server, app_arc, _server) = mock_servarr_api(
@@ -4763,7 +4726,7 @@ mod test {
None, None,
Some(json!([])), Some(json!([])),
None, None,
SonarrEvent::SearchNewSeries(None), SonarrEvent::SearchNewSeries("test term".into()),
None, None,
Some("term=test%20term"), Some("term=test%20term"),
) )
@@ -4772,7 +4735,7 @@ mod test {
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_sonarr_event(SonarrEvent::SearchNewSeries(None)) .handle_sonarr_event(SonarrEvent::SearchNewSeries("test term".into()))
.await .await
.is_ok()); .is_ok());
@@ -4790,61 +4753,6 @@ mod test {
); );
} }
#[tokio::test]
async fn test_handle_search_new_series_event_no_panic_on_race_condition() {
let resource = format!(
"{}?term=test%20term",
SonarrEvent::SearchNewSeries(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 sonarr_config = ServarrConfig {
host,
port,
api_token: "test1234".to_owned(),
..ServarrConfig::default()
};
app.config.sonarr = Some(sonarr_config);
let app_arc = Arc::new(Mutex::new(app));
app_arc
.lock()
.await
.push_navigation_stack(ActiveSonarrBlock::Series.into());
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_sonarr_event(SonarrEvent::SearchNewSeries(None))
.await
.is_ok());
async_server.assert_async().await;
assert!(app_arc
.lock()
.await
.data
.sonarr_data
.add_searched_series
.is_none());
assert_eq!(
app_arc.lock().await.get_current_route(),
ActiveSonarrBlock::Series.into()
);
}
#[tokio::test] #[tokio::test]
async fn test_handle_start_sonarr_task_event() { async fn test_handle_start_sonarr_task_event() {
let response = json!({ "test": "test"}); let response = json!({ "test": "test"});