[py] add name of what is being executed to large JS executions

titusfortner · titusfortner · commit b869b998d4d3 · 2023-01-04T09:41:59.000-06:00
diff --git a/py/selenium/webdriver/remote/webdriver.py b/py/selenium/webdriver/remote/webdriver.py
@@ -842,7 +842,7 @@ def find_elements(self, by=By.ID, value: Optional[str] = None) -> List[WebElemen
         if isinstance(by, RelativeBy):
             _pkg = ".".join(__name__.split(".")[:-1])
             raw_function = pkgutil.get_data(_pkg, "findElements.js").decode("utf8")
-            find_element_js = f"return ({raw_function}).apply(null, arguments);"
+            find_element_js = f"/* findElements */return ({raw_function}).apply(null, arguments);"
             return self.execute_script(find_element_js, by.to_dict())
 
         if by == By.ID:
diff --git a/py/selenium/webdriver/remote/webelement.py b/py/selenium/webdriver/remote/webelement.py
@@ -95,7 +95,7 @@ def click(self) -> None:
     def submit(self):
         """Submits a form."""
         script = (
-            "var form = arguments[0];\n"
+            "/* submitForm */var form = arguments[0];\n"
             'while (form.nodeName != "FORM" && form.parentNode) {\n'
             "  form = form.parentNode;\n"
             "}\n"
@@ -174,7 +174,9 @@ def get_attribute(self, name) -> str:
         """
         if getAttribute_js is None:
             _load_js()
-        attribute_value = self.parent.execute_script(f"return ({getAttribute_js}).apply(null, arguments);", self, name)
+        attribute_value = self.parent.execute_script(
+            f"/* getAttribute */return ({getAttribute_js}).apply(null, arguments);", self, name
+        )
         return attribute_value
 
     def is_selected(self) -> bool:
@@ -255,7 +257,7 @@ def is_displayed(self) -> bool:
         # Only go into this conditional for browsers that don't use the atom themselves
         if isDisplayed_js is None:
             _load_js()
-        return self.parent.execute_script(f"return ({isDisplayed_js}).apply(null, arguments);", self)
+        return self.parent.execute_script(f"/* isDisplayed */return ({isDisplayed_js}).apply(null, arguments);", self)
 
     @property
     def location_once_scrolled_into_view(self) -> dict:
