chrome/browser/notifications/notification_exceptions_table_model_unittest.cc

   1 // Copyright (c) 2010 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 "chrome/browser/notifications/notification_exceptions_table_model.h"
   6
   7 #include "base/utf_string_conversions.h"
   8 #include "chrome/test/testing_profile.h"
   9 #include "content/browser/browser_thread.h"
  10 #include "content/browser/renderer_host/test_render_view_host.h"
  11 #include "grit/generated_resources.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13 #include "ui/base/l10n/l10n_util.h"
  14
  15 class NotificationExceptionsTableModelTest : public RenderViewHostTestHarness {
  16  public:
  17   NotificationExceptionsTableModelTest()
  18      : ui_thread_(BrowserThread::UI, MessageLoop::current()) {
  19   }
  20
  21   virtual ~NotificationExceptionsTableModelTest() {
  22   }
  23
  24   virtual void SetUp() {
  25     RenderViewHostTestHarness::SetUp();
  26     service_ = profile()->GetDesktopNotificationService();
  27     ResetModel();
  28   }
  29
  30   virtual void TearDown() {
  31     model_.reset(NULL);
  32     RenderViewHostTestHarness::TearDown();
  33   }
  34
  35   virtual void ResetModel() {
  36     model_.reset(new NotificationExceptionsTableModel(service_));
  37   }
  38
  39   virtual void FillData() {
  40     service_->GrantPermission(GURL("http://e-allowed2.com"));
  41     service_->GrantPermission(GURL("http://allowed.com"));
  42
  43     service_->DenyPermission(GURL("http://denied2.com"));
  44     service_->DenyPermission(GURL("http://denied.com"));
  45     service_->DenyPermission(GURL("http://f-denied3.com"));
  46
  47     ResetModel();
  48   }
  49
  50  protected:
  51   BrowserThread ui_thread_;
  52   scoped_ptr<NotificationExceptionsTableModel> model_;
  53   DesktopNotificationService* service_;
  54 };
  55
  56 TEST_F(NotificationExceptionsTableModelTest, CanCreate) {
  57   EXPECT_EQ(0, model_->RowCount());
  58 }
  59
  60 TEST_F(NotificationExceptionsTableModelTest, RemoveAll) {
  61   FillData();
  62   EXPECT_EQ(2u, service_->GetAllowedOrigins().size());
  63   EXPECT_EQ(3u, service_->GetBlockedOrigins().size());
  64   EXPECT_EQ(5, model_->RowCount());
  65
  66   model_->RemoveAll();
  67   EXPECT_EQ(0, model_->RowCount());
  68
  69   EXPECT_EQ(0u, service_->GetAllowedOrigins().size());
  70   EXPECT_EQ(0u, service_->GetBlockedOrigins().size());
  71 }
  72
  73 TEST_F(NotificationExceptionsTableModelTest, AlphabeticalOrder) {
  74   FillData();
  75   EXPECT_EQ(5, model_->RowCount());
  76
  77   EXPECT_EQ(ASCIIToUTF16("allowed.com"),
  78             model_->GetText(0, IDS_EXCEPTIONS_HOSTNAME_HEADER));
  79   EXPECT_EQ(l10n_util::GetStringUTF16(IDS_EXCEPTIONS_ALLOW_BUTTON),
  80             model_->GetText(0, IDS_EXCEPTIONS_ACTION_HEADER));
  81
  82   EXPECT_EQ(ASCIIToUTF16("denied.com"),
  83             model_->GetText(1, IDS_EXCEPTIONS_HOSTNAME_HEADER));
  84   EXPECT_EQ(l10n_util::GetStringUTF16(IDS_EXCEPTIONS_BLOCK_BUTTON),
  85             model_->GetText(1, IDS_EXCEPTIONS_ACTION_HEADER));
  86
  87   EXPECT_EQ(ASCIIToUTF16("denied2.com"),
  88             model_->GetText(2, IDS_EXCEPTIONS_HOSTNAME_HEADER));
  89   EXPECT_EQ(l10n_util::GetStringUTF16(IDS_EXCEPTIONS_BLOCK_BUTTON),
  90             model_->GetText(2, IDS_EXCEPTIONS_ACTION_HEADER));
  91
  92   EXPECT_EQ(ASCIIToUTF16("e-allowed2.com"),
  93             model_->GetText(3, IDS_EXCEPTIONS_HOSTNAME_HEADER));
  94   EXPECT_EQ(l10n_util::GetStringUTF16(IDS_EXCEPTIONS_ALLOW_BUTTON),
  95             model_->GetText(3, IDS_EXCEPTIONS_ACTION_HEADER));
  96
  97   EXPECT_EQ(ASCIIToUTF16("f-denied3.com"),
  98             model_->GetText(4, IDS_EXCEPTIONS_HOSTNAME_HEADER));
  99   EXPECT_EQ(l10n_util::GetStringUTF16(IDS_EXCEPTIONS_BLOCK_BUTTON),
 100             model_->GetText(4, IDS_EXCEPTIONS_ACTION_HEADER));
 101 }
 102
 103 TEST_F(NotificationExceptionsTableModelTest, RemoveRows) {
 104   FillData();
 105   EXPECT_EQ(5, model_->RowCount());
 106
 107   {
 108     RemoveRowsTableModel::Rows rows;
 109     rows.insert(0);  // allowed.com
 110     rows.insert(3);  // e-allowed2.com
 111     model_->RemoveRows(rows);
 112   }
 113   EXPECT_EQ(3, model_->RowCount());
 114   EXPECT_EQ(0u, service_->GetAllowedOrigins().size());
 115   EXPECT_EQ(3u, service_->GetBlockedOrigins().size());
 116
 117   {
 118     RemoveRowsTableModel::Rows rows;
 119     rows.insert(0);
 120     rows.insert(1);
 121     rows.insert(2);
 122     model_->RemoveRows(rows);
 123   }
 124   EXPECT_EQ(0, model_->RowCount());
 125   EXPECT_EQ(0u, service_->GetAllowedOrigins().size());
 126   EXPECT_EQ(0u, service_->GetBlockedOrigins().size());
 127 }