feat(network): Support for adding a new series to Sonarr
This commit is contained in:
@@ -1,9 +1,68 @@
|
|||||||
|
use strum::IntoEnumIterator;
|
||||||
|
|
||||||
use crate::models::{
|
use crate::models::{
|
||||||
sonarr_models::{Episode, SonarrHistoryItem, SonarrRelease},
|
servarr_models::RootFolder,
|
||||||
|
sonarr_models::{Episode, SeriesMonitor, SeriesType, SonarrHistoryItem, SonarrRelease},
|
||||||
|
stateful_list::StatefulList,
|
||||||
stateful_table::StatefulTable,
|
stateful_table::StatefulTable,
|
||||||
ScrollableText,
|
HorizontallyScrollableText, ScrollableText,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
use super::sonarr_data::SonarrData;
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
#[path = "modals_tests.rs"]
|
||||||
|
mod modals_tests;
|
||||||
|
|
||||||
|
#[derive(Default)]
|
||||||
|
pub struct AddSeriesModal {
|
||||||
|
pub root_folder_list: StatefulList<RootFolder>,
|
||||||
|
pub monitor_list: StatefulList<SeriesMonitor>,
|
||||||
|
pub quality_profile_list: StatefulList<String>,
|
||||||
|
pub language_profile_list: StatefulList<String>,
|
||||||
|
pub series_type_list: StatefulList<SeriesType>,
|
||||||
|
pub use_season_folder: bool,
|
||||||
|
pub tags: HorizontallyScrollableText,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl From<&SonarrData> for AddSeriesModal {
|
||||||
|
fn from(sonarr_data: &SonarrData) -> AddSeriesModal {
|
||||||
|
let mut add_series_modal = AddSeriesModal {
|
||||||
|
use_season_folder: true,
|
||||||
|
..AddSeriesModal::default()
|
||||||
|
};
|
||||||
|
add_series_modal
|
||||||
|
.monitor_list
|
||||||
|
.set_items(Vec::from_iter(SeriesMonitor::iter()));
|
||||||
|
add_series_modal
|
||||||
|
.series_type_list
|
||||||
|
.set_items(Vec::from_iter(SeriesType::iter()));
|
||||||
|
let mut quality_profile_names: Vec<String> = sonarr_data
|
||||||
|
.quality_profile_map
|
||||||
|
.right_values()
|
||||||
|
.cloned()
|
||||||
|
.collect();
|
||||||
|
quality_profile_names.sort();
|
||||||
|
add_series_modal
|
||||||
|
.quality_profile_list
|
||||||
|
.set_items(quality_profile_names);
|
||||||
|
let mut language_profile_names: Vec<String> = sonarr_data
|
||||||
|
.language_profiles_map
|
||||||
|
.right_values()
|
||||||
|
.cloned()
|
||||||
|
.collect();
|
||||||
|
language_profile_names.sort();
|
||||||
|
add_series_modal
|
||||||
|
.language_profile_list
|
||||||
|
.set_items(language_profile_names);
|
||||||
|
add_series_modal
|
||||||
|
.root_folder_list
|
||||||
|
.set_items(sonarr_data.root_folders.items.to_vec());
|
||||||
|
|
||||||
|
add_series_modal
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Default)]
|
#[derive(Default)]
|
||||||
pub struct EpisodeDetailsModal {
|
pub struct EpisodeDetailsModal {
|
||||||
pub episode_details: ScrollableText,
|
pub episode_details: ScrollableText,
|
||||||
|
|||||||
@@ -0,0 +1,58 @@
|
|||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use bimap::BiMap;
|
||||||
|
use strum::IntoEnumIterator;
|
||||||
|
|
||||||
|
use crate::models::{
|
||||||
|
servarr_data::sonarr::{modals::AddSeriesModal, sonarr_data::SonarrData},
|
||||||
|
servarr_models::RootFolder,
|
||||||
|
sonarr_models::{SeriesMonitor, SeriesType},
|
||||||
|
};
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_add_series_modal_from_sonarr_data() {
|
||||||
|
let root_folder = RootFolder {
|
||||||
|
id: 1,
|
||||||
|
path: "/nfs".to_owned(),
|
||||||
|
accessible: true,
|
||||||
|
free_space: 219902325555200,
|
||||||
|
unmapped_folders: None,
|
||||||
|
};
|
||||||
|
let mut sonarr_data = SonarrData {
|
||||||
|
quality_profile_map: BiMap::from_iter([
|
||||||
|
(2222, "HD - 1080p".to_owned()),
|
||||||
|
(1111, "Any".to_owned()),
|
||||||
|
]),
|
||||||
|
language_profiles_map: BiMap::from_iter([
|
||||||
|
(2222, "English".to_owned()),
|
||||||
|
(1111, "Any".to_owned()),
|
||||||
|
]),
|
||||||
|
..SonarrData::default()
|
||||||
|
};
|
||||||
|
sonarr_data
|
||||||
|
.root_folders
|
||||||
|
.set_items(vec![root_folder.clone()]);
|
||||||
|
|
||||||
|
let add_series_modal = AddSeriesModal::from(&sonarr_data);
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
add_series_modal.monitor_list.items,
|
||||||
|
Vec::from_iter(SeriesMonitor::iter())
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
add_series_modal.series_type_list.items,
|
||||||
|
Vec::from_iter(SeriesType::iter())
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
add_series_modal.quality_profile_list.items,
|
||||||
|
vec!["Any".to_owned(), "HD - 1080p".to_owned()]
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
add_series_modal.language_profile_list.items,
|
||||||
|
vec!["Any".to_owned(), "English".to_owned()]
|
||||||
|
);
|
||||||
|
assert_eq!(add_series_modal.root_folder_list.items, vec![root_folder]);
|
||||||
|
assert!(add_series_modal.tags.text.is_empty());
|
||||||
|
assert!(add_series_modal.use_season_folder);
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -6,14 +6,15 @@ use crate::models::{
|
|||||||
servarr_data::modals::IndexerTestResultModalItem,
|
servarr_data::modals::IndexerTestResultModalItem,
|
||||||
servarr_models::{DiskSpace, Indexer, QueueEvent, RootFolder},
|
servarr_models::{DiskSpace, Indexer, QueueEvent, RootFolder},
|
||||||
sonarr_models::{
|
sonarr_models::{
|
||||||
BlocklistItem, DownloadRecord, IndexerSettings, Season, Series, SonarrHistoryItem, SonarrTask,
|
AddSeriesSearchResult, BlocklistItem, DownloadRecord, IndexerSettings, Season, Series,
|
||||||
|
SonarrHistoryItem, SonarrTask,
|
||||||
},
|
},
|
||||||
stateful_list::StatefulList,
|
stateful_list::StatefulList,
|
||||||
stateful_table::StatefulTable,
|
stateful_table::StatefulTable,
|
||||||
HorizontallyScrollableText, Route, ScrollableText,
|
HorizontallyScrollableText, Route, ScrollableText,
|
||||||
};
|
};
|
||||||
|
|
||||||
use super::modals::SeasonDetailsModal;
|
use super::modals::{AddSeriesModal, SeasonDetailsModal};
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
#[path = "sonarr_data_tests.rs"]
|
#[path = "sonarr_data_tests.rs"]
|
||||||
@@ -21,6 +22,9 @@ mod sonarr_data_tests;
|
|||||||
|
|
||||||
pub struct SonarrData {
|
pub struct SonarrData {
|
||||||
pub add_list_exclusion: bool,
|
pub add_list_exclusion: bool,
|
||||||
|
pub add_searched_series: Option<StatefulTable<AddSeriesSearchResult>>,
|
||||||
|
pub add_series_modal: Option<AddSeriesModal>,
|
||||||
|
pub add_series_search: Option<HorizontallyScrollableText>,
|
||||||
pub blocklist: StatefulTable<BlocklistItem>,
|
pub blocklist: StatefulTable<BlocklistItem>,
|
||||||
pub delete_series_files: bool,
|
pub delete_series_files: bool,
|
||||||
pub downloads: StatefulTable<DownloadRecord>,
|
pub downloads: StatefulTable<DownloadRecord>,
|
||||||
@@ -58,6 +62,9 @@ impl Default for SonarrData {
|
|||||||
fn default() -> SonarrData {
|
fn default() -> SonarrData {
|
||||||
SonarrData {
|
SonarrData {
|
||||||
add_list_exclusion: false,
|
add_list_exclusion: false,
|
||||||
|
add_searched_series: None,
|
||||||
|
add_series_search: None,
|
||||||
|
add_series_modal: None,
|
||||||
blocklist: StatefulTable::default(),
|
blocklist: StatefulTable::default(),
|
||||||
downloads: StatefulTable::default(),
|
downloads: StatefulTable::default(),
|
||||||
delete_series_files: false,
|
delete_series_files: false,
|
||||||
|
|||||||
@@ -49,6 +49,9 @@ mod tests {
|
|||||||
let sonarr_data = SonarrData::default();
|
let sonarr_data = SonarrData::default();
|
||||||
|
|
||||||
assert!(!sonarr_data.add_list_exclusion);
|
assert!(!sonarr_data.add_list_exclusion);
|
||||||
|
assert!(sonarr_data.add_searched_series.is_none());
|
||||||
|
assert!(sonarr_data.add_series_search.is_none());
|
||||||
|
assert!(sonarr_data.add_series_modal.is_none());
|
||||||
assert!(sonarr_data.blocklist.is_empty());
|
assert!(sonarr_data.blocklist.is_empty());
|
||||||
assert!(!sonarr_data.delete_series_files);
|
assert!(!sonarr_data.delete_series_files);
|
||||||
assert!(sonarr_data.downloads.is_empty());
|
assert!(sonarr_data.downloads.is_empty());
|
||||||
|
|||||||
@@ -27,19 +27,46 @@ mod sonarr_models_tests;
|
|||||||
pub struct AddSeriesBody {
|
pub struct AddSeriesBody {
|
||||||
pub tvdb_id: i64,
|
pub tvdb_id: i64,
|
||||||
pub title: String,
|
pub title: String,
|
||||||
|
pub monitored: bool,
|
||||||
pub root_folder_path: String,
|
pub root_folder_path: String,
|
||||||
pub quality_profile_id: i64,
|
pub quality_profile_id: i64,
|
||||||
pub series_type: SeriesType,
|
|
||||||
pub season_folder: bool,
|
|
||||||
pub language_profile_id: i64,
|
pub language_profile_id: i64,
|
||||||
|
pub series_type: String,
|
||||||
|
pub season_folder: bool,
|
||||||
pub tags: Vec<i64>,
|
pub tags: Vec<i64>,
|
||||||
pub add_options: AddSeriesOptions,
|
pub add_options: AddSeriesOptions,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Derivative, Serialize, Deserialize, Debug, Default, Clone, PartialEq, Eq)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct AddSeriesSearchResult {
|
||||||
|
#[serde(deserialize_with = "super::from_i64")]
|
||||||
|
pub tvdb_id: i64,
|
||||||
|
pub title: HorizontallyScrollableText,
|
||||||
|
pub status: Option<String>,
|
||||||
|
pub ended: bool,
|
||||||
|
pub overview: Option<String>,
|
||||||
|
pub genres: Vec<String>,
|
||||||
|
#[serde(deserialize_with = "super::from_i64")]
|
||||||
|
pub year: i64,
|
||||||
|
pub network: Option<String>,
|
||||||
|
#[serde(deserialize_with = "super::from_i64")]
|
||||||
|
pub runtime: i64,
|
||||||
|
pub ratings: Option<Rating>,
|
||||||
|
pub statistics: Option<AddSeriesSearchResultStatistics>,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Derivative, Serialize, Deserialize, Debug, Default, Clone, PartialEq, Eq)]
|
||||||
|
#[serde(rename_all = "camelCase")]
|
||||||
|
pub struct AddSeriesSearchResultStatistics {
|
||||||
|
#[serde(deserialize_with = "super::from_i64")]
|
||||||
|
pub season_count: i64,
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Default, Clone, Serialize, Debug, PartialEq, Eq)]
|
#[derive(Default, Clone, Serialize, Debug, PartialEq, Eq)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
pub struct AddSeriesOptions {
|
pub struct AddSeriesOptions {
|
||||||
pub monitor: SeriesMonitor,
|
pub monitor: String,
|
||||||
pub search_for_cutoff_unmet_episodes: bool,
|
pub search_for_cutoff_unmet_episodes: bool,
|
||||||
pub search_for_missing_episodes: bool,
|
pub search_for_missing_episodes: bool,
|
||||||
}
|
}
|
||||||
@@ -258,9 +285,9 @@ pub struct Series {
|
|||||||
)]
|
)]
|
||||||
#[serde(rename_all = "camelCase")]
|
#[serde(rename_all = "camelCase")]
|
||||||
pub enum SeriesMonitor {
|
pub enum SeriesMonitor {
|
||||||
Unknown,
|
|
||||||
#[default]
|
#[default]
|
||||||
All,
|
All,
|
||||||
|
Unknown,
|
||||||
Future,
|
Future,
|
||||||
Missing,
|
Missing,
|
||||||
Existing,
|
Existing,
|
||||||
|
|||||||
@@ -299,7 +299,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.tags
|
.tags
|
||||||
.text
|
.text
|
||||||
.clone();
|
.clone();
|
||||||
let tag_ids_vec = self.extract_and_add_tag_ids_vec(tags).await;
|
let tag_ids_vec = self.extract_and_add_radarr_tag_ids_vec(tags).await;
|
||||||
let mut app = self.app.lock().await;
|
let mut app = self.app.lock().await;
|
||||||
let AddMovieModal {
|
let AddMovieModal {
|
||||||
root_folder_list,
|
root_folder_list,
|
||||||
@@ -1037,7 +1037,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.tags
|
.tags
|
||||||
.text
|
.text
|
||||||
.clone();
|
.clone();
|
||||||
let tag_ids_vec = self.extract_and_add_tag_ids_vec(tags).await;
|
let tag_ids_vec = self.extract_and_add_radarr_tag_ids_vec(tags).await;
|
||||||
let mut app = self.app.lock().await;
|
let mut app = self.app.lock().await;
|
||||||
|
|
||||||
let params = {
|
let params = {
|
||||||
@@ -1222,7 +1222,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.tags
|
.tags
|
||||||
.text
|
.text
|
||||||
.clone();
|
.clone();
|
||||||
let tag_ids_vec = self.extract_and_add_tag_ids_vec(tags).await;
|
let tag_ids_vec = self.extract_and_add_radarr_tag_ids_vec(tags).await;
|
||||||
let mut app = self.app.lock().await;
|
let mut app = self.app.lock().await;
|
||||||
|
|
||||||
let params = {
|
let params = {
|
||||||
@@ -2237,7 +2237,7 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn extract_and_add_tag_ids_vec(&mut self, edit_tags: String) -> Vec<i64> {
|
async fn extract_and_add_radarr_tag_ids_vec(&mut self, edit_tags: String) -> Vec<i64> {
|
||||||
let tags_map = self.app.lock().await.data.radarr_data.tags_map.clone();
|
let tags_map = self.app.lock().await.data.radarr_data.tags_map.clone();
|
||||||
let tags = edit_tags.clone();
|
let tags = edit_tags.clone();
|
||||||
let missing_tags_vec = edit_tags
|
let missing_tags_vec = edit_tags
|
||||||
|
|||||||
@@ -4876,7 +4876,7 @@ mod test {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_extract_and_add_tag_ids_vec() {
|
async fn test_extract_and_add_radarr_tag_ids_vec() {
|
||||||
let app_arc = Arc::new(Mutex::new(App::default()));
|
let app_arc = Arc::new(Mutex::new(App::default()));
|
||||||
let tags = " test,hi ,, usenet ".to_owned();
|
let tags = " test,hi ,, usenet ".to_owned();
|
||||||
{
|
{
|
||||||
@@ -4890,13 +4890,13 @@ mod test {
|
|||||||
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
network.extract_and_add_tag_ids_vec(tags).await,
|
network.extract_and_add_radarr_tag_ids_vec(tags).await,
|
||||||
vec![2, 3, 1]
|
vec![2, 3, 1]
|
||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_extract_and_add_tag_ids_vec_add_missing_tags_first() {
|
async fn test_extract_and_add_radarr_tag_ids_vec_add_missing_tags_first() {
|
||||||
let (async_server, app_arc, _server) = mock_servarr_api(
|
let (async_server, app_arc, _server) = mock_servarr_api(
|
||||||
RequestMethod::Post,
|
RequestMethod::Post,
|
||||||
Some(json!({ "label": "testing" })),
|
Some(json!({ "label": "testing" })),
|
||||||
@@ -4919,7 +4919,7 @@ mod test {
|
|||||||
}
|
}
|
||||||
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
let tag_ids_vec = network.extract_and_add_tag_ids_vec(tags).await;
|
let tag_ids_vec = network.extract_and_add_radarr_tag_ids_vec(tags).await;
|
||||||
|
|
||||||
async_server.assert_async().await;
|
async_server.assert_async().await;
|
||||||
assert_eq!(tag_ids_vec, vec![1, 2, 3]);
|
assert_eq!(tag_ids_vec, vec![1, 2, 3]);
|
||||||
|
|||||||
@@ -9,7 +9,7 @@ use crate::{
|
|||||||
servarr_data::{
|
servarr_data::{
|
||||||
modals::IndexerTestResultModalItem,
|
modals::IndexerTestResultModalItem,
|
||||||
sonarr::{
|
sonarr::{
|
||||||
modals::{EpisodeDetailsModal, SeasonDetailsModal},
|
modals::{AddSeriesModal, EpisodeDetailsModal, SeasonDetailsModal},
|
||||||
sonarr_data::ActiveSonarrBlock,
|
sonarr_data::ActiveSonarrBlock,
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
@@ -18,10 +18,10 @@ use crate::{
|
|||||||
QualityProfile, QueueEvent, RootFolder, SecurityConfig, Tag, Update,
|
QualityProfile, QueueEvent, RootFolder, SecurityConfig, Tag, Update,
|
||||||
},
|
},
|
||||||
sonarr_models::{
|
sonarr_models::{
|
||||||
BlocklistResponse, DeleteSeriesParams, DownloadRecord, DownloadsResponse, Episode,
|
AddSeriesBody, AddSeriesOptions, AddSeriesSearchResult, BlocklistResponse,
|
||||||
IndexerSettings, Series, SonarrCommandBody, SonarrHistoryItem, SonarrHistoryWrapper,
|
DeleteSeriesParams, DownloadRecord, DownloadsResponse, Episode, IndexerSettings, Series,
|
||||||
SonarrRelease, SonarrReleaseDownloadBody, SonarrSerdeable, SonarrTask, SonarrTaskName,
|
SonarrCommandBody, SonarrHistoryItem, SonarrHistoryWrapper, SonarrRelease,
|
||||||
SystemStatus,
|
SonarrReleaseDownloadBody, SonarrSerdeable, SonarrTask, SonarrTaskName, SystemStatus,
|
||||||
},
|
},
|
||||||
stateful_table::StatefulTable,
|
stateful_table::StatefulTable,
|
||||||
HorizontallyScrollableText, Route, Scrollable, ScrollableText,
|
HorizontallyScrollableText, Route, Scrollable, ScrollableText,
|
||||||
@@ -38,6 +38,7 @@ mod sonarr_network_tests;
|
|||||||
#[derive(Debug, Eq, PartialEq, Clone)]
|
#[derive(Debug, Eq, PartialEq, Clone)]
|
||||||
pub enum SonarrEvent {
|
pub enum SonarrEvent {
|
||||||
AddRootFolder(Option<String>),
|
AddRootFolder(Option<String>),
|
||||||
|
AddSeries(Option<AddSeriesBody>),
|
||||||
AddTag(String),
|
AddTag(String),
|
||||||
ClearBlocklist,
|
ClearBlocklist,
|
||||||
DeleteBlocklistItem(Option<i64>),
|
DeleteBlocklistItem(Option<i64>),
|
||||||
@@ -120,9 +121,10 @@ impl NetworkResource for SonarrEvent {
|
|||||||
SonarrEvent::GetTasks => "/system/task",
|
SonarrEvent::GetTasks => "/system/task",
|
||||||
SonarrEvent::GetUpdates => "/update",
|
SonarrEvent::GetUpdates => "/update",
|
||||||
SonarrEvent::HealthCheck => "/health",
|
SonarrEvent::HealthCheck => "/health",
|
||||||
SonarrEvent::ListSeries | SonarrEvent::GetSeriesDetails(_) | SonarrEvent::DeleteSeries(_) => {
|
SonarrEvent::AddSeries(_)
|
||||||
"/series"
|
| SonarrEvent::ListSeries
|
||||||
}
|
| SonarrEvent::GetSeriesDetails(_)
|
||||||
|
| SonarrEvent::DeleteSeries(_) => "/series",
|
||||||
SonarrEvent::MarkHistoryItemAsFailed(_) => "/history/failed",
|
SonarrEvent::MarkHistoryItemAsFailed(_) => "/history/failed",
|
||||||
SonarrEvent::TestIndexer(_) => "/indexer/test",
|
SonarrEvent::TestIndexer(_) => "/indexer/test",
|
||||||
SonarrEvent::TestAllIndexers => "/indexer/testall",
|
SonarrEvent::TestAllIndexers => "/indexer/testall",
|
||||||
@@ -146,6 +148,10 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.add_sonarr_root_folder(path)
|
.add_sonarr_root_folder(path)
|
||||||
.await
|
.await
|
||||||
.map(SonarrSerdeable::from),
|
.map(SonarrSerdeable::from),
|
||||||
|
SonarrEvent::AddSeries(body) => self
|
||||||
|
.add_sonarr_series(body)
|
||||||
|
.await
|
||||||
|
.map(SonarrSerdeable::from),
|
||||||
SonarrEvent::AddTag(tag) => self.add_sonarr_tag(tag).await.map(SonarrSerdeable::from),
|
SonarrEvent::AddTag(tag) => self.add_sonarr_tag(tag).await.map(SonarrSerdeable::from),
|
||||||
SonarrEvent::ClearBlocklist => self
|
SonarrEvent::ClearBlocklist => self
|
||||||
.clear_sonarr_blocklist()
|
.clear_sonarr_blocklist()
|
||||||
@@ -327,6 +333,106 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn add_sonarr_series(
|
||||||
|
&mut self,
|
||||||
|
add_series_body_option: Option<AddSeriesBody>,
|
||||||
|
) -> Result<Value> {
|
||||||
|
info!("Adding new series to Sonarr");
|
||||||
|
let event = SonarrEvent::AddSeries(None);
|
||||||
|
let body = if let Some(add_series_body) = add_series_body_option {
|
||||||
|
add_series_body
|
||||||
|
} else {
|
||||||
|
let tags = self
|
||||||
|
.app
|
||||||
|
.lock()
|
||||||
|
.await
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.add_series_modal
|
||||||
|
.as_ref()
|
||||||
|
.unwrap()
|
||||||
|
.tags
|
||||||
|
.text
|
||||||
|
.clone();
|
||||||
|
let tag_ids_vec = self.extract_and_add_sonarr_tag_ids_vec(tags).await;
|
||||||
|
let mut app = self.app.lock().await;
|
||||||
|
let AddSeriesModal {
|
||||||
|
root_folder_list,
|
||||||
|
monitor_list,
|
||||||
|
quality_profile_list,
|
||||||
|
language_profile_list,
|
||||||
|
series_type_list,
|
||||||
|
use_season_folder,
|
||||||
|
..
|
||||||
|
} = app.data.sonarr_data.add_series_modal.as_ref().unwrap();
|
||||||
|
let season_folder = *use_season_folder;
|
||||||
|
let (tvdb_id, title) = {
|
||||||
|
let AddSeriesSearchResult { tvdb_id, title, .. } = app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.add_searched_series
|
||||||
|
.as_ref()
|
||||||
|
.unwrap()
|
||||||
|
.current_selection()
|
||||||
|
.clone();
|
||||||
|
(tvdb_id, title.text)
|
||||||
|
};
|
||||||
|
let quality_profile = quality_profile_list.current_selection();
|
||||||
|
let quality_profile_id = *app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.quality_profile_map
|
||||||
|
.iter()
|
||||||
|
.filter(|(_, value)| *value == quality_profile)
|
||||||
|
.map(|(key, _)| key)
|
||||||
|
.next()
|
||||||
|
.unwrap();
|
||||||
|
let language_profile = language_profile_list.current_selection();
|
||||||
|
let language_profile_id = *app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.language_profiles_map
|
||||||
|
.iter()
|
||||||
|
.filter(|(_, value)| *value == language_profile)
|
||||||
|
.map(|(key, _)| key)
|
||||||
|
.next()
|
||||||
|
.unwrap();
|
||||||
|
|
||||||
|
let path = root_folder_list.current_selection().path.clone();
|
||||||
|
let monitor = monitor_list.current_selection().to_string();
|
||||||
|
let series_type = series_type_list.current_selection().to_string();
|
||||||
|
|
||||||
|
app.data.sonarr_data.add_series_modal = None;
|
||||||
|
|
||||||
|
AddSeriesBody {
|
||||||
|
tvdb_id,
|
||||||
|
title,
|
||||||
|
monitored: true,
|
||||||
|
root_folder_path: path,
|
||||||
|
quality_profile_id,
|
||||||
|
language_profile_id,
|
||||||
|
series_type,
|
||||||
|
season_folder,
|
||||||
|
tags: tag_ids_vec,
|
||||||
|
add_options: AddSeriesOptions {
|
||||||
|
monitor,
|
||||||
|
search_for_cutoff_unmet_episodes: true,
|
||||||
|
search_for_missing_episodes: true,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
debug!("Add series body: {body:?}");
|
||||||
|
|
||||||
|
let request_props = self
|
||||||
|
.request_props_from(event, RequestMethod::Post, Some(body), None, None)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
self
|
||||||
|
.handle_request::<AddSeriesBody, Value>(request_props, |_, _| ())
|
||||||
|
.await
|
||||||
|
}
|
||||||
|
|
||||||
async fn add_sonarr_tag(&mut self, tag: String) -> Result<Tag> {
|
async fn add_sonarr_tag(&mut self, tag: String) -> Result<Tag> {
|
||||||
info!("Adding a new Sonarr tag");
|
info!("Adding a new Sonarr tag");
|
||||||
let event = SonarrEvent::AddTag(String::new());
|
let event = SonarrEvent::AddTag(String::new());
|
||||||
@@ -1703,6 +1809,36 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn extract_and_add_sonarr_tag_ids_vec(&mut self, edit_tags: String) -> Vec<i64> {
|
||||||
|
let tags_map = self.app.lock().await.data.sonarr_data.tags_map.clone();
|
||||||
|
let tags = edit_tags.clone();
|
||||||
|
let missing_tags_vec = edit_tags
|
||||||
|
.split(',')
|
||||||
|
.filter(|&tag| !tag.is_empty() && tags_map.get_by_right(tag.trim()).is_none())
|
||||||
|
.collect::<Vec<&str>>();
|
||||||
|
|
||||||
|
for tag in missing_tags_vec {
|
||||||
|
self
|
||||||
|
.add_sonarr_tag(tag.trim().to_owned())
|
||||||
|
.await
|
||||||
|
.expect("Unable to add tag");
|
||||||
|
}
|
||||||
|
|
||||||
|
let app = self.app.lock().await;
|
||||||
|
tags
|
||||||
|
.split(',')
|
||||||
|
.filter(|tag| !tag.is_empty())
|
||||||
|
.map(|tag| {
|
||||||
|
*app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.tags_map
|
||||||
|
.get_by_right(tag.trim())
|
||||||
|
.unwrap()
|
||||||
|
})
|
||||||
|
.collect()
|
||||||
|
}
|
||||||
|
|
||||||
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
|
||||||
|
|||||||
@@ -11,13 +11,21 @@ mod test {
|
|||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
use serde_json::json;
|
use serde_json::json;
|
||||||
use serde_json::{Number, Value};
|
use serde_json::{Number, Value};
|
||||||
|
use strum::IntoEnumIterator;
|
||||||
use tokio::sync::Mutex;
|
use tokio::sync::Mutex;
|
||||||
use tokio_util::sync::CancellationToken;
|
use tokio_util::sync::CancellationToken;
|
||||||
|
|
||||||
|
use crate::models::sonarr_models::{
|
||||||
|
AddSeriesBody, AddSeriesOptions, AddSeriesSearchResult, AddSeriesSearchResultStatistics,
|
||||||
|
SeriesMonitor,
|
||||||
|
};
|
||||||
|
|
||||||
use crate::app::App;
|
use crate::app::App;
|
||||||
use crate::models::radarr_models::IndexerTestResult;
|
use crate::models::radarr_models::IndexerTestResult;
|
||||||
use crate::models::servarr_data::modals::IndexerTestResultModalItem;
|
use crate::models::servarr_data::modals::IndexerTestResultModalItem;
|
||||||
use crate::models::servarr_data::sonarr::modals::{EpisodeDetailsModal, SeasonDetailsModal};
|
use crate::models::servarr_data::sonarr::modals::{
|
||||||
|
AddSeriesModal, EpisodeDetailsModal, SeasonDetailsModal,
|
||||||
|
};
|
||||||
use crate::models::servarr_data::sonarr::sonarr_data::ActiveSonarrBlock;
|
use crate::models::servarr_data::sonarr::sonarr_data::ActiveSonarrBlock;
|
||||||
use crate::models::servarr_models::{
|
use crate::models::servarr_models::{
|
||||||
DiskSpace, HostConfig, Indexer, IndexerField, Language, LogResponse, Quality, QualityProfile,
|
DiskSpace, HostConfig, Indexer, IndexerField, Language, LogResponse, Quality, QualityProfile,
|
||||||
@@ -33,7 +41,7 @@ mod test {
|
|||||||
use crate::models::sonarr_models::{SonarrTask, SystemStatus};
|
use crate::models::sonarr_models::{SonarrTask, SystemStatus};
|
||||||
use crate::models::stateful_table::StatefulTable;
|
use crate::models::stateful_table::StatefulTable;
|
||||||
use crate::models::{sonarr_models::SonarrSerdeable, stateful_table::SortOption};
|
use crate::models::{sonarr_models::SonarrSerdeable, stateful_table::SortOption};
|
||||||
use crate::models::{HorizontallyScrollableText, ScrollableText};
|
use crate::models::{HorizontallyScrollableText, Scrollable, ScrollableText};
|
||||||
|
|
||||||
use crate::network::sonarr_network::get_episode_status;
|
use crate::network::sonarr_network::get_episode_status;
|
||||||
use crate::{
|
use crate::{
|
||||||
@@ -138,6 +146,7 @@ mod test {
|
|||||||
#[rstest]
|
#[rstest]
|
||||||
fn test_resource_series(
|
fn test_resource_series(
|
||||||
#[values(
|
#[values(
|
||||||
|
SonarrEvent::AddSeries(None),
|
||||||
SonarrEvent::ListSeries,
|
SonarrEvent::ListSeries,
|
||||||
SonarrEvent::GetSeriesDetails(None),
|
SonarrEvent::GetSeriesDetails(None),
|
||||||
SonarrEvent::DeleteSeries(None)
|
SonarrEvent::DeleteSeries(None)
|
||||||
@@ -321,6 +330,267 @@ mod test {
|
|||||||
.is_none());
|
.is_none());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_add_sonarr_series_event() {
|
||||||
|
let (async_server, app_arc, _server) = mock_servarr_api(
|
||||||
|
RequestMethod::Post,
|
||||||
|
Some(json!({
|
||||||
|
"tvdbId": 1234,
|
||||||
|
"title": "Test",
|
||||||
|
"monitored": true,
|
||||||
|
"rootFolderPath": "/nfs2",
|
||||||
|
"qualityProfileId": 2222,
|
||||||
|
"languageProfileId": 2222,
|
||||||
|
"seriesType": "standard",
|
||||||
|
"seasonFolder": true,
|
||||||
|
"tags": [1, 2],
|
||||||
|
"addOptions": {
|
||||||
|
"monitor": "all",
|
||||||
|
"searchForCutoffUnmetEpisodes": true,
|
||||||
|
"searchForMissingEpisodes": true
|
||||||
|
}
|
||||||
|
})),
|
||||||
|
Some(json!({})),
|
||||||
|
None,
|
||||||
|
SonarrEvent::AddSeries(None),
|
||||||
|
None,
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
{
|
||||||
|
let mut app = app_arc.lock().await;
|
||||||
|
let mut add_series_modal = AddSeriesModal {
|
||||||
|
use_season_folder: true,
|
||||||
|
tags: "usenet, testing".into(),
|
||||||
|
..AddSeriesModal::default()
|
||||||
|
};
|
||||||
|
add_series_modal.root_folder_list.set_items(vec![
|
||||||
|
RootFolder {
|
||||||
|
id: 1,
|
||||||
|
path: "/nfs".to_owned(),
|
||||||
|
accessible: true,
|
||||||
|
free_space: 219902325555200,
|
||||||
|
unmapped_folders: None,
|
||||||
|
},
|
||||||
|
RootFolder {
|
||||||
|
id: 2,
|
||||||
|
path: "/nfs2".to_owned(),
|
||||||
|
accessible: true,
|
||||||
|
free_space: 21990232555520,
|
||||||
|
unmapped_folders: None,
|
||||||
|
},
|
||||||
|
]);
|
||||||
|
add_series_modal.root_folder_list.state.select(Some(1));
|
||||||
|
add_series_modal
|
||||||
|
.quality_profile_list
|
||||||
|
.set_items(vec!["HD - 1080p".to_owned()]);
|
||||||
|
add_series_modal
|
||||||
|
.language_profile_list
|
||||||
|
.set_items(vec!["English".to_owned()]);
|
||||||
|
add_series_modal
|
||||||
|
.monitor_list
|
||||||
|
.set_items(Vec::from_iter(SeriesMonitor::iter()));
|
||||||
|
add_series_modal
|
||||||
|
.series_type_list
|
||||||
|
.set_items(Vec::from_iter(SeriesType::iter()));
|
||||||
|
app.data.sonarr_data.add_series_modal = Some(add_series_modal);
|
||||||
|
app.data.sonarr_data.quality_profile_map =
|
||||||
|
BiMap::from_iter([(2222, "HD - 1080p".to_owned())]);
|
||||||
|
app.data.sonarr_data.language_profiles_map = BiMap::from_iter([(2222, "English".to_owned())]);
|
||||||
|
app.data.sonarr_data.tags_map =
|
||||||
|
BiMap::from_iter([(1, "usenet".to_owned()), (2, "testing".to_owned())]);
|
||||||
|
let mut add_searched_series = StatefulTable::default();
|
||||||
|
add_searched_series.set_items(vec![add_series_search_result()]);
|
||||||
|
app.data.sonarr_data.add_searched_series = Some(add_searched_series);
|
||||||
|
}
|
||||||
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
|
assert!(network
|
||||||
|
.handle_sonarr_event(SonarrEvent::AddSeries(None))
|
||||||
|
.await
|
||||||
|
.is_ok());
|
||||||
|
|
||||||
|
async_server.assert_async().await;
|
||||||
|
assert!(app_arc
|
||||||
|
.lock()
|
||||||
|
.await
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.add_series_modal
|
||||||
|
.is_none());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_add_sonarr_series_event_uses_provided_body() {
|
||||||
|
let (async_server, app_arc, _server) = mock_servarr_api(
|
||||||
|
RequestMethod::Post,
|
||||||
|
Some(json!({
|
||||||
|
"tvdbId": 1234,
|
||||||
|
"title": "Test",
|
||||||
|
"monitored": true,
|
||||||
|
"rootFolderPath": "/nfs2",
|
||||||
|
"qualityProfileId": 2222,
|
||||||
|
"languageProfileId": 2222,
|
||||||
|
"seriesType": "standard",
|
||||||
|
"seasonFolder": true,
|
||||||
|
"tags": [1, 2],
|
||||||
|
"addOptions": {
|
||||||
|
"monitor": "standard",
|
||||||
|
"searchForCutoffUnmetEpisodes": true,
|
||||||
|
"searchForMissingEpisodes": true
|
||||||
|
}
|
||||||
|
})),
|
||||||
|
Some(json!({})),
|
||||||
|
None,
|
||||||
|
SonarrEvent::AddSeries(None),
|
||||||
|
None,
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
let body = AddSeriesBody {
|
||||||
|
tvdb_id: 1234,
|
||||||
|
title: "Test".to_owned(),
|
||||||
|
monitored: true,
|
||||||
|
root_folder_path: "/nfs2".to_owned(),
|
||||||
|
quality_profile_id: 2222,
|
||||||
|
language_profile_id: 2222,
|
||||||
|
series_type: "standard".to_owned(),
|
||||||
|
season_folder: true,
|
||||||
|
tags: vec![1, 2],
|
||||||
|
add_options: AddSeriesOptions {
|
||||||
|
monitor: "standard".to_owned(),
|
||||||
|
search_for_cutoff_unmet_episodes: true,
|
||||||
|
search_for_missing_episodes: true,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
|
assert!(network
|
||||||
|
.handle_sonarr_event(SonarrEvent::AddSeries(Some(body)))
|
||||||
|
.await
|
||||||
|
.is_ok());
|
||||||
|
|
||||||
|
async_server.assert_async().await;
|
||||||
|
assert!(app_arc
|
||||||
|
.lock()
|
||||||
|
.await
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.add_series_modal
|
||||||
|
.is_none());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_handle_add_sonarr_series_event_reuse_existing_table_if_search_already_performed() {
|
||||||
|
let (async_server, app_arc, _server) = mock_servarr_api(
|
||||||
|
RequestMethod::Post,
|
||||||
|
Some(json!({
|
||||||
|
"tvdbId": 5678,
|
||||||
|
"title": "Test",
|
||||||
|
"monitored": true,
|
||||||
|
"rootFolderPath": "/nfs2",
|
||||||
|
"qualityProfileId": 2222,
|
||||||
|
"languageProfileId": 2222,
|
||||||
|
"seriesType": "standard",
|
||||||
|
"seasonFolder": true,
|
||||||
|
"tags": [1, 2],
|
||||||
|
"addOptions": {
|
||||||
|
"monitor": "all",
|
||||||
|
"searchForCutoffUnmetEpisodes": true,
|
||||||
|
"searchForMissingEpisodes": true
|
||||||
|
}
|
||||||
|
})),
|
||||||
|
Some(json!({})),
|
||||||
|
None,
|
||||||
|
SonarrEvent::AddSeries(None),
|
||||||
|
None,
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
|
||||||
|
{
|
||||||
|
let mut app = app_arc.lock().await;
|
||||||
|
let mut add_series_modal = AddSeriesModal {
|
||||||
|
use_season_folder: true,
|
||||||
|
tags: "usenet, testing".into(),
|
||||||
|
..AddSeriesModal::default()
|
||||||
|
};
|
||||||
|
add_series_modal.root_folder_list.set_items(vec![
|
||||||
|
RootFolder {
|
||||||
|
id: 1,
|
||||||
|
path: "/nfs".to_owned(),
|
||||||
|
accessible: true,
|
||||||
|
free_space: 219902325555200,
|
||||||
|
unmapped_folders: None,
|
||||||
|
},
|
||||||
|
RootFolder {
|
||||||
|
id: 2,
|
||||||
|
path: "/nfs2".to_owned(),
|
||||||
|
accessible: true,
|
||||||
|
free_space: 21990232555520,
|
||||||
|
unmapped_folders: None,
|
||||||
|
},
|
||||||
|
]);
|
||||||
|
add_series_modal.root_folder_list.state.select(Some(1));
|
||||||
|
add_series_modal
|
||||||
|
.quality_profile_list
|
||||||
|
.set_items(vec!["HD - 1080p".to_owned()]);
|
||||||
|
add_series_modal
|
||||||
|
.language_profile_list
|
||||||
|
.set_items(vec!["English".to_owned()]);
|
||||||
|
add_series_modal
|
||||||
|
.monitor_list
|
||||||
|
.set_items(Vec::from_iter(SeriesMonitor::iter()));
|
||||||
|
add_series_modal
|
||||||
|
.series_type_list
|
||||||
|
.set_items(Vec::from_iter(SeriesType::iter()));
|
||||||
|
app.data.sonarr_data.add_series_modal = Some(add_series_modal);
|
||||||
|
app.data.sonarr_data.quality_profile_map =
|
||||||
|
BiMap::from_iter([(2222, "HD - 1080p".to_owned())]);
|
||||||
|
app.data.sonarr_data.language_profiles_map = BiMap::from_iter([(2222, "English".to_owned())]);
|
||||||
|
app.data.sonarr_data.tags_map =
|
||||||
|
BiMap::from_iter([(1, "usenet".to_owned()), (2, "testing".to_owned())]);
|
||||||
|
let secondary_search_result = AddSeriesSearchResult {
|
||||||
|
tvdb_id: 5678,
|
||||||
|
..add_series_search_result()
|
||||||
|
};
|
||||||
|
let mut add_searched_series = StatefulTable::default();
|
||||||
|
add_searched_series.set_items(vec![add_series_search_result(), secondary_search_result]);
|
||||||
|
add_searched_series.scroll_to_bottom();
|
||||||
|
app.data.sonarr_data.add_searched_series = Some(add_searched_series);
|
||||||
|
}
|
||||||
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
|
assert!(network
|
||||||
|
.handle_sonarr_event(SonarrEvent::AddSeries(None))
|
||||||
|
.await
|
||||||
|
.is_ok());
|
||||||
|
|
||||||
|
async_server.assert_async().await;
|
||||||
|
assert!(app_arc
|
||||||
|
.lock()
|
||||||
|
.await
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.add_series_modal
|
||||||
|
.is_none());
|
||||||
|
assert_eq!(
|
||||||
|
app_arc
|
||||||
|
.lock()
|
||||||
|
.await
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.add_searched_series
|
||||||
|
.as_ref()
|
||||||
|
.unwrap()
|
||||||
|
.current_selection()
|
||||||
|
.tvdb_id,
|
||||||
|
5678
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::test]
|
#[tokio::test]
|
||||||
async fn test_handle_add_sonarr_tag() {
|
async fn test_handle_add_sonarr_tag() {
|
||||||
let tag_json = json!({ "id": 3, "label": "testing" });
|
let tag_json = json!({ "id": 3, "label": "testing" });
|
||||||
@@ -4851,6 +5121,64 @@ mod test {
|
|||||||
async_server.assert_async().await;
|
async_server.assert_async().await;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_extract_and_add_sonarr_tag_ids_vec() {
|
||||||
|
let app_arc = Arc::new(Mutex::new(App::default()));
|
||||||
|
let tags = " test,hi ,, usenet ".to_owned();
|
||||||
|
{
|
||||||
|
let mut app = app_arc.lock().await;
|
||||||
|
app.data.sonarr_data.tags_map = BiMap::from_iter([
|
||||||
|
(1, "usenet".to_owned()),
|
||||||
|
(2, "test".to_owned()),
|
||||||
|
(3, "hi".to_owned()),
|
||||||
|
]);
|
||||||
|
}
|
||||||
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
|
assert_eq!(
|
||||||
|
network.extract_and_add_sonarr_tag_ids_vec(tags).await,
|
||||||
|
vec![2, 3, 1]
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[tokio::test]
|
||||||
|
async fn test_extract_and_add_sonarr_tag_ids_vec_add_missing_tags_first() {
|
||||||
|
let (async_server, app_arc, _server) = mock_servarr_api(
|
||||||
|
RequestMethod::Post,
|
||||||
|
Some(json!({ "label": "testing" })),
|
||||||
|
Some(json!({ "id": 3, "label": "testing" })),
|
||||||
|
None,
|
||||||
|
SonarrEvent::GetTags,
|
||||||
|
None,
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.await;
|
||||||
|
let tags = "usenet, test, testing".to_owned();
|
||||||
|
{
|
||||||
|
let mut app = app_arc.lock().await;
|
||||||
|
app.data.sonarr_data.add_series_modal = Some(AddSeriesModal {
|
||||||
|
tags: tags.clone().into(),
|
||||||
|
..AddSeriesModal::default()
|
||||||
|
});
|
||||||
|
app.data.sonarr_data.tags_map =
|
||||||
|
BiMap::from_iter([(1, "usenet".to_owned()), (2, "test".to_owned())]);
|
||||||
|
}
|
||||||
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
|
let tag_ids_vec = network.extract_and_add_sonarr_tag_ids_vec(tags).await;
|
||||||
|
|
||||||
|
async_server.assert_async().await;
|
||||||
|
assert_eq!(tag_ids_vec, vec![1, 2, 3]);
|
||||||
|
assert_eq!(
|
||||||
|
app_arc.lock().await.data.sonarr_data.tags_map,
|
||||||
|
BiMap::from_iter([
|
||||||
|
(1, "usenet".to_owned()),
|
||||||
|
(2, "test".to_owned()),
|
||||||
|
(3, "testing".to_owned())
|
||||||
|
])
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
#[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()));
|
||||||
@@ -5084,6 +5412,26 @@ mod test {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn add_series_search_result() -> AddSeriesSearchResult {
|
||||||
|
AddSeriesSearchResult {
|
||||||
|
tvdb_id: 1234,
|
||||||
|
title: HorizontallyScrollableText::from("Test"),
|
||||||
|
status: Some("continuing".to_owned()),
|
||||||
|
ended: false,
|
||||||
|
overview: Some("New series blah blah blah".to_owned()),
|
||||||
|
genres: genres(),
|
||||||
|
year: 2023,
|
||||||
|
network: Some("Prime Video".to_owned()),
|
||||||
|
runtime: 60,
|
||||||
|
ratings: Some(rating()),
|
||||||
|
statistics: Some(add_series_search_result_statistics()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn add_series_search_result_statistics() -> AddSeriesSearchResultStatistics {
|
||||||
|
AddSeriesSearchResultStatistics { season_count: 3 }
|
||||||
|
}
|
||||||
|
|
||||||
fn blocklist_item() -> BlocklistItem {
|
fn blocklist_item() -> BlocklistItem {
|
||||||
BlocklistItem {
|
BlocklistItem {
|
||||||
id: 1,
|
id: 1,
|
||||||
@@ -5151,6 +5499,10 @@ mod test {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn genres() -> Vec<String> {
|
||||||
|
vec!["cool".to_owned(), "family".to_owned(), "fun".to_owned()]
|
||||||
|
}
|
||||||
|
|
||||||
fn history_data() -> SonarrHistoryData {
|
fn history_data() -> SonarrHistoryData {
|
||||||
SonarrHistoryData {
|
SonarrHistoryData {
|
||||||
dropped_path: Some("/nfs/nzbget/completed/series/Coolness/something.cool.mkv".to_owned()),
|
dropped_path: Some("/nfs/nzbget/completed/series/Coolness/something.cool.mkv".to_owned()),
|
||||||
|
|||||||
Reference in New Issue
Block a user