fix(sonarr): Construct and pass the AddRootFolderBody alongside all AddRootFolder events when publishing to the networking channel

This commit is contained in:
2024-12-17 21:48:52 -07:00
parent 43410fac60
commit 23971cbb76
7 changed files with 351 additions and 73 deletions
+5 -22
View File
@@ -39,7 +39,7 @@ mod sonarr_network_tests;
#[derive(Debug, Eq, PartialEq, Clone)]
pub enum SonarrEvent {
AddRootFolder(Option<String>),
AddRootFolder(AddRootFolderBody),
AddSeries(Option<AddSeriesBody>),
AddTag(String),
ClearBlocklist,
@@ -357,31 +357,14 @@ impl<'a, 'b> Network<'a, 'b> {
}
}
async fn add_sonarr_root_folder(&mut self, root_folder: Option<String>) -> Result<Value> {
async fn add_sonarr_root_folder(&mut self, add_root_folder_body: AddRootFolderBody) -> Result<Value> {
info!("Adding new root folder to Sonarr");
let event = SonarrEvent::AddRootFolder(None);
let body = if let Some(path) = root_folder {
AddRootFolderBody { path }
} else {
let mut app = self.app.lock().await;
let path = app
.data
.sonarr_data
.edit_root_folder
.as_ref()
.unwrap()
.text
.clone();
let event = SonarrEvent::AddRootFolder(add_root_folder_body.clone());
app.data.sonarr_data.edit_root_folder = None;
AddRootFolderBody { path }
};
debug!("Add root folder body: {body:?}");
debug!("Add root folder body: {add_root_folder_body:?}");
let request_props = self
.request_props_from(event, RequestMethod::Post, Some(body), None, None)
.request_props_from(event, RequestMethod::Post, Some(add_root_folder_body), None, None)
.await;
self
+6 -42
View File
@@ -18,7 +18,7 @@ mod test {
use crate::models::sonarr_models::{
AddSeriesBody, AddSeriesOptions, AddSeriesSearchResult, AddSeriesSearchResultStatistics,
DownloadStatus, EditSeriesParams, IndexerSettings, MonitorEpisodeBody, SeriesMonitor,
SonarrHistoryEventType,
SonarrHistoryEventType
};
use crate::app::{App, ServarrConfig};
@@ -28,10 +28,7 @@ mod test {
AddSeriesModal, EditSeriesModal, EpisodeDetailsModal, SeasonDetailsModal,
};
use crate::models::servarr_data::sonarr::sonarr_data::ActiveSonarrBlock;
use crate::models::servarr_models::{
DiskSpace, EditIndexerParams, HostConfig, Indexer, IndexerField, Language, LogResponse,
Quality, QualityProfile, QualityWrapper, QueueEvent, RootFolder, SecurityConfig, Tag, Update,
};
use crate::models::servarr_models::{AddRootFolderBody, DiskSpace, EditIndexerParams, HostConfig, Indexer, IndexerField, Language, LogResponse, Quality, QualityProfile, QualityWrapper, QueueEvent, RootFolder, SecurityConfig, Tag, Update};
use crate::models::sonarr_models::{
BlocklistItem, DeleteSeriesParams, DownloadRecord, DownloadsResponse, Episode, EpisodeFile,
MediaInfo, SonarrRelease, SonarrReleaseDownloadBody, SonarrTaskName,
@@ -250,7 +247,7 @@ mod test {
#[values(
SonarrEvent::GetRootFolders,
SonarrEvent::DeleteRootFolder(None),
SonarrEvent::AddRootFolder(None)
SonarrEvent::AddRootFolder(AddRootFolderBody::default())
)]
event: SonarrEvent,
) {
@@ -310,6 +307,7 @@ mod test {
#[tokio::test]
async fn test_handle_add_sonarr_root_folder_event() {
let expected_add_root_folder_body = AddRootFolderBody { path: "/nfs/test".to_owned() };
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Post,
Some(json!({
@@ -317,49 +315,15 @@ mod test {
})),
Some(json!({})),
None,
SonarrEvent::AddRootFolder(None),
SonarrEvent::AddRootFolder(expected_add_root_folder_body.clone()),
None,
None,
)
.await;
app_arc.lock().await.data.sonarr_data.edit_root_folder = Some("/nfs/test".into());
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_sonarr_event(SonarrEvent::AddRootFolder(None))
.await
.is_ok());
async_server.assert_async().await;
assert!(app_arc
.lock()
.await
.data
.sonarr_data
.edit_root_folder
.is_none());
}
#[tokio::test]
async fn test_handle_add_sonarr_root_folder_event_uses_provided_path() {
let (async_server, app_arc, _server) = mock_servarr_api(
RequestMethod::Post,
Some(json!({
"path": "/test/test"
})),
Some(json!({})),
None,
SonarrEvent::AddRootFolder(None),
None,
None,
)
.await;
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
assert!(network
.handle_sonarr_event(SonarrEvent::AddRootFolder(Some("/test/test".to_owned())))
.handle_sonarr_event(SonarrEvent::AddRootFolder(expected_add_root_folder_body))
.await
.is_ok());