fix(sonarr): Provide the task name directly alongside all StartTask events when publishing to the networking channel
This commit is contained in:
@@ -252,7 +252,7 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrCommand> for SonarrCliHandler<'a, '
|
|||||||
SonarrCommand::StartTask { task_name } => {
|
SonarrCommand::StartTask { task_name } => {
|
||||||
let resp = self
|
let resp = self
|
||||||
.network
|
.network
|
||||||
.handle_network_event(SonarrEvent::StartTask(Some(task_name)).into())
|
.handle_network_event(SonarrEvent::StartTask(task_name).into())
|
||||||
.await?;
|
.await?;
|
||||||
serde_json::to_string_pretty(&resp)?
|
serde_json::to_string_pretty(&resp)?
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -624,7 +624,7 @@ mod tests {
|
|||||||
mock_network
|
mock_network
|
||||||
.expect_handle_network_event()
|
.expect_handle_network_event()
|
||||||
.with(eq::<NetworkEvent>(
|
.with(eq::<NetworkEvent>(
|
||||||
SonarrEvent::StartTask(Some(expected_task_name)).into(),
|
SonarrEvent::StartTask(expected_task_name).into(),
|
||||||
))
|
))
|
||||||
.times(1)
|
.times(1)
|
||||||
.returning(|_| {
|
.returning(|_| {
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ use crate::app::App;
|
|||||||
use crate::event::Key;
|
use crate::event::Key;
|
||||||
use crate::handlers::{handle_prompt_toggle, KeyEventHandler};
|
use crate::handlers::{handle_prompt_toggle, KeyEventHandler};
|
||||||
use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, SYSTEM_DETAILS_BLOCKS};
|
use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, SYSTEM_DETAILS_BLOCKS};
|
||||||
|
use crate::models::sonarr_models::SonarrTaskName;
|
||||||
use crate::models::stateful_list::StatefulList;
|
use crate::models::stateful_list::StatefulList;
|
||||||
use crate::models::Scrollable;
|
use crate::models::Scrollable;
|
||||||
use crate::network::sonarr_network::SonarrEvent;
|
use crate::network::sonarr_network::SonarrEvent;
|
||||||
@@ -18,6 +19,18 @@ pub(super) struct SystemDetailsHandler<'a, 'b> {
|
|||||||
_context: Option<ActiveSonarrBlock>,
|
_context: Option<ActiveSonarrBlock>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl<'a, 'b> SystemDetailsHandler<'a, 'b> {
|
||||||
|
fn extract_task_name(&self) -> SonarrTaskName {
|
||||||
|
self
|
||||||
|
.app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.tasks
|
||||||
|
.current_selection()
|
||||||
|
.task_name
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for SystemDetailsHandler<'a, 'b> {
|
impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for SystemDetailsHandler<'a, 'b> {
|
||||||
fn accepts(active_block: ActiveSonarrBlock) -> bool {
|
fn accepts(active_block: ActiveSonarrBlock) -> bool {
|
||||||
SYSTEM_DETAILS_BLOCKS.contains(&active_block)
|
SYSTEM_DETAILS_BLOCKS.contains(&active_block)
|
||||||
@@ -137,7 +150,8 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for SystemDetailsHandler
|
|||||||
}
|
}
|
||||||
ActiveSonarrBlock::SystemTaskStartConfirmPrompt => {
|
ActiveSonarrBlock::SystemTaskStartConfirmPrompt => {
|
||||||
if self.app.data.sonarr_data.prompt_confirm {
|
if self.app.data.sonarr_data.prompt_confirm {
|
||||||
self.app.data.sonarr_data.prompt_confirm_action = Some(SonarrEvent::StartTask(None));
|
self.app.data.sonarr_data.prompt_confirm_action =
|
||||||
|
Some(SonarrEvent::StartTask(self.extract_task_name()));
|
||||||
}
|
}
|
||||||
|
|
||||||
self.app.pop_navigation_stack();
|
self.app.pop_navigation_stack();
|
||||||
@@ -174,7 +188,8 @@ impl<'a, 'b> KeyEventHandler<'a, 'b, ActiveSonarrBlock> for SystemDetailsHandler
|
|||||||
&& self.key == DEFAULT_KEYBINDINGS.confirm.key
|
&& self.key == DEFAULT_KEYBINDINGS.confirm.key
|
||||||
{
|
{
|
||||||
self.app.data.sonarr_data.prompt_confirm = true;
|
self.app.data.sonarr_data.prompt_confirm = true;
|
||||||
self.app.data.sonarr_data.prompt_confirm_action = Some(SonarrEvent::StartTask(None));
|
self.app.data.sonarr_data.prompt_confirm_action =
|
||||||
|
Some(SonarrEvent::StartTask(self.extract_task_name()));
|
||||||
self.app.pop_navigation_stack();
|
self.app.pop_navigation_stack();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use pretty_assertions::assert_str_eq;
|
use pretty_assertions::{assert_eq, assert_str_eq};
|
||||||
use strum::IntoEnumIterator;
|
use strum::IntoEnumIterator;
|
||||||
|
|
||||||
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
use crate::app::key_binding::DEFAULT_KEYBINDINGS;
|
||||||
@@ -12,10 +12,11 @@ mod tests {
|
|||||||
ActiveSonarrBlock, SYSTEM_DETAILS_BLOCKS,
|
ActiveSonarrBlock, SYSTEM_DETAILS_BLOCKS,
|
||||||
};
|
};
|
||||||
use crate::models::servarr_models::QueueEvent;
|
use crate::models::servarr_models::QueueEvent;
|
||||||
use crate::models::sonarr_models::SonarrTask;
|
use crate::models::sonarr_models::{SonarrTask, SonarrTaskName};
|
||||||
use crate::models::{HorizontallyScrollableText, ScrollableText};
|
use crate::models::{HorizontallyScrollableText, ScrollableText};
|
||||||
|
|
||||||
mod test_handle_scroll_up_and_down {
|
mod test_handle_scroll_up_and_down {
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
|
|
||||||
use crate::models::{HorizontallyScrollableText, ScrollableText};
|
use crate::models::{HorizontallyScrollableText, ScrollableText};
|
||||||
@@ -245,6 +246,7 @@ mod tests {
|
|||||||
use crate::{extended_stateful_iterable_vec, test_iterable_home_and_end};
|
use crate::{extended_stateful_iterable_vec, test_iterable_home_and_end};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
|
|
||||||
test_iterable_home_and_end!(
|
test_iterable_home_and_end!(
|
||||||
test_log_details_home_end,
|
test_log_details_home_end,
|
||||||
@@ -693,6 +695,11 @@ mod tests {
|
|||||||
let mut app = App::default();
|
let mut app = App::default();
|
||||||
app.data.sonarr_data.updates = ScrollableText::with_string("Test".to_owned());
|
app.data.sonarr_data.updates = ScrollableText::with_string("Test".to_owned());
|
||||||
app.data.sonarr_data.prompt_confirm = true;
|
app.data.sonarr_data.prompt_confirm = true;
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.tasks
|
||||||
|
.set_items(vec![SonarrTask::default()]);
|
||||||
app.push_navigation_stack(ActiveSonarrBlock::SystemTasks.into());
|
app.push_navigation_stack(ActiveSonarrBlock::SystemTasks.into());
|
||||||
app.push_navigation_stack(ActiveSonarrBlock::SystemTaskStartConfirmPrompt.into());
|
app.push_navigation_stack(ActiveSonarrBlock::SystemTaskStartConfirmPrompt.into());
|
||||||
|
|
||||||
@@ -707,7 +714,7 @@ mod tests {
|
|||||||
assert!(app.data.sonarr_data.prompt_confirm);
|
assert!(app.data.sonarr_data.prompt_confirm);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.data.sonarr_data.prompt_confirm_action,
|
app.data.sonarr_data.prompt_confirm_action,
|
||||||
Some(SonarrEvent::StartTask(None))
|
Some(SonarrEvent::StartTask(SonarrTaskName::default()))
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
@@ -848,6 +855,7 @@ mod tests {
|
|||||||
}
|
}
|
||||||
|
|
||||||
mod test_handle_key_char {
|
mod test_handle_key_char {
|
||||||
|
use pretty_assertions::assert_eq;
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
|
|
||||||
use crate::network::sonarr_network::SonarrEvent;
|
use crate::network::sonarr_network::SonarrEvent;
|
||||||
@@ -914,6 +922,11 @@ mod tests {
|
|||||||
let mut app = App::default();
|
let mut app = App::default();
|
||||||
app.push_navigation_stack(ActiveSonarrBlock::System.into());
|
app.push_navigation_stack(ActiveSonarrBlock::System.into());
|
||||||
app.data.sonarr_data.updates = ScrollableText::with_string("Test".to_owned());
|
app.data.sonarr_data.updates = ScrollableText::with_string("Test".to_owned());
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.tasks
|
||||||
|
.set_items(vec![SonarrTask::default()]);
|
||||||
app.push_navigation_stack(ActiveSonarrBlock::SystemTasks.into());
|
app.push_navigation_stack(ActiveSonarrBlock::SystemTasks.into());
|
||||||
app.push_navigation_stack(ActiveSonarrBlock::SystemTaskStartConfirmPrompt.into());
|
app.push_navigation_stack(ActiveSonarrBlock::SystemTaskStartConfirmPrompt.into());
|
||||||
|
|
||||||
@@ -928,7 +941,7 @@ mod tests {
|
|||||||
assert!(app.data.sonarr_data.prompt_confirm);
|
assert!(app.data.sonarr_data.prompt_confirm);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.data.sonarr_data.prompt_confirm_action,
|
app.data.sonarr_data.prompt_confirm_action,
|
||||||
Some(SonarrEvent::StartTask(None))
|
Some(SonarrEvent::StartTask(SonarrTaskName::default()))
|
||||||
);
|
);
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
app.get_current_route(),
|
app.get_current_route(),
|
||||||
@@ -948,6 +961,26 @@ mod tests {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_extract_task_name() {
|
||||||
|
let mut app = App::default();
|
||||||
|
app
|
||||||
|
.data
|
||||||
|
.sonarr_data
|
||||||
|
.tasks
|
||||||
|
.set_items(vec![SonarrTask::default()]);
|
||||||
|
|
||||||
|
let task_name = SystemDetailsHandler::with(
|
||||||
|
DEFAULT_KEYBINDINGS.esc.key,
|
||||||
|
&mut app,
|
||||||
|
ActiveSonarrBlock::SystemTasks,
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
.extract_task_name();
|
||||||
|
|
||||||
|
assert_eq!(task_name, SonarrTaskName::default());
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_system_details_handler_not_ready_when_loading() {
|
fn test_system_details_handler_not_ready_when_loading() {
|
||||||
let mut app = App::default();
|
let mut app = App::default();
|
||||||
|
|||||||
@@ -83,7 +83,7 @@ pub enum SonarrEvent {
|
|||||||
ListSeries,
|
ListSeries,
|
||||||
MarkHistoryItemAsFailed(i64),
|
MarkHistoryItemAsFailed(i64),
|
||||||
SearchNewSeries(String),
|
SearchNewSeries(String),
|
||||||
StartTask(Option<SonarrTaskName>),
|
StartTask(SonarrTaskName),
|
||||||
TestIndexer(Option<i64>),
|
TestIndexer(Option<i64>),
|
||||||
TestAllIndexers,
|
TestAllIndexers,
|
||||||
ToggleSeasonMonitoring(Option<(i64, i64)>),
|
ToggleSeasonMonitoring(Option<(i64, i64)>),
|
||||||
@@ -2025,23 +2025,9 @@ impl<'a, 'b> Network<'a, 'b> {
|
|||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn start_sonarr_task(&mut self, task: Option<SonarrTaskName>) -> Result<Value> {
|
async fn start_sonarr_task(&mut self, task: SonarrTaskName) -> Result<Value> {
|
||||||
let event = SonarrEvent::StartTask(None);
|
let event = SonarrEvent::StartTask(task);
|
||||||
let task_name = if let Some(t_name) = task {
|
let task_name = task.to_string();
|
||||||
t_name
|
|
||||||
} else {
|
|
||||||
self
|
|
||||||
.app
|
|
||||||
.lock()
|
|
||||||
.await
|
|
||||||
.data
|
|
||||||
.sonarr_data
|
|
||||||
.tasks
|
|
||||||
.current_selection()
|
|
||||||
.task_name
|
|
||||||
}
|
|
||||||
.to_string();
|
|
||||||
|
|
||||||
info!("Starting Sonarr task: {task_name}");
|
info!("Starting Sonarr task: {task_name}");
|
||||||
|
|
||||||
let body = CommandBody { name: task_name };
|
let body = CommandBody { name: task_name };
|
||||||
|
|||||||
@@ -192,7 +192,7 @@ mod test {
|
|||||||
fn test_resource_command(
|
fn test_resource_command(
|
||||||
#[values(
|
#[values(
|
||||||
SonarrEvent::GetQueuedEvents,
|
SonarrEvent::GetQueuedEvents,
|
||||||
SonarrEvent::StartTask(None),
|
SonarrEvent::StartTask(SonarrTaskName::default()),
|
||||||
SonarrEvent::TriggerAutomaticEpisodeSearch(None),
|
SonarrEvent::TriggerAutomaticEpisodeSearch(None),
|
||||||
SonarrEvent::TriggerAutomaticSeasonSearch(None),
|
SonarrEvent::TriggerAutomaticSeasonSearch(None),
|
||||||
SonarrEvent::TriggerAutomaticSeriesSearch(None),
|
SonarrEvent::TriggerAutomaticSeriesSearch(None),
|
||||||
@@ -4763,7 +4763,7 @@ mod test {
|
|||||||
})),
|
})),
|
||||||
Some(response.clone()),
|
Some(response.clone()),
|
||||||
None,
|
None,
|
||||||
SonarrEvent::StartTask(None),
|
SonarrEvent::StartTask(SonarrTaskName::ApplicationUpdateCheck),
|
||||||
None,
|
None,
|
||||||
None,
|
None,
|
||||||
)
|
)
|
||||||
@@ -4781,34 +4781,9 @@ mod test {
|
|||||||
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
||||||
|
|
||||||
if let SonarrSerdeable::Value(value) = network
|
if let SonarrSerdeable::Value(value) = network
|
||||||
.handle_sonarr_event(SonarrEvent::StartTask(None))
|
.handle_sonarr_event(SonarrEvent::StartTask(
|
||||||
.await
|
SonarrTaskName::ApplicationUpdateCheck,
|
||||||
.unwrap()
|
))
|
||||||
{
|
|
||||||
async_server.assert_async().await;
|
|
||||||
assert_eq!(value, response);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[tokio::test]
|
|
||||||
async fn test_handle_start_sonarr_task_event_uses_provided_task_name() {
|
|
||||||
let response = json!({ "test": "test"});
|
|
||||||
let (async_server, app_arc, _server) = mock_servarr_api(
|
|
||||||
RequestMethod::Post,
|
|
||||||
Some(json!({
|
|
||||||
"name": "ApplicationUpdateCheck"
|
|
||||||
})),
|
|
||||||
Some(response.clone()),
|
|
||||||
None,
|
|
||||||
SonarrEvent::StartTask(None),
|
|
||||||
None,
|
|
||||||
None,
|
|
||||||
)
|
|
||||||
.await;
|
|
||||||
let mut network = Network::new(&app_arc, CancellationToken::new(), Client::new());
|
|
||||||
|
|
||||||
if let SonarrSerdeable::Value(value) = network
|
|
||||||
.handle_sonarr_event(SonarrEvent::StartTask(Some(SonarrTaskName::default())))
|
|
||||||
.await
|
.await
|
||||||
.unwrap()
|
.unwrap()
|
||||||
{
|
{
|
||||||
|
|||||||
Reference in New Issue
Block a user