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
|
// Copyright (c) 2012 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/common/extensions/chrome_extensions_client.h"
#include <memory>
#include <set>
#include <string>
#include "base/path_service.h"
#include "chrome/common/chrome_paths.h"
#include "extensions/common/extension.h"
#include "extensions/common/file_util.h"
#include "extensions/common/manifest.h"
#include "extensions/common/manifest_handler.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extensions {
class ChromeExtensionsClientTest : public testing::Test {
public:
void SetUp() override {
extensions_client_.reset(new ChromeExtensionsClient());
ExtensionsClient::Set(extensions_client_.get());
}
private:
std::unique_ptr<ChromeExtensionsClient> extensions_client_;
};
// Test that a browser action extension returns a path to an icon.
TEST_F(ChromeExtensionsClientTest, GetBrowserImagePaths) {
base::FilePath install_dir;
ASSERT_TRUE(base::PathService::Get(chrome::DIR_TEST_DATA, &install_dir));
install_dir = install_dir.AppendASCII("extensions")
.AppendASCII("api_test")
.AppendASCII("browser_action")
.AppendASCII("basics");
std::string error;
scoped_refptr<Extension> extension(file_util::LoadExtension(
install_dir, Manifest::UNPACKED, Extension::NO_FLAGS, &error));
ASSERT_TRUE(extension.get());
// The extension contains one icon.
std::set<base::FilePath> paths =
ExtensionsClient::Get()->GetBrowserImagePaths(extension.get());
ASSERT_EQ(1u, paths.size());
EXPECT_EQ("icon.png", paths.begin()->BaseName().AsUTF8Unsafe());
}
// Test that extensions with zero-length action icons will not load.
TEST_F(ChromeExtensionsClientTest, CheckZeroLengthActionIconFiles) {
base::FilePath install_dir;
ASSERT_TRUE(base::PathService::Get(chrome::DIR_TEST_DATA, &install_dir));
// Try to install an extension with a zero-length browser action icon file.
base::FilePath ext_dir = install_dir.AppendASCII("extensions")
.AppendASCII("bad")
.AppendASCII("Extensions")
.AppendASCII("gggggggggggggggggggggggggggggggg");
std::string error;
scoped_refptr<Extension> extension2(file_util::LoadExtension(
ext_dir, Manifest::UNPACKED, Extension::NO_FLAGS, &error));
EXPECT_FALSE(extension2.get());
EXPECT_STREQ("Could not load icon 'icon.png' for browser action.",
error.c_str());
// Try to install an extension with a zero-length page action icon file.
ext_dir = install_dir.AppendASCII("extensions")
.AppendASCII("bad")
.AppendASCII("Extensions")
.AppendASCII("hhhhhhhhhhhhhhhhhhhhhhhhhhhhhhhh");
scoped_refptr<Extension> extension3(file_util::LoadExtension(
ext_dir, Manifest::UNPACKED, Extension::NO_FLAGS, &error));
EXPECT_FALSE(extension3.get());
EXPECT_STREQ("Could not load icon 'icon.png' for page action.",
error.c_str());
}
// Test that the ManifestHandlerRegistry handler map hasn't overflowed.
// If this test fails, increase ManifestHandlerRegistry::kHandlerMax.
TEST_F(ChromeExtensionsClientTest, CheckManifestHandlerRegistryForOverflow) {
ManifestHandlerRegistry* registry = ManifestHandlerRegistry::Get();
ASSERT_TRUE(registry);
ASSERT_LT(0u, registry->handlers_.size());
EXPECT_LE(registry->handlers_.size(), ManifestHandlerRegistry::kHandlerMax);
}
} // namespace extensions
|
