| Index: chrome/browser/extensions/user_script_loader_unittest.cc
|
| diff --git a/chrome/browser/extensions/user_script_loader_unittest.cc b/chrome/browser/extensions/user_script_loader_unittest.cc
|
| deleted file mode 100644
|
| index 4e6d8eee4b300e26fd02b7acdd2c428e1d84b422..0000000000000000000000000000000000000000
|
| --- a/chrome/browser/extensions/user_script_loader_unittest.cc
|
| +++ /dev/null
|
| @@ -1,264 +0,0 @@
|
| -// Copyright 2014 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/user_script_loader.h"
|
| -
|
| -#include <set>
|
| -#include <string>
|
| -
|
| -#include "base/files/file_path.h"
|
| -#include "base/files/file_util.h"
|
| -#include "base/files/scoped_temp_dir.h"
|
| -#include "base/message_loop/message_loop.h"
|
| -#include "base/strings/string_util.h"
|
| -#include "chrome/browser/chrome_notification_types.h"
|
| -#include "chrome/test/base/testing_profile.h"
|
| -#include "content/public/browser/notification_observer.h"
|
| -#include "content/public/browser/notification_registrar.h"
|
| -#include "content/public/browser/notification_service.h"
|
| -#include "content/public/test/test_browser_thread.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -using content::BrowserThread;
|
| -using extensions::URLPatternSet;
|
| -
|
| -namespace {
|
| -
|
| -static void AddPattern(URLPatternSet* extent, const std::string& pattern) {
|
| - int schemes = URLPattern::SCHEME_ALL;
|
| - extent->AddPattern(URLPattern(schemes, pattern));
|
| -}
|
| -}
|
| -
|
| -namespace extensions {
|
| -
|
| -// Test bringing up a script loader on a specific directory, putting a script
|
| -// in there, etc.
|
| -
|
| -class UserScriptLoaderTest : public testing::Test,
|
| - public content::NotificationObserver {
|
| - public:
|
| - UserScriptLoaderTest() : shared_memory_(NULL) {}
|
| -
|
| - void SetUp() override {
|
| - ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
|
| -
|
| - // Register for all user script notifications.
|
| - registrar_.Add(this,
|
| - extensions::NOTIFICATION_USER_SCRIPTS_UPDATED,
|
| - content::NotificationService::AllSources());
|
| -
|
| - // UserScriptLoader posts tasks to the file thread so make the current
|
| - // thread look like one.
|
| - file_thread_.reset(new content::TestBrowserThread(
|
| - BrowserThread::FILE, base::MessageLoop::current()));
|
| - ui_thread_.reset(new content::TestBrowserThread(
|
| - BrowserThread::UI, base::MessageLoop::current()));
|
| - }
|
| -
|
| - void TearDown() override {
|
| - file_thread_.reset();
|
| - ui_thread_.reset();
|
| - }
|
| -
|
| - void Observe(int type,
|
| - const content::NotificationSource& source,
|
| - const content::NotificationDetails& details) override {
|
| - DCHECK(type == extensions::NOTIFICATION_USER_SCRIPTS_UPDATED);
|
| -
|
| - shared_memory_ = content::Details<base::SharedMemory>(details).ptr();
|
| - if (base::MessageLoop::current() == &message_loop_)
|
| - base::MessageLoop::current()->Quit();
|
| - }
|
| -
|
| - // Directory containing user scripts.
|
| - base::ScopedTempDir temp_dir_;
|
| -
|
| - content::NotificationRegistrar registrar_;
|
| -
|
| - // MessageLoop used in tests.
|
| - base::MessageLoopForUI message_loop_;
|
| -
|
| - scoped_ptr<content::TestBrowserThread> file_thread_;
|
| - scoped_ptr<content::TestBrowserThread> ui_thread_;
|
| -
|
| - // Updated to the script shared memory when we get notified.
|
| - base::SharedMemory* shared_memory_;
|
| -};
|
| -
|
| -// Test that we get notified even when there are no scripts.
|
| -TEST_F(UserScriptLoaderTest, NoScripts) {
|
| - TestingProfile profile;
|
| - UserScriptLoader loader(&profile,
|
| - std::string() /* owner_extension_id */,
|
| - true /* listen_for_extension_system_loaded */);
|
| - loader.StartLoad();
|
| - message_loop_.PostTask(FROM_HERE, base::MessageLoop::QuitClosure());
|
| - message_loop_.Run();
|
| -
|
| - ASSERT_TRUE(shared_memory_ != NULL);
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, Parse1) {
|
| - const std::string text(
|
| - "// This is my awesome script\n"
|
| - "// It does stuff.\n"
|
| - "// ==UserScript== trailing garbage\n"
|
| - "// @name foobar script\n"
|
| - "// @namespace http://www.google.com/\n"
|
| - "// @include *mail.google.com*\n"
|
| - "// \n"
|
| - "// @othergarbage\n"
|
| - "// @include *mail.yahoo.com*\r\n"
|
| - "// @include \t *mail.msn.com*\n" // extra spaces after "@include" OK
|
| - "//@include not-recognized\n" // must have one space after "//"
|
| - "// ==/UserScript== trailing garbage\n"
|
| - "\n"
|
| - "\n"
|
| - "alert('hoo!');\n");
|
| -
|
| - UserScript script;
|
| - EXPECT_TRUE(UserScriptLoader::ParseMetadataHeader(text, &script));
|
| - ASSERT_EQ(3U, script.globs().size());
|
| - EXPECT_EQ("*mail.google.com*", script.globs()[0]);
|
| - EXPECT_EQ("*mail.yahoo.com*", script.globs()[1]);
|
| - EXPECT_EQ("*mail.msn.com*", script.globs()[2]);
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, Parse2) {
|
| - const std::string text("default to @include *");
|
| -
|
| - UserScript script;
|
| - EXPECT_TRUE(UserScriptLoader::ParseMetadataHeader(text, &script));
|
| - ASSERT_EQ(1U, script.globs().size());
|
| - EXPECT_EQ("*", script.globs()[0]);
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, Parse3) {
|
| - const std::string text(
|
| - "// ==UserScript==\n"
|
| - "// @include *foo*\n"
|
| - "// ==/UserScript=="); // no trailing newline
|
| -
|
| - UserScript script;
|
| - UserScriptLoader::ParseMetadataHeader(text, &script);
|
| - ASSERT_EQ(1U, script.globs().size());
|
| - EXPECT_EQ("*foo*", script.globs()[0]);
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, Parse4) {
|
| - const std::string text(
|
| - "// ==UserScript==\n"
|
| - "// @match http://*.mail.google.com/*\n"
|
| - "// @match \t http://mail.yahoo.com/*\n"
|
| - "// ==/UserScript==\n");
|
| -
|
| - URLPatternSet expected_patterns;
|
| - AddPattern(&expected_patterns, "http://*.mail.google.com/*");
|
| - AddPattern(&expected_patterns, "http://mail.yahoo.com/*");
|
| -
|
| - UserScript script;
|
| - EXPECT_TRUE(UserScriptLoader::ParseMetadataHeader(text, &script));
|
| - EXPECT_EQ(0U, script.globs().size());
|
| - EXPECT_EQ(expected_patterns, script.url_patterns());
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, Parse5) {
|
| - const std::string text(
|
| - "// ==UserScript==\n"
|
| - "// @match http://*mail.google.com/*\n"
|
| - "// ==/UserScript==\n");
|
| -
|
| - // Invalid @match value.
|
| - UserScript script;
|
| - EXPECT_FALSE(UserScriptLoader::ParseMetadataHeader(text, &script));
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, Parse6) {
|
| - const std::string text(
|
| - "// ==UserScript==\n"
|
| - "// @include http://*.mail.google.com/*\n"
|
| - "// @match \t http://mail.yahoo.com/*\n"
|
| - "// ==/UserScript==\n");
|
| -
|
| - // Allowed to match @include and @match.
|
| - UserScript script;
|
| - EXPECT_TRUE(UserScriptLoader::ParseMetadataHeader(text, &script));
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, Parse7) {
|
| - // Greasemonkey allows there to be any leading text before the comment marker.
|
| - const std::string text(
|
| - "// ==UserScript==\n"
|
| - "adsasdfasf// @name hello\n"
|
| - " // @description\twiggity woo\n"
|
| - "\t// @match \t http://mail.yahoo.com/*\n"
|
| - "// ==/UserScript==\n");
|
| -
|
| - UserScript script;
|
| - EXPECT_TRUE(UserScriptLoader::ParseMetadataHeader(text, &script));
|
| - ASSERT_EQ("hello", script.name());
|
| - ASSERT_EQ("wiggity woo", script.description());
|
| - ASSERT_EQ(1U, script.url_patterns().patterns().size());
|
| - EXPECT_EQ("http://mail.yahoo.com/*",
|
| - script.url_patterns().begin()->GetAsString());
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, Parse8) {
|
| - const std::string text(
|
| - "// ==UserScript==\n"
|
| - "// @name myscript\n"
|
| - "// @match http://www.google.com/*\n"
|
| - "// @exclude_match http://www.google.com/foo*\n"
|
| - "// ==/UserScript==\n");
|
| -
|
| - UserScript script;
|
| - EXPECT_TRUE(UserScriptLoader::ParseMetadataHeader(text, &script));
|
| - ASSERT_EQ("myscript", script.name());
|
| - ASSERT_EQ(1U, script.url_patterns().patterns().size());
|
| - EXPECT_EQ("http://www.google.com/*",
|
| - script.url_patterns().begin()->GetAsString());
|
| - ASSERT_EQ(1U, script.exclude_url_patterns().patterns().size());
|
| - EXPECT_EQ("http://www.google.com/foo*",
|
| - script.exclude_url_patterns().begin()->GetAsString());
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, SkipBOMAtTheBeginning) {
|
| - base::FilePath path = temp_dir_.path().AppendASCII("script.user.js");
|
| - const std::string content("\xEF\xBB\xBF alert('hello');");
|
| - size_t written = base::WriteFile(path, content.c_str(), content.size());
|
| - ASSERT_EQ(written, content.size());
|
| -
|
| - UserScript user_script;
|
| - user_script.js_scripts().push_back(
|
| - UserScript::File(temp_dir_.path(), path.BaseName(), GURL()));
|
| -
|
| - UserScriptList user_scripts;
|
| - user_scripts.push_back(user_script);
|
| -
|
| - UserScriptLoader::LoadScriptsForTest(&user_scripts);
|
| -
|
| - EXPECT_EQ(content.substr(3),
|
| - user_scripts[0].js_scripts()[0].GetContent().as_string());
|
| -}
|
| -
|
| -TEST_F(UserScriptLoaderTest, LeaveBOMNotAtTheBeginning) {
|
| - base::FilePath path = temp_dir_.path().AppendASCII("script.user.js");
|
| - const std::string content("alert('here's a BOOM: \xEF\xBB\xBF');");
|
| - size_t written = base::WriteFile(path, content.c_str(), content.size());
|
| - ASSERT_EQ(written, content.size());
|
| -
|
| - UserScript user_script;
|
| - user_script.js_scripts().push_back(UserScript::File(
|
| - temp_dir_.path(), path.BaseName(), GURL()));
|
| -
|
| - UserScriptList user_scripts;
|
| - user_scripts.push_back(user_script);
|
| -
|
| - UserScriptLoader::LoadScriptsForTest(&user_scripts);
|
| -
|
| - EXPECT_EQ(content, user_scripts[0].js_scripts()[0].GetContent().as_string());
|
| -}
|
| -
|
| -} // namespace extensions
|
|
|