feat: add option to wait for promise in Element.apply method (#135)

* feat: add option to wait for promise in Element.apply method

* Update element.py

Co-authored-by: Stephan Lensky <[email protected]>

* reorder doc

---------

Co-authored-by: Stephan Lensky <[email protected]>

Author: nathanfallet

CHANGELOG.md

@@ -12,6 +12,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - Added `speed` in `Tab.scroll_down` and `Tab.scroll_up` methods to control the scroll speed @nathanfallet
+- Allow to wait for promise in `Element.apply` method @nathanfallet
 
 ### Changed
 

zendriver/core/element.py

@@ -444,7 +444,7 @@ def __call__(self, js_method):
         """
         return self.apply(f"(e) => e['{js_method}']()")
 
-    async def apply(self, js_function, return_by_value=True):
+    async def apply(self, js_function, return_by_value=True, *, await_promise=False):
         """
         apply javascript to this element. the given js_function string should accept the js element as parameter,
         and can be a arrow function, or function declaration.
@@ -457,6 +457,8 @@ async def apply(self, js_function, return_by_value=True):
         :type js_function: str
         :param return_by_value:
         :type return_by_value:
+        :param await_promise: when True, waits for the promise to resolve before returning
+        :type await_promise: bool
         :return:
         :rtype:
         """
@@ -474,6 +476,7 @@ async def apply(self, js_function, return_by_value=True):
                 ],
                 return_by_value=True,
                 user_gesture=True,
+                await_promise=await_promise,
             )
         )
         if result and result[0]: