managarr/src/cli/sonarr/list_command_handler_tests.rs

#[cfg(test)]
mod tests {
  use crate::Cli;
  use crate::cli::{
    Command,
    sonarr::{SonarrCommand, list_command_handler::SonarrListCommand},
  };
  use clap::CommandFactory;
  use pretty_assertions::assert_eq;

  #[test]
  fn test_sonarr_list_command_from() {
    let command = SonarrListCommand::Series;

    let result = Command::from(command.clone());

    assert_eq!(result, Command::Sonarr(SonarrCommand::List(command)));
  }

  mod cli {
    use super::*;
    use clap::{Parser, error::ErrorKind};
    use pretty_assertions::assert_eq;
    use rstest::rstest;

    #[rstest]
    fn test_list_commands_have_no_arg_requirements(
      #[values(
        "blocklist",
        "series",
        "disk-space",
        "quality-profiles",
        "indexers",
        "queued-events",
        "root-folders",
        "tags",
        "tasks",
        "updates",
        "language-profiles"
      )]
      subcommand: &str,
    ) {
      let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "list", subcommand]);

      assert_ok!(&result);
    }

    #[test]
    fn test_list_episodes_requires_series_id() {
      let result = Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "episodes"]);

      assert_err!(&result);
      assert_eq!(
        result.unwrap_err().kind(),
        ErrorKind::MissingRequiredArgument
      );
    }

    #[test]
    fn test_list_episode_files_requires_series_id() {
      let result =
        Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "episode-files"]);

      assert_err!(&result);
      assert_eq!(
        result.unwrap_err().kind(),
        ErrorKind::MissingRequiredArgument
      );
    }

    #[test]
    fn test_list_episode_history_requires_series_id() {
      let result =
        Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "episode-history"]);

      assert_err!(&result);
      assert_eq!(
        result.unwrap_err().kind(),
        ErrorKind::MissingRequiredArgument
      );
    }

    #[test]
    fn test_list_episode_history_success() {
      let expected_args = SonarrListCommand::EpisodeHistory { episode_id: 1 };
      let result = Cli::try_parse_from([
        "managarr",
        "sonarr",
        "list",
        "episode-history",
        "--episode-id",
        "1",
      ]);

      assert_ok!(&result);

      let Some(Command::Sonarr(SonarrCommand::List(episode_history_command))) =
        result.unwrap().command
      else {
        panic!("Unexpected command type");
      };
      assert_eq!(episode_history_command, expected_args);
    }

    #[test]
    fn test_list_downloads_count_flag_requires_arguments() {
      let result =
        Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "downloads", "--count"]);

      assert_err!(&result);
      assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
    }

    #[test]
    fn test_list_downloads_default_values() {
      let expected_args = SonarrListCommand::Downloads { count: 500 };
      let result = Cli::try_parse_from(["managarr", "sonarr", "list", "downloads"]);

      assert_ok!(&result);

      let Some(Command::Sonarr(SonarrCommand::List(downloads_command))) = result.unwrap().command
      else {
        panic!("Unexpected command type");
      };
      assert_eq!(downloads_command, expected_args);
    }

    #[test]
    fn test_list_history_events_flag_requires_arguments() {
      let result =
        Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "history", "--events"]);

      assert_err!(&result);
      assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
    }

    #[test]
    fn test_list_history_default_values() {
      let expected_args = SonarrListCommand::History { events: 500 };
      let result = Cli::try_parse_from(["managarr", "sonarr", "list", "history"]);

      assert_ok!(&result);

      let Some(Command::Sonarr(SonarrCommand::List(history_command))) = result.unwrap().command
      else {
        panic!("Unexpected command type");
      };
      assert_eq!(history_command, expected_args);
    }

    #[test]
    fn test_list_logs_events_flag_requires_arguments() {
      let result =
        Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "logs", "--events"]);

      assert_err!(&result);
      assert_eq!(result.unwrap_err().kind(), ErrorKind::InvalidValue);
    }

    #[test]
    fn test_list_logs_default_values() {
      let expected_args = SonarrListCommand::Logs {
        events: 500,
        output_in_log_format: false,
      };
      let result = Cli::try_parse_from(["managarr", "sonarr", "list", "logs"]);

      assert_ok!(&result);

      let Some(Command::Sonarr(SonarrCommand::List(logs_command))) = result.unwrap().command else {
        panic!("Unexpected command type");
      };
      assert_eq!(logs_command, expected_args);
    }

    #[test]
    fn test_list_episodes_success() {
      let expected_args = SonarrListCommand::Episodes { series_id: 1 };
      let result =
        Cli::try_parse_from(["managarr", "sonarr", "list", "episodes", "--series-id", "1"]);

      assert_ok!(&result);

      let Some(Command::Sonarr(SonarrCommand::List(episodes_command))) = result.unwrap().command
      else {
        panic!("Unexpected command type");
      };
      assert_eq!(episodes_command, expected_args);
    }

    #[test]
    fn test_list_episode_files_success() {
      let expected_args = SonarrListCommand::EpisodeFiles { series_id: 1 };
      let result = Cli::try_parse_from([
        "managarr",
        "sonarr",
        "list",
        "episode-files",
        "--series-id",
        "1",
      ]);

      assert_ok!(&result);

      let Some(Command::Sonarr(SonarrCommand::List(episode_files_command))) =
        result.unwrap().command
      else {
        panic!("Unexpected command type");
      };
      assert_eq!(episode_files_command, expected_args);
    }

    #[test]
    fn test_season_history_requires_series_id() {
      let result = Cli::command().try_get_matches_from([
        "managarr",
        "sonarr",
        "list",
        "season-history",
        "--season-number",
        "1",
      ]);

      assert_err!(&result);
      assert_eq!(
        result.unwrap_err().kind(),
        ErrorKind::MissingRequiredArgument
      );
    }

    #[test]
    fn test_season_history_requires_season_number() {
      let result = Cli::command().try_get_matches_from([
        "managarr",
        "sonarr",
        "list",
        "season-history",
        "--series-id",
        "1",
      ]);

      assert_err!(&result);
      assert_eq!(
        result.unwrap_err().kind(),
        ErrorKind::MissingRequiredArgument
      );
    }

    #[test]
    fn test_season_history_requirements_satisfied() {
      let result = Cli::command().try_get_matches_from([
        "managarr",
        "sonarr",
        "list",
        "season-history",
        "--series-id",
        "1",
        "--season-number",
        "1",
      ]);

      assert_ok!(&result);
    }

    #[test]
    fn test_list_series_history_requires_series_id() {
      let result =
        Cli::command().try_get_matches_from(["managarr", "sonarr", "list", "series-history"]);

      assert_err!(&result);
      assert_eq!(
        result.unwrap_err().kind(),
        ErrorKind::MissingRequiredArgument
      );
    }

    #[test]
    fn test_list_series_history_success() {
      let expected_args = SonarrListCommand::SeriesHistory { series_id: 1 };
      let result = Cli::try_parse_from([
        "managarr",
        "sonarr",
        "list",
        "series-history",
        "--series-id",
        "1",
      ]);

      assert_ok!(&result);

      let Some(Command::Sonarr(SonarrCommand::List(series_command))) = result.unwrap().command
      else {
        panic!("Unexpected command type");
      };
      assert_eq!(series_command, expected_args);
    }
  }

  mod handler {

    use std::sync::Arc;

    use mockall::predicate::eq;
    use rstest::rstest;
    use serde_json::json;
    use tokio::sync::Mutex;

    use crate::cli::CliCommandHandler;
    use crate::cli::sonarr::list_command_handler::{SonarrListCommand, SonarrListCommandHandler};
    use crate::models::Serdeable;
    use crate::models::sonarr_models::SonarrSerdeable;
    use crate::network::sonarr_network::SonarrEvent;
    use crate::{
      app::App,
      network::{MockNetworkTrait, NetworkEvent},
    };

    #[rstest]
    #[case(SonarrListCommand::Blocklist, SonarrEvent::GetBlocklist)]
    #[case(SonarrListCommand::DiskSpace, SonarrEvent::GetDiskSpace)]
    #[case(SonarrListCommand::Indexers, SonarrEvent::GetIndexers)]
    #[case(SonarrListCommand::QualityProfiles, SonarrEvent::GetQualityProfiles)]
    #[case(SonarrListCommand::QueuedEvents, SonarrEvent::GetQueuedEvents)]
    #[case(SonarrListCommand::RootFolders, SonarrEvent::GetRootFolders)]
    #[case(SonarrListCommand::Series, SonarrEvent::ListSeries)]
    #[case(SonarrListCommand::Tags, SonarrEvent::GetTags)]
    #[case(SonarrListCommand::Tasks, SonarrEvent::GetTasks)]
    #[case(SonarrListCommand::Updates, SonarrEvent::GetUpdates)]
    #[case(SonarrListCommand::LanguageProfiles, SonarrEvent::GetLanguageProfiles)]
    #[tokio::test]
    async fn test_handle_list_command(
      #[case] list_command: SonarrListCommand,
      #[case] expected_sonarr_event: SonarrEvent,
    ) {
      let mut mock_network = MockNetworkTrait::new();
      mock_network
        .expect_handle_network_event()
        .with(eq::<NetworkEvent>(expected_sonarr_event.into()))
        .times(1)
        .returning(|_| {
          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
            json!({"testResponse": "response"}),
          )))
        });
      let app_arc = Arc::new(Mutex::new(App::test_default()));

      let result = SonarrListCommandHandler::with(&app_arc, list_command, &mut mock_network)
        .handle()
        .await;

      assert_ok!(&result);
    }

    #[tokio::test]
    async fn test_handle_list_episodes_command() {
      let expected_series_id = 1;
      let mut mock_network = MockNetworkTrait::new();
      mock_network
        .expect_handle_network_event()
        .with(eq::<NetworkEvent>(
          SonarrEvent::GetEpisodes(expected_series_id).into(),
        ))
        .times(1)
        .returning(|_| {
          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
            json!({"testResponse": "response"}),
          )))
        });
      let app_arc = Arc::new(Mutex::new(App::test_default()));
      let list_episodes_command = SonarrListCommand::Episodes { series_id: 1 };

      let result =
        SonarrListCommandHandler::with(&app_arc, list_episodes_command, &mut mock_network)
          .handle()
          .await;

      assert_ok!(&result);
    }

    #[tokio::test]
    async fn test_handle_list_episode_files_command() {
      let expected_series_id = 1;
      let mut mock_network = MockNetworkTrait::new();
      mock_network
        .expect_handle_network_event()
        .with(eq::<NetworkEvent>(
          SonarrEvent::GetEpisodeFiles(expected_series_id).into(),
        ))
        .times(1)
        .returning(|_| {
          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
            json!({"testResponse": "response"}),
          )))
        });
      let app_arc = Arc::new(Mutex::new(App::test_default()));
      let list_episode_files_command = SonarrListCommand::EpisodeFiles { series_id: 1 };

      let result =
        SonarrListCommandHandler::with(&app_arc, list_episode_files_command, &mut mock_network)
          .handle()
          .await;

      assert_ok!(&result);
    }

    #[tokio::test]
    async fn test_handle_list_downloads_command() {
      let expected_count = 1000;
      let mut mock_network = MockNetworkTrait::new();
      mock_network
        .expect_handle_network_event()
        .with(eq::<NetworkEvent>(
          SonarrEvent::GetDownloads(expected_count).into(),
        ))
        .times(1)
        .returning(|_| {
          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
            json!({"testResponse": "response"}),
          )))
        });
      let app_arc = Arc::new(Mutex::new(App::test_default()));
      let list_downloads_command = SonarrListCommand::Downloads { count: 1000 };

      let result =
        SonarrListCommandHandler::with(&app_arc, list_downloads_command, &mut mock_network)
          .handle()
          .await;

      assert_ok!(&result);
    }

    #[tokio::test]
    async fn test_handle_list_history_command() {
      let expected_events = 1000;
      let mut mock_network = MockNetworkTrait::new();
      mock_network
        .expect_handle_network_event()
        .with(eq::<NetworkEvent>(
          SonarrEvent::GetHistory(expected_events).into(),
        ))
        .times(1)
        .returning(|_| {
          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
            json!({"testResponse": "response"}),
          )))
        });
      let app_arc = Arc::new(Mutex::new(App::test_default()));
      let list_history_command = SonarrListCommand::History { events: 1000 };

      let result =
        SonarrListCommandHandler::with(&app_arc, list_history_command, &mut mock_network)
          .handle()
          .await;

      assert_ok!(&result);
    }

    #[tokio::test]
    async fn test_handle_list_logs_command() {
      let expected_events = 1000;
      let mut mock_network = MockNetworkTrait::new();
      mock_network
        .expect_handle_network_event()
        .with(eq::<NetworkEvent>(
          SonarrEvent::GetLogs(expected_events).into(),
        ))
        .times(1)
        .returning(|_| {
          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
            json!({"testResponse": "response"}),
          )))
        });
      let app_arc = Arc::new(Mutex::new(App::test_default()));
      let list_logs_command = SonarrListCommand::Logs {
        events: 1000,
        output_in_log_format: false,
      };

      let result = SonarrListCommandHandler::with(&app_arc, list_logs_command, &mut mock_network)
        .handle()
        .await;

      assert_ok!(&result);
    }

    #[tokio::test]
    async fn test_handle_list_series_history_command() {
      let expected_series_id = 1;
      let mut mock_network = MockNetworkTrait::new();
      mock_network
        .expect_handle_network_event()
        .with(eq::<NetworkEvent>(
          SonarrEvent::GetSeriesHistory(expected_series_id).into(),
        ))
        .times(1)
        .returning(|_| {
          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
            json!({"testResponse": "response"}),
          )))
        });
      let app_arc = Arc::new(Mutex::new(App::test_default()));
      let list_series_history_command = SonarrListCommand::SeriesHistory { series_id: 1 };

      let result =
        SonarrListCommandHandler::with(&app_arc, list_series_history_command, &mut mock_network)
          .handle()
          .await;

      assert_ok!(&result);
    }

    #[tokio::test]
    async fn test_handle_list_episode_history_command() {
      let expected_episode_id = 1;
      let mut mock_network = MockNetworkTrait::new();
      mock_network
        .expect_handle_network_event()
        .with(eq::<NetworkEvent>(
          SonarrEvent::GetEpisodeHistory(expected_episode_id).into(),
        ))
        .times(1)
        .returning(|_| {
          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
            json!({"testResponse": "response"}),
          )))
        });
      let app_arc = Arc::new(Mutex::new(App::test_default()));
      let list_episode_history_command = SonarrListCommand::EpisodeHistory { episode_id: 1 };

      let result =
        SonarrListCommandHandler::with(&app_arc, list_episode_history_command, &mut mock_network)
          .handle()
          .await;

      assert_ok!(&result);
    }

    #[tokio::test]
    async fn test_list_season_history_command() {
      let expected_series_id = 1;
      let expected_season_number = 1;
      let mut mock_network = MockNetworkTrait::new();
      mock_network
        .expect_handle_network_event()
        .with(eq::<NetworkEvent>(
          SonarrEvent::GetSeasonHistory((expected_series_id, expected_season_number)).into(),
        ))
        .times(1)
        .returning(|_| {
          Ok(Serdeable::Sonarr(SonarrSerdeable::Value(
            json!({"testResponse": "response"}),
          )))
        });
      let app_arc = Arc::new(Mutex::new(App::test_default()));
      let list_season_history_command = SonarrListCommand::SeasonHistory {
        series_id: 1,
        season_number: 1,
      };

      let result =
        SonarrListCommandHandler::with(&app_arc, list_season_history_command, &mut mock_network)
          .handle()
          .await;

      assert_ok!(&result);
    }
  }
}