Add __getitem__ to XPathLocator

cspell.config.yaml

@@ -35,6 +35,8 @@ words:
   - yaspeller
 
     # Code words
+  - argnames
+  - argvalues
   - docstrings
   - conftest
   - kwargs

docs/CHANGELOG.rst

@@ -3,6 +3,10 @@ Version history
 
 We follow `Semantic Versions <https://semver.org/>`_.
 
+0.8.5 (10.02.25)
+*******************************************************************************
+- Add ability to get related xpath locators by index for ``XPathLocator``
+
 0.8.4 (30.01.25)
 *******************************************************************************
 - Add escaping single and double quotes in the: ``ElementWithTextLocator``,

pomcorn/locators/base_locators.py

@@ -152,6 +152,28 @@ def __or__(self, other: XPathLocator) -> XPathLocator:
         """
         return XPathLocator(query=f"({self.query} | {other.query})")
 
+    def __getitem__(self, index: int) -> XPathLocator:
+        """Allow get related xpath locator by index.
+
+        Example:
+            div_locator = XPathLocator("//div") --> `//div`
+            div_locator[0]  --> `(//div)[1]`   # xpath numeration starts with 1
+            div_locator[-1] --> `(//div)[last()]`
+
+        """
+        query = f"({self.query})"
+
+        if index >= 0:
+            # `+1` is used here because numeration in xpath starts with 1
+            query += f"[{index + 1}]"
+        elif index == -1:
+            # To avoid ugly locators with `...)[last() - 0]`
+            query += "[last()]"
+        else:
+            query += f"[last() - {abs(index + 1)}]"
+
+        return XPathLocator(query)
+
     def __bool__(self) -> bool:
         """Return whether query of current locator is empty or not."""
         return bool(self.related_query)

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pomcorn"
-version = "0.8.4"
+version = "0.8.5"
 description = "Base implementation of Page Object Model"
 authors = [
   "Saritasa <[email protected]>",

tests/locators/test_get_item.py

@@ -0,0 +1,20 @@
+import pytest
+
+from pomcorn.locators.base_locators import XPathLocator
+
+TEST_QUERY = "//span[text()='Users']"
+
+
+@pytest.mark.parametrize(
+    argnames=["index", "result"],
+    argvalues=[
+        [-2, f"({TEST_QUERY})[last() - 1]"],
+        [-1, f"({TEST_QUERY})[last()]"],
+        [0, f"({TEST_QUERY})[1]"],
+        [1, f"({TEST_QUERY})[2]"],
+    ],
+)
+def test_getting_related_xpath_locators_by_index(index: int, result: str):
+    """Check that getting related xpath locators by index works correct."""
+    locator = XPathLocator(TEST_QUERY)
+    assert locator[index].query == result