managarr/src/handlers/sonarr_handlers/library/delete_series_handler_tests.rs

#[cfg(test)]
mod tests {
  use pretty_assertions::assert_eq;
  use rstest::rstest;
  use strum::IntoEnumIterator;

  use crate::app::App;
  use crate::app::key_binding::DEFAULT_KEYBINDINGS;
  use crate::event::Key;
  use crate::handlers::KeyEventHandler;
  use crate::handlers::sonarr_handlers::library::delete_series_handler::DeleteSeriesHandler;
  use crate::handlers::sonarr_handlers::sonarr_handler_test_utils::utils::series;
  use crate::models::servarr_data::sonarr::sonarr_data::{ActiveSonarrBlock, DELETE_SERIES_BLOCKS};
  use crate::models::sonarr_models::{DeleteSeriesParams, Series};

  mod test_handle_scroll_up_and_down {
    use pretty_assertions::assert_eq;
    use rstest::rstest;

    use crate::models::BlockSelectionState;
    use crate::models::servarr_data::sonarr::sonarr_data::DELETE_SERIES_SELECTION_BLOCKS;

    use super::*;

    #[rstest]
    fn test_delete_series_prompt_scroll(#[values(Key::Up, Key::Down)] key: Key) {
      let mut app = App::test_default();
      app.data.sonarr_data.selected_block =
        BlockSelectionState::new(DELETE_SERIES_SELECTION_BLOCKS);
      app.data.sonarr_data.selected_block.down();

      DeleteSeriesHandler::new(key, &mut app, ActiveSonarrBlock::DeleteSeriesPrompt, None).handle();

      if key == Key::Up {
        assert_eq!(
          app.data.sonarr_data.selected_block.get_active_block(),
          ActiveSonarrBlock::DeleteSeriesToggleDeleteFile
        );
      } else {
        assert_eq!(
          app.data.sonarr_data.selected_block.get_active_block(),
          ActiveSonarrBlock::DeleteSeriesConfirmPrompt
        );
      }
    }

    #[rstest]
    fn test_delete_series_prompt_scroll_no_op_when_not_ready(
      #[values(Key::Up, Key::Down)] key: Key,
    ) {
      let mut app = App::test_default();
      app.is_loading = true;
      app.data.sonarr_data.selected_block =
        BlockSelectionState::new(DELETE_SERIES_SELECTION_BLOCKS);
      app.data.sonarr_data.selected_block.down();

      DeleteSeriesHandler::new(key, &mut app, ActiveSonarrBlock::DeleteSeriesPrompt, None).handle();

      assert_eq!(
        app.data.sonarr_data.selected_block.get_active_block(),
        ActiveSonarrBlock::DeleteSeriesToggleAddListExclusion
      );
    }
  }

  mod test_handle_left_right_action {
    use rstest::rstest;

    use super::*;

    #[rstest]
    fn test_left_right_prompt_toggle(#[values(Key::Left, Key::Right)] key: Key) {
      let mut app = App::test_default();
      app.push_navigation_stack(ActiveSonarrBlock::DeleteSeriesPrompt.into());

      DeleteSeriesHandler::new(key, &mut app, ActiveSonarrBlock::DeleteSeriesPrompt, None).handle();

      assert!(app.data.sonarr_data.prompt_confirm);

      DeleteSeriesHandler::new(key, &mut app, ActiveSonarrBlock::DeleteSeriesPrompt, None).handle();

      assert!(!app.data.sonarr_data.prompt_confirm);
    }
  }

  mod test_handle_submit {
    use pretty_assertions::assert_eq;

    use crate::models::BlockSelectionState;
    use crate::models::servarr_data::sonarr::sonarr_data::DELETE_SERIES_SELECTION_BLOCKS;
    use crate::network::sonarr_network::SonarrEvent;

    use super::*;
    use crate::assert_navigation_popped;

    const SUBMIT_KEY: Key = DEFAULT_KEYBINDINGS.submit.key;

    #[test]
    fn test_delete_series_prompt_prompt_decline_submit() {
      let mut app = App::test_default();
      app.push_navigation_stack(ActiveSonarrBlock::Series.into());
      app.push_navigation_stack(ActiveSonarrBlock::DeleteSeriesPrompt.into());
      app.data.sonarr_data.selected_block =
        BlockSelectionState::new(DELETE_SERIES_SELECTION_BLOCKS);
      app
        .data
        .sonarr_data
        .selected_block
        .set_index(0, DELETE_SERIES_SELECTION_BLOCKS.len() - 1);
      app.data.sonarr_data.delete_series_files = true;
      app.data.sonarr_data.add_list_exclusion = true;

      DeleteSeriesHandler::new(
        SUBMIT_KEY,
        &mut app,
        ActiveSonarrBlock::DeleteSeriesPrompt,
        None,
      )
      .handle();

      assert_navigation_popped!(app, ActiveSonarrBlock::Series.into());
      assert_none!(app.data.sonarr_data.prompt_confirm_action);
      assert!(!app.data.sonarr_data.prompt_confirm);
      assert!(!app.data.sonarr_data.delete_series_files);
      assert!(!app.data.sonarr_data.add_list_exclusion);
    }

