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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/lifetime/application_lifetime.h"
#include "base/command_line.h"
#include "base/test/mock_callback.h"
#include "chrome/browser/browser_process.h"
#include "chrome/browser/chrome_browser_main.h"
#include "chrome/browser/first_run/scoped_relaunch_chrome_browser_override.h"
#include "chrome/browser/lifetime/application_lifetime_desktop.h"
#include "chrome/browser/lifetime/browser_shutdown.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/tabs/tab_strip_model.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/common/pref_names.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/prefs/testing_pref_service.h"
#include "content/public/browser/browser_task_traits.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_test.h"
#include "content/public/test/browser_test_utils.h"
#include "content/public/test/test_utils.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
// A gMock matcher that is satisfied when its argument is a command line
// containing a given switch.
MATCHER_P(HasSwitch, switch_name, "") {
return arg.HasSwitch(switch_name);
}
} // namespace
// Test Fixture for testing chrome::AttemptRestart method.
//
// The SetupCommandLine method takes care of initiating the browser with the
// incognito or guest switch. This switch is accessed via GetParam() method and
// the switches are passed during the testcase declaration which invokes the
// same test once per switch.
//
// The SetUpInProcessBrowserTestFixture method, creates a mock callback which
// gets hooked when chrome tries to Relaunch the browser. The hooking is thanks
// to the upgrade_util::ScopedRelaunchChromeBrowserOverride. Our mock callback
// runs the MATCHER_P.
//
// It is template parameterized on the type of switches::kIncognito and
// switches::kGuest which is const char*
class AttemptRestartTest : public InProcessBrowserTest,
public testing::WithParamInterface<const char*> {
protected:
AttemptRestartTest() = default;
// InProcessBrowserTest:
void SetUpCommandLine(base::CommandLine* command_line) override {
command_line->AppendSwitch(GetParam());
}
void SetUpInProcessBrowserTestFixture() override {
// Expect a browser relaunch late in browser shutdown.
EXPECT_CALL(mock_relaunch_callback_,
Run(testing::Not(HasSwitch(GetParam()))));
}
private:
base::MockCallback<upgrade_util::RelaunchChromeBrowserCallback>
mock_relaunch_callback_;
upgrade_util::ScopedRelaunchChromeBrowserOverride relaunch_chrome_override_{
mock_relaunch_callback_.Get()};
};
INSTANTIATE_TEST_SUITE_P(,
AttemptRestartTest,
testing::Values(switches::kIncognito,
switches::kGuest));
IN_PROC_BROWSER_TEST_P(AttemptRestartTest, AttemptRestartWithOTRProfiles) {
// We will now attempt restart, prior to (crbug.com/999085)
// the new session after restart defaulted to the browser type
// of the last session. Now, we will restart always to regular mode.
chrome::AttemptRestart();
}
// Ensures that a relaunch actually happens without interaction when a
// beforeunload handler is in place when RelaunchIgnoreUnloadHandlers is called.
class RelaunchIgnoreUnloadHandlersTest : public InProcessBrowserTest {
protected:
RelaunchIgnoreUnloadHandlersTest() = default;
// InProcessBrowserTest:
void SetUpInProcessBrowserTestFixture() override {
// Expect a browser relaunch late in browser shutdown.
EXPECT_CALL(mock_relaunch_callback_, Run(::testing::_));
}
private:
base::MockCallback<upgrade_util::RelaunchChromeBrowserCallback>
mock_relaunch_callback_;
upgrade_util::ScopedRelaunchChromeBrowserOverride relaunch_chrome_override_{
mock_relaunch_callback_.Get()};
};
IN_PROC_BROWSER_TEST_F(RelaunchIgnoreUnloadHandlersTest, Do) {
ASSERT_TRUE(
ui_test_utils::NavigateToURL(browser(), GURL(url::kAboutBlankURL)));
content::WebContents* tab =
browser()->tab_strip_model()->GetActiveWebContents();
ASSERT_TRUE(
content::ExecJs(tab,
"window.addEventListener('beforeunload',"
"function(event) { event.returnValue = 'Foo'; });"));
content::PrepContentsForBeforeUnloadTest(tab);
content::GetUIThreadTaskRunner({})->PostTask(
FROM_HERE, base::BindOnce(&chrome::RelaunchIgnoreUnloadHandlers));
ui_test_utils::WaitForBrowserToClose(browser());
}
using ApplicationLifetimeTest = InProcessBrowserTest;
IN_PROC_BROWSER_TEST_F(ApplicationLifetimeTest, AttemptRestart) {
ASSERT_TRUE(g_browser_process);
PrefService* testing_pref_service = g_browser_process->local_state();
EXPECT_FALSE(testing_pref_service->GetBoolean(prefs::kWasRestarted));
chrome::AttemptRestart();
EXPECT_TRUE(testing_pref_service->GetBoolean(prefs::kWasRestarted));
// Cancel the effects of us calling chrome::AttemptRestart. Otherwise
// this test and tests ran after this one will fail.
browser_shutdown::SetTryingToQuit(false);
}
|
