feat(network): Added support for fetching series details for a given series ID in Sonarr

This commit is contained in:
2024-11-19 16:56:48 -07:00
parent 540db5993b
commit 5e63c34a9f
4 changed files with 118 additions and 17 deletions
+2
View File
@@ -320,6 +320,7 @@ pub enum SonarrSerdeable {
Releases(Vec<Release>), Releases(Vec<Release>),
SecurityConfig(SecurityConfig), SecurityConfig(SecurityConfig),
SeriesVec(Vec<Series>), SeriesVec(Vec<Series>),
Series(Series),
SystemStatus(SystemStatus), SystemStatus(SystemStatus),
BlocklistResponse(BlocklistResponse), BlocklistResponse(BlocklistResponse),
LogResponse(LogResponse), LogResponse(LogResponse),
@@ -351,6 +352,7 @@ serde_enum_from!(
Releases(Vec<Release>), Releases(Vec<Release>),
SecurityConfig(SecurityConfig), SecurityConfig(SecurityConfig),
SeriesVec(Vec<Series>), SeriesVec(Vec<Series>),
Series(Series),
SystemStatus(SystemStatus), SystemStatus(SystemStatus),
BlocklistResponse(BlocklistResponse), BlocklistResponse(BlocklistResponse),
LogResponse(LogResponse), LogResponse(LogResponse),
+15 -3
View File
@@ -144,15 +144,27 @@ mod tests {
} }
#[test] #[test]
fn test_sonarr_serdeable_from_series() { fn test_sonarr_serdeable_from_series_vec() {
let series = vec![Series { let series_vec = vec![Series {
id: 1, id: 1,
..Series::default() ..Series::default()
}]; }];
let sonarr_serdeable: SonarrSerdeable = series_vec.clone().into();
assert_eq!(sonarr_serdeable, SonarrSerdeable::SeriesVec(series_vec));
}
#[test]
fn test_sonarr_serdeable_from_series() {
let series = Series {
id: 1,
..Series::default()
};
let sonarr_serdeable: SonarrSerdeable = series.clone().into(); let sonarr_serdeable: SonarrSerdeable = series.clone().into();
assert_eq!(sonarr_serdeable, SonarrSerdeable::SeriesVec(series)); assert_eq!(sonarr_serdeable, SonarrSerdeable::Series(series));
} }
#[test] #[test]
+26 -1
View File
@@ -43,6 +43,7 @@ pub enum SonarrEvent {
GetQueuedEvents, GetQueuedEvents,
GetSeasonReleases(Option<(i64, i64)>), GetSeasonReleases(Option<(i64, i64)>),
GetSecurityConfig, GetSecurityConfig,
GetSeriesDetails(Option<i64>),
GetStatus, GetStatus,
HealthCheck, HealthCheck,
ListSeries, ListSeries,
@@ -65,7 +66,7 @@ impl NetworkResource for SonarrEvent {
SonarrEvent::GetSeasonReleases(_) => "/release", SonarrEvent::GetSeasonReleases(_) => "/release",
SonarrEvent::GetStatus => "/system/status", SonarrEvent::GetStatus => "/system/status",
SonarrEvent::HealthCheck => "/health", SonarrEvent::HealthCheck => "/health",
SonarrEvent::ListSeries => "/series", SonarrEvent::ListSeries | SonarrEvent::GetSeriesDetails(_) => "/series",
} }
} }
} }
@@ -129,6 +130,10 @@ impl<'a, 'b> Network<'a, 'b> {
.get_sonarr_security_config() .get_sonarr_security_config()
.await .await
.map(SonarrSerdeable::from), .map(SonarrSerdeable::from),
SonarrEvent::GetSeriesDetails(series_id) => self
.get_series_details(series_id)
.await
.map(SonarrSerdeable::from),
SonarrEvent::GetStatus => self.get_sonarr_status().await.map(SonarrSerdeable::from), SonarrEvent::GetStatus => self.get_sonarr_status().await.map(SonarrSerdeable::from),
SonarrEvent::HealthCheck => self SonarrEvent::HealthCheck => self
.get_sonarr_healthcheck() .get_sonarr_healthcheck()
@@ -608,6 +613,26 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
} }
async fn get_series_details(&mut self, series_id: Option<i64>) -> Result<Series> {
let (id, _) = self.extract_series_id(series_id).await;
info!("Fetching details for Sonarr series with ID: {id}");
let event = SonarrEvent::GetSeriesDetails(series_id);
let request_props = self
.request_props_from(
event,
RequestMethod::Get,
None::<()>,
Some(format!("/{id}")),
None,
)
.await;
self
.handle_request::<(), Series>(request_props, |_, _| ())
.await
}
async fn list_series(&mut self) -> Result<Vec<Series>> { async fn list_series(&mut self) -> Result<Vec<Series>> {
info!("Fetching Sonarr library"); info!("Fetching Sonarr library");
let event = SonarrEvent::ListSeries; let event = SonarrEvent::ListSeries;
+75 -13
View File
@@ -137,7 +137,9 @@ mod test {
} }
#[rstest] #[rstest]
fn test_resource_series(#[values(SonarrEvent::ListSeries)] event: SonarrEvent) { fn test_resource_series(
#[values(SonarrEvent::ListSeries, SonarrEvent::GetSeriesDetails(None))] event: SonarrEvent,
) {
assert_str_eq!(event.resource(), "/series"); assert_str_eq!(event.resource(), "/series");
} }
@@ -1010,7 +1012,8 @@ mod test {
#[tokio::test] #[tokio::test]
#[should_panic(expected = "Season details modal is empty")] #[should_panic(expected = "Season details modal is empty")]
async fn test_handle_get_episode_details_event_requires_season_details_modal_to_be_some() { async fn test_handle_get_episode_details_event_requires_season_details_modal_to_be_some_when_in_tui_mode(
) {
let (_async_server, app_arc, _server) = mock_servarr_api( let (_async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get, RequestMethod::Get,
None, None,
@@ -1518,7 +1521,7 @@ mod test {
#[rstest] #[rstest]
#[tokio::test] #[tokio::test]
async fn test_handle_get_series_event(#[values(true, false)] use_custom_sorting: bool) { async fn test_handle_list_series_event(#[values(true, false)] use_custom_sorting: bool) {
let mut series_1: Value = serde_json::from_str(SERIES_JSON).unwrap(); let mut series_1: Value = serde_json::from_str(SERIES_JSON).unwrap();
let mut series_2: Value = serde_json::from_str(SERIES_JSON).unwrap(); let mut series_2: Value = serde_json::from_str(SERIES_JSON).unwrap();
*series_1.get_mut("id").unwrap() = json!(1); *series_1.get_mut("id").unwrap() = json!(1);
@@ -1597,6 +1600,75 @@ mod test {
} }
} }
#[tokio::test]
async fn test_handle_get_series_details_event() {
let expected_series: Series = serde_json::from_str(SERIES_JSON).unwrap();
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(serde_json::from_str(SERIES_JSON).unwrap()),
None,
SonarrEvent::GetSeriesDetails(None),
Some("/1"),
None,
)
.await;
app_arc
.lock()
.await
.data
.sonarr_data
.series
.set_items(vec![series()]);
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Series(series) = network
.handle_sonarr_event(SonarrEvent::GetSeriesDetails(None))
.await
.unwrap()
{
async_server.assert_async().await;
assert_eq!(series, expected_series);
}
}
#[tokio::test]
async fn test_handle_get_series_details_event_uses_provided_series_id() {
let expected_series: Series = Series {
id: 2,
..serde_json::from_str(SERIES_JSON).unwrap()
};
let mut response: Value = serde_json::from_str(SERIES_JSON).unwrap();
*response.get_mut("id").unwrap() = json!(2);
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(response),
None,
SonarrEvent::GetSeriesDetails(Some(2)),
Some("/2"),
None,
)
.await;
app_arc
.lock()
.await
.data
.sonarr_data
.series
.set_items(vec![series()]);
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Series(series) = network
.handle_sonarr_event(SonarrEvent::GetSeriesDetails(Some(2)))
.await
.unwrap()
{
async_server.assert_async().await;
assert_eq!(series, expected_series);
}
}
#[tokio::test] #[tokio::test]
async fn test_handle_get_series_event_no_op_while_user_is_selecting_sort_options() { async fn test_handle_get_series_event_no_op_while_user_is_selecting_sort_options() {
let mut series_1: Value = serde_json::from_str(SERIES_JSON).unwrap(); let mut series_1: Value = serde_json::from_str(SERIES_JSON).unwrap();
@@ -2183,14 +2255,4 @@ mod test {
quality: quality_wrapper(), quality: quality_wrapper(),
} }
} }
fn render<T>(state: &mut TreeState, items: &[TreeItem<T>])
where
T: ToText + Clone + Default + Display + Hash + PartialEq + Eq,
{
let tree = Tree::new(items).unwrap();
let area = Rect::new(0, 0, 10, 4);
let mut buffer = Buffer::empty(area);
StatefulWidget::render(tree, area, &mut buffer, state);
}
} }