1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include <string>
6 #include <vector>
7
8 #include "base/file_path.h"
9 #include "base/file_util.h"
10 #include "base/memory/scoped_ptr.h"
11 #include "base/path_service.h"
12 #include "chrome/browser/extensions/convert_user_script.h"
13 #include "chrome/common/chrome_paths.h"
14 #include "chrome/common/extensions/extension.h"
15 #include "testing/gtest/include/gtest/gtest.h"
16
TEST(ExtensionFromUserScript,Basic)17 TEST(ExtensionFromUserScript, Basic) {
18 FilePath test_file;
19 ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &test_file));
20 test_file = test_file.AppendASCII("extensions")
21 .AppendASCII("user_script_basic.user.js");
22
23 std::string error;
24 scoped_refptr<Extension> extension(ConvertUserScriptToExtension(
25 test_file, GURL("http://www.google.com/foo"), &error));
26
27 ASSERT_TRUE(extension.get());
28 EXPECT_EQ("", error);
29
30 // Validate generated extension metadata.
31 EXPECT_EQ("My user script", extension->name());
32 EXPECT_EQ("2.2.2", extension->VersionString());
33 EXPECT_EQ("Does totally awesome stuff.", extension->description());
34 EXPECT_EQ("IhCFCg9PMQTAcJdc9ytUP99WME+4yh6aMnM1uupkovo=",
35 extension->public_key());
36
37 ASSERT_EQ(1u, extension->content_scripts().size());
38 const UserScript& script = extension->content_scripts()[0];
39 ASSERT_EQ(2u, script.globs().size());
40 EXPECT_EQ("http://www.google.com/*", script.globs().at(0));
41 EXPECT_EQ("http://www.yahoo.com/*", script.globs().at(1));
42 ASSERT_EQ(1u, script.exclude_globs().size());
43 EXPECT_EQ("*foo*", script.exclude_globs().at(0));
44 ASSERT_EQ(1u, script.url_patterns().size());
45 EXPECT_EQ("http://www.google.com/*", script.url_patterns()[0].GetAsString());
46
47 // Make sure the files actually exist on disk.
48 EXPECT_TRUE(file_util::PathExists(
49 extension->path().Append(script.js_scripts()[0].relative_path())));
50 EXPECT_TRUE(file_util::PathExists(
51 extension->path().Append(Extension::kManifestFilename)));
52
53 // Cleanup
54 file_util::Delete(extension->path(), true);
55 }
56
TEST(ExtensionFromUserScript,NoMetdata)57 TEST(ExtensionFromUserScript, NoMetdata) {
58 FilePath test_file;
59 ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &test_file));
60 test_file = test_file.AppendASCII("extensions")
61 .AppendASCII("user_script_no_metadata.user.js");
62
63 std::string error;
64 scoped_refptr<Extension> extension(ConvertUserScriptToExtension(
65 test_file, GURL("http://www.google.com/foo/bar.user.js?monkey"), &error));
66
67 ASSERT_TRUE(extension.get());
68 EXPECT_EQ("", error);
69
70 // Validate generated extension metadata.
71 EXPECT_EQ("bar.user.js", extension->name());
72 EXPECT_EQ("1.0", extension->VersionString());
73 EXPECT_EQ("", extension->description());
74 EXPECT_EQ("k1WxKx54hX6tfl5gQaXD/m4d9QUMwRdXWM4RW+QkWcY=",
75 extension->public_key());
76
77 ASSERT_EQ(1u, extension->content_scripts().size());
78 const UserScript& script = extension->content_scripts()[0];
79 ASSERT_EQ(1u, script.globs().size());
80 EXPECT_EQ("*", script.globs()[0]);
81 EXPECT_EQ(0u, script.exclude_globs().size());
82 ASSERT_EQ(2u, script.url_patterns().size());
83 EXPECT_EQ("http://*/*", script.url_patterns()[0].GetAsString());
84 EXPECT_EQ("https://*/*", script.url_patterns()[1].GetAsString());
85
86 // Make sure the files actually exist on disk.
87 EXPECT_TRUE(file_util::PathExists(
88 extension->path().Append(script.js_scripts()[0].relative_path())));
89 EXPECT_TRUE(file_util::PathExists(
90 extension->path().Append(Extension::kManifestFilename)));
91
92 // Cleanup
93 file_util::Delete(extension->path(), true);
94 }
95
TEST(ExtensionFromUserScript,NotUTF8)96 TEST(ExtensionFromUserScript, NotUTF8) {
97 FilePath test_file;
98
99 ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &test_file));
100 test_file = test_file.AppendASCII("extensions")
101 .AppendASCII("user_script_not_utf8.user.js");
102
103 std::string error;
104 scoped_refptr<Extension> extension(ConvertUserScriptToExtension(
105 test_file, GURL("http://www.google.com/foo/bar.user.js?monkey"), &error));
106
107 ASSERT_FALSE(extension.get());
108 EXPECT_EQ("User script must be UTF8 encoded.", error);
109 }
110