feat(readarr): add Readarr tags and expose the command on the CLI
This commit is contained in:
5 files changed
+268
-1
No files matched your search
@@ -1,5 +1,6 @@
|
||||
use anyhow::Result;
|
||||
use log::info;
|
||||
use serde_json::{Value, json};
|
||||
|
||||
use super::{NetworkEvent, NetworkResource};
|
||||
use crate::models::readarr_models::{ReadarrSerdeable, ReadarrTaskName};
|
||||
@@ -15,6 +16,7 @@ mod readarr_network_tests;
|
||||
|
||||
#[derive(Debug, Eq, PartialEq, Clone)]
|
||||
pub enum ReadarrEvent {
|
||||
AddTag(String),
|
||||
GetDiskSpace,
|
||||
GetHostConfig,
|
||||
GetLogs(u64),
|
||||
@@ -44,7 +46,7 @@ impl NetworkResource for ReadarrEvent {
|
||||
ReadarrEvent::GetRootFolders => "/rootfolder",
|
||||
ReadarrEvent::GetStatus => "/system/status",
|
||||
ReadarrEvent::GetTasks => "/system/task",
|
||||
ReadarrEvent::GetTags => "/tag",
|
||||
ReadarrEvent::AddTag(_) | ReadarrEvent::GetTags => "/tag",
|
||||
ReadarrEvent::GetUpdates => "/update",
|
||||
}
|
||||
}
|
||||
@@ -62,6 +64,7 @@ impl Network<'_, '_> {
|
||||
readarr_event: ReadarrEvent,
|
||||
) -> Result<ReadarrSerdeable> {
|
||||
match readarr_event {
|
||||
ReadarrEvent::AddTag(tag) => self.add_readarr_tag(tag).await.map(ReadarrSerdeable::from),
|
||||
ReadarrEvent::GetDiskSpace => self
|
||||
.get_readarr_diskspace()
|
||||
.await
|
||||
@@ -177,4 +180,25 @@ impl Network<'_, '_> {
|
||||
})
|
||||
.await
|
||||
}
|
||||
|
||||
async fn add_readarr_tag(&mut self, tag: String) -> Result<Tag> {
|
||||
info!("Adding a new Readarr tag");
|
||||
let event = ReadarrEvent::AddTag(String::new());
|
||||
|
||||
let request_props = self
|
||||
.request_props_from(
|
||||
event,
|
||||
RequestMethod::Post,
|
||||
Some(json!({ "label": tag })),
|
||||
None,
|
||||
None,
|
||||
)
|
||||
.await;
|
||||
|
||||
self
|
||||
.handle_request::<Value, Tag>(request_props, |tag, mut app| {
|
||||
app.data.readarr_data.tags_map.insert(tag.id, tag.label);
|
||||
})
|
||||
.await
|
||||
}
|
||||
}
|
||||
@@ -22,6 +22,7 @@ mod tests {
|
||||
#[case(ReadarrEvent::GetRootFolders, "/rootfolder")]
|
||||
#[case(ReadarrEvent::GetStatus, "/system/status")]
|
||||
#[case(ReadarrEvent::GetTasks, "/system/task")]
|
||||
#[case(ReadarrEvent::AddTag(String::new()), "/tag")]
|
||||
#[case(ReadarrEvent::GetTags, "/tag")]
|
||||
#[case(ReadarrEvent::GetUpdates, "/update")]
|
||||
fn test_resource(#[case] event: ReadarrEvent, #[case] expected_uri: &str) {
|
||||
@@ -345,4 +346,68 @@ mod tests {
|
||||
assert_err!(result);
|
||||
assert_eq!(app.lock().await.data.readarr_data.tags_map, seeded_map);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_handle_add_readarr_tag_event() {
|
||||
let tag_json = json!({
|
||||
"id": 2,
|
||||
"label": "managarr-verify"
|
||||
});
|
||||
let response: Tag = serde_json::from_value(tag_json.clone()).unwrap();
|
||||
let (mock, app, _server) = MockServarrApi::post()
|
||||
.with_request_body(json!({ "label": "managarr-verify" }))
|
||||
.returns(tag_json)
|
||||
.build_for(ReadarrEvent::AddTag("managarr-verify".to_owned()))
|
||||
.await;
|
||||
{
|
||||
let mut app = app.lock().await;
|
||||
app.server_tabs.set_index(3);
|
||||
app.data.readarr_data.tags_map = BiMap::from_iter([(1i64, "huntarr-missing".to_owned())]);
|
||||
}
|
||||
let mut network = test_network(&app);
|
||||
|
||||
let result = network
|
||||
.handle_readarr_event(ReadarrEvent::AddTag("managarr-verify".to_owned()))
|
||||
.await;
|
||||
|
||||
mock.assert_async().await;
|
||||
|
||||
let ReadarrSerdeable::Tag(tag) = result.unwrap() else {
|
||||
panic!("Expected Tag");
|
||||
};
|
||||
|
||||
assert_eq!(tag, response);
|
||||
assert_eq!(
|
||||
app.lock().await.data.readarr_data.tags_map,
|
||||
BiMap::from_iter([
|
||||
(1i64, "huntarr-missing".to_owned()),
|
||||
(2i64, "managarr-verify".to_owned())
|
||||
])
|
||||
);
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_handle_add_readarr_tag_event_failure() {
|
||||
let seeded_map = BiMap::from_iter([(1i64, "huntarr-missing".to_owned())]);
|
||||
let (mock, app, _server) = MockServarrApi::post()
|
||||
.with_request_body(json!({ "label": "managarr-verify" }))
|
||||
.returns(json!({}))
|
||||
.status(500)
|
||||
.build_for(ReadarrEvent::AddTag("managarr-verify".to_owned()))
|
||||
.await;
|
||||
{
|
||||
let mut app = app.lock().await;
|
||||
app.server_tabs.set_index(3);
|
||||
app.data.readarr_data.tags_map = seeded_map.clone();
|
||||
}
|
||||
let mut network = test_network(&app);
|
||||
|
||||
let result = network
|
||||
.handle_readarr_event(ReadarrEvent::AddTag("managarr-verify".to_owned()))
|
||||
.await;
|
||||
|
||||
mock.assert_async().await;
|
||||
assert_err!(result);
|
||||
assert_eq!(app.lock().await.data.readarr_data.tags_map, seeded_map);
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user