[chromedriver] Added option to make element references W3C compliant.

chrome/test/chromedriver/client/chromedriver.py

 # Copyright 2013 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import sys
 import platform
 
 import command_executor
 from command_executor import Command
 from webelement import WebElement
 
+ELEMENT_KEY_W3C = "element-6066-11e4-a52e-4f735466cecf"
+ELEMENT_KEY = "ELEMENT"
 
 class ChromeDriverException(Exception):
   pass
 class NoSuchElement(ChromeDriverException):
   pass
 class NoSuchFrame(ChromeDriverException):
   pass
 class UnknownCommand(ChromeDriverException):
   pass
 class StaleElementReference(ChromeDriverException):
@@ skipping 179 matching lines @@
     if isinstance(response['status'], basestring):
       self.w3c_compliant = True
     elif isinstance(response['status'], int):
       self.w3c_compliant = False
     else:
       raise UnknownError("unexpected response")
 
     self._session_id = response['sessionId']
     self.capabilities = self._UnwrapValue(response['value'])
 
+
   def _WrapValue(self, value):
     """Wrap value from client side for chromedriver side."""
     if isinstance(value, dict):
       converted = {}
       for key, val in value.items():
         converted[key] = self._WrapValue(val)
       return converted
     elif isinstance(value, WebElement):
-      return {'ELEMENT': value._id}
+      if (self.w3c_compliant):
+        return {ELEMENT_KEY_W3C: value._id}
+      else:
+        return {ELEMENT_KEY: value._id}
     elif isinstance(value, list):
       return list(self._WrapValue(item) for item in value)
     else:
       return value
 
   def _UnwrapValue(self, value):
-    """Unwrap value from chromedriver side for client side."""
     if isinstance(value, dict):
-      if (len(value) == 1 and 'ELEMENT' in value
-          and isinstance(value['ELEMENT'], basestring)):
-        return WebElement(self, value['ELEMENT'])
+      if (self.w3c_compliant and len(value) == 1
+          and ELEMENT_KEY_W3C in value
+          and isinstance(
+            value[ELEMENT_KEY_W3C], basestring)):
+        return WebElement(self, value[ELEMENT_KEY_W3C])
+      elif (len(value) == 1 and ELEMENT_KEY in value
+            and isinstance(value[ELEMENT_KEY], basestring)):
+        return WebElement(self, value[ELEMENT_KEY])
       else:
         unwraped = {}
         for key, val in value.items():
           unwraped[key] = self._UnwrapValue(val)
         return unwraped
     elif isinstance(value, list):
       return list(self._UnwrapValue(item) for item in value)
     else:
       return value
 
@@ skipping 231 matching lines @@
 
   def GetNetworkConnection(self):
     return self.ExecuteCommand(Command.GET_NETWORK_CONNECTION)
 
   def DeleteNetworkConditions(self):
     self.ExecuteCommand(Command.DELETE_NETWORK_CONDITIONS)
 
   def SetNetworkConnection(self, connection_type):
     params = {'parameters': {'type': connection_type}}
     self.ExecuteCommand(Command.SET_NETWORK_CONNECTION, params)