Unit test for back/forward for url fragments. Improvements in the testing...

chrome_frame/test/chrome_frame_unittests.h

 // Copyright (c) 2006-2008 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_TEST_CHROME_FRAME_UNITTESTS_H_
 #define CHROME_FRAME_TEST_CHROME_FRAME_UNITTESTS_H_
 
 #include <atlbase.h>
 #include <atlcom.h>
 #include <string>
 #include <exdisp.h>
@@ skipping 110 matching lines @@
       ALLOW_THIS_IN_INITIALIZER_LIST(
           onloaderror_(this, &WebBrowserEventSink::OnLoadErrorInternal)),
       ALLOW_THIS_IN_INITIALIZER_LIST(
           onload_(this, &WebBrowserEventSink::OnLoadInternal)) {
   }
 
   ~WebBrowserEventSink() {
     Uninitialize();
   }
 
-  void Uninitialize() {
-    chrome_frame_ = NULL;
-    if (web_browser2_.get()) {
-      DispEventUnadvise(web_browser2_);
-      web_browser2_->Quit();
-      web_browser2_.Release();
-      // Give IE some time to quit and release our references
-      Sleep(1000);
-    }
-  }
+  void Uninitialize();
 
   // Helper function to launch IE and navigate to a URL.
   // Returns S_OK on success, S_FALSE if the test was not run, other
   // errors on failure.
   HRESULT LaunchIEAndNavigate(const std::wstring& navigate_url);
 
   HRESULT Navigate(const std::wstring& navigate_url);
 
+  // Set input focus to chrome frame window.
+  void SetFocusToChrome();
+
+  // Send keyboard input to the renderer window hosted in chrome using
+  // SendInput API
+  void SendInputToChrome(const std::string& input_string);
+
 BEGIN_COM_MAP(WebBrowserEventSink)
 END_COM_MAP()
 
 BEGIN_SINK_MAP(WebBrowserEventSink)
   SINK_ENTRY_INFO(0, DIID_DWebBrowserEvents2, DISPID_BEFORENAVIGATE2,
                  OnBeforeNavigate2Internal, &kBeforeNavigate2Info)
   SINK_ENTRY_INFO(0, DIID_DWebBrowserEvents2, DISPID_DOWNLOADBEGIN,
                   OnDownloadBegin, &kVoidMethodInfo)
   SINK_ENTRY_INFO(0, DIID_DWebBrowserEvents2, DISPID_NAVIGATECOMPLETE2,
                  OnNavigateComplete2Internal, &kNavigateComplete2Info)
@@ skipping 46 matching lines @@
   IWebBrowser2* web_browser2() {
     return web_browser2_.get();
   }
 
  protected:
   // IChromeFrame callbacks
   HRESULT OnLoadInternal(const VARIANT* param);
   HRESULT OnLoadErrorInternal(const VARIANT* param);
   HRESULT OnMessageInternal(const VARIANT* param);
 
-   void ConnectToChromeFrame();
+  void ConnectToChromeFrame();
+  HWND GetChromeRendererWindow();
 
  public:
   ScopedComPtr<IWebBrowser2> web_browser2_;
   ScopedComPtr<IChromeFrame> chrome_frame_;
   DispCallback<WebBrowserEventSink> onmessage_;
   DispCallback<WebBrowserEventSink> onloaderror_;
   DispCallback<WebBrowserEventSink> onload_;
 
  protected:
   static _ATL_FUNC_INFO kBeforeNavigate2Info;
   static _ATL_FUNC_INFO kNavigateComplete2Info;
   static _ATL_FUNC_INFO kNavigateErrorInfo;
   static _ATL_FUNC_INFO kNewWindow3Info;
   static _ATL_FUNC_INFO kVoidMethodInfo;
 };
 
 #endif  // CHROME_FRAME_TEST_CHROME_FRAME_UNITTESTS_H_