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
content / public / test / javascript_test_observer.cc [blame]
// Copyright 2014 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/public/test/javascript_test_observer.h"
#include "base/run_loop.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/test_utils.h"
namespace content {
TestMessageHandler::TestMessageHandler() : ok_(true) {
}
TestMessageHandler::~TestMessageHandler() {
}
void TestMessageHandler::SetError(const std::string& message) {
ok_ = false;
error_message_ = message;
}
void TestMessageHandler::Reset() {
ok_ = true;
error_message_.clear();
}
JavascriptTestObserver::JavascriptTestObserver(WebContents* web_contents,
TestMessageHandler* handler)
: WebContentsObserver(web_contents),
handler_(handler),
running_(false),
finished_(false) {
Reset();
}
JavascriptTestObserver::~JavascriptTestObserver() {
}
bool JavascriptTestObserver::Run() {
// Messages may have arrived before Run was called.
if (!finished_) {
CHECK(!running_);
running_ = true;
loop_.Run();
running_ = false;
}
return handler_->ok();
}
void JavascriptTestObserver::Reset() {
CHECK(!running_);
running_ = false;
finished_ = false;
handler_->Reset();
}
void JavascriptTestObserver::DomOperationResponse(
RenderFrameHost* render_frame_host,
const std::string& json_string) {
// We might receive responses for other script execution, but we only
// care about the test finished message.
TestMessageHandler::MessageResponse response =
handler_->HandleMessage(json_string);
if (response == TestMessageHandler::DONE) {
EndTest();
} else {
Continue();
}
}
void JavascriptTestObserver::Continue() {
}
void JavascriptTestObserver::EndTest() {
finished_ = true;
if (running_) {
running_ = false;
loop_.QuitWhenIdle();
}
}
} // namespace content