Update find_elements to match spec

Author: AutomatedTester

py/selenium/webdriver/remote/webdriver.py

@@ -677,7 +677,17 @@ def find_element(self, by=By.ID, value=None):
  """
  if not By.is_valid(by) or not isinstance(value, str):
  raise InvalidSelectorException("Invalid locator values passed in")
-
+ if by == By.ID:
+ by = By.CSS_SELECTOR
+ value = '[id="%s"]' % value
+ elif by == By.TAG_NAME:
+ by = By.CSS_SELECTOR
+ elif by == By.CLASS_NAME:
+ by = By.CSS_SELECTOR
+ value = ".%s" % value
+ elif by == By.NAME:
+ by = By.CSS_SELECTOR
+ value = '[name="%s"]' % value
  return self.execute(Command.FIND_ELEMENT,
  {'using': by, 'value': value})['value']
 
@@ -693,6 +703,18 @@ def find_elements(self, by=By.ID, value=None):
  if not By.is_valid(by) or not isinstance(value, str):
  raise InvalidSelectorException("Invalid locator values passed in")
 
+ if by == By.ID:
+ by = By.CSS_SELECTOR
+ value = '[id="%s"]' % value
+ elif by == By.TAG_NAME:
+ by = By.CSS_SELECTOR
+ elif by == By.CLASS_NAME:
+ by = By.CSS_SELECTOR
+ value = ".%s" % value
+ elif by == By.NAME:
+ by = By.CSS_SELECTOR
+ value = '[name="%s"]' % value
+
  return self.execute(Command.FIND_ELEMENTS,
  {'using': by, 'value': value})['value']
  @property

py/selenium/webdriver/remote/webelement.py

@@ -450,13 +450,37 @@ def find_element(self, by=By.ID, value=None):
  if not By.is_valid(by) or not isinstance(value, str):
  raise InvalidSelectorException("Invalid locator values passed in")
 
+ if by == By.ID:
+ by = By.CSS_SELECTOR
+ value = '[id="%s"]' % value
+ elif by == By.TAG_NAME:
+ by = By.CSS_SELECTOR
+ elif by == By.CLASS_NAME:
+ by = By.CSS_SELECTOR
+ value = ".%s" % value
+ elif by == By.NAME:
+ by = By.CSS_SELECTOR
+ value = '[name="%s"]' % value
+
  return self._execute(Command.FIND_CHILD_ELEMENT,
  {"using": by, "value": value})['value']
 
  def find_elements(self, by=By.ID, value=None):
  if not By.is_valid(by) or not isinstance(value, str):
  raise InvalidSelectorException("Invalid locator values passed in")
 
+ if by == By.ID:
+ by = By.CSS_SELECTOR
+ value = '[id="%s"]' % value
+ elif by == By.TAG_NAME:
+ by = By.CSS_SELECTOR
+ elif by == By.CLASS_NAME:
+ by = By.CSS_SELECTOR
+ value = ".%s" % value
+ elif by == By.NAME:
+ by = By.CSS_SELECTOR
+ value = '[name="%s"]' % value
+
  return self._execute(Command.FIND_CHILD_ELEMENTS,
  {"using": by, "value": value})['value']