feat(network): Support for fetching and listing all Sonarr tags

This commit is contained in:
2024-11-22 15:05:56 -07:00
parent 208acafc73
commit c5328917de
4 changed files with 63 additions and 5 deletions
+2 -2
View File
@@ -242,7 +242,7 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
.map(RadarrSerdeable::from), .map(RadarrSerdeable::from),
RadarrEvent::GetStatus => self.get_radarr_status().await.map(RadarrSerdeable::from), RadarrEvent::GetStatus => self.get_radarr_status().await.map(RadarrSerdeable::from),
RadarrEvent::GetTags => self.get_tags().await.map(RadarrSerdeable::from), RadarrEvent::GetTags => self.get_radarr_tags().await.map(RadarrSerdeable::from),
RadarrEvent::GetTasks => self.get_tasks().await.map(RadarrSerdeable::from), RadarrEvent::GetTasks => self.get_tasks().await.map(RadarrSerdeable::from),
RadarrEvent::GetUpdates => self.get_updates().await.map(RadarrSerdeable::from), RadarrEvent::GetUpdates => self.get_updates().await.map(RadarrSerdeable::from),
RadarrEvent::HealthCheck => self RadarrEvent::HealthCheck => self
@@ -1837,7 +1837,7 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
} }
async fn get_tags(&mut self) -> Result<Vec<Tag>> { async fn get_radarr_tags(&mut self) -> Result<Vec<Tag>> {
info!("Fetching Radarr tags"); info!("Fetching Radarr tags");
let event = RadarrEvent::GetTags; let event = RadarrEvent::GetTags;
+1 -1
View File
@@ -2633,7 +2633,7 @@ mod test {
} }
#[tokio::test] #[tokio::test]
async fn test_handle_get_tags_event() { async fn test_handle_get_radarr_tags_event() {
let tags_json = json!([{ let tags_json = json!([{
"id": 2222, "id": 2222,
"label": "usenet" "label": "usenet"
+21 -1
View File
@@ -58,6 +58,7 @@ pub enum SonarrEvent {
GetSeriesDetails(Option<i64>), GetSeriesDetails(Option<i64>),
GetSeriesHistory(Option<i64>), GetSeriesHistory(Option<i64>),
GetStatus, GetStatus,
GetTags,
HealthCheck, HealthCheck,
ListSeries, ListSeries,
} }
@@ -65,7 +66,7 @@ pub enum SonarrEvent {
impl NetworkResource for SonarrEvent { impl NetworkResource for SonarrEvent {
fn resource(&self) -> &'static str { fn resource(&self) -> &'static str {
match &self { match &self {
SonarrEvent::AddTag(_) | SonarrEvent::DeleteTag(_) => "/tag", SonarrEvent::AddTag(_) | SonarrEvent::DeleteTag(_) | SonarrEvent::GetTags => "/tag",
SonarrEvent::ClearBlocklist => "/blocklist/bulk", SonarrEvent::ClearBlocklist => "/blocklist/bulk",
SonarrEvent::DeleteBlocklistItem(_) => "/blocklist", SonarrEvent::DeleteBlocklistItem(_) => "/blocklist",
SonarrEvent::GetAllIndexerSettings => "/config/indexer", SonarrEvent::GetAllIndexerSettings => "/config/indexer",
@@ -195,6 +196,7 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
.map(SonarrSerdeable::from), .map(SonarrSerdeable::from),
SonarrEvent::GetStatus => self.get_sonarr_status().await.map(SonarrSerdeable::from), SonarrEvent::GetStatus => self.get_sonarr_status().await.map(SonarrSerdeable::from),
SonarrEvent::GetTags => self.get_sonarr_tags().await.map(SonarrSerdeable::from),
SonarrEvent::HealthCheck => self SonarrEvent::HealthCheck => self
.get_sonarr_healthcheck() .get_sonarr_healthcheck()
.await .await
@@ -1112,6 +1114,24 @@ impl<'a, 'b> Network<'a, 'b> {
.await .await
} }
async fn get_sonarr_tags(&mut self) -> Result<Vec<Tag>> {
info!("Fetching Sonarr tags");
let event = SonarrEvent::GetTags;
let request_props = self
.request_props_from(event, RequestMethod::Get, None::<()>, None, None)
.await;
self
.handle_request::<(), Vec<Tag>>(request_props, |tags_vec, mut app| {
app.data.sonarr_data.tags_map = tags_vec
.into_iter()
.map(|tag| (tag.id, tag.label))
.collect();
})
.await
}
async fn extract_series_id(&mut self, series_id: Option<i64>) -> (i64, String) { async fn extract_series_id(&mut self, series_id: Option<i64>) -> (i64, String) {
let series_id = if let Some(id) = series_id { let series_id = if let Some(id) = series_id {
id id
+39 -1
View File
@@ -140,7 +140,12 @@ mod test {
#[rstest] #[rstest]
fn test_resource_tag( fn test_resource_tag(
#[values(SonarrEvent::AddTag(String::new()), SonarrEvent::DeleteTag(0))] event: SonarrEvent, #[values(
SonarrEvent::AddTag(String::new()),
SonarrEvent::DeleteTag(0),
SonarrEvent::GetTags
)]
event: SonarrEvent,
) { ) {
assert_str_eq!(event.resource(), "/tag"); assert_str_eq!(event.resource(), "/tag");
} }
@@ -3588,6 +3593,39 @@ mod test {
} }
} }
#[tokio::test]
async fn test_handle_get_sonarr_tags_event() {
let tags_json = json!([{
"id": 2222,
"label": "usenet"
}]);
let response: Vec<Tag> = serde_json::from_value(tags_json.clone()).unwrap();
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Get,
None,
Some(tags_json),
None,
SonarrEvent::GetTags,
None,
None,
)
.await;
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
if let SonarrSerdeable::Tags(tags) = network
.handle_sonarr_event(SonarrEvent::GetTags)
.await
.unwrap()
{
async_server.assert_async().await;
assert_eq!(
app_arc.lock().await.data.sonarr_data.tags_map,
BiMap::from_iter([(2222i64, "usenet".to_owned())])
);
assert_eq!(tags, response);
}
}
#[tokio::test] #[tokio::test]
async fn test_extract_series_id() { async fn test_extract_series_id() {
let app_arc = Arc::new(Mutex::new(App::default())); let app_arc = Arc::new(Mutex::new(App::default()));