1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
// Copyright (c) 2011 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.
#include "chrome/browser/ui/webui/web_ui_test_handler.h"
#include "base/bind.h"
#include "base/bind_helpers.h"
#include "base/utf_string_conversions.h"
#include "base/values.h"
#include "chrome/common/render_messages.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/browser/renderer_host/render_view_host.h"
#include "content/public/browser/notification_details.h"
#include "content/public/browser/notification_registrar.h"
#include "content/public/browser/notification_types.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_ui.h"
WebUITestHandler::WebUITestHandler()
: test_done_(false),
test_succeeded_(false),
run_test_done_(false),
run_test_succeeded_(false),
is_waiting_(false) {
}
void WebUITestHandler::PreloadJavaScript(const string16& js_text,
RenderViewHost* preload_host) {
DCHECK(preload_host);
preload_host->Send(new ChromeViewMsg_WebUIJavaScript(
preload_host->routing_id(), string16(), js_text, 0,
false));
}
void WebUITestHandler::RunJavaScript(const string16& js_text) {
web_ui()->GetWebContents()->GetRenderViewHost()->ExecuteJavascriptInWebFrame(
string16(), js_text);
}
bool WebUITestHandler::RunJavaScriptTestWithResult(const string16& js_text) {
test_succeeded_ = false;
run_test_succeeded_ = false;
RenderViewHost* rvh = web_ui()->GetWebContents()->GetRenderViewHost();
content::NotificationRegistrar notification_registrar;
notification_registrar.Add(
this, content::NOTIFICATION_EXECUTE_JAVASCRIPT_RESULT,
content::Source<RenderViewHost>(rvh));
rvh->ExecuteJavascriptInWebFrameNotifyResult(string16(), js_text);
return WaitForResult();
}
void WebUITestHandler::RegisterMessages() {
web_ui()->RegisterMessageCallback("testResult",
base::Bind(&WebUITestHandler::HandleTestResult, base::Unretained(this)));
}
void WebUITestHandler::HandleTestResult(const ListValue* test_result) {
// Quit the message loop if |is_waiting_| so waiting process can get result or
// error. To ensure this gets done, do this before ASSERT* calls.
if (is_waiting_)
MessageLoopForUI::current()->Quit();
SCOPED_TRACE("WebUITestHandler::HandleTestResult");
EXPECT_FALSE(test_done_);
test_done_ = true;
test_succeeded_ = false;
ASSERT_TRUE(test_result->GetBoolean(0, &test_succeeded_));
if (!test_succeeded_) {
std::string message;
ASSERT_TRUE(test_result->GetString(1, &message));
LOG(ERROR) << message;
}
}
void WebUITestHandler::Observe(int type,
const content::NotificationSource& source,
const content::NotificationDetails& details) {
// Quit the message loop if |is_waiting_| so waiting process can get result or
// error. To ensure this gets done, do this before ASSERT* calls.
if (is_waiting_)
MessageLoopForUI::current()->Quit();
ASSERT_EQ(content::NOTIFICATION_EXECUTE_JAVASCRIPT_RESULT, type);
SCOPED_TRACE("WebUITestHandler::Observe");
EXPECT_FALSE(run_test_done_);
run_test_done_ = true;
run_test_succeeded_ = false;
Value* value = content::Details<std::pair<int, Value*> >(details)->second;
ASSERT_TRUE(value->GetAsBoolean(&run_test_succeeded_));
}
bool WebUITestHandler::WaitForResult() {
SCOPED_TRACE("WebUITestHandler::WaitForResult");
test_done_ = false;
run_test_done_ = false;
is_waiting_ = true;
// Either sync test completion or the testDone() will cause message loop
// to quit.
ui_test_utils::RunMessageLoop();
// Run a second message loop when not |run_test_done_| so that the sync test
// completes, or |run_test_succeeded_| but not |test_done_| so async tests
// complete.
if (!run_test_done_ || (run_test_succeeded_ && !test_done_)) {
ui_test_utils::RunMessageLoop();
}
is_waiting_ = false;
// To succeed the test must execute as well as pass the test.
return run_test_succeeded_ && test_succeeded_;
}
|
