test: Created snapshot tests for all Radarr UI modules

src/ui/radarr_ui/library/edit_movie_ui_tests.rs

@@ -1,19 +1,15 @@
 #[cfg(test)]
 mod tests {
-  use bimap::BiMap;
   use strum::IntoEnumIterator;
 
   use crate::app::App;
   use crate::models::BlockSelectionState;
-  use crate::models::radarr_models::Movie;
-  use crate::models::servarr_data::radarr::modals::EditMovieModal;
   use crate::models::servarr_data::radarr::radarr_data::{
     ActiveRadarrBlock, EDIT_MOVIE_BLOCKS, EDIT_MOVIE_SELECTION_BLOCKS,
   };
-  use crate::models::stateful_table::StatefulTable;
   use crate::ui::DrawUi;
   use crate::ui::radarr_ui::library::edit_movie_ui::EditMovieUi;
-  use crate::ui::ui_test_utils::test_utils::render_to_string_with_app;
+  use crate::ui::ui_test_utils::test_utils::{TerminalSize, render_to_string_with_app};
 
   #[test]
   fn test_edit_movie_ui_accepts() {
@@ -26,27 +22,39 @@ mod tests {
     });
   }
 
-  #[test]
-  fn test_edit_movie_ui_renders_edit_movie_modal() {
-    let mut app = App::test_default();
-    app.push_navigation_stack(ActiveRadarrBlock::EditMoviePathInput.into());
-    app.data.radarr_data.quality_profile_map =
-      BiMap::from_iter(vec![(1, "HD - 1080p".to_owned()), (2, "Any".to_owned())]);
-    app.data.radarr_data.selected_block = BlockSelectionState::new(EDIT_MOVIE_SELECTION_BLOCKS);
-    app.data.radarr_data.movies = StatefulTable::default();
-    app.data.radarr_data.movies.set_items(vec![Movie {
-      id: 1,
-      title: "Test Movie".into(),
-      path: "/movies/test".into(),
-      quality_profile_id: 1,
-      ..Movie::default()
-    }]);
-    app.data.radarr_data.edit_movie_modal = Some(EditMovieModal::from(&app.data.radarr_data));
+  mod snapshot_tests {
+    use rstest::rstest;
+    use super::*;
 
-    let output = render_to_string_with_app(120, 30, &mut app, |f, app| {
-      EditMovieUi::draw(f, app, f.area());
-    });
+    #[rstest]
+    #[case(ActiveRadarrBlock::EditMoviePrompt, None, 0)]
+    #[case(ActiveRadarrBlock::EditMoviePrompt, Some(ActiveRadarrBlock::MovieDetails), 0)]
+    #[case(ActiveRadarrBlock::EditMoviePrompt, Some(ActiveRadarrBlock::MovieHistory), 1)]
+    #[case(ActiveRadarrBlock::EditMoviePrompt, Some(ActiveRadarrBlock::FileInfo), 2)]
+    #[case(ActiveRadarrBlock::EditMoviePrompt, Some(ActiveRadarrBlock::Cast), 3)]
+    #[case(ActiveRadarrBlock::EditMoviePrompt, Some(ActiveRadarrBlock::Crew), 4)]
+    #[case(ActiveRadarrBlock::EditMoviePrompt, Some(ActiveRadarrBlock::ManualSearch), 5)]
+    fn test_edit_movie_ui_renders_edit_movie_modal(
+      #[case] active_radarr_block: ActiveRadarrBlock,
+      #[case] context: Option<ActiveRadarrBlock>,
+      #[case] index: usize,
+    ) {
+      let mut app = App::test_default_fully_populated();
+      app.push_navigation_stack((active_radarr_block, context).into());
+      if context.is_some() {
+        app.data.radarr_data.movie_info_tabs.set_index(index);
+      }
+      app.data.radarr_data.selected_block = BlockSelectionState::new(EDIT_MOVIE_SELECTION_BLOCKS);
 
-    insta::assert_snapshot!(output);
+      let output = render_to_string_with_app(TerminalSize::Large, &mut app, |f, app| {
+        EditMovieUi::draw(f, app, f.area());
+      });
+
+      if let Some(context) = context {
+        insta::assert_snapshot!(format!("{}_{}", active_radarr_block.to_string(), context.to_string()), output);
+      } else {
+        insta::assert_snapshot!(active_radarr_block.to_string(), output);
+      }
+    }
   }
-}
+}