Dark-Alex-17/managarr
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(network): Support for searching for new series

Browse Source
This commit is contained in:
Alex Clarke
2024-11-24 14:54:41 -07:00
parent 53a59cdb4c
commit da3bb795b7
5 changed files with 304 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/network/sonarr_network_tests.rs
+186 -2
View File
@@ -5,7 +5,7 @@ mod test {
use bimap::BiMap;
use chrono::{DateTime, Utc};
use indoc::formatdoc;
use mockito::Matcher;
use mockito::{Matcher, Server};
use pretty_assertions::{assert_eq, assert_str_eq};
use reqwest::Client;
use rstest::rstest;
@@ -20,7 +20,7 @@ mod test {
SeriesMonitor,
};
use crate::app::App;
use crate::app::{App, ServarrConfig};
use crate::models::radarr_models::IndexerTestResult;
use crate::models::servarr_data::modals::IndexerTestResultModalItem;
use crate::models::servarr_data::sonarr::modals::{
@@ -251,6 +251,7 @@ mod test {
#[case(SonarrEvent::GetTasks, "/system/task")]
#[case(SonarrEvent::GetUpdates, "/update")]
#[case(SonarrEvent::MarkHistoryItemAsFailed(0), "/history/failed")]
#[case(SonarrEvent::SearchNewSeries(None), "/series/lookup")]
#[case(SonarrEvent::TestIndexer(None), "/indexer/test")]
#[case(SonarrEvent::TestAllIndexers, "/indexer/testall")]
fn test_resource(#[case] event: SonarrEvent, #[case] expected_uri: String) {
@@ -4392,6 +4393,189 @@ mod test {
async_server.assert_async().await;
}
#[tokio::test]
async fn test_handle_search_new_series_event() {
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;
app_arc.lock().await.data.sonarr_data.add_series_search = Some("test term".into());
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(None))
.await
.unwrap()
{
async_server.assert_async().await;
assert!(app_arc
.lock()
.await
.data
.sonarr_data
.add_searched_series
.is_some());
assert_eq!(
app_arc
.lock()
.await
.data
.sonarr_data
.add_searched_series
.as_ref()
.unwrap()
.items,
vec![add_series_search_result()]
);
assert_eq!(add_series_search_results, vec![add_series_search_result()]);
}
}
#[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]
async fn test_handle_search_new_series_event_no_results() {
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(json!([])),
None,
SonarrEvent::SearchNewSeries(None),
None,
Some("term=test%20term"),
)
.await;
app_arc.lock().await.data.sonarr_data.add_series_search = Some("test term".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::AddSeriesEmptySearchResults.into()
);
}
#[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]
async fn test_handle_start_sonarr_task_event() {
let response = json!({ "test": "test"});