Added tests for models

2023-08-08 10:50:05 -06:00
parent a939346d30
commit 4962f3d5c3
3 changed files with 841 additions and 460 deletions
@@ -497,6 +497,7 @@ pub mod radarr_test_utils {
 }

 #[cfg(test)]
+mod tests {
  mod radarr_data_tests {
    use pretty_assertions::assert_eq;

@@ -539,7 +540,6 @@ mod radarr_data_tests {
    }
  }

-#[cfg(test)]
  mod active_radarr_block_tests {
    use pretty_assertions::assert_eq;

@@ -596,8 +596,7 @@ mod active_radarr_block_tests {
    }
  }

-#[cfg(test)]
-mod tests {
+  mod radarr_tests {
    use std::time::Duration;

    use pretty_assertions::assert_eq;
@@ -1036,3 +1035,4 @@ mod tests {
      (app, sync_network_rx)
    }
  }
+}
@@ -29,20 +29,12 @@ pub trait Scrollable {

 macro_rules! stateful_iterable {
  ($name:ident, $state:ty) => {
+    #[derive(Default)]
    pub struct $name<T> {
      pub state: $state,
      pub items: Vec<T>,
    }

-    impl<T> Default for $name<T> {
-      fn default() -> $name<T> {
-        $name {
-          state: <$state>::default(),
-          items: Vec::new(),
-        }
-      }
-    }
-
    impl<T> Scrollable for $name<T> {
      fn scroll_down(&mut self) {
        let selected_row = match self.state.selected() {
@@ -147,7 +139,7 @@ impl ScrollableText {

 impl Scrollable for ScrollableText {
  fn scroll_down(&mut self) {
-    if self.offset < self.items.len() as u16 {
+    if self.offset < (self.items.len() - 1) as u16 {
      self.offset += 1;
    }
  }
@@ -228,7 +220,7 @@ impl HorizontallyScrollableText {
  }
 }

-#[derive(Clone)]
+#[derive(Clone, PartialEq, Eq, Debug)]
 pub struct TabRoute {
  pub title: String,
  pub route: Route,
@@ -275,3 +267,392 @@ impl TabState {
    }
  }
 }
+
+#[cfg(test)]
+mod tests {
+  use std::cell::RefCell;
+
+  use pretty_assertions::{assert_eq, assert_str_eq};
+
+  use crate::app::radarr::ActiveRadarrBlock;
+  use crate::models::{
+    HorizontallyScrollableText, Scrollable, ScrollableText, StatefulTable, TabRoute, TabState,
+  };
+
+  #[test]
+  fn test_stateful_table_scroll() {
+    let mut stateful_table = create_test_stateful_table();
+
+    assert_eq!(stateful_table.state.selected(), Some(0));
+
+    stateful_table.scroll_down();
+
+    assert_eq!(stateful_table.state.selected(), Some(1));
+
+    stateful_table.scroll_down();
+
+    assert_eq!(stateful_table.state.selected(), Some(0));
+
+    stateful_table.scroll_up();
+
+    assert_eq!(stateful_table.state.selected(), Some(1));
+
+    stateful_table.scroll_up();
+
+    assert_eq!(stateful_table.state.selected(), Some(0));
+
+    stateful_table.scroll_to_bottom();
+
+    assert_eq!(stateful_table.state.selected(), Some(1));
+
+    stateful_table.scroll_to_top();
+
+    assert_eq!(stateful_table.state.selected(), Some(0));
+  }
+
+  #[test]
+  fn test_stateful_table_set_items() {
+    let items_vec = vec!["Test 1", "Test 2", "Test 3"];
+    let mut stateful_table: StatefulTable<&str> = StatefulTable::default();
+
+    stateful_table.set_items(items_vec.clone());
+
+    assert_eq!(stateful_table.state.selected(), Some(0));
+
+    stateful_table.state.select(Some(1));
+    stateful_table.set_items(items_vec.clone());
+
+    assert_eq!(stateful_table.state.selected(), Some(1));
+
+    stateful_table.state.select(Some(3));
+    stateful_table.set_items(items_vec);
+
+    assert_eq!(stateful_table.state.selected(), Some(2));
+  }
+
+  #[test]
+  fn test_stateful_table_current_selection() {
+    let mut stateful_table = create_test_stateful_table();
+
+    assert_str_eq!(stateful_table.current_selection(), &stateful_table.items[0]);
+
+    stateful_table.state.select(Some(1));
+
+    assert_str_eq!(stateful_table.current_selection(), &stateful_table.items[1]);
+  }
+
+  #[test]
+  fn test_stateful_table_current_selection_clone() {
+    let mut stateful_table = create_test_stateful_table();
+
+    assert_str_eq!(
+      stateful_table.current_selection_clone(),
+      stateful_table.items[0]
+    );
+
+    stateful_table.state.select(Some(1));
+
+    assert_str_eq!(
+      stateful_table.current_selection_clone(),
+      stateful_table.items[1]
+    );
+  }
+
+  #[test]
+  fn test_stateful_table_select_index() {
+    let mut stateful_table = create_test_stateful_table();
+
+    assert_eq!(stateful_table.state.selected(), Some(0));
+
+    stateful_table.select_index(Some(1));
+
+    assert_eq!(stateful_table.state.selected(), Some(1));
+
+    stateful_table.select_index(None);
+
+    assert_eq!(stateful_table.state.selected(), None);
+  }
+
+  #[test]
+  fn test_stateful_table_scroll_up() {
+    let mut stateful_table = create_test_stateful_table();
+
+    assert_eq!(stateful_table.state.selected(), Some(0));
+
+    stateful_table.scroll_up();
+
+    assert_eq!(stateful_table.state.selected(), Some(1));
+
+    stateful_table.scroll_up();
+
+    assert_eq!(stateful_table.state.selected(), Some(0));
+  }
+
+  #[test]
+  fn test_scrollable_text_with_string() {
+    let scrollable_text = ScrollableText::with_string("Test \n String \n".to_owned());
+
+    assert_eq!(scrollable_text.items.len(), 3);
+    assert_eq!(scrollable_text.items, vec!["Test ", " String ", ""]);
+    assert_eq!(scrollable_text.offset, 0);
+  }
+
+  #[test]
+  fn test_scrollable_text_get_text() {
+    let test_text = "Test \nString";
+    let scrollable_text = ScrollableText::with_string(test_text.to_owned());
+
+    assert_str_eq!(scrollable_text.get_text(), test_text);
+  }
+
+  #[test]
+  fn test_scrollable_text_scroll() {
+    let mut scrollable_text = ScrollableText::with_string("Test \nString".to_owned());
+
+    scrollable_text.scroll_down();
+
+    assert_eq!(scrollable_text.offset, 1);
+
+    scrollable_text.scroll_down();
+
+    assert_eq!(scrollable_text.offset, 1);
+
+    scrollable_text.scroll_up();
+
+    assert_eq!(scrollable_text.offset, 0);
+
+    scrollable_text.scroll_up();
+
+    assert_eq!(scrollable_text.offset, 0);
+
+    scrollable_text.scroll_to_bottom();
+
+    assert_eq!(scrollable_text.offset, 1);
+
+    scrollable_text.scroll_to_top();
+
+    assert_eq!(scrollable_text.offset, 0);
+  }
+
+  #[test]
+  fn test_horizontally_scrollable_text_from() {
+    let test_text = "Test string";
+    let horizontally_scrollable_text = HorizontallyScrollableText::from(test_text.to_owned());
+
+    assert_eq!(*horizontally_scrollable_text.offset.borrow(), 0);
+    assert_str_eq!(
+      horizontally_scrollable_text.text,
+      format!("{}        ", test_text)
+    );
+  }
+
+  #[test]
+  fn test_horizontally_scrollable_text_to_string() {
+    let test_text = "Test string";
+    let horizontally_scrollable_text = HorizontallyScrollableText::from(test_text.to_owned());
+
+    assert_str_eq!(
+      horizontally_scrollable_text.to_string(),
+      format!("{}        ", test_text)
+    );
+
+    let horizontally_scrollable_text = HorizontallyScrollableText {
+      text: test_text.to_owned(),
+      offset: RefCell::new(test_text.len() - 1),
+    };
+
+    assert_str_eq!(horizontally_scrollable_text.to_string(), "g");
+
+    let horizontally_scrollable_text = HorizontallyScrollableText {
+      text: test_text.to_owned(),
+      offset: RefCell::new(test_text.len()),
+    };
+
+    assert!(horizontally_scrollable_text.to_string().is_empty());
+  }
+
+  #[test]
+  fn test_horizontally_scrollable_text_new() {
+    let test_text = "Test string";
+    let horizontally_scrollable_text = HorizontallyScrollableText::new(test_text.to_owned());
+
+    assert_eq!(*horizontally_scrollable_text.offset.borrow(), 0);
+    assert_str_eq!(
+      horizontally_scrollable_text.text,
+      format!("{}        ", test_text)
+    );
+  }
+
+  #[test]
+  fn test_horizontally_scrollable_text_scroll_text() {
+    let horizontally_scrollable_text = HorizontallyScrollableText::from("Test string".to_owned());
+
+    assert_eq!(*horizontally_scrollable_text.offset.borrow(), 0);
+
+    for i in 1..horizontally_scrollable_text.text.len() {
+      horizontally_scrollable_text.scroll_text();
+
+      assert_eq!(*horizontally_scrollable_text.offset.borrow(), i);
+    }
+
+    horizontally_scrollable_text.scroll_text();
+
+    assert_eq!(*horizontally_scrollable_text.offset.borrow(), 0);
+  }
+
+  #[test]
+  fn test_horizontally_scrollable_text_reset_offset() {
+    let horizontally_scrollable_text = HorizontallyScrollableText {
+      text: "Test string".to_owned(),
+      offset: RefCell::new(1),
+    };
+
+    horizontally_scrollable_text.reset_offset();
+
+    assert_eq!(*horizontally_scrollable_text.offset.borrow(), 0);
+  }
+
+  #[test]
+  fn test_horizontally_scrollable_text_scroll_or_reset() {
+    let width = 3;
+    let test_text = "Test string";
+    let horizontally_scrollable_text = HorizontallyScrollableText::from(test_text.to_owned());
+
+    horizontally_scrollable_text.scroll_or_reset(width, true);
+
+    assert_eq!(*horizontally_scrollable_text.offset.borrow(), 1);
+
+    horizontally_scrollable_text.scroll_or_reset(width, false);
+
+    assert_eq!(*horizontally_scrollable_text.offset.borrow(), 0);
+
+    horizontally_scrollable_text.scroll_or_reset(width, true);
+
+    assert_eq!(*horizontally_scrollable_text.offset.borrow(), 1);
+
+    horizontally_scrollable_text.scroll_or_reset(test_text.len(), false);
+
+    assert_eq!(*horizontally_scrollable_text.offset.borrow(), 0);
+  }
+
+  #[test]
+  fn test_horizontally_scrollable_text_stationary_style() {
+    let test_text = "Test string";
+    let horizontally_scrollable_text = HorizontallyScrollableText::from(test_text.to_owned());
+
+    assert_eq!(horizontally_scrollable_text.stationary_style(), test_text);
+  }
+
+  #[test]
+  fn test_tab_state_new() {
+    let tab_state = TabState::new(create_test_tab_routes());
+
+    assert_eq!(tab_state.index, 0);
+  }
+
+  #[test]
+  fn test_tab_state_set_index() {
+    let mut tab_state = TabState::new(create_test_tab_routes());
+
+    let result = tab_state.set_index(1);
+
+    assert_eq!(result, &create_test_tab_routes()[1]);
+    assert_eq!(tab_state.index, 1);
+  }
+
+  #[test]
+  fn test_tab_state_get_active_route() {
+    let second_tab = create_test_tab_routes()[1].clone().route;
+    let tab_state = TabState {
+      tabs: create_test_tab_routes(),
+      index: 1,
+    };
+
+    let active_route = tab_state.get_active_route();
+
+    assert_eq!(active_route, &second_tab);
+  }
+
+  #[test]
+  fn test_tab_state_get_active_tab_help() {
+    let second_tab_help = create_test_tab_routes()[1].clone().help;
+    let tab_state = TabState {
+      tabs: create_test_tab_routes(),
+      index: 1,
+    };
+
+    let tab_help = tab_state.get_active_tab_help();
+
+    assert_str_eq!(tab_help, second_tab_help);
+  }
+
+  #[test]
+  fn test_tab_state_get_active_tab_contextual_help() {
+    let second_tab_contextual_help = create_test_tab_routes()[1].clone().contextual_help.unwrap();
+    let tab_state = TabState {
+      tabs: create_test_tab_routes(),
+      index: 1,
+    };
+
+    let tab_contextual_help = tab_state.get_active_tab_contextual_help();
+
+    assert!(tab_contextual_help.is_some());
+    assert_str_eq!(tab_contextual_help.unwrap(), second_tab_contextual_help);
+  }
+
+  #[test]
+  fn test_tab_state_next() {
+    let tab_routes = create_test_tab_routes();
+    let mut tab_state = TabState::new(create_test_tab_routes());
+
+    assert_eq!(tab_state.get_active_route(), &tab_routes[0].route);
+
+    tab_state.next();
+
+    assert_eq!(tab_state.get_active_route(), &tab_routes[1].route);
+
+    tab_state.next();
+
+    assert_eq!(tab_state.get_active_route(), &tab_routes[0].route);
+  }
+
+  #[test]
+  fn test_tab_state_previous() {
+    let tab_routes = create_test_tab_routes();
+    let mut tab_state = TabState::new(create_test_tab_routes());
+
+    assert_eq!(tab_state.get_active_route(), &tab_routes[0].route);
+
+    tab_state.previous();
+
+    assert_eq!(tab_state.get_active_route(), &tab_routes[1].route);
+
+    tab_state.previous();
+
+    assert_eq!(tab_state.get_active_route(), &tab_routes[0].route);
+  }
+
+  fn create_test_tab_routes() -> Vec<TabRoute> {
+    vec![
+      TabRoute {
+        title: "Test 1".to_owned(),
+        route: ActiveRadarrBlock::Movies.into(),
+        help: "Help for Test 1".to_owned(),
+        contextual_help: Some("Contextual Help for Test 1".to_owned()),
+      },
+      TabRoute {
+        title: "Test 2".to_owned(),
+        route: ActiveRadarrBlock::Collections.into(),
+        help: "Help for Test 2".to_owned(),
+        contextual_help: Some("Contextual Help for Test 2".to_owned()),
+      },
+    ]
+  }
+
+  fn create_test_stateful_table() -> StatefulTable<&'static str> {
+    let mut stateful_table = StatefulTable::default();
+    stateful_table.set_items(vec!["Test 1", "Test 2"]);
+
+    stateful_table
+  }
+}
@@ -399,7 +399,7 @@ fn draw_movie_releases<B: Backend>(f: &mut Frame<'_, B>, app: &mut App, content_
      );
      let size = convert_to_gb(size.as_u64().unwrap());
      let rejected_str = if *rejected { "⛔" } else { "" };
-      let mut peers = if seeders.is_none() || leechers.is_none() {
+      let peers = if seeders.is_none() || leechers.is_none() {
        Text::default()
      } else {
        let seeders = seeders.clone().unwrap().as_u64().unwrap();