summaryrefslogtreecommitdiffstats
path: root/chrome/common/extensions/api/extension_api_unittest.cc
blob: 70982f790cb1e7bc9db9560a16bf95935405b48c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
// Copyright (c) 2012 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/common/extensions/api/extension_api.h"

#include <string>

#include "base/file_path.h"
#include "base/memory/ref_counted.h"
#include "base/memory/scoped_ptr.h"
#include "base/values.h"
#include "chrome/common/extensions/extension.h"
#include "testing/gtest/include/gtest/gtest.h"

using extensions::ExtensionAPI;

TEST(ExtensionAPI, IsPrivileged) {
  ExtensionAPI* extension_api = ExtensionAPI::GetInstance();
  EXPECT_FALSE(extension_api->IsPrivileged("extension.connect"));
  EXPECT_FALSE(extension_api->IsPrivileged("extension.onConnect"));

  // Properties are not supported yet.
  EXPECT_TRUE(extension_api->IsPrivileged("extension.lastError"));

  // Default unknown names to privileged for paranoia's sake.
  EXPECT_TRUE(extension_api->IsPrivileged(""));
  EXPECT_TRUE(extension_api->IsPrivileged("<unknown-namespace>"));
  EXPECT_TRUE(extension_api->IsPrivileged("extension.<unknown-member>"));

  // Exists, but privileged.
  EXPECT_TRUE(extension_api->IsPrivileged("extension.getViews"));
  EXPECT_TRUE(extension_api->IsPrivileged("history.search"));

  // Whole APIs that are unprivileged.
  EXPECT_FALSE(extension_api->IsPrivileged("storage.local"));
  EXPECT_FALSE(extension_api->IsPrivileged("storage.local.onChanged"));
  EXPECT_FALSE(extension_api->IsPrivileged("storage.local.set"));
  EXPECT_FALSE(extension_api->IsPrivileged("storage.local.MAX_ITEMS"));
  EXPECT_FALSE(extension_api->IsPrivileged("storage.set"));
}

TEST(ExtensionAPI, IsWholeAPIPrivileged) {
  ExtensionAPI* extension_api = ExtensionAPI::GetInstance();

  // Completely unprivileged.
  EXPECT_FALSE(extension_api->IsWholeAPIPrivileged("storage"));

  // Partially unprivileged.
  EXPECT_FALSE(extension_api->IsWholeAPIPrivileged("extension"));
  EXPECT_FALSE(extension_api->IsWholeAPIPrivileged("test"));

  // Nothing unprivileged.
  EXPECT_TRUE(extension_api->IsWholeAPIPrivileged("history"));

  // Paranoid above... paranoid here, too.
  EXPECT_TRUE(extension_api->IsWholeAPIPrivileged(""));
  EXPECT_TRUE(extension_api->IsWholeAPIPrivileged("<unknown-namespace>"));
}

TEST(ExtensionAPI, Depends) {
  // Fake extension with the "ttsEngine" permission but not the "tts"
  // permission; it must load TTS.
  DictionaryValue manifest;
  manifest.SetString("name", "test extension");
  manifest.SetString("version", "1.0");
  {
    scoped_ptr<ListValue> permissions(new ListValue());
    permissions->Append(Value::CreateStringValue("ttsEngine"));
    manifest.Set("permissions", permissions.release());
  }

  std::string error;
  scoped_refptr<Extension> extension(Extension::Create(
      FilePath(), Extension::LOAD, manifest, Extension::NO_FLAGS, &error));
  CHECK(extension.get());
  CHECK(error.empty());

  ExtensionAPI::SchemaMap schemas;
  ExtensionAPI::GetInstance()->GetSchemasForExtension(
      *extension, ExtensionAPI::ALL, &schemas);
  EXPECT_EQ(1u, schemas.count("tts"));
}