cef/tests/unittests/test_handler.cc

128 lines
3.0 KiB
C++

// 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.
#include "test_handler.h"
#include "include/cef_runnable.h"
namespace {
void NotifyEvent(base::WaitableEvent* event)
{
event->Signal();
}
} // namespace
// TestHandler
TestHandler::TestHandler()
: browser_hwnd_(NULL),
completion_event_(true, false)
{
}
TestHandler::~TestHandler()
{
}
void TestHandler::OnAfterCreated(CefRefPtr<CefBrowser> browser)
{
AutoLock lock_scope(this);
if(!browser->IsPopup())
{
// Keep the main child window, but not popup windows
browser_ = browser;
browser_hwnd_ = browser->GetWindowHandle();
}
}
void TestHandler::OnBeforeClose(CefRefPtr<CefBrowser> browser)
{
AutoLock lock_scope(this);
if(browser_hwnd_ == browser->GetWindowHandle())
{
// Free the browser pointer so that the browser can be destroyed
browser_ = NULL;
browser_hwnd_ = NULL;
// Signal that the test is now complete.
completion_event_.Signal();
}
}
bool TestHandler::OnBeforeResourceLoad(CefRefPtr<CefBrowser> browser,
CefRefPtr<CefRequest> request,
CefString& redirectUrl,
CefRefPtr<CefStreamReader>& resourceStream,
CefRefPtr<CefResponse> response,
int loadFlags)
{
AutoLock lock_scope(this);
if(resource_map_.size() > 0) {
CefString url = request->GetURL();
ResourceMap::const_iterator it = resource_map_.find(url);
if(it != resource_map_.end()) {
// Return the previously mapped resource
resourceStream = CefStreamReader::CreateForData(
(void*)it->second.first.c_str(), it->second.first.length());
response->SetMimeType(it->second.second);
response->SetStatus(200);
}
}
return false;
}
void TestHandler::ExecuteTest()
{
// Run the test
RunTest();
// Wait for the test to complete
completion_event_.Wait();
// Reset the event so the same test can be executed again.
completion_event_.Reset();
}
void TestHandler::DestroyTest()
{
AutoLock lock_scope(this);
if(browser_hwnd_ != NULL)
browser_->CloseBrowser();
}
void TestHandler::CreateBrowser(const CefString& url)
{
CefWindowInfo windowInfo;
CefBrowserSettings settings;
#if defined(OS_WIN)
windowInfo.SetAsPopup(NULL, "CefUnitTest");
windowInfo.m_dwStyle |= WS_VISIBLE;
#endif
CefBrowser::CreateBrowser(windowInfo, this, url, settings);
}
void TestHandler::AddResource(const CefString& key, const std::string& value,
const CefString& mimeType)
{
resource_map_.insert(std::make_pair(key, std::make_pair(value, mimeType)));
}
void TestHandler::ClearResources()
{
resource_map_.clear();
}
// global functions
void WaitForThread(CefThreadId thread_id)
{
base::WaitableEvent event(true, false);
CefPostTask(thread_id, NewCefRunnableFunction(&NotifyEvent, &event));
event.Wait();
}