First batch of context menu tests...

chrome_frame/chrome_frame_plugin.h

 // Copyright (c) 2009 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.
 
 #ifndef CHROME_FRAME_CHROME_FRAME_PLUGIN_H_
 #define CHROME_FRAME_CHROME_FRAME_PLUGIN_H_
 
 #include "base/ref_counted.h"
 #include "base/win_util.h"
 #include "chrome_frame/chrome_frame_automation.h"
@@ skipping 89 matching lines @@
       return;
     }
 
     // TrackPopupMenuEx call will fail on IE on Vista running
     // in low integrity mode. We DO seem to be able to enumerate the menu
     // though, so just clone it and show the copy:
     HMENU copy = UtilCloneContextMenu(static_cast<HMENU>(menu_handle));
     if (!copy)
       return;
 
-    T* pThis = static_cast<T*>(this);
-    if (pThis->PreProcessContextMenu(copy)) {
+    T* self = static_cast<T*>(this);
+    if (self->PreProcessContextMenu(copy)) {
+      // In order for the context menu to handle keyboard input, give the
+      // ActiveX window focus.
+      ignore_setfocus_ = true;
+      SetFocus(GetWindow());
+      ignore_setfocus_ = false;
       UINT flags = align_flags | TPM_LEFTBUTTON | TPM_RETURNCMD | TPM_RECURSE;
       UINT selected = TrackPopupMenuEx(copy, flags, params.screen_x,
                                        params.screen_y, GetWindow(), NULL);
-      if (selected != 0 && !pThis->HandleContextMenuCommand(selected, params)) {
+      // Menu is over now give focus back to chrome
+      GiveFocusToChrome();
+      if (selected != 0 && !self->HandleContextMenuCommand(selected, params)) {
         automation_client_->SendContextMenuCommandToChromeFrame(selected);
       }
     }
 
     DestroyMenu(copy);
   }
 
   LRESULT OnSetFocus(UINT message, WPARAM wparam, LPARAM lparam,
                      BOOL& handled) {  // NO_LINT
     if (!ignore_setfocus_ && automation_client_ != NULL) {
@@ skipping 90 matching lines @@
 
   // List of functions to enable for automation, or a single entry "*" to
   // enable all functions for automation.  Ignored unless is_privileged_ is
   // true.  Defaults to the empty list, meaning automation will not be
   // turned on.
   std::vector<std::string> functions_enabled_;
 };
 
 #endif  // CHROME_FRAME_CHROME_FRAME_PLUGIN_H_