diff options
Diffstat (limited to 'chrome/common/extensions/extension_unittest.cc')
| -rw-r--r-- | chrome/common/extensions/extension_unittest.cc | 256 |
1 files changed, 256 insertions, 0 deletions
diff --git a/chrome/common/extensions/extension_unittest.cc b/chrome/common/extensions/extension_unittest.cc new file mode 100644 index 0000000..7f0f288 --- /dev/null +++ b/chrome/common/extensions/extension_unittest.cc @@ -0,0 +1,256 @@ +// Copyright (c) 2006-2008 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 "base/file_path.h" +#include "base/string_util.h" +#include "base/path_service.h" +#include "chrome/common/chrome_paths.h" +#include "chrome/common/extensions/extension.h" +#include "chrome/common/extensions/extension_error_reporter.h" +#include "chrome/common/json_value_serializer.h" +#include "testing/gtest/include/gtest/gtest.h" + +class ExtensionTest : public testing::Test { +}; + +TEST(ExtensionTest, InitFromValueInvalid) { +#if defined(OS_WIN) + FilePath path(FILE_PATH_LITERAL("c:\\foo")); +#elif defined(OS_POSIX) + FilePath path(FILE_PATH_LITERAL("/foo")); +#endif + Extension extension(path); + std::string error; + ExtensionErrorReporter::Init(false); + + // Start with a valid extension manifest + FilePath extensions_path; + ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &extensions_path)); + extensions_path = extensions_path.AppendASCII("extensions") + .AppendASCII("good") + .AppendASCII("extension1") + .AppendASCII("1") + .AppendASCII(Extension::kManifestFilename); + + JSONFileValueSerializer serializer(extensions_path); + scoped_ptr<DictionaryValue> valid_value( + static_cast<DictionaryValue*>(serializer.Deserialize(&error))); + ASSERT_TRUE(valid_value.get()); + ASSERT_EQ("", error); + ASSERT_TRUE(extension.InitFromValue(*valid_value, true, &error)); + ASSERT_EQ("", error); + + scoped_ptr<DictionaryValue> input_value; + + // Test missing and invalid ids + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->Remove(Extension::kIdKey, NULL); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_EQ(Extension::kInvalidIdError, error); + + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->SetInteger(Extension::kIdKey, 42); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_EQ(Extension::kInvalidIdError, error); + + // Test missing and invalid versions + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->Remove(Extension::kVersionKey, NULL); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_EQ(Extension::kInvalidVersionError, error); + + input_value->SetInteger(Extension::kVersionKey, 42); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_EQ(Extension::kInvalidVersionError, error); + + // Test missing and invalid names + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->Remove(Extension::kNameKey, NULL); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_EQ(Extension::kInvalidNameError, error); + + input_value->SetInteger(Extension::kNameKey, 42); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_EQ(Extension::kInvalidNameError, error); + + // Test invalid description + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->SetInteger(Extension::kDescriptionKey, 42); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_EQ(Extension::kInvalidDescriptionError, error); + + // Test invalid user scripts list + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->SetInteger(Extension::kContentScriptsKey, 42); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_EQ(Extension::kInvalidContentScriptsListError, error); + + // Test invalid user script item + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + ListValue* content_scripts = NULL; + input_value->GetList(Extension::kContentScriptsKey, &content_scripts); + ASSERT_FALSE(NULL == content_scripts); + content_scripts->Set(0, Value::CreateIntegerValue(42)); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidContentScriptError)); + + // Test missing and invalid matches array + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->GetList(Extension::kContentScriptsKey, &content_scripts); + DictionaryValue* user_script = NULL; + content_scripts->GetDictionary(0, &user_script); + user_script->Remove(Extension::kMatchesKey, NULL); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidMatchesError)); + + user_script->Set(Extension::kMatchesKey, Value::CreateIntegerValue(42)); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidMatchesError)); + + ListValue* matches = new ListValue; + user_script->Set(Extension::kMatchesKey, matches); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidMatchCountError)); + + // Test invalid match element + matches->Set(0, Value::CreateIntegerValue(42)); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidMatchError)); + + // Test missing and invalid files array + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->GetList(Extension::kContentScriptsKey, &content_scripts); + content_scripts->GetDictionary(0, &user_script); + user_script->Remove(Extension::kJsKey, NULL); + user_script->Remove(Extension::kCssKey, NULL); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kMissingFileError)); + + user_script->Set(Extension::kJsKey, Value::CreateIntegerValue(42)); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidJsListError)); + + user_script->Set(Extension::kCssKey, new ListValue); + user_script->Set(Extension::kJsKey, new ListValue); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kMissingFileError)); + user_script->Remove(Extension::kCssKey, NULL); + + ListValue* files = new ListValue; + user_script->Set(Extension::kJsKey, files); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kMissingFileError)); + + // Test invalid file element + files->Set(0, Value::CreateIntegerValue(42)); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidJsError)); + + user_script->Remove(Extension::kJsKey, NULL); + // Test the css element + user_script->Set(Extension::kCssKey, Value::CreateIntegerValue(42)); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidCssListError)); + + // Test invalid file element + ListValue* css_files = new ListValue; + user_script->Set(Extension::kCssKey, css_files); + css_files->Set(0, Value::CreateIntegerValue(42)); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidCssError)); + + // Test missing and invalid permissions array + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + EXPECT_TRUE(extension.InitFromValue(*input_value, true, &error)); + ListValue* permissions = NULL; + input_value->GetList(Extension::kPermissionsKey, &permissions); + ASSERT_FALSE(NULL == permissions); + + permissions = new ListValue; + input_value->Set(Extension::kPermissionsKey, permissions); + EXPECT_TRUE(extension.InitFromValue(*input_value, true, &error)); + const std::vector<std::string>* error_vector = + ExtensionErrorReporter::GetInstance()->GetErrors(); + const std::string log_error = error_vector->at(error_vector->size() - 1); + EXPECT_TRUE(MatchPattern(log_error, + Extension::kInvalidPermissionCountWarning)); + + input_value->Set(Extension::kPermissionsKey, Value::CreateIntegerValue(9)); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidPermissionsError)); + + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->GetList(Extension::kPermissionsKey, &permissions); + permissions->Set(0, Value::CreateIntegerValue(24)); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidPermissionError)); + + permissions->Set(0, Value::CreateStringValue("www.google.com")); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidPermissionError)); + + // Test permissions scheme. + input_value.reset(static_cast<DictionaryValue*>(valid_value->DeepCopy())); + input_value->GetList(Extension::kPermissionsKey, &permissions); + permissions->Set(0, Value::CreateStringValue("file:///C:/foo.txt")); + EXPECT_FALSE(extension.InitFromValue(*input_value, true, &error)); + EXPECT_TRUE(MatchPattern(error, Extension::kInvalidPermissionSchemeError)); +} + +TEST(ExtensionTest, InitFromValueValid) { +#if defined(OS_WIN) + FilePath path(FILE_PATH_LITERAL("C:\\foo")); +#elif defined(OS_POSIX) + FilePath path(FILE_PATH_LITERAL("/foo")); +#endif + Extension extension(path); + std::string error; + DictionaryValue input_value; + + // Test minimal extension + input_value.SetString(Extension::kIdKey, + "00123456789ABCDEF0123456789ABCDEF0123456"); + input_value.SetString(Extension::kVersionKey, "1.0.0.0"); + input_value.SetString(Extension::kNameKey, "my extension"); + + EXPECT_TRUE(extension.InitFromValue(input_value, true, &error)); + EXPECT_EQ("", error); + EXPECT_EQ("00123456789abcdef0123456789abcdef0123456", extension.id()); + EXPECT_EQ("1.0.0.0", extension.VersionString()); + EXPECT_EQ("my extension", extension.name()); + EXPECT_EQ("chrome-extension://00123456789abcdef0123456789abcdef0123456/", + extension.url().spec()); + EXPECT_EQ(path.value(), extension.path().value()); +} + +TEST(ExtensionTest, GetResourceURLAndPath) { +#if defined(OS_WIN) + FilePath path(FILE_PATH_LITERAL("C:\\foo")); +#elif defined(OS_POSIX) + FilePath path(FILE_PATH_LITERAL("/foo")); +#endif + Extension extension(path); + DictionaryValue input_value; + input_value.SetString(Extension::kIdKey, + "00123456789ABCDEF0123456789ABCDEF0123456"); + input_value.SetString(Extension::kVersionKey, "1.0.0.0"); + input_value.SetString(Extension::kNameKey, "my extension"); + EXPECT_TRUE(extension.InitFromValue(input_value, true, NULL)); + + EXPECT_EQ(extension.url().spec() + "bar/baz.js", + Extension::GetResourceURL(extension.url(), "bar/baz.js").spec()); + EXPECT_EQ(extension.url().spec() + "baz.js", + Extension::GetResourceURL(extension.url(), "bar/../baz.js").spec()); + EXPECT_EQ(extension.url().spec() + "baz.js", + Extension::GetResourceURL(extension.url(), "../baz.js").spec()); + + EXPECT_EQ(path.Append(FILE_PATH_LITERAL("bar")) + .Append(FILE_PATH_LITERAL("baz.js")).value(), + Extension::GetResourcePath(extension.path(), "bar/baz.js").value()); + EXPECT_EQ(path.Append(FILE_PATH_LITERAL("baz.js")).value(), + Extension::GetResourcePath(extension.path(), "bar/../baz.js") + .value()); + EXPECT_EQ(FilePath().value(), + Extension::GetResourcePath(extension.path(), "../baz.js").value()); +} |