blob: becdd834ab46789f6874fbeafca165f74e98a839 (
plain)
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
|
// Copyright 2014 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 "content/public/test/javascript_test_observer.h"
#include "content/public/browser/notification_types.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)
: handler_(handler),
running_(false),
finished_(false) {
Reset();
registrar_.Add(this,
NOTIFICATION_DOM_OPERATION_RESPONSE,
Source<WebContents>(web_contents));
}
JavascriptTestObserver::~JavascriptTestObserver() {
}
bool JavascriptTestObserver::Run() {
// Messages may have arrived before Run was called.
if (!finished_) {
CHECK(!running_);
running_ = true;
RunMessageLoop();
running_ = false;
}
return handler_->ok();
}
void JavascriptTestObserver::Reset() {
CHECK(!running_);
running_ = false;
finished_ = false;
handler_->Reset();
}
void JavascriptTestObserver::Observe(
int type,
const NotificationSource& source,
const NotificationDetails& details) {
CHECK(type == NOTIFICATION_DOM_OPERATION_RESPONSE);
Details<std::string> dom_op_result(details);
// We might receive responses for other script execution, but we only
// care about the test finished message.
TestMessageHandler::MessageResponse response =
handler_->HandleMessage(*dom_op_result.ptr());
if (response == TestMessageHandler::DONE) {
EndTest();
} else {
Continue();
}
}
void JavascriptTestObserver::Continue() {
}
void JavascriptTestObserver::EndTest() {
finished_ = true;
if (running_) {
running_ = false;
base::MessageLoopForUI::current()->QuitWhenIdle();
}
}
} // namespace content
|
