Add CalendarMonth widget

Author: LightOfHeaven1994

README.md

@@ -17,6 +17,8 @@ Button - <https://www.patternfly.org/v4/components/button>
 
 Bullet Chart - <https://www.patternfly.org/v4/charts/bullet-chart>
 
+Calendar month - <https://www.patternfly.org/v4/components/calendar-month>
+
 Card - <https://www.patternfly.org/v4/components/card>
 
 Chip Group - <https://www.patternfly.org/v4/components/chip-group>

src/widgetastic_patternfly4/__init__.py

@@ -2,6 +2,7 @@
 from .breadcrumb import BreadCrumb
 from .bulletchart import BulletChart
 from .button import Button
+from .calendarmonth import CalendarMonth
 from .card import Card
 from .card import CardCheckBox
 from .card import CardForCardGroup
@@ -64,6 +65,7 @@
     "BreadCrumb",
     "Button",
     "BulletChart",
+    "CalendarMonth",
     "CheckboxMenu",
     "CheckboxSelect",
     "Chip",

src/widgetastic_patternfly4/calendarmonth.py

@@ -0,0 +1,98 @@
+from selenium.common.exceptions import NoSuchElementException
+from widgetastic.utils import ParametrizedLocator
+from widgetastic.widget import Table
+from widgetastic.widget import TextInput
+from widgetastic.widget import Widget
+
+from .select import Select
+
+
+class BaseCalendarMonth:
+    """Represents calendar month component.
+
+    https://www.patternfly.org/v4/components/calendar-month
+    """
+
+    CALENDAR_HEADER = ".//div[@class='pf-c-calendar-month__header']"
+    MONTH_SELECT_LOCATOR = (
+        f"{CALENDAR_HEADER}//div[contains(@class, 'header-month')]"
+        f"//div[@data-ouia-component-type='PF4/Select']"
+    )
+    YEAR_INPUT_LOCATOR = f"{CALENDAR_HEADER}//div[contains(@class, 'header-year')]/input"
+
+    PREV_BUTTON_LOCATOR = f"{CALENDAR_HEADER}//div[contains(@class, 'prev-month')]"
+    NEXT_BUTTON_LOCATOR = f"{CALENDAR_HEADER}//div[contains(@class, 'next-month')]"
+
+    TABLE = ".//table"
+    SELECTED_DATE_LOCATOR = f"{TABLE}/tbody//td[contains(@class, 'pf-m-selected')]"
+
+    def prev(self):
+        return self.browser.click(self.PREV_BUTTON_LOCATOR)
+
+    def next(self):
+        return self.browser.click(self.NEXT_BUTTON_LOCATOR)
+
+    @property
+    def year(self):
+        el = self.browser.element(self.YEAR_INPUT_LOCATOR)
+        return el.get_attribute("value")
+
+    @year.setter
+    def year(self, value):
+        year_input_widget = TextInput(self, locator=self.YEAR_INPUT_LOCATOR)
+        year_input_widget.fill(value)
+
+    @property
+    def month(self):
+        el = self.browser.element(self.MONTH_SELECT_LOCATOR)
+        return self.browser.text(el)
+
+    @month.setter
+    def month(self, value):
+        month_select_widget = Select(self, locator=self.MONTH_SELECT_LOCATOR)
+        month_select_widget.item_select(value)
+
+    @property
+    def day(self):
+        try:
+            el = self.browser.element(self.SELECTED_DATE_LOCATOR)
+        except NoSuchElementException:
+            return ""
+        return self.browser.text(el)
+
+    @day.setter
+    def day(self, value):
+        table_widget = Table(self, locator=self.TABLE)
+        for row in table_widget:
+            for _, col in row:
+                if col.text == value:
+                    col.click()
+                    return
+
+    def read(self):
+        """Returns the currently selected date in format DD MONTH YYYY."""
+        return f"{self.day} {self.month} {self.year}"
+
+    def fill(self, items):
+        """Fills a Calendar with all items.
+        Example dictionary: {'day': '22', 'month': 'November', 'year': '2023'}"
+
+        Args:
+            items: A dictionary containing what items to select (True) or deselect (False)
+        """
+        if type(items) is not dict:
+            raise TypeError("Fill value has to be dictionary type. ")
+        if "day" in items:
+            self.day = items["day"]
+        if "month" in items:
+            self.month = items["month"]
+        if "year" in items:
+            self.year = items["year"]
+
+
+class CalendarMonth(BaseCalendarMonth, Widget):
+    ROOT = ParametrizedLocator("{@locator}")
+
+    def __init__(self, parent, locator, logger=None):
+        super().__init__(parent, logger=logger)
+        self.locator = locator

testing/test_calendarmonth.py

@@ -0,0 +1,74 @@
+import calendar
+
+import pytest
+from widgetastic.widget import View
+
+from widgetastic_patternfly4 import CalendarMonth
+
+TESTING_PAGE_URL = "https://patternfly-react.surge.sh/components/calendar-month"
+
+MONTHS_LIST = list(calendar.month_name[1:])
+
+
+@pytest.fixture
+def calendar_month_view(browser):
+    class TestView(View):
+        ROOT = ".//div[@id='ws-react-c-calendar-month-selectable-date']"
+        calendar = CalendarMonth(locator=".//div[@class='pf-c-calendar-month']")
+
+    return TestView(browser)
+
+
+def test_year_selection(calendar_month_view):
+    assert calendar_month_view.calendar.year
+    calendar_month_view.calendar.year = "2023"
+    assert calendar_month_view.calendar.year == "2023"
+
+
+def test_month_selection(calendar_month_view):
+    assert calendar_month_view.calendar.month
+    calendar_month_view.calendar.month = "December"
+    assert calendar_month_view.calendar.month == "December"
+
+
+def test_day_selection(calendar_month_view):
+    calendar_month_view.calendar.day = "20"
+    assert calendar_month_view.calendar.day == "20"
+    calendar_month_view.calendar.next()
+    assert not calendar_month_view.calendar.day
+
+
+def _get_proper_month_index(index):
+    max_index = len(MONTHS_LIST) - 1
+    if index < 0:
+        index = max_index
+    elif index > max_index:
+        index = 0
+
+    return index
+
+
+def test_month_navigation(calendar_month_view):
+    current_month = calendar_month_view.calendar.month
+    prev_month_index = _get_proper_month_index(MONTHS_LIST.index(current_month) - 1)
+    next_month_index = _get_proper_month_index(MONTHS_LIST.index(current_month) + 1)
+
+    calendar_month_view.calendar.prev()
+    prev_month = calendar_month_view.calendar.month
+    assert prev_month == MONTHS_LIST[prev_month_index]
+
+    # reset to current default month
+    calendar_month_view.calendar.month = current_month
+
+    calendar_month_view.calendar.next()
+    next_month = calendar_month_view.calendar.month
+    assert next_month == MONTHS_LIST[next_month_index]
+
+
+def test_fill_and_read(calendar_month_view):
+    calendar_month_view.calendar.fill({"day": "10", "month": "February", "year": "2023"})
+    result = calendar_month_view.calendar.read()
+    assert result == "10 February 2023"
+
+    with pytest.raises(TypeError):
+        calendar_month_view.calendar.fill("foo")