130 lines
3.1 KiB
C++
130 lines
3.1 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()
|
|
{
|
|
Lock();
|
|
#if defined(OS_WIN)
|
|
if(browser_hwnd_ != NULL)
|
|
PostMessage(browser_hwnd_, WM_CLOSE, 0, 0);
|
|
#endif
|
|
Unlock();
|
|
}
|
|
|
|
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();
|
|
}
|