OLD | NEW |
| (Empty) |
1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include "chrome/browser/tab_contents/test_tab_contents.h" | |
6 | |
7 #include <utility> | |
8 | |
9 #include "chrome/browser/browser_url_handler.h" | |
10 #include "chrome/browser/renderer_host/mock_render_process_host.h" | |
11 #include "chrome/browser/renderer_host/render_view_host.h" | |
12 #include "chrome/browser/renderer_host/site_instance.h" | |
13 #include "chrome/browser/renderer_host/test/test_render_view_host.h" | |
14 #include "chrome/browser/tab_contents/infobar_delegate.h" | |
15 #include "chrome/common/notification_details.h" | |
16 #include "chrome/common/notification_source.h" | |
17 #include "chrome/common/page_transition_types.h" | |
18 | |
19 TestTabContents::TestTabContents(Profile* profile, SiteInstance* instance) | |
20 : TabContents(profile, instance, MSG_ROUTING_NONE, NULL, NULL), | |
21 transition_cross_site(false) { | |
22 // Listen for infobar events so we can call InfoBarClosed() on the infobar | |
23 // delegates and give them an opportunity to delete themselves. (Since we | |
24 // have no InfobarContainer in TestTabContents, InfoBarClosed() is not called | |
25 // most likely leading to the infobar delegates being leaked.) | |
26 Source<TabContents> source(this); | |
27 registrar_.Add(this, NotificationType::TAB_CONTENTS_INFOBAR_REMOVED, | |
28 source); | |
29 registrar_.Add(this, NotificationType::TAB_CONTENTS_INFOBAR_REPLACED, | |
30 source); | |
31 } | |
32 | |
33 void TestTabContents::Observe(NotificationType type, | |
34 const NotificationSource& source, | |
35 const NotificationDetails& details) { | |
36 // TabContents does not handle TAB_CONTENTS_INFOBAR_* so we don't pass it | |
37 // these notifications. | |
38 switch (type.value) { | |
39 case NotificationType::TAB_CONTENTS_INFOBAR_REMOVED: | |
40 Details<InfoBarDelegate>(details).ptr()->InfoBarClosed(); | |
41 break; | |
42 case NotificationType::TAB_CONTENTS_INFOBAR_REPLACED: | |
43 Details<std::pair<InfoBarDelegate*, InfoBarDelegate*> >(details).ptr()-> | |
44 first->InfoBarClosed(); | |
45 break; | |
46 default: | |
47 TabContents::Observe(type, source, details); | |
48 break; | |
49 } | |
50 } | |
51 | |
52 TestRenderViewHost* TestTabContents::pending_rvh() const { | |
53 return static_cast<TestRenderViewHost*>( | |
54 render_manager_.pending_render_view_host_); | |
55 } | |
56 | |
57 bool TestTabContents::CreateRenderViewForRenderManager( | |
58 RenderViewHost* render_view_host) { | |
59 // This will go to a TestRenderViewHost. | |
60 render_view_host->CreateRenderView(string16()); | |
61 return true; | |
62 } | |
63 | |
64 TabContents* TestTabContents::Clone() { | |
65 TabContents* tc = new TestTabContents( | |
66 profile(), SiteInstance::CreateSiteInstance(profile())); | |
67 tc->controller().CopyStateFrom(controller_); | |
68 return tc; | |
69 } | |
70 | |
71 void TestTabContents::NavigateAndCommit(const GURL& url) { | |
72 controller().LoadURL(url, GURL(), PageTransition::LINK); | |
73 GURL loaded_url(url); | |
74 bool reverse_on_redirect = false; | |
75 BrowserURLHandler::RewriteURLIfNecessary( | |
76 &loaded_url, profile(), &reverse_on_redirect); | |
77 | |
78 // LoadURL created a navigation entry, now simulate the RenderView sending | |
79 // a notification that it actually navigated. | |
80 CommitPendingNavigation(); | |
81 } | |
82 | |
83 void TestTabContents::CommitPendingNavigation() { | |
84 // If we are doing a cross-site navigation, this simulates the current RVH | |
85 // notifying that it has unloaded so the pending RVH is resumed and can | |
86 // navigate. | |
87 ProceedWithCrossSiteNavigation(); | |
88 TestRenderViewHost* rvh = pending_rvh(); | |
89 if (!rvh) | |
90 rvh = static_cast<TestRenderViewHost*>(render_manager_.current_host()); | |
91 | |
92 const NavigationEntry* entry = controller().pending_entry(); | |
93 DCHECK(entry); | |
94 int page_id = entry->page_id(); | |
95 if (page_id == -1) { | |
96 // It's a new navigation, assign a never-seen page id to it. | |
97 page_id = | |
98 static_cast<MockRenderProcessHost*>(rvh->process())->max_page_id() + 1; | |
99 } | |
100 rvh->SendNavigate(page_id, entry->url()); | |
101 } | |
102 | |
103 void TestTabContents::ProceedWithCrossSiteNavigation() { | |
104 if (!pending_rvh()) | |
105 return; | |
106 render_manager_.ShouldClosePage(true, true); | |
107 } | |
OLD | NEW |