feat: Completed support for viewing Lidarr artist details
This commit is contained in:
@@ -0,0 +1,106 @@
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use mockito::Matcher;
|
||||
use pretty_assertions::assert_eq;
|
||||
use serde_json::{json, Value};
|
||||
use crate::models::lidarr_models::{Album, LidarrSerdeable};
|
||||
use crate::network::lidarr_network::lidarr_network_test_utils::test_utils::{ALBUM_JSON};
|
||||
use crate::network::lidarr_network::LidarrEvent;
|
||||
use crate::network::network_tests::test_utils::{test_network, MockServarrApi};
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_handle_get_albums_event() {
|
||||
let albums_json = json!([{
|
||||
"id": 1,
|
||||
"title": "Test Album",
|
||||
"foreignAlbumId": "test-foreign-album-id",
|
||||
"monitored": true,
|
||||
"anyReleaseOk": true,
|
||||
"profileId": 1,
|
||||
"duration": 180,
|
||||
"albumType": "Album",
|
||||
"genres": ["Classical"],
|
||||
"ratings": {"votes": 15, "value": 8.4},
|
||||
"releaseDate": "2023-01-01T00:00:00Z",
|
||||
"statistics": {
|
||||
"trackFileCount": 10,
|
||||
"trackCount": 10,
|
||||
"totalTrackCount": 10,
|
||||
"sizeOnDisk": 1024,
|
||||
"percentOfTracks": 99.9
|
||||
}
|
||||
}]);
|
||||
let response: Vec<Album> = serde_json::from_value(albums_json.clone()).unwrap();
|
||||
let (mock, app, _server) = MockServarrApi::get()
|
||||
.returns(albums_json)
|
||||
.query("artistId=1")
|
||||
.build_for(LidarrEvent::GetAlbums(1))
|
||||
.await;
|
||||
app.lock().await.server_tabs.set_index(2);
|
||||
let mut network = test_network(&app);
|
||||
|
||||
let result = network.handle_lidarr_event(LidarrEvent::GetAlbums(1)).await;
|
||||
|
||||
mock.assert_async().await;
|
||||
|
||||
let LidarrSerdeable::Albums(albums) = result.unwrap() else {
|
||||
panic!("Expected Albums");
|
||||
};
|
||||
|
||||
assert_eq!(albums, response);
|
||||
assert!(!app.lock().await.data.lidarr_data.albums.is_empty());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_handle_toggle_album_monitoring_event() {
|
||||
let mut expected_body: Value = serde_json::from_str(ALBUM_JSON).unwrap();
|
||||
*expected_body.get_mut("monitored").unwrap() = json!(false);
|
||||
let (get_mock, app, mut server) = MockServarrApi::get()
|
||||
.returns(serde_json::from_str(ALBUM_JSON).unwrap())
|
||||
.path("/1")
|
||||
.build_for(LidarrEvent::GetAlbums(1))
|
||||
.await;
|
||||
let put_mock = server
|
||||
.mock("PUT", "/api/v1/album/1")
|
||||
.match_body(Matcher::Json(expected_body))
|
||||
.match_header("X-Api-Key", "test1234")
|
||||
.with_status(202)
|
||||
.create_async()
|
||||
.await;
|
||||
app.lock().await.server_tabs.set_index(2);
|
||||
let mut network = test_network(&app);
|
||||
|
||||
assert_ok!(
|
||||
network
|
||||
.handle_lidarr_event(LidarrEvent::ToggleAlbumMonitoring(1))
|
||||
.await
|
||||
);
|
||||
|
||||
get_mock.assert_async().await;
|
||||
put_mock.assert_async().await;
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_handle_get_album_details_event() {
|
||||
let expected_album: Album = serde_json::from_str(ALBUM_JSON).unwrap();
|
||||
let (mock, app, _server) = MockServarrApi::get()
|
||||
.returns(serde_json::from_str(ALBUM_JSON).unwrap())
|
||||
.path("/1")
|
||||
.build_for(LidarrEvent::GetAlbumDetails(1))
|
||||
.await;
|
||||
app.lock().await.server_tabs.set_index(2);
|
||||
let mut network = test_network(&app);
|
||||
|
||||
let result = network
|
||||
.handle_lidarr_event(LidarrEvent::GetAlbumDetails(1))
|
||||
.await;
|
||||
|
||||
mock.assert_async().await;
|
||||
|
||||
let LidarrSerdeable::Album(album) = result.unwrap() else {
|
||||
panic!("Expected Album");
|
||||
};
|
||||
|
||||
assert_eq!(album, expected_album);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,121 @@
|
||||
use crate::models::lidarr_models::{Album};
|
||||
use crate::network::lidarr_network::LidarrEvent;
|
||||
use crate::network::{Network, RequestMethod};
|
||||
use anyhow::Result;
|
||||
use log::{debug, info, warn};
|
||||
use serde_json::{Value, json};
|
||||
|
||||
#[cfg(test)]
|
||||
#[path = "lidarr_albums_network_tests.rs"]
|
||||
mod lidarr_albums_network_tests;
|
||||
|
||||
impl Network<'_, '_> {
|
||||
pub(in crate::network::lidarr_network) async fn get_albums(
|
||||
&mut self,
|
||||
artist_id: i64,
|
||||
) -> Result<Vec<Album>> {
|
||||
info!("Fetching albums for Lidarr artist with ID: {artist_id}");
|
||||
let event = LidarrEvent::GetAlbums(artist_id);
|
||||
|
||||
let request_props = self
|
||||
.request_props_from(
|
||||
event,
|
||||
RequestMethod::Get,
|
||||
None::<()>,
|
||||
None,
|
||||
Some(format!("artistId={artist_id}")),
|
||||
)
|
||||
.await;
|
||||
|
||||
self
|
||||
.handle_request::<(), Vec<Album>>(request_props, |mut albums_vec, mut app| {
|
||||
albums_vec.sort_by(|a, b| a.id.cmp(&b.id));
|
||||
app.data.lidarr_data.albums.set_items(albums_vec);
|
||||
})
|
||||
.await
|
||||
}
|
||||
|
||||
pub(in crate::network::lidarr_network) async fn get_album_details(
|
||||
&mut self,
|
||||
album_id: i64,
|
||||
) -> Result<Album> {
|
||||
info!("Fetching details for Lidarr album with ID: {album_id}");
|
||||
let event = LidarrEvent::GetAlbumDetails(album_id);
|
||||
|
||||
let request_props = self
|
||||
.request_props_from(
|
||||
event,
|
||||
RequestMethod::Get,
|
||||
None::<()>,
|
||||
Some(format!("/{album_id}")),
|
||||
None,
|
||||
)
|
||||
.await;
|
||||
|
||||
self
|
||||
.handle_request::<(), Album>(request_props, |_, _| ())
|
||||
.await
|
||||
}
|
||||
|
||||
pub(in crate::network::lidarr_network) async fn toggle_album_monitoring(
|
||||
&mut self,
|
||||
album_id: i64,
|
||||
) -> Result<()> {
|
||||
let event = LidarrEvent::ToggleAlbumMonitoring(album_id);
|
||||
info!("Toggling album monitoring for album with ID: {album_id}");
|
||||
info!("Fetching album details for album with ID: {album_id}");
|
||||
|
||||
let detail_event = LidarrEvent::GetAlbums(0);
|
||||
let request_props = self
|
||||
.request_props_from(
|
||||
detail_event,
|
||||
RequestMethod::Get,
|
||||
None::<()>,
|
||||
Some(format!("/{album_id}")),
|
||||
None,
|
||||
)
|
||||
.await;
|
||||
|
||||
let mut response = String::new();
|
||||
|
||||
self
|
||||
.handle_request::<(), Value>(request_props, |detailed_album_body, _| {
|
||||
response = detailed_album_body.to_string()
|
||||
})
|
||||
.await?;
|
||||
|
||||
info!("Constructing toggle album monitoring body");
|
||||
|
||||
match serde_json::from_str::<Value>(&response) {
|
||||
Ok(mut detailed_album_body) => {
|
||||
let monitored = detailed_album_body
|
||||
.get("monitored")
|
||||
.unwrap()
|
||||
.as_bool()
|
||||
.unwrap();
|
||||
|
||||
*detailed_album_body.get_mut("monitored").unwrap() = json!(!monitored);
|
||||
|
||||
debug!("Toggle album monitoring body: {detailed_album_body:?}");
|
||||
|
||||
let request_props = self
|
||||
.request_props_from(
|
||||
event,
|
||||
RequestMethod::Put,
|
||||
Some(detailed_album_body),
|
||||
Some(format!("/{album_id}")),
|
||||
None,
|
||||
)
|
||||
.await;
|
||||
|
||||
self
|
||||
.handle_request::<Value, ()>(request_props, |_, _| ())
|
||||
.await
|
||||
}
|
||||
Err(_) => {
|
||||
warn!("Request for detailed album body was interrupted");
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user