extensions/browser/api/device_permissions_prompt_unittest.cc

   1 // Copyright 2015 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/memory/ref_counted.h"
   6 #include "base/strings/utf_string_conversions.h"
   7 #include "extensions/browser/api/device_permissions_prompt.h"
   8 #include "extensions/common/extension.h"
   9 #include "extensions/common/extension_builder.h"
  10 #include "extensions/common/value_builder.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 namespace extensions {
  14
  15 namespace {
  16
  17 class DevicePermissionsPromptTest : public testing::Test {};
  18
  19 TEST_F(DevicePermissionsPromptTest, HidPromptMessages) {
  20   scoped_refptr<Extension> extension =
  21       ExtensionBuilder()
  22           .SetManifest(DictionaryBuilder()
  23                            .Set("name", "Test Application")
  24                            .Set("manifest_version", 2)
  25                            .Set("version", "1.0"))
  26           .Build();
  27
  28   scoped_refptr<DevicePermissionsPrompt::Prompt> prompt =
  29       DevicePermissionsPrompt::CreateHidPromptForTest(extension.get(), false);
  30   EXPECT_EQ(base::ASCIIToUTF16("Select a HID device"), prompt->GetHeading());
  31   EXPECT_EQ(
  32       base::ASCIIToUTF16(
  33           "The application \"Test Application\" is requesting access to one of "
  34           "your devices."),
  35       prompt->GetPromptMessage());
  36
  37   prompt =
  38       DevicePermissionsPrompt::CreateHidPromptForTest(extension.get(), true);
  39   EXPECT_EQ(base::ASCIIToUTF16("Select HID devices"), prompt->GetHeading());
  40   EXPECT_EQ(
  41       base::ASCIIToUTF16(
  42           "The application \"Test Application\" is requesting access to one or "
  43           "more of your devices."),
  44       prompt->GetPromptMessage());
  45 }
  46
  47 TEST_F(DevicePermissionsPromptTest, UsbPromptMessages) {
  48   scoped_refptr<Extension> extension =
  49       ExtensionBuilder()
  50           .SetManifest(DictionaryBuilder()
  51                            .Set("name", "Test Application")
  52                            .Set("manifest_version", 2)
  53                            .Set("version", "1.0"))
  54           .Build();
  55
  56   scoped_refptr<DevicePermissionsPrompt::Prompt> prompt =
  57       DevicePermissionsPrompt::CreateUsbPromptForTest(extension.get(), false);
  58   EXPECT_EQ(base::ASCIIToUTF16("Select a USB device"), prompt->GetHeading());
  59   EXPECT_EQ(
  60       base::ASCIIToUTF16(
  61           "The application \"Test Application\" is requesting access to one of "
  62           "your devices."),
  63       prompt->GetPromptMessage());
  64
  65   prompt =
  66       DevicePermissionsPrompt::CreateUsbPromptForTest(extension.get(), true);
  67   EXPECT_EQ(base::ASCIIToUTF16("Select USB devices"), prompt->GetHeading());
  68   EXPECT_EQ(
  69       base::ASCIIToUTF16(
  70           "The application \"Test Application\" is requesting access to one or "
  71           "more of your devices."),
  72       prompt->GetPromptMessage());
  73 }
  74
  75 }  // namespace
  76
  77 }  // namespace extensions