From aa13735533a6ae0d0a1297c536dad12d79c88737 Mon Sep 17 00:00:00 2001
From: Alex Clarke <alex.j.tusa@gmail.com>
Date: Fri, 22 Nov 2024 17:01:00 -0700
Subject: [PATCH] feat(cli): Support for starting a Sonarr task

---
 src/cli/sonarr/mod.rs                  | 18 +++++++
 src/cli/sonarr/sonarr_command_tests.rs | 69 +++++++++++++++++++++++++-
 2 files changed, 86 insertions(+), 1 deletion(-)

diff --git a/src/cli/sonarr/mod.rs b/src/cli/sonarr/mod.rs
index 658c56e..870223e 100644
--- a/src/cli/sonarr/mod.rs
+++ b/src/cli/sonarr/mod.rs
@@ -10,6 +10,7 @@ use tokio::sync::Mutex;
 
 use crate::{
   app::App,
+  models::sonarr_models::SonarrTaskName,
   network::{sonarr_network::SonarrEvent, NetworkTrait},
 };
 
@@ -79,6 +80,16 @@ pub enum SonarrCommand {
     #[arg(long, help = "The season number to search for", required = true)]
     season_number: i64,
   },
+  #[command(about = "Start the specified Sonarr task")]
+  StartTask {
+    #[arg(
+      long,
+      help = "The name of the task to trigger",
+      value_enum,
+      required = true
+    )]
+    task_name: SonarrTaskName,
+  },
 }
 
 impl From<SonarrCommand> for Command {
@@ -167,6 +178,13 @@ impl<'a, 'b> CliCommandHandler<'a, 'b, SonarrCommand> for SonarrCliHandler<'a, '
           .await?;
         serde_json::to_string_pretty(&resp)?
       }
+      SonarrCommand::StartTask { task_name } => {
+        let resp = self
+          .network
+          .handle_network_event(SonarrEvent::StartTask(Some(task_name)).into())
+          .await?;
+        serde_json::to_string_pretty(&resp)?
+      }
     };
 
     Ok(result)
diff --git a/src/cli/sonarr/sonarr_command_tests.rs b/src/cli/sonarr/sonarr_command_tests.rs
index be2f4b5..fc0988e 100644
--- a/src/cli/sonarr/sonarr_command_tests.rs
+++ b/src/cli/sonarr/sonarr_command_tests.rs
@@ -126,6 +126,44 @@ mod tests {
 
       assert!(result.is_ok());
     }
+
+    #[rstest]
+    fn test_start_task_requires_task_name() {
+      let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "start-task"]);
+
+      assert!(result.is_err());
+      assert_eq!(
+        result.unwrap_err().kind(),
+        ErrorKind::MissingRequiredArgument
+      );
+    }
+
+    #[rstest]
+    fn test_start_task_task_name_validation() {
+      let result = Cli::command().try_get_matches_from([
+        "managarr",
+        "sonarr",
+        "start-task",
+        "--task-name",
+        "test",
+      ]);
+
+      assert!(result.is_err());
+      assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
+    }
+
+    #[test]
+    fn test_start_task_requirements_satisfied() {
+      let result = Cli::command().try_get_matches_from([
+        "managarr",
+        "sonarr",
+        "start-task",
+        "--task-name",
+        "application-update-check",
+      ]);
+
+      assert!(result.is_ok());
+    }
   }
 
   mod handler {
@@ -146,7 +184,9 @@ mod tests {
         CliCommandHandler,
       },
       models::{
-        sonarr_models::{BlocklistItem, BlocklistResponse, Series, SonarrSerdeable},
+        sonarr_models::{
+          BlocklistItem, BlocklistResponse, Series, SonarrSerdeable, SonarrTaskName,
+        },
         Serdeable,
       },
       network::{sonarr_network::SonarrEvent, MockNetworkTrait, NetworkEvent},
@@ -370,5 +410,32 @@ mod tests {
 
       assert!(result.is_ok());
     }
+
+    #[tokio::test]
+    async fn test_start_task_command() {
+      let expected_task_name = SonarrTaskName::ApplicationUpdateCheck;
+      let mut mock_network = MockNetworkTrait::new();
+      mock_network
+        .expect_handle_network_event()
+        .with(eq::<NetworkEvent>(
+          SonarrEvent::StartTask(Some(expected_task_name)).into(),
+        ))
+        .times(1)
+        .returning(|_| {
+          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
+            json!({"testResponse": "response"}),
+          )))
+        });
+      let app_arc = Arc::new(Mutex::new(App::default()));
+      let start_task_command = SonarrCommand::StartTask {
+        task_name: SonarrTaskName::ApplicationUpdateCheck,
+      };
+
+      let result = SonarrCliHandler::with(&app_arc, start_task_command, &mut mock_network)
+        .handle()
+        .await;
+
+      assert!(result.is_ok());
+    }
   }
 }