feat(widget-calendar): Add tests for calendar widget

Cargo.toml

@@ -19,6 +19,7 @@ rust-version = "1.59.0"
 
 [features]
 default = ["crossterm"]
+all-widgets = ["widget-calendar"]
 widget-calendar = ["time"]
 
 [package.metadata.docs.rs]

Makefile.toml

@@ -100,11 +100,11 @@ args = [
 ]
 
 [tasks.test-crossterm]
-env = { TUI_FEATURES = "serde,crossterm" }
+env = { TUI_FEATURES = "serde,crossterm,all-widgets" }
 run_task = "test"
 
 [tasks.test-termion]
-env = { TUI_FEATURES = "serde,termion" }
+env = { TUI_FEATURES = "serde,termion,all-widgets" }
 run_task = "test"
 
 [tasks.test]

src/widgets/calendar.rs

@@ -211,3 +211,35 @@ impl Default for CalendarEventStore {
         Self(HashMap::with_capacity(4))
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::style::Color;
+    use time::Month;
+
+    #[test]
+    fn event_store() {
+        let a = (
+            Date::from_calendar_date(2023, Month::January, 1).unwrap(),
+            Style::default(),
+        );
+        let b = (
+            Date::from_calendar_date(2023, Month::January, 2).unwrap(),
+            Style::default().bg(Color::Red).fg(Color::Blue),
+        );
+        let mut s = CalendarEventStore::default();
+        s.add(b.0, b.1);
+
+        assert_eq!(
+            s.get_style(a.0),
+            a.1,
+            "Date not added to the styler should look up as Style::default()"
+        );
+        assert_eq!(
+            s.get_style(b.0),
+            b.1,
+            "Date added to styler should return the provided style"
+        );
+    }
+}

tests/widgets_calendar.rs

@@ -0,0 +1,114 @@
+#![cfg(feature = "widget-calendar")]
+use ratatui::{
+    backend::TestBackend,
+    buffer::Buffer,
+    style::Style,
+    widgets::{
+        calendar::{CalendarEventStore, Monthly},
+        Block, Borders, Paragraph, Widget, Wrap,
+    },
+    Terminal,
+};
+
+use time::{Date, Month};
+
+fn test_render<'a, W: Widget>(widget: W, expected: Buffer, size: (u16, u16)) {
+    let backend = TestBackend::new(size.0, size.1);
+    let mut terminal = Terminal::new(backend).unwrap();
+
+    terminal
+        .draw(|f| f.render_widget(widget, f.size()))
+        .unwrap();
+    terminal.backend().assert_buffer(&expected);
+}
+
+#[test]
+fn days_layout() {
+    let c = Monthly::new(
+        Date::from_calendar_date(2023, Month::January, 1).unwrap(),
+        CalendarEventStore::default(),
+    );
+    let expected = Buffer::with_lines(vec![
+        "  1  2  3  4  5  6  7",
+        "  8  9 10 11 12 13 14",
+        " 15 16 17 18 19 20 21",
+        " 22 23 24 25 26 27 28",
+        " 29 30 31",
+    ]);
+    test_render(c, expected, (21, 5));
+}
+
+#[test]
+fn days_layout_show_surrounding() {
+    let c = Monthly::new(
+        Date::from_calendar_date(2023, Month::December, 1).unwrap(),
+        CalendarEventStore::default(),
+    )
+    .show_surrounding(Style::default());
+    let expected = Buffer::with_lines(vec![
+        " 26 27 28 29 30  1  2",
+        "  3  4  5  6  7  8  9",
+        " 10 11 12 13 14 15 16",
+        " 17 18 19 20 21 22 23",
+        " 24 25 26 27 28 29 30",
+        " 31  1  2  3  4  5  6",
+    ]);
+    test_render(c, expected, (21, 6));
+}
+
+#[test]
+fn show_month_header() {
+    let c = Monthly::new(
+        Date::from_calendar_date(2023, Month::January, 1).unwrap(),
+        CalendarEventStore::default(),
+    )
+    .show_month_header(Style::default());
+    let expected = Buffer::with_lines(vec![
+        "     January 2023    ",
+        "  1  2  3  4  5  6  7",
+        "  8  9 10 11 12 13 14",
+        " 15 16 17 18 19 20 21",
+        " 22 23 24 25 26 27 28",
+        " 29 30 31",
+    ]);
+    test_render(c, expected, (21, 6));
+}
+
+#[test]
+fn show_weekdays_header() {
+    let c = Monthly::new(
+        Date::from_calendar_date(2023, Month::January, 1).unwrap(),
+        CalendarEventStore::default(),
+    )
+    .show_weekdays_header(Style::default());
+    let expected = Buffer::with_lines(vec![
+        " Su Mo Tu We Th Fr Sa",
+        "  1  2  3  4  5  6  7",
+        "  8  9 10 11 12 13 14",
+        " 15 16 17 18 19 20 21",
+        " 22 23 24 25 26 27 28",
+        " 29 30 31",
+    ]);
+    test_render(c, expected, (21, 6));
+}
+
+#[test]
+fn show_combo() {
+    let c = Monthly::new(
+        Date::from_calendar_date(2023, Month::January, 1).unwrap(),
+        CalendarEventStore::default(),
+    )
+    .show_weekdays_header(Style::default())
+    .show_month_header(Style::default())
+    .show_surrounding(Style::default());
+    let expected = Buffer::with_lines(vec![
+        "     January 2023    ",
+        " Su Mo Tu We Th Fr Sa",
+        "  1  2  3  4  5  6  7",
+        "  8  9 10 11 12 13 14",
+        " 15 16 17 18 19 20 21",
+        " 22 23 24 25 26 27 28",
+        " 29 30 31  1  2  3  4",
+    ]);
+    test_render(c, expected, (21, 7));
+}