| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/json/json_file_value_serializer.h" | 5 #include "base/json/json_file_value_serializer.h" |
| 6 #include "base/message_loop.h" | 6 #include "base/message_loop.h" |
| 7 #include "base/path_service.h" | 7 #include "base/path_service.h" |
| 8 #include "chrome/browser/extensions/extension_info_map.h" | 8 #include "chrome/browser/extensions/extension_info_map.h" |
| 9 #include "chrome/common/chrome_paths.h" | 9 #include "chrome/common/chrome_paths.h" |
| 10 #include "chrome/common/extensions/extension.h" | 10 #include "chrome/common/extensions/extension.h" |
| (...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 47 #endif | 47 #endif |
| 48 | 48 |
| 49 DictionaryValue manifest; | 49 DictionaryValue manifest; |
| 50 manifest.SetString(keys::kVersion, "1.0.0.0"); | 50 manifest.SetString(keys::kVersion, "1.0.0.0"); |
| 51 manifest.SetString(keys::kName, name); | 51 manifest.SetString(keys::kName, name); |
| 52 | 52 |
| 53 std::string error; | 53 std::string error; |
| 54 scoped_refptr<Extension> extension = Extension::Create( | 54 scoped_refptr<Extension> extension = Extension::Create( |
| 55 path.AppendASCII(name), Manifest::INVALID_LOCATION, manifest, | 55 path.AppendASCII(name), Manifest::INVALID_LOCATION, manifest, |
| 56 Extension::NO_FLAGS, &error); | 56 Extension::NO_FLAGS, &error); |
| 57 EXPECT_TRUE(extension) << error; | 57 EXPECT_TRUE(extension.get()) << error; |
| 58 | 58 |
| 59 return extension; | 59 return extension; |
| 60 } | 60 } |
| 61 | 61 |
| 62 static scoped_refptr<Extension> LoadManifest(const std::string& dir, | 62 static scoped_refptr<Extension> LoadManifest(const std::string& dir, |
| 63 const std::string& test_file) { | 63 const std::string& test_file) { |
| 64 base::FilePath path; | 64 base::FilePath path; |
| 65 PathService::Get(chrome::DIR_TEST_DATA, &path); | 65 PathService::Get(chrome::DIR_TEST_DATA, &path); |
| 66 path = path.AppendASCII("extensions") | 66 path = path.AppendASCII("extensions") |
| 67 .AppendASCII(dir) | 67 .AppendASCII(dir) |
| 68 .AppendASCII(test_file); | 68 .AppendASCII(test_file); |
| 69 | 69 |
| 70 JSONFileValueSerializer serializer(path); | 70 JSONFileValueSerializer serializer(path); |
| 71 scoped_ptr<Value> result(serializer.Deserialize(NULL, NULL)); | 71 scoped_ptr<Value> result(serializer.Deserialize(NULL, NULL)); |
| 72 if (!result) | 72 if (!result) |
| 73 return NULL; | 73 return NULL; |
| 74 | 74 |
| 75 std::string error; | 75 std::string error; |
| 76 scoped_refptr<Extension> extension = Extension::Create( | 76 scoped_refptr<Extension> extension = Extension::Create( |
| 77 path, Manifest::INVALID_LOCATION, | 77 path, Manifest::INVALID_LOCATION, |
| 78 *static_cast<DictionaryValue*>(result.get()), | 78 *static_cast<DictionaryValue*>(result.get()), |
| 79 Extension::NO_FLAGS, &error); | 79 Extension::NO_FLAGS, &error); |
| 80 EXPECT_TRUE(extension) << error; | 80 EXPECT_TRUE(extension.get()) << error; |
| 81 | 81 |
| 82 return extension; | 82 return extension; |
| 83 } | 83 } |
| 84 | 84 |
| 85 // Test that the ExtensionInfoMap handles refcounting properly. | 85 // Test that the ExtensionInfoMap handles refcounting properly. |
| 86 TEST_F(ExtensionInfoMapTest, RefCounting) { | 86 TEST_F(ExtensionInfoMapTest, RefCounting) { |
| 87 scoped_refptr<ExtensionInfoMap> info_map(new ExtensionInfoMap()); | 87 scoped_refptr<ExtensionInfoMap> info_map(new ExtensionInfoMap()); |
| 88 | 88 |
| 89 // New extensions should have a single reference holding onto them. | 89 // New extensions should have a single reference holding onto them. |
| 90 scoped_refptr<Extension> extension1(CreateExtension("extension1")); | 90 scoped_refptr<Extension> extension1(CreateExtension("extension1")); |
| 91 scoped_refptr<Extension> extension2(CreateExtension("extension2")); | 91 scoped_refptr<Extension> extension2(CreateExtension("extension2")); |
| 92 scoped_refptr<Extension> extension3(CreateExtension("extension3")); | 92 scoped_refptr<Extension> extension3(CreateExtension("extension3")); |
| 93 EXPECT_TRUE(extension1->HasOneRef()); | 93 EXPECT_TRUE(extension1->HasOneRef()); |
| 94 EXPECT_TRUE(extension2->HasOneRef()); | 94 EXPECT_TRUE(extension2->HasOneRef()); |
| 95 EXPECT_TRUE(extension3->HasOneRef()); | 95 EXPECT_TRUE(extension3->HasOneRef()); |
| 96 | 96 |
| 97 // Add a ref to each extension and give it to the info map. | 97 // Add a ref to each extension and give it to the info map. |
| 98 info_map->AddExtension(extension1, base::Time(), false); | 98 info_map->AddExtension(extension1.get(), base::Time(), false); |
| 99 info_map->AddExtension(extension2, base::Time(), false); | 99 info_map->AddExtension(extension2.get(), base::Time(), false); |
| 100 info_map->AddExtension(extension3, base::Time(), false); | 100 info_map->AddExtension(extension3.get(), base::Time(), false); |
| 101 | 101 |
| 102 // Release extension1, and the info map should have the only ref. | 102 // Release extension1, and the info map should have the only ref. |
| 103 const Extension* weak_extension1 = extension1; | 103 const Extension* weak_extension1 = extension1.get(); |
| 104 extension1 = NULL; | 104 extension1 = NULL; |
| 105 EXPECT_TRUE(weak_extension1->HasOneRef()); | 105 EXPECT_TRUE(weak_extension1->HasOneRef()); |
| 106 | 106 |
| 107 // Remove extension2, and the extension2 object should have the only ref. | 107 // Remove extension2, and the extension2 object should have the only ref. |
| 108 info_map->RemoveExtension( | 108 info_map->RemoveExtension( |
| 109 extension2->id(), extension_misc::UNLOAD_REASON_UNINSTALL); | 109 extension2->id(), extension_misc::UNLOAD_REASON_UNINSTALL); |
| 110 EXPECT_TRUE(extension2->HasOneRef()); | 110 EXPECT_TRUE(extension2->HasOneRef()); |
| 111 | 111 |
| 112 // Delete the info map, and the extension3 object should have the only ref. | 112 // Delete the info map, and the extension3 object should have the only ref. |
| 113 info_map = NULL; | 113 info_map = NULL; |
| 114 EXPECT_TRUE(extension3->HasOneRef()); | 114 EXPECT_TRUE(extension3->HasOneRef()); |
| 115 } | 115 } |
| 116 | 116 |
| 117 // Tests that we can query a few extension properties from the ExtensionInfoMap. | 117 // Tests that we can query a few extension properties from the ExtensionInfoMap. |
| 118 TEST_F(ExtensionInfoMapTest, Properties) { | 118 TEST_F(ExtensionInfoMapTest, Properties) { |
| 119 scoped_refptr<ExtensionInfoMap> info_map(new ExtensionInfoMap()); | 119 scoped_refptr<ExtensionInfoMap> info_map(new ExtensionInfoMap()); |
| 120 | 120 |
| 121 scoped_refptr<Extension> extension1(CreateExtension("extension1")); | 121 scoped_refptr<Extension> extension1(CreateExtension("extension1")); |
| 122 scoped_refptr<Extension> extension2(CreateExtension("extension2")); | 122 scoped_refptr<Extension> extension2(CreateExtension("extension2")); |
| 123 | 123 |
| 124 info_map->AddExtension(extension1, base::Time(), false); | 124 info_map->AddExtension(extension1.get(), base::Time(), false); |
| 125 info_map->AddExtension(extension2, base::Time(), false); | 125 info_map->AddExtension(extension2.get(), base::Time(), false); |
| 126 | 126 |
| 127 EXPECT_EQ(2u, info_map->extensions().size()); | 127 EXPECT_EQ(2u, info_map->extensions().size()); |
| 128 EXPECT_EQ(extension1.get(), info_map->extensions().GetByID(extension1->id())); | 128 EXPECT_EQ(extension1.get(), info_map->extensions().GetByID(extension1->id())); |
| 129 EXPECT_EQ(extension2.get(), info_map->extensions().GetByID(extension2->id())); | 129 EXPECT_EQ(extension2.get(), info_map->extensions().GetByID(extension2->id())); |
| 130 } | 130 } |
| 131 | 131 |
| 132 // Tests CheckURLAccessToExtensionPermission given both extension and app URLs. | 132 // Tests CheckURLAccessToExtensionPermission given both extension and app URLs. |
| 133 TEST_F(ExtensionInfoMapTest, CheckPermissions) { | 133 TEST_F(ExtensionInfoMapTest, CheckPermissions) { |
| 134 scoped_refptr<ExtensionInfoMap> info_map(new ExtensionInfoMap()); | 134 scoped_refptr<ExtensionInfoMap> info_map(new ExtensionInfoMap()); |
| 135 | 135 |
| 136 scoped_refptr<Extension> app(LoadManifest("manifest_tests", | 136 scoped_refptr<Extension> app(LoadManifest("manifest_tests", |
| 137 "valid_app.json")); | 137 "valid_app.json")); |
| 138 scoped_refptr<Extension> extension(LoadManifest("manifest_tests", | 138 scoped_refptr<Extension> extension(LoadManifest("manifest_tests", |
| 139 "tabs_extension.json")); | 139 "tabs_extension.json")); |
| 140 | 140 |
| 141 GURL app_url("http://www.google.com/mail/foo.html"); | 141 GURL app_url("http://www.google.com/mail/foo.html"); |
| 142 WebSecurityOrigin app_origin = WebSecurityOrigin::create( | 142 WebSecurityOrigin app_origin = WebSecurityOrigin::create( |
| 143 GURL("http://www.google.com/mail/foo.html")); | 143 GURL("http://www.google.com/mail/foo.html")); |
| 144 ASSERT_TRUE(app->is_app()); | 144 ASSERT_TRUE(app->is_app()); |
| 145 ASSERT_TRUE(app->web_extent().MatchesURL(app_url)); | 145 ASSERT_TRUE(app->web_extent().MatchesURL(app_url)); |
| 146 | 146 |
| 147 info_map->AddExtension(app, base::Time(), false); | 147 info_map->AddExtension(app.get(), base::Time(), false); |
| 148 info_map->AddExtension(extension, base::Time(), false); | 148 info_map->AddExtension(extension.get(), base::Time(), false); |
| 149 | 149 |
| 150 // The app should have the notifications permission, either from a | 150 // The app should have the notifications permission, either from a |
| 151 // chrome-extension URL or from its web extent. | 151 // chrome-extension URL or from its web extent. |
| 152 const Extension* match = info_map->extensions().GetExtensionOrAppByURL( | 152 const Extension* match = info_map->extensions().GetExtensionOrAppByURL( |
| 153 ExtensionURLInfo(app_origin, app->GetResourceURL("a.html"))); | 153 ExtensionURLInfo(app_origin, app->GetResourceURL("a.html"))); |
| 154 EXPECT_TRUE(match && | 154 EXPECT_TRUE(match && |
| 155 match->HasAPIPermission(APIPermission::kNotification)); | 155 match->HasAPIPermission(APIPermission::kNotification)); |
| 156 match = info_map->extensions().GetExtensionOrAppByURL( | 156 match = info_map->extensions().GetExtensionOrAppByURL( |
| 157 ExtensionURLInfo(app_origin, app_url)); | 157 ExtensionURLInfo(app_origin, app_url)); |
| 158 EXPECT_TRUE(match && | 158 EXPECT_TRUE(match && |
| (...skipping 16 matching lines...) Expand all Loading... |
| 175 EXPECT_FALSE(match); | 175 EXPECT_FALSE(match); |
| 176 | 176 |
| 177 // Sandboxed origins should not have any permissions. | 177 // Sandboxed origins should not have any permissions. |
| 178 match = info_map->extensions().GetExtensionOrAppByURL(ExtensionURLInfo( | 178 match = info_map->extensions().GetExtensionOrAppByURL(ExtensionURLInfo( |
| 179 WebSecurityOrigin::createFromString(WebString::fromUTF8("null")), | 179 WebSecurityOrigin::createFromString(WebString::fromUTF8("null")), |
| 180 app_url)); | 180 app_url)); |
| 181 EXPECT_FALSE(match); | 181 EXPECT_FALSE(match); |
| 182 } | 182 } |
| 183 | 183 |
| 184 } // namespace | 184 } // namespace |
| OLD | NEW |