Branch CEF3 files from /branches/cef3 to /trunk/cef3 (issue #564).

git-svn-id: https://chromiumembedded.googlecode.com/svn/trunk@571 5089003a-bbd8-11dd-ad1f-f1f9622dbc98

tests/unittests/test_handler.h

@@ -0,0 +1,117 @@
+// Copyright (c) 2011 The Chromium Embedded Framework 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 CEF_TESTS_UNITTESTS_TEST_HANDLER_H_
+#define CEF_TESTS_UNITTESTS_TEST_HANDLER_H_
+#pragma once
+
+#include <map>
+#include <string>
+#include <utility>
+
+#include "include/cef_browser.h"
+#include "include/cef_client.h"
+#include "include/cef_frame.h"
+#include "include/cef_task.h"
+#include "base/synchronization/waitable_event.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+class TrackCallback {
+ public:
+  TrackCallback(): gotit_(false) {}
+  void yes() { gotit_ = true; }
+  bool isSet() { return gotit_; }
+  void reset() { gotit_ = false; }
+  operator bool() const { return gotit_; }
+ protected:
+  bool gotit_;
+};
+
+// Base implementation of CefClient for unit tests. Add new interfaces as needed
+// by test cases.
+class TestHandler : public CefClient,
+                    public CefDisplayHandler,
+                    public CefLifeSpanHandler,
+                    public CefLoadHandler,
+                    public CefRequestHandler {
+ public:
+  TestHandler();
+  virtual ~TestHandler();
+
+  // Implement this method to run the test
+  virtual void RunTest() =0;
+
+  // CefClient methods. Add new methods as needed by test cases.
+  virtual CefRefPtr<CefDisplayHandler> GetDisplayHandler() OVERRIDE {
+    return this;
+  }
+  virtual CefRefPtr<CefLifeSpanHandler> GetLifeSpanHandler() OVERRIDE {
+    return this;
+  }
+  virtual CefRefPtr<CefLoadHandler> GetLoadHandler() OVERRIDE {
+    return this;
+  }
+  virtual CefRefPtr<CefRequestHandler> GetRequestHandler() OVERRIDE {
+    return this;
+  }
+
+  // CefLifeSpanHandler methods
+  virtual void OnAfterCreated(CefRefPtr<CefBrowser> browser) OVERRIDE;
+  virtual void OnBeforeClose(CefRefPtr<CefBrowser> browser) OVERRIDE;
+
+  // CefRequestHandler methods
+  virtual CefRefPtr<CefResourceHandler> GetResourceHandler(
+      CefRefPtr<CefBrowser> browser,
+      CefRefPtr<CefFrame> frame,
+      CefRefPtr<CefRequest> request) OVERRIDE;
+
+  CefRefPtr<CefBrowser> GetBrowser() { return browser_; }
+  int GetBrowserId() { return browser_id_; }
+
+  // Called by the test function to execute the test.  This method blocks until
+  // the test is complete. Do not reference the object after this method
+  // returns.
+  void ExecuteTest();
+
+ protected:
+  // Destroy the browser window. Once the window is destroyed test completion
+  // will be signaled.
+  void DestroyTest();
+
+  void CreateBrowser(const CefString& url);
+
+  void AddResource(const CefString& url,
+                   const std::string& content,
+                   const CefString& mimeType);
+  void ClearResources();
+
+ private:
+  // The child browser window
+  CefRefPtr<CefBrowser> browser_;
+
+  // The browser window identifier
+  int browser_id_;
+
+  // Handle used to notify when the test is complete
+  base::WaitableEvent completion_event_;
+
+  // Map of resources that can be automatically loaded
+  typedef std::map<CefString, std::pair<std::string, CefString> > ResourceMap;
+  ResourceMap resource_map_;
+
+  // Include the default reference counting implementation.
+  IMPLEMENT_REFCOUNTING(TestHandler);
+  // Include the default locking implementation.
+  IMPLEMENT_LOCKING(TestHandler);
+};
+
+
+// Post a task to the specified thread and wait for the task to execute as
+// indication that all previously pending tasks on that thread have completed.
+void WaitForThread(CefThreadId thread_id);
+
+#define WaitForIOThread() WaitForThread(TID_IO)
+#define WaitForUIThread() WaitForThread(TID_UI)
+
+#endif  // CEF_TESTS_UNITTESTS_TEST_HANDLER_H_