chrome/browser/notifications/desktop_notification_profile_util_unittest.cc

   1 // Copyright 2014 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/bind.h"
   6 #include "chrome/browser/notifications/desktop_notification_profile_util.h"
   7 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
   8 #include "chrome/test/base/testing_profile.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 class DesktopNotificationServiceTest : public ChromeRenderViewHostTestHarness {
  12  protected:
  13   void SetUp() override { ChromeRenderViewHostTestHarness::SetUp(); }
  14 };
  15
  16
  17 TEST_F(DesktopNotificationServiceTest, GetNotificationsSettings) {
  18   DesktopNotificationProfileUtil::GrantPermission(profile(),
  19                                                   GURL("http://allowed2.com"));
  20   DesktopNotificationProfileUtil::GrantPermission(profile(),
  21                                                   GURL("http://allowed.com"));
  22   DesktopNotificationProfileUtil::DenyPermission(profile(),
  23                                                  GURL("http://denied2.com"));
  24   DesktopNotificationProfileUtil::DenyPermission(profile(),
  25                                                  GURL("http://denied.com"));
  26
  27   ContentSettingsForOneType settings;
  28   DesktopNotificationProfileUtil::GetNotificationsSettings(
  29       profile(), &settings);
  30   // |settings| contains the default setting and 4 exceptions.
  31   ASSERT_EQ(5u, settings.size());
  32
  33   EXPECT_EQ(ContentSettingsPattern::FromURLNoWildcard(
  34                 GURL("http://allowed.com")),
  35             settings[0].primary_pattern);
  36   EXPECT_EQ(CONTENT_SETTING_ALLOW,
  37             settings[0].setting);
  38   EXPECT_EQ(ContentSettingsPattern::FromURLNoWildcard(
  39                 GURL("http://allowed2.com")),
  40             settings[1].primary_pattern);
  41   EXPECT_EQ(CONTENT_SETTING_ALLOW,
  42             settings[1].setting);
  43   EXPECT_EQ(ContentSettingsPattern::FromURLNoWildcard(
  44                 GURL("http://denied.com")),
  45             settings[2].primary_pattern);
  46   EXPECT_EQ(CONTENT_SETTING_BLOCK,
  47             settings[2].setting);
  48   EXPECT_EQ(ContentSettingsPattern::FromURLNoWildcard(
  49                 GURL("http://denied2.com")),
  50             settings[3].primary_pattern);
  51   EXPECT_EQ(CONTENT_SETTING_BLOCK,
  52             settings[3].setting);
  53   EXPECT_EQ(ContentSettingsPattern::Wildcard(),
  54             settings[4].primary_pattern);
  55   EXPECT_EQ(CONTENT_SETTING_ASK,
  56             settings[4].setting);
  57 }