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
92
93
94
95
96
97
|
// Copyright (c) 2006-2008 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/browser_with_test_window_test.h"
#if defined(OS_WIN)
#include <ole2.h>
#endif // defined(OS_WIN)
#include "chrome/browser/browser.h"
#include "chrome/browser/tab_contents/navigation_entry.h"
#include "chrome/common/render_messages.h"
#include "chrome/test/testing_profile.h"
BrowserWithTestWindowTest::BrowserWithTestWindowTest()
: rph_factory_(),
rvh_factory_(&rph_factory_) {
#if defined(OS_WIN)
OleInitialize(NULL);
#endif
}
void BrowserWithTestWindowTest::SetUp() {
// NOTE: I have a feeling we're going to want virtual methods for creating
// these, as such they're in SetUp instead of the constructor.
profile_.reset(new TestingProfile());
browser_.reset(new Browser(Browser::TYPE_NORMAL, profile()));
window_.reset(new TestBrowserWindow(browser()));
browser_->set_window(window_.get());
}
BrowserWithTestWindowTest::~BrowserWithTestWindowTest() {
// Make sure we close all tabs, otherwise Browser isn't happy in its
// destructor.
browser()->CloseAllTabs();
// A Task is leaked if we don't destroy everything, then run the message
// loop.
browser_.reset(NULL);
window_.reset(NULL);
profile_.reset(NULL);
MessageLoop::current()->PostTask(FROM_HERE, new MessageLoop::QuitTask);
MessageLoop::current()->Run();
#if defined(OS_WIN)
OleUninitialize();
#endif
}
TestRenderViewHost* BrowserWithTestWindowTest::TestRenderViewHostForTab(
TabContents* tab_contents) {
return static_cast<TestRenderViewHost*>(
tab_contents->render_view_host());
}
void BrowserWithTestWindowTest::AddTab(Browser* browser, const GURL& url) {
TabContents* new_tab = browser->AddTabWithURL(
url, GURL(), PageTransition::TYPED, 0, TabStripModel::ADD_SELECTED, NULL,
std::string());
CommitPendingLoad(&new_tab->controller());
}
void BrowserWithTestWindowTest::CommitPendingLoad(
NavigationController* controller) {
if (!controller->pending_entry())
return; // Nothing to commit.
TestRenderViewHost* test_rvh =
TestRenderViewHostForTab(controller->tab_contents());
MockRenderProcessHost* mock_rph = static_cast<MockRenderProcessHost*>(
test_rvh->process());
// For new navigations, we need to send a larger page ID. For renavigations,
// we need to send the preexisting page ID. We can tell these apart because
// renavigations will have a pending_entry_index while new ones won't (they'll
// just have a standalong pending_entry that isn't in the list already).
if (controller->pending_entry_index() >= 0) {
test_rvh->SendNavigate(controller->pending_entry()->page_id(),
controller->pending_entry()->url());
} else {
test_rvh->SendNavigate(mock_rph->max_page_id() + 1,
controller->pending_entry()->url());
}
}
void BrowserWithTestWindowTest::NavigateAndCommit(
NavigationController* controller,
const GURL& url) {
controller->LoadURL(url, GURL(), 0);
CommitPendingLoad(controller);
}
void BrowserWithTestWindowTest::NavigateAndCommitActiveTab(const GURL& url) {
NavigateAndCommit(&browser()->GetSelectedTabContents()->controller(), url);
}
|
