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
|
// 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/test/test_navigation_observer.h"
#include "chrome/test/ui_test_utils.h"
#include "testing/gtest/include/gtest/gtest.h"
TestNavigationObserver::JsInjectionReadyObserver::JsInjectionReadyObserver() {
}
TestNavigationObserver::JsInjectionReadyObserver::~JsInjectionReadyObserver() {
}
TestNavigationObserver::TestNavigationObserver(
NavigationController* controller,
TestNavigationObserver::JsInjectionReadyObserver*
js_injection_ready_observer,
int number_of_navigations)
: navigation_started_(false),
navigation_entry_committed_(false),
navigations_completed_(0),
number_of_navigations_(number_of_navigations),
js_injection_ready_observer_(js_injection_ready_observer),
done_(false),
running_(false) {
RegisterAsObserver(controller);
}
TestNavigationObserver::~TestNavigationObserver() {
}
void TestNavigationObserver::WaitForObservation() {
if (!done_) {
EXPECT_FALSE(running_);
running_ = true;
ui_test_utils::RunMessageLoop();
}
}
TestNavigationObserver::TestNavigationObserver(
TestNavigationObserver::JsInjectionReadyObserver*
js_injection_ready_observer,
int number_of_navigations)
: navigation_started_(false),
navigation_entry_committed_(false),
navigations_completed_(0),
number_of_navigations_(number_of_navigations),
js_injection_ready_observer_(js_injection_ready_observer),
done_(false),
running_(false) {
}
void TestNavigationObserver::RegisterAsObserver(
NavigationController* controller) {
registrar_.Add(this, content::NOTIFICATION_NAV_ENTRY_COMMITTED,
Source<NavigationController>(controller));
registrar_.Add(this, content::NOTIFICATION_LOAD_START,
Source<NavigationController>(controller));
registrar_.Add(this, content::NOTIFICATION_LOAD_STOP,
Source<NavigationController>(controller));
}
void TestNavigationObserver::Observe(
int type, const NotificationSource& source,
const NotificationDetails& details) {
switch (type) {
case content::NOTIFICATION_NAV_ENTRY_COMMITTED:
if (!navigation_entry_committed_ && js_injection_ready_observer_)
js_injection_ready_observer_->OnJsInjectionReady();
navigation_started_ = true;
navigation_entry_committed_ = true;
break;
case content::NOTIFICATION_LOAD_START:
navigation_started_ = true;
break;
case content::NOTIFICATION_LOAD_STOP:
if (navigation_started_ &&
++navigations_completed_ == number_of_navigations_) {
navigation_started_ = false;
done_ = true;
if (running_)
MessageLoopForUI::current()->Quit();
}
break;
default:
NOTREACHED();
}
}
|
