1// Copyright 2013 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 "base/message_loop/message_loop.h"
6#include "base/path_service.h"
7#include "content/public/test/test_browser_thread.h"
8#include "extensions/browser/info_map.h"
9#include "extensions/common/extension.h"
10#include "extensions/common/extension_paths.h"
11#include "extensions/common/manifest_constants.h"
12#include "testing/gtest/include/gtest/gtest.h"
13
14using content::BrowserThread;
15
16namespace keys = extensions::manifest_keys;
17
18namespace extensions {
19
20class InfoMapTest : public testing::Test {
21 public:
22  InfoMapTest()
23      : ui_thread_(BrowserThread::UI, &message_loop_),
24        io_thread_(BrowserThread::IO, &message_loop_) {}
25
26 private:
27  base::MessageLoop message_loop_;
28  content::TestBrowserThread ui_thread_;
29  content::TestBrowserThread io_thread_;
30};
31
32// Returns a barebones test Extension object with the given name.
33static scoped_refptr<Extension> CreateExtension(const std::string& name) {
34  base::FilePath path;
35  PathService::Get(DIR_TEST_DATA, &path);
36
37  base::DictionaryValue manifest;
38  manifest.SetString(keys::kVersion, "1.0.0.0");
39  manifest.SetString(keys::kName, name);
40
41  std::string error;
42  scoped_refptr<Extension> extension =
43      Extension::Create(path.AppendASCII(name),
44                        Manifest::INVALID_LOCATION,
45                        manifest,
46                        Extension::NO_FLAGS,
47                        &error);
48  EXPECT_TRUE(extension.get()) << error;
49
50  return extension;
51}
52
53// Test that the InfoMap handles refcounting properly.
54TEST_F(InfoMapTest, RefCounting) {
55  scoped_refptr<InfoMap> info_map(new InfoMap());
56
57  // New extensions should have a single reference holding onto them.
58  scoped_refptr<Extension> extension1(CreateExtension("extension1"));
59  scoped_refptr<Extension> extension2(CreateExtension("extension2"));
60  scoped_refptr<Extension> extension3(CreateExtension("extension3"));
61  EXPECT_TRUE(extension1->HasOneRef());
62  EXPECT_TRUE(extension2->HasOneRef());
63  EXPECT_TRUE(extension3->HasOneRef());
64
65  // Add a ref to each extension and give it to the info map.
66  info_map->AddExtension(extension1.get(), base::Time(), false, false);
67  info_map->AddExtension(extension2.get(), base::Time(), false, false);
68  info_map->AddExtension(extension3.get(), base::Time(), false, false);
69
70  // Release extension1, and the info map should have the only ref.
71  const Extension* weak_extension1 = extension1.get();
72  extension1 = NULL;
73  EXPECT_TRUE(weak_extension1->HasOneRef());
74
75  // Remove extension2, and the extension2 object should have the only ref.
76  info_map->RemoveExtension(
77      extension2->id(), extensions::UnloadedExtensionInfo::REASON_UNINSTALL);
78  EXPECT_TRUE(extension2->HasOneRef());
79
80  // Delete the info map, and the extension3 object should have the only ref.
81  info_map = NULL;
82  EXPECT_TRUE(extension3->HasOneRef());
83}
84
85// Tests that we can query a few extension properties from the InfoMap.
86TEST_F(InfoMapTest, Properties) {
87  scoped_refptr<InfoMap> info_map(new InfoMap());
88
89  scoped_refptr<Extension> extension1(CreateExtension("extension1"));
90  scoped_refptr<Extension> extension2(CreateExtension("extension2"));
91
92  info_map->AddExtension(extension1.get(), base::Time(), false, false);
93  info_map->AddExtension(extension2.get(), base::Time(), false, false);
94
95  EXPECT_EQ(2u, info_map->extensions().size());
96  EXPECT_EQ(extension1.get(), info_map->extensions().GetByID(extension1->id()));
97  EXPECT_EQ(extension2.get(), info_map->extensions().GetByID(extension2->id()));
98}
99
100// Tests that extension URLs are properly mapped to local file paths.
101TEST_F(InfoMapTest, MapUrlToLocalFilePath) {
102  scoped_refptr<InfoMap> info_map(new InfoMap());
103  scoped_refptr<Extension> app(CreateExtension("platform_app"));
104  info_map->AddExtension(app.get(), base::Time(), false, false);
105
106  // Non-extension URLs don't map to anything.
107  base::FilePath non_extension_path;
108  GURL non_extension_url("http://not-an-extension.com/");
109  EXPECT_FALSE(info_map->MapUrlToLocalFilePath(
110      non_extension_url, false, &non_extension_path));
111  EXPECT_TRUE(non_extension_path.empty());
112
113  // Valid resources return a valid path.
114  base::FilePath valid_path;
115  GURL valid_url = app->GetResourceURL("manifest.json");
116  EXPECT_TRUE(info_map->MapUrlToLocalFilePath(
117      valid_url, true /* use_blocking_api */, &valid_path));
118  EXPECT_FALSE(valid_path.empty());
119
120  // A file must exist to be mapped to a path using the blocking API.
121  base::FilePath does_not_exist_path;
122  GURL does_not_exist_url = app->GetResourceURL("does-not-exist.html");
123  EXPECT_FALSE(info_map->MapUrlToLocalFilePath(
124      does_not_exist_url, true /* use_blocking_api */, &does_not_exist_path));
125  EXPECT_TRUE(does_not_exist_path.empty());
126
127  // A file does not need to exist to be mapped to a path with the non-blocking
128  // API. This avoids hitting the disk to see if it exists.
129  EXPECT_TRUE(info_map->MapUrlToLocalFilePath(
130      does_not_exist_url, false /* use_blocking_api */, &does_not_exist_path));
131  EXPECT_FALSE(does_not_exist_path.empty());
132}
133
134}  // namespace extensions
135