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
|
// 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/base/test_html_dialog_observer.h"
#include "chrome/common/chrome_notification_types.h"
#include "content/test/js_injection_ready_observer.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/public/browser/notification_service.h"
#include "content/browser/tab_contents/navigation_controller.h"
#include "content/browser/tab_contents/tab_contents.h"
#include "content/browser/webui/web_ui.h"
#include "content/public/browser/notification_details.h"
#include "content/public/browser/notification_source.h"
#include "content/public/browser/notification_types.h"
TestHtmlDialogObserver::TestHtmlDialogObserver(
JsInjectionReadyObserver* js_injection_ready_observer)
: js_injection_ready_observer_(js_injection_ready_observer),
web_ui_(NULL), done_(false), running_(false) {
registrar_.Add(this, chrome::NOTIFICATION_HTML_DIALOG_SHOWN,
content::NotificationService::AllSources());
}
TestHtmlDialogObserver::~TestHtmlDialogObserver() {
}
void TestHtmlDialogObserver::Observe(
int type,
const content::NotificationSource& source,
const content::NotificationDetails& details) {
switch (type) {
case chrome::NOTIFICATION_HTML_DIALOG_SHOWN:
if (js_injection_ready_observer_) {
js_injection_ready_observer_->OnJsInjectionReady(
content::Details<RenderViewHost>(details).ptr());
}
web_ui_ = content::Source<WebUI>(source).ptr();
registrar_.Remove(this, chrome::NOTIFICATION_HTML_DIALOG_SHOWN,
content::NotificationService::AllSources());
// Wait for navigation on the new WebUI instance to complete. This depends
// on receiving the notification of the HtmlDialog being shown before the
// NavigationController finishes loading. The HtmlDialog notification is
// issued from html_dialog_ui.cc on RenderView creation which results from
// the call to render_manager_.Navigate in the method
// TabContents::NavigateToEntry. The new RenderView is later told to
// navigate in this method, ensuring that this is not a race condition.
registrar_.Add(this, content::NOTIFICATION_LOAD_STOP,
content::Source<NavigationController>(
&web_ui_->tab_contents()->controller()));
break;
case content::NOTIFICATION_LOAD_STOP:
DCHECK(web_ui_);
registrar_.Remove(this, content::NOTIFICATION_LOAD_STOP,
content::Source<NavigationController>(
&web_ui_->tab_contents()->controller()));
done_ = true;
// If the message loop is running stop it.
if (running_) {
running_ = false;
MessageLoopForUI::current()->Quit();
}
break;
default:
NOTREACHED();
};
}
WebUI* TestHtmlDialogObserver::GetWebUI() {
if (!done_) {
EXPECT_FALSE(running_);
running_ = true;
ui_test_utils::RunMessageLoop();
}
return web_ui_;
}
|
