Split InputByLabelLocator for two cases

pomcorn/locators/xpath_locators.py

@@ -200,13 +200,18 @@ def __init__(self, text: str, exact: bool = False):
         super().__init__(text=text, element="button", exact=exact)
 
 
-class InputByLabelLocator(XPathLocator):
+class InputInLabelLocator(XPathLocator):
     """Locator to looking for input with label by XPath.
 
     Specify the query as the string
     ``//label[contains(., "label")]//input``, where ``label`` is the text of
     the input label.
 
+    Example:
+        <label>Title</label>
+            <input value="Value">
+        </label>
+
     """
 
     def __init__(self, label: str):
@@ -216,6 +221,28 @@ def __init__(self, label: str):
         )
 
 
+class InputByLabelLocator(XPathLocator):
+    """Locator to looking for input next to label by XPath.
+
+    Specify the query as the string
+    ``//label[contains(., "label")]/following-sibling::input``, where ``label``
+    is the text of the input label.
+
+    Example:
+        <div>
+            <label for="InputWithLabel">Title</label>
+            <input id="InputWithLabel" value="Value">
+        </div>
+
+    """
+
+    def __init__(self, label: str):
+        """Init XPathLocator."""
+        super().__init__(
+            query=f'//label[contains(., "{label}")]/following-sibling::input',
+        )
+
+
 class TextAreaByLabelLocator(XPathLocator):
     """Locator to looking for textarea with label by XPath.