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
|
// Copyright 2025 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/extensions/app_tab_helper.h"
#include "base/run_loop.h"
#include "chrome/browser/extensions/extension_service_test_with_install.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/test/base/test_browser_window.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/test_navigation_observer.h"
#include "content/public/test/web_contents_tester.h"
#include "extensions/browser/extension_registrar.h"
#include "extensions/test/test_extension_dir.h"
#include "url/origin.h"
namespace extensions {
class AppTabHelperUnitTest : public ExtensionServiceTestWithInstall {
public:
void SetUp() override {
ExtensionServiceTestWithInstall::SetUp();
InitializeEmptyExtensionService();
std::unique_ptr<content::WebContents> web_contents(
content::WebContentsTester::CreateTestWebContents(profile(), nullptr));
web_contents_tester_ = content::WebContentsTester::For(web_contents.get());
AppTabHelper::CreateForWebContents(web_contents.get());
app_tab_helper_ = AppTabHelper::FromWebContents(web_contents.get());
browser()->tab_strip_model()->AppendWebContents(std::move(web_contents),
true);
}
void TearDown() override {
app_tab_helper_ = nullptr;
web_contents_tester_ = nullptr;
// Remove any tabs in the tab strip to avoid test crashes.
if (browser_) {
while (!browser_->tab_strip_model()->empty()) {
browser_->tab_strip_model()->DetachAndDeleteWebContentsAt(0);
}
}
ExtensionServiceTestBase::TearDown();
}
Browser* browser() {
if (!browser_) {
Browser::CreateParams params(profile(), true);
browser_window_ = std::make_unique<TestBrowserWindow>();
params.window = browser_window_.get();
browser_ = Browser::DeprecatedCreateOwnedForTesting(params);
}
return browser_.get();
}
AppTabHelper* app_tab_helper() { return app_tab_helper_; }
private:
// The browser and accompanying window.
std::unique_ptr<TestBrowserWindow> browser_window_;
std::unique_ptr<Browser> browser_;
raw_ptr<content::WebContentsTester> web_contents_tester_ = nullptr;
raw_ptr<AppTabHelper> app_tab_helper_ = nullptr;
};
TEST_F(AppTabHelperUnitTest, ClearsExtensionOnUnload) {
const Extension* extension =
PackAndInstallCRX(data_dir().AppendASCII("hosted_app"), INSTALL_NEW);
ASSERT_TRUE(extension);
app_tab_helper()->SetExtensionApp(extension);
EXPECT_EQ(extension->id(), app_tab_helper()->GetExtensionAppId());
EXPECT_TRUE(app_tab_helper()->is_app());
registrar()->RemoveExtension(extension->id(),
UnloadedExtensionReason::TERMINATE);
base::RunLoop().RunUntilIdle();
EXPECT_EQ(ExtensionId(), app_tab_helper()->GetExtensionAppId());
}
} // namespace extensions
|
