Add missing None check in get_attribute.

SeleniumHQ · Aug 12, 2016 · 4326cb0 · 4326cb0
1 parent 4b46465
commit 4326cb0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/py/selenium/webdriver/remote/webelement.py b/py/selenium/webdriver/remote/webelement.py
@@ -159,8 +159,9 @@ def get_attribute(self, name):
         else:
             resp = self._execute(Command.GET_ELEMENT_ATTRIBUTE, {'name': name})
             attributeValue = resp.get('value')
-            if name != 'value' and attributeValue.lower() in ('true', 'false'):
-                attributeValue = attributeValue.lower()
+            if attributeValue is not None:
+                if name != 'value' and attributeValue.lower() in ('true', 'false'):
+                    attributeValue = attributeValue.lower()
         return attributeValue
 
     def is_selected(self):