[py] Add element that mutated when using logging mutation

SeleniumHQ · Dec 18, 2020 · cfb9783 · cfb9783
1 parent 96e5d9a
commit cfb9783
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/py/selenium/webdriver/remote/webdriver.py b/py/selenium/webdriver/remote/webdriver.py
@@ -1517,8 +1517,10 @@ async def log_mutation_events(self):
                 yield event
 
             payload = json.loads(evnt.value.payload)
-            elements = self.find_elements(By.CSS_SELECTOR, "*[data-__webdriver_id={}".format(payload['target']))
-            # event["element"] = elements[0]
+            elements: list = self.find_elements(By.CSS_SELECTOR, "*[data-__webdriver_id={}".format(payload['target']))
+            if not elements:
+                elements.append(None)
+            event["element"] = elements[0]
             event["attribute_name"] = payload['name']
             event["current_value"] = payload['value']
             event["old_value"] = payload['oldValue']