From b2a38a20b145e44f22c5d4973f5920f076da715b Mon Sep 17 00:00:00 2001
From: Jakub Rusz <jrusz@redhat.com>
Date: Wed, 13 Mar 2024 09:46:02 +0100
Subject: [PATCH] Add Expandable section

---
 README.md                                     |  1 +
 src/widgetastic_patternfly5/__init__.py       |  2 +
 .../components/expandable_section.py          | 60 +++++++++++++++++++
 testing/components/test_expandable_section.py | 28 +++++++++
 4 files changed, 91 insertions(+)
 create mode 100644 src/widgetastic_patternfly5/components/expandable_section.py
 create mode 100644 testing/components/test_expandable_section.py

diff --git a/README.md b/README.md
index b0e9ab1..e1ef261 100644
--- a/README.md
+++ b/README.md
@@ -37,6 +37,7 @@ itteration of [widgetastic.patternfly4](https://github.com/RedHatQE/widgetastic.
 - [description-list](https://www.patternfly.org/components/description-list)
 - [drawer](https://www.patternfly.org/components/drawer)
 - [dual-list-selector](https://www.patternfly.org/components/dual-list-selector)
+- [expandable-section](https://www.patternfly.org/components/expandable-section)
 - forms
   - [form-select](https://www.patternfly.org/components/forms/form-select)
   - [radio](https://www.patternfly.org/components/forms/radio)
diff --git a/src/widgetastic_patternfly5/__init__.py b/src/widgetastic_patternfly5/__init__.py
index 462dd88..e73ebc7 100644
--- a/src/widgetastic_patternfly5/__init__.py
+++ b/src/widgetastic_patternfly5/__init__.py
@@ -25,6 +25,7 @@
 from .components.drawer import Drawer
 from .components.dual_list_selector import DualListSelector
 from .components.dual_list_selector import SearchDualListSelector
+from .components.expandable_section import ExpandableSection
 from .components.forms.form_select import FormSelect
 from .components.forms.form_select import FormSelectDisabled
 from .components.forms.form_select import FormSelectOptionDisabled
@@ -137,4 +138,5 @@
     "SwitchDisabled",
     "Tab",
     "Title",
+    "ExpandableSection",
 ]
diff --git a/src/widgetastic_patternfly5/components/expandable_section.py b/src/widgetastic_patternfly5/components/expandable_section.py
new file mode 100644
index 0000000..71c0a85
--- /dev/null
+++ b/src/widgetastic_patternfly5/components/expandable_section.py
@@ -0,0 +1,60 @@
+from wait_for import wait_for_decorator
+from widgetastic.widget import View
+
+
+class ExpandableSection(View):
+    """Represents the Patternfly Expandable section widget.
+
+    Expands itself automatically when any child widget gets accessed, ensuring
+    that the widget is visible.
+
+    https://www.patternfly.org/components/expandable-section
+
+    NOTE: When using this in your code you need to set the ROOT yourself!
+
+    Example code with a button underneath the section:
+
+        class MySection(ExpandableSection):
+            ROOT = ".//div[contains(@data-testid, 'my-expandable-section')]"
+            my-button = Button(locator=".//button")
+
+    After creating the view you can run `MySection.my-button.click()` and it
+    will automatically expand and click on your button.
+    """
+
+    ROOT = './/div[contains(@class, "-c-expandable-section")]'
+    BUTTON_LOCATOR = ".//button"
+
+    @property
+    def is_expanded(self):
+        """Returns a boolean."""
+        if self.browser.get_attribute("aria-expanded", self.BUTTON_LOCATOR) == "true":
+            return True
+        else:
+            return False
+
+    def click(self):
+        """Clicks the expandable section button."""
+        return self.browser.click(self.BUTTON_LOCATOR)
+
+    def expand(self):
+        """Expands the section (checks if not expanded already first)."""
+        if not self.is_expanded:
+
+            @wait_for_decorator(timeout=3)
+            def _click():
+                self.click()
+                return self.is_expanded
+
+    def collapse(self):
+        """Collapses the section (checks if expanded already first)."""
+        if self.is_expanded:
+
+            @wait_for_decorator(timeout=3)
+            def _click():
+                self.click()
+                return not self.is_expanded
+
+    def child_widget_accessed(self, widget):
+        # Expand the section
+        self.expand()
diff --git a/testing/components/test_expandable_section.py b/testing/components/test_expandable_section.py
new file mode 100644
index 0000000..00ba1da
--- /dev/null
+++ b/testing/components/test_expandable_section.py
@@ -0,0 +1,28 @@
+from widgetastic.widget import Text
+from widgetastic.widget import View
+
+from widgetastic_patternfly5 import ExpandableSection
+
+TESTING_PAGE_URL = "https://patternfly-react-main.surge.sh/components/expandable-section"
+
+
+class SectionView(View):
+    @View.nested
+    class basic(ExpandableSection):
+        text = Text(".//div[contains(@class, '-c-expandable-section__content')]")
+
+
+def test_section_expansion(browser):
+    view = SectionView(browser)
+    view.basic.expand()
+    assert view.basic.is_expanded
+    view.basic.collapse()
+    assert not view.basic.is_expanded
+
+
+def test_section_text(browser):
+    view = SectionView(browser)
+    assert view.basic.is_displayed
+    assert view.basic.text.is_displayed
+    assert view.basic.text.read() == "This content is visible only when the component is expanded."
+    view.basic.collapse()