    #[test]
    fn test_delete_series_confirm_prompt_prompt_confirmation_submit() {
      let mut app = App::test_default();
      app.push_navigation_stack(ActiveSonarrBlock::Series.into());
      app.push_navigation_stack(ActiveSonarrBlock::DeleteSeriesPrompt.into());
      app.data.sonarr_data.prompt_confirm = true;
      app.data.sonarr_data.delete_series_files = true;
      app.data.sonarr_data.add_list_exclusion = true;
      app.data.sonarr_data.series.set_items(vec![
        Series {
          id: 999,
          ..series()
        },
        series(),
      ]);
      app.data.sonarr_data.series.select_index(Some(1));
      let expected_delete_series_params = DeleteSeriesParams {
        id: 1,
        delete_series_files: true,
        add_list_exclusion: true,
      };
      app.data.sonarr_data.selected_block =
        BlockSelectionState::new(DELETE_SERIES_SELECTION_BLOCKS);
      app
        .data
        .sonarr_data
        .selected_block
        .set_index(0, DELETE_SERIES_SELECTION_BLOCKS.len() - 1);

      DeleteSeriesHandler::new(
        SUBMIT_KEY,
        &mut app,
        ActiveSonarrBlock::DeleteSeriesPrompt,
        None,
      )
      .handle();

      assert_navigation_popped!(app, ActiveSonarrBlock::Series.into());
      assert_eq!(
        app.data.sonarr_data.prompt_confirm_action,
        Some(SonarrEvent::DeleteSeries(expected_delete_series_params))
      );
      assert!(app.should_refresh);
      assert!(app.data.sonarr_data.prompt_confirm);
      assert!(!app.data.sonarr_data.delete_series_files);
      assert!(!app.data.sonarr_data.add_list_exclusion);
    }

    #[test]
    fn test_delete_series_confirm_prompt_prompt_confirmation_submit_no_op_when_not_ready() {
      let mut app = App::test_default();
      app.is_loading = true;
      app.push_navigation_stack(ActiveSonarrBlock::Series.into());
      app.push_navigation_stack(ActiveSonarrBlock::DeleteSeriesPrompt.into());
      app.data.sonarr_data.prompt_confirm = true;
      app.data.sonarr_data.delete_series_files = true;
      app.data.sonarr_data.add_list_exclusion = true;

      DeleteSeriesHandler::new(
        SUBMIT_KEY,
        &mut app,
        ActiveSonarrBlock::DeleteSeriesPrompt,
        None,
      )
      .handle();

      assert_eq!(
        app.get_current_route(),
        ActiveSonarrBlock::DeleteSeriesPrompt.into()
      );
      assert_none!(app.data.sonarr_data.prompt_confirm_action);
      assert!(!app.should_refresh);
      assert!(app.data.sonarr_data.prompt_confirm);
      assert!(app.data.sonarr_data.delete_series_files);
      assert!(app.data.sonarr_data.add_list_exclusion);
    }

    #[test]
    fn test_delete_series_toggle_delete_files_submit() {
      let current_route = ActiveSonarrBlock::DeleteSeriesPrompt.into();
      let mut app = App::test_default();
      app.data.sonarr_data.selected_block =
        BlockSelectionState::new(DELETE_SERIES_SELECTION_BLOCKS);
      app.push_navigation_stack(ActiveSonarrBlock::DeleteSeriesPrompt.into());

      DeleteSeriesHandler::new(
        SUBMIT_KEY,
        &mut app,
        ActiveSonarrBlock::DeleteSeriesPrompt,
        None,
      )
      .handle();

      assert_eq!(app.get_current_route(), current_route);
      assert_eq!(app.data.sonarr_data.delete_series_files, true);

      DeleteSeriesHandler::new(
        SUBMIT_KEY,
        &mut app,
        ActiveSonarrBlock::DeleteSeriesPrompt,
        None,
      )
      .handle();

      assert_eq!(app.get_current_route(), current_route);
      assert_eq!(app.data.sonarr_data.delete_series_files, false);
    }
  }

  mod test_handle_esc {
    use super::*;
    use crate::assert_navigation_popped;
    use rstest::rstest;

    const ESC_KEY: Key = DEFAULT_KEYBINDINGS.esc.key;

    #[rstest]
    fn test_delete_series_prompt_esc(#[values(true, false)] is_ready: bool) {
      let mut app = App::test_default();
      app.is_loading = is_ready;
      app.push_navigation_stack(ActiveSonarrBlock::Series.into());
      app.push_navigation_stack(ActiveSonarrBlock::DeleteSeriesPrompt.into());
      app.data.sonarr_data.prompt_confirm = true;
      app.data.sonarr_data.delete_series_files = true;
      app.data.sonarr_data.add_list_exclusion = true;

      DeleteSeriesHandler::new(
        ESC_KEY,
        &mut app,
        ActiveSonarrBlock::DeleteSeriesPrompt,
        None,
      )
      .handle();

      assert_navigation_popped!(app, ActiveSonarrBlock::Series.into());
      assert!(!app.data.sonarr_data.prompt_confirm);
      assert!(!app.data.sonarr_data.delete_series_files);
      assert!(!app.data.sonarr_data.add_list_exclusion);
    }
  }

  mod test_handle_key_char {
    use crate::{
      assert_navigation_popped,
      models::{
        BlockSelectionState, servarr_data::sonarr::sonarr_data::DELETE_SERIES_SELECTION_BLOCKS,
      },
      network::sonarr_network::SonarrEvent,
    };
    use pretty_assertions::assert_eq;

    use super::*;

    #[test]
    fn test_delete_series_confirm_prompt_prompt_confirm() {
      let mut app = App::test_default();
      app.push_navigation_stack(ActiveSonarrBlock::Series.into());
      app.push_navigation_stack(ActiveSonarrBlock::DeleteSeriesPrompt.into());
      app.data.sonarr_data.delete_series_files = true;
      app.data.sonarr_data.add_list_exclusion = true;
      app.data.sonarr_data.series.set_items(vec![
        Series {
          id: 999,
          ..series()
        },
        series(),
      ]);
      app.data.sonarr_data.series.select_index(Some(1));
      let expected_delete_series_params = DeleteSeriesParams {
        id: 1,
        delete_series_files: true,
        add_list_exclusion: true,
      };
      app.data.sonarr_data.selected_block =
        BlockSelectionState::new(DELETE_SERIES_SELECTION_BLOCKS);
      app
        .data
        .sonarr_data
        .selected_block
        .set_index(0, DELETE_SERIES_SELECTION_BLOCKS.len() - 1);

      DeleteSeriesHandler::new(
        DEFAULT_KEYBINDINGS.confirm.key,
        &mut app,
        ActiveSonarrBlock::DeleteSeriesPrompt,
        None,
      )
      .handle();

      assert_navigation_popped!(app, ActiveSonarrBlock::Series.into());
      assert_eq!(
        app.data.sonarr_data.prompt_confirm_action,
        Some(SonarrEvent::DeleteSeries(expected_delete_series_params))
      );
      assert!(app.should_refresh);
      assert!(app.data.sonarr_data.prompt_confirm);
      assert!(!app.data.sonarr_data.delete_series_files);
      assert!(!app.data.sonarr_data.add_list_exclusion);
    }
  }

  #[test]
  fn test_delete_series_handler_accepts() {
    ActiveSonarrBlock::iter().for_each(|active_sonarr_block| {
      if DELETE_SERIES_BLOCKS.contains(&active_sonarr_block) {
        assert!(DeleteSeriesHandler::accepts(active_sonarr_block));
      } else {
        assert!(!DeleteSeriesHandler::accepts(active_sonarr_block));
      }
    });
  }

  #[rstest]
  fn test_delete_series_handler_ignore_special_keys(
    #[values(true, false)] ignore_special_keys_for_textbox_input: bool,
  ) {
    let mut app = App::test_default();
    app.ignore_special_keys_for_textbox_input = ignore_special_keys_for_textbox_input;
    let handler = DeleteSeriesHandler::new(
      DEFAULT_KEYBINDINGS.esc.key,
      &mut app,
      ActiveSonarrBlock::default(),
      None,
    );

    assert_eq!(
      handler.ignore_special_keys(),
      ignore_special_keys_for_textbox_input
    );
  }

  #[test]
  fn test_build_delete_series_params() {
    let mut app = App::test_default();
    app.data.sonarr_data.series.set_items(vec![
      Series {
        id: 999,
        ..series()
      },
      series(),
    ]);
    app.data.sonarr_data.series.select_index(Some(1));
    app.data.sonarr_data.delete_series_files = true;
    app.data.sonarr_data.add_list_exclusion = true;
    let expected_delete_series_params = DeleteSeriesParams {
      id: 1,
      delete_series_files: true,
      add_list_exclusion: true,
    };

    let delete_series_params = DeleteSeriesHandler::new(
      DEFAULT_KEYBINDINGS.esc.key,
      &mut app,
      ActiveSonarrBlock::DeleteSeriesPrompt,
      None,
    )
    .build_delete_series_params();

    assert_eq!(delete_series_params, expected_delete_series_params);
    assert!(!app.data.sonarr_data.delete_series_files);
    assert!(!app.data.sonarr_data.add_list_exclusion);
  }

  #[test]
  fn test_delete_series_handler_not_ready_when_loading() {
    let mut app = App::test_default();
    app.is_loading = true;

    let handler = DeleteSeriesHandler::new(
      DEFAULT_KEYBINDINGS.esc.key,
      &mut app,
      ActiveSonarrBlock::DeleteSeriesPrompt,
      None,
    );

    assert!(!handler.is_ready());
  }

  #[test]
  fn test_delete_series_handler_ready_when_not_loading() {
    let mut app = App::test_default();
    app.is_loading = false;

    let handler = DeleteSeriesHandler::new(
      DEFAULT_KEYBINDINGS.esc.key,
      &mut app,
      ActiveSonarrBlock::DeleteSeriesPrompt,
      None,
    );

    assert!(handler.is_ready());
  }
}