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
|
// Copyright 2017 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/browser/extensions/install_verifier.h"
#include "base/macros.h"
#include "base/values.h"
#include "chrome/browser/extensions/extension_management.h"
#include "chrome/browser/extensions/extension_service_test_base.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/test/base/testing_profile.h"
#include "components/sync_preferences/testing_pref_service_syncable.h"
#include "extensions/browser/pref_names.h"
#include "extensions/common/extension.h"
#include "extensions/common/extension_builder.h"
#include "extensions/common/extension_urls.h"
#include "extensions/common/manifest.h"
#include "extensions/common/value_builder.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extensions {
class InstallVerifierTest : public ExtensionServiceTestBase {
public:
InstallVerifierTest() = default;
~InstallVerifierTest() override = default;
void SetUp() override {
ExtensionServiceTestBase::SetUp();
InitializeExtensionService(ExtensionServiceInitParams());
}
// Adds an extension as being allowed by policy.
void AddExtensionAsPolicyInstalled(const ExtensionId& id) {
std::unique_ptr<base::DictionaryValue> extension_entry =
DictionaryBuilder().Set("installation_mode", "allowed").Build();
testing_profile()->GetTestingPrefService()->SetManagedPref(
pref_names::kExtensionManagement,
DictionaryBuilder().Set(id, std::move(extension_entry)).Build());
EXPECT_TRUE(ExtensionManagementFactory::GetForBrowserContext(profile())
->IsInstallationExplicitlyAllowed(id));
}
private:
ScopedInstallVerifierBypassForTest force_install_verification{
ScopedInstallVerifierBypassForTest::kForceOn};
std::unique_ptr<ExtensionManagement> extension_management_;
std::unique_ptr<TestingPrefServiceSimple> pref_service_;
DISALLOW_COPY_AND_ASSIGN(InstallVerifierTest);
};
// Test the behavior of the InstallVerifier for various extensions.
TEST_F(InstallVerifierTest, TestIsFromStoreAndMustRemainDisabled) {
enum FromStoreStatus {
FROM_STORE,
NOT_FROM_STORE,
};
enum MustRemainDisabledStatus {
MUST_REMAIN_DISABLED,
CAN_BE_ENABLED,
};
GURL store_update_url = extension_urls::GetWebstoreUpdateUrl();
GURL non_store_update_url("https://example.com");
struct {
const char* test_name;
Manifest::Location location;
base::Optional<GURL> update_url;
FromStoreStatus expected_from_store_status;
MustRemainDisabledStatus expected_must_remain_disabled_status;
} test_cases[] = {
{"internal from store", Manifest::INTERNAL, store_update_url, FROM_STORE,
CAN_BE_ENABLED},
{"internal non-store update url", Manifest::INTERNAL,
non_store_update_url, NOT_FROM_STORE, MUST_REMAIN_DISABLED},
{"internal no update url", Manifest::INTERNAL, base::nullopt,
NOT_FROM_STORE, MUST_REMAIN_DISABLED},
{"unpacked from store", Manifest::UNPACKED, store_update_url, FROM_STORE,
CAN_BE_ENABLED},
{"unpacked non-store update url", Manifest::UNPACKED,
non_store_update_url, NOT_FROM_STORE, CAN_BE_ENABLED},
{"unpacked no update url", Manifest::UNPACKED, base::nullopt,
NOT_FROM_STORE, CAN_BE_ENABLED},
{"external from store", Manifest::EXTERNAL_POLICY_DOWNLOAD,
store_update_url, FROM_STORE, CAN_BE_ENABLED},
{"external non-store update url", Manifest::EXTERNAL_POLICY_DOWNLOAD,
non_store_update_url, NOT_FROM_STORE, CAN_BE_ENABLED},
};
InstallVerifier* install_verifier = InstallVerifier::Get(profile());
for (const auto& test_case : test_cases) {
SCOPED_TRACE(test_case.test_name);
ExtensionBuilder extension_builder(test_case.test_name);
extension_builder.SetLocation(test_case.location);
if (test_case.update_url) {
extension_builder.SetManifestKey("update_url",
test_case.update_url->spec());
}
scoped_refptr<const Extension> extension = extension_builder.Build();
if (Manifest::IsPolicyLocation(test_case.location))
AddExtensionAsPolicyInstalled(extension->id());
EXPECT_EQ(test_case.expected_from_store_status == FROM_STORE,
InstallVerifier::IsFromStore(*extension));
disable_reason::DisableReason disable_reason;
base::string16 error;
EXPECT_EQ(
test_case.expected_must_remain_disabled_status == MUST_REMAIN_DISABLED,
install_verifier->MustRemainDisabled(extension.get(), &disable_reason,
&error))
<< error;
}
}
} // namespace extensions
|
