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
|
// Copyright 2023 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/supervised_user/extension_icon_loader.h"
#include <memory>
#include "base/command_line.h"
#include "base/files/file_path.h"
#include "base/functional/bind.h"
#include "base/json/json_file_value_serializer.h"
#include "base/memory/raw_ptr.h"
#include "base/path_service.h"
#include "base/run_loop.h"
#include "base/values.h"
#include "chrome/browser/extensions/test_extension_system.h"
#include "chrome/common/chrome_paths.h"
#include "chrome/test/base/browser_with_test_window_test.h"
#include "extensions/browser/extension_registrar.h"
#include "extensions/browser/extension_util.h"
#include "extensions/common/extension.h"
#include "extensions/common/extension_builder.h"
#include "extensions/common/manifest.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/gfx/image/image_skia.h"
class ExtensionIconLoaderTest : public BrowserWithTestWindowTest {
public:
ExtensionIconLoaderTest() = default;
ExtensionIconLoaderTest(const ExtensionIconLoaderTest&) = delete;
ExtensionIconLoaderTest& operator=(const ExtensionIconLoaderTest&) = delete;
~ExtensionIconLoaderTest() override = default;
void SetUp() override {
BrowserWithTestWindowTest::SetUp();
extensions::TestExtensionSystem* extension_system =
static_cast<extensions::TestExtensionSystem*>(
extensions::ExtensionSystem::Get(profile()));
extension_system->CreateExtensionService(
base::CommandLine::ForCurrentProcess(), base::FilePath(), false);
}
void OnIconFetched(base::RunLoop* run_loop, const gfx::ImageSkia& icon) {
run_loop->Quit();
loaded_icon_ = icon;
}
const gfx::ImageSkia& loaded_icon() { return loaded_icon_; }
extensions::ExtensionRegistrar* extension_registrar() {
return extensions::ExtensionRegistrar::Get(profile());
}
protected:
gfx::ImageSkia loaded_icon_;
};
TEST_F(ExtensionIconLoaderTest, LoadExtensionWithIcon) {
// Create and add an extension from a test manifest with an icon.
base::FilePath test_file;
ASSERT_TRUE(base::PathService::Get(chrome::DIR_TEST_DATA, &test_file));
test_file = test_file.AppendASCII("extensions/simple_with_icon");
int error_code = 0;
std::string error;
JSONFileValueDeserializer deserializer(
test_file.AppendASCII("manifest.json"));
std::unique_ptr<base::Value> valid_value =
deserializer.Deserialize(&error_code, &error);
scoped_refptr<extensions::Extension> extension =
extensions::Extension::Create(
test_file, extensions::mojom::ManifestLocation::kUnpacked,
valid_value->GetDict(), extensions::Extension::NO_FLAGS, &error);
extension_registrar()->AddExtension(extension);
extensions::ExtensionIconLoader loader;
base::RunLoop run_loop;
loader.Load(*extension, profile(),
base::BindOnce(&ExtensionIconLoaderTest::OnIconFetched,
base::Unretained(this), &run_loop));
run_loop.Run();
// Check that the loaded icon is not a default icon.
EXPECT_FALSE(loaded_icon().isNull());
EXPECT_NE(loaded_icon().bitmap(),
extensions::util::GetDefaultExtensionIcon().bitmap());
EXPECT_NE(loaded_icon().bitmap(),
extensions::util::GetDefaultAppIcon().bitmap());
}
TEST_F(ExtensionIconLoaderTest, LoadDefaultAppIcon) {
// Create an app with no icon.
scoped_refptr<const extensions::Extension> extension =
extensions::ExtensionBuilder(
"extension", extensions::ExtensionBuilder::Type::PLATFORM_APP)
.Build();
extension_registrar()->AddExtension(extension);
extensions::ExtensionIconLoader loader;
base::RunLoop run_loop;
loader.Load(*extension, profile(),
base::BindOnce(&ExtensionIconLoaderTest::OnIconFetched,
base::Unretained(this), &run_loop));
run_loop.Run();
// Check that the default app icon is loaded.
EXPECT_EQ(loaded_icon().bitmap(),
extensions::util::GetDefaultAppIcon().bitmap());
}
TEST_F(ExtensionIconLoaderTest, LoadDefaultExtensionIcon) {
// Create an extension with no icon.
scoped_refptr<const extensions::Extension> extension =
extensions::ExtensionBuilder("extension").Build();
extension_registrar()->AddExtension(extension);
extensions::ExtensionIconLoader loader;
base::RunLoop run_loop;
loader.Load(*extension, profile(),
base::BindOnce(&ExtensionIconLoaderTest::OnIconFetched,
base::Unretained(this), &run_loop));
run_loop.Run();
// Check that the default extension icon is loaded.
EXPECT_EQ(loaded_icon().bitmap(),
extensions::util::GetDefaultExtensionIcon().bitmap());
}
|
