mirror of
https://bitbucket.org/chromiumembedded/cef
synced 2024-12-16 03:19:21 +01:00
27fac646d9
git-svn-id: https://chromiumembedded.googlecode.com/svn/trunk@1305 5089003a-bbd8-11dd-ad1f-f1f9622dbc98
228 lines
5.7 KiB
C++
228 lines
5.7 KiB
C++
// Copyright (c) 2013 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 "tests/unittests/test_handler.h"
|
|
#include "base/logging.h"
|
|
#include "include/cef_command_line.h"
|
|
#include "include/cef_runnable.h"
|
|
#include "include/cef_stream.h"
|
|
#include "include/wrapper/cef_stream_resource_handler.h"
|
|
|
|
namespace {
|
|
|
|
void NotifyEvent(base::WaitableEvent* event) {
|
|
event->Signal();
|
|
}
|
|
|
|
} // namespace
|
|
|
|
|
|
// TestHandler::CompletionState
|
|
|
|
TestHandler::CompletionState::CompletionState(int total)
|
|
: total_(total),
|
|
count_(0),
|
|
event_(true, false) {
|
|
}
|
|
|
|
void TestHandler::CompletionState::TestComplete() {
|
|
if (++count_ == total_) {
|
|
// Signal that the test is now complete.
|
|
event_.Signal();
|
|
count_ = 0;
|
|
}
|
|
}
|
|
|
|
void TestHandler::CompletionState::WaitForTests() {
|
|
// Wait for the test to complete
|
|
event_.Wait();
|
|
|
|
// Reset the event so the same test can be executed again.
|
|
event_.Reset();
|
|
}
|
|
|
|
|
|
// TestHandler::Collection
|
|
|
|
TestHandler::Collection::Collection(CompletionState* completion_state)
|
|
: completion_state_(completion_state) {
|
|
DCHECK(completion_state_);
|
|
}
|
|
|
|
void TestHandler::Collection::AddTestHandler(TestHandler* test_handler) {
|
|
DCHECK_EQ(test_handler->completion_state_, completion_state_);
|
|
handler_list_.push_back(test_handler);
|
|
}
|
|
|
|
void TestHandler::Collection::ExecuteTests() {
|
|
DCHECK_GT(handler_list_.size(), 0UL);
|
|
|
|
TestHandlerList::const_iterator it;
|
|
|
|
it = handler_list_.begin();
|
|
for (; it != handler_list_.end(); ++it)
|
|
(*it)->SetupTest();
|
|
|
|
completion_state_->WaitForTests();
|
|
|
|
it = handler_list_.begin();
|
|
for (; it != handler_list_.end(); ++it)
|
|
(*it)->RunTest();
|
|
|
|
completion_state_->WaitForTests();
|
|
}
|
|
|
|
|
|
// TestHandler
|
|
|
|
int TestHandler::browser_count_ = 0;
|
|
|
|
TestHandler::TestHandler(CompletionState* completion_state)
|
|
: browser_id_(0) {
|
|
if (completion_state) {
|
|
completion_state_ = completion_state;
|
|
completion_state_owned_ = false;
|
|
} else {
|
|
completion_state_ = new CompletionState(1);
|
|
completion_state_owned_ = true;
|
|
}
|
|
}
|
|
|
|
TestHandler::~TestHandler() {
|
|
if (completion_state_owned_)
|
|
delete completion_state_;
|
|
}
|
|
|
|
void TestHandler::OnAfterCreated(CefRefPtr<CefBrowser> browser) {
|
|
browser_count_++;
|
|
|
|
AutoLock lock_scope(this);
|
|
if (!browser->IsPopup()) {
|
|
// Keep the main child window, but not popup windows
|
|
browser_ = browser;
|
|
browser_id_ = browser->GetIdentifier();
|
|
}
|
|
}
|
|
|
|
void TestHandler::OnBeforeClose(CefRefPtr<CefBrowser> browser) {
|
|
{
|
|
AutoLock lock_scope(this);
|
|
if (browser_id_ == browser->GetIdentifier()) {
|
|
// Free the browser pointer so that the browser can be destroyed
|
|
browser_ = NULL;
|
|
browser_id_ = 0;
|
|
|
|
// Signal that the test is now complete.
|
|
completion_state_->TestComplete();
|
|
}
|
|
}
|
|
|
|
browser_count_--;
|
|
}
|
|
|
|
CefRefPtr<CefResourceHandler> TestHandler::GetResourceHandler(
|
|
CefRefPtr<CefBrowser> browser,
|
|
CefRefPtr<CefFrame> frame,
|
|
CefRefPtr<CefRequest> request) {
|
|
AutoLock lock_scope(this);
|
|
|
|
if (resource_map_.size() > 0) {
|
|
CefString url = request->GetURL();
|
|
|
|
// Ignore the query component, if any.
|
|
std::string urlStr = url;
|
|
int idx = urlStr.find('?');
|
|
if (idx > 0)
|
|
urlStr = urlStr.substr(0, idx);
|
|
|
|
ResourceMap::const_iterator it = resource_map_.find(urlStr);
|
|
if (it != resource_map_.end()) {
|
|
// Return the previously mapped resource
|
|
CefRefPtr<CefStreamReader> stream =
|
|
CefStreamReader::CreateForData(
|
|
static_cast<void*>(const_cast<char*>(it->second.first.c_str())),
|
|
it->second.first.length());
|
|
return new CefStreamResourceHandler(it->second.second, stream);
|
|
}
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
void TestHandler::ExecuteTest() {
|
|
DCHECK_EQ(completion_state_->total(), 1);
|
|
|
|
// Run the test
|
|
RunTest();
|
|
|
|
// Wait for the test to complete
|
|
completion_state_->WaitForTests();
|
|
}
|
|
|
|
void TestHandler::SetupComplete() {
|
|
// Signal that the test setup is complete.
|
|
completion_state_->TestComplete();
|
|
}
|
|
|
|
void TestHandler::DestroyTest() {
|
|
AutoLock lock_scope(this);
|
|
if (browser_id_ != 0)
|
|
browser_->GetHost()->CloseBrowser(false);
|
|
}
|
|
|
|
void TestHandler::CreateBrowser(const CefString& url) {
|
|
CefWindowInfo windowInfo;
|
|
CefBrowserSettings settings;
|
|
#if defined(OS_WIN)
|
|
windowInfo.SetAsPopup(NULL, "CefUnitTest");
|
|
windowInfo.style |= WS_VISIBLE;
|
|
#endif
|
|
CefBrowserHost::CreateBrowser(windowInfo, this, url, settings);
|
|
}
|
|
|
|
void TestHandler::AddResource(const std::string& url,
|
|
const std::string& content,
|
|
const std::string& mimeType) {
|
|
// Ignore the query component, if any.
|
|
std::string urlStr = url;
|
|
int idx = urlStr.find('?');
|
|
if (idx > 0)
|
|
urlStr = urlStr.substr(0, idx);
|
|
|
|
resource_map_.insert(
|
|
std::make_pair(urlStr, std::make_pair(content, 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();
|
|
}
|
|
|
|
void WaitForThread(CefRefPtr<CefTaskRunner> task_runner) {
|
|
base::WaitableEvent event(true, false);
|
|
task_runner->PostTask(NewCefRunnableFunction(&NotifyEvent, &event));
|
|
event.Wait();
|
|
}
|
|
|
|
bool TestFailed() {
|
|
CefRefPtr<CefCommandLine> command_line =
|
|
CefCommandLine::GetGlobalCommandLine();
|
|
if (command_line->HasSwitch("single-process")) {
|
|
// Check for a failure on the current test only.
|
|
return ::testing::UnitTest::GetInstance()->current_test_info()->result()->
|
|
Failed();
|
|
} else {
|
|
// Check for any global failure.
|
|
return ::testing::UnitTest::GetInstance()->Failed();
|
|
}
|
|
}
|