Updating IE driver getActiveElement command to be spec-compliant

jimevans · jimevans · commit 6ccbe0a5c845 · 2017-11-28T09:10:59.000-08:00
If there is no active element, this method may now return a null results,
as in compliance with the W3C WebDriver specification, which relies on the
DOM definition of active element. Language bindings maintainers should
take note that this is a behavior change, and that the command can now
return null values. Adjust your language bindings accordingly.
diff --git a/cpp/iedriver/CommandHandlers/GetActiveElementCommandHandler.cpp b/cpp/iedriver/CommandHandlers/GetActiveElementCommandHandler.cpp
@@ -29,9 +29,9 @@ GetActiveElementCommandHandler::~GetActiveElementCommandHandler(void) {
 }
 
 void GetActiveElementCommandHandler::ExecuteInternal(
-    const IECommandExecutor& executor,
-    const ParametersMap& command_parameters,
-    Response* response) {
+  const IECommandExecutor& executor,
+  const ParametersMap& command_parameters,
+  Response* response) {
   BrowserHandle browser_wrapper;
   int status_code = executor.GetCurrentBrowser(&browser_wrapper);
   if (status_code != WD_SUCCESS) {
@@ -46,24 +46,27 @@ void GetActiveElementCommandHandler::ExecuteInternal(
     return;
   }
 
-  CComPtr<IHTMLElement> element;
-  doc->get_activeElement(&element);
+  CComPtr<IHTMLElement> element(NULL);
+  HRESULT hr = doc->get_activeElement(&element);
 
-  // For some contentEditable frames, the <body> element will be the
-  // active element. However, to properly have focus, we must explicitly
-  // set focus to the element.
-  CComPtr<IHTMLBodyElement> body_element;
-  HRESULT body_hr = element->QueryInterface<IHTMLBodyElement>(&body_element);
-  if (body_element) {
-    CComPtr<IHTMLElement2> body_element2;
-    body_element->QueryInterface<IHTMLElement2>(&body_element2);
-    body_element2->focus();
+  if (FAILED(hr)) {
+    // For some contentEditable frames, the <body> element will be the
+    // active element. However, to properly have focus, we must explicitly
+    // set focus to the element.
+    CComPtr<IHTMLBodyElement> body_element;
+    HRESULT body_hr = element->QueryInterface<IHTMLBodyElement>(&body_element);
+    if (body_element) {
+      CComPtr<IHTMLElement2> body_element2;
+      body_element->QueryInterface<IHTMLElement2>(&body_element2);
+      body_element2->focus();
+    }
   }
 
-  // If we don't have an element at this point, just return the
-  // body element so that we don't return a NULL pointer.
+  // If we don't have an element at this point, we should return a
+  // null result, as that's what document.activeElement() returns.
   if (!element) {
-    doc->get_body(&element);
+    response->SetSuccessResponse(Json::Value::null);
+    return;
   }
 
   if (element) {
