extensions/browser/warning_service_unittest.cc

   1 // Copyright (c) 2012 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 "extensions/browser/warning_service.h"
   6
   7 #include "content/public/test/test_browser_context.h"
   8 #include "extensions/browser/extensions_test.h"
   9 #include "testing/gmock/include/gmock/gmock.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace extensions {
  13
  14 namespace {
  15
  16 class TestWarningService : public WarningService {
  17  public:
  18   explicit TestWarningService(content::BrowserContext* browser_context)
  19       : WarningService(browser_context) {
  20   }
  21   ~TestWarningService() override {}
  22
  23   void AddWarning(const Warning& warning) {
  24     WarningSet warnings;
  25     warnings.insert(warning);
  26     AddWarnings(warnings);
  27   }
  28 };
  29
  30 class MockObserver : public WarningService::Observer {
  31  public:
  32   virtual ~MockObserver() {}
  33   MOCK_METHOD1(ExtensionWarningsChanged, void(const ExtensionIdSet&));
  34 };
  35
  36 typedef ExtensionsTest WarningServiceTest;
  37
  38 const char* ext1_id = "extension1";
  39 const char* ext2_id = "extension2";
  40 const Warning::WarningType warning_1 = Warning::kNetworkDelay;
  41 const Warning::WarningType warning_2 = Warning::kNetworkConflict;
  42
  43 }  // namespace
  44
  45 // Check that inserting a warning triggers notifications, whereas inserting
  46 // the same warning again is silent.
  47 TEST_F(WarningServiceTest, SetWarning) {
  48   content::TestBrowserContext browser_context;
  49   TestWarningService warning_service(&browser_context);
  50   MockObserver observer;
  51   warning_service.AddObserver(&observer);
  52
  53   ExtensionIdSet affected_extensions;
  54   affected_extensions.insert(ext1_id);
  55   // Insert warning for the first time.
  56   EXPECT_CALL(observer, ExtensionWarningsChanged(affected_extensions));
  57   warning_service.AddWarning(
  58       Warning::CreateNetworkDelayWarning(ext1_id));
  59   testing::Mock::VerifyAndClearExpectations(&warning_service);
  60
  61   // Second insertion of same warning does not trigger anything.
  62   warning_service.AddWarning(Warning::CreateNetworkDelayWarning(ext1_id));
  63   testing::Mock::VerifyAndClearExpectations(&warning_service);
  64
  65   warning_service.RemoveObserver(&observer);
  66 }
  67
  68 // Check that ClearWarnings deletes exactly the specified warnings and
  69 // triggers notifications where appropriate.
  70 TEST_F(WarningServiceTest, ClearWarnings) {
  71   content::TestBrowserContext browser_context;
  72   TestWarningService warning_service(&browser_context);
  73   MockObserver observer;
  74   warning_service.AddObserver(&observer);
  75
  76   // Insert two unique warnings in one batch.
  77   std::set<std::string> affected_extensions;
  78   affected_extensions.insert(ext1_id);
  79   affected_extensions.insert(ext2_id);
  80   EXPECT_CALL(observer, ExtensionWarningsChanged(affected_extensions));
  81   WarningSet warning_set;
  82   warning_set.insert(Warning::CreateNetworkDelayWarning(ext1_id));
  83   warning_set.insert(Warning::CreateNetworkConflictWarning(ext2_id));
  84   warning_service.AddWarnings(warning_set);
  85   testing::Mock::VerifyAndClearExpectations(&warning_service);
  86
  87   // Remove one warning and check that the badge remains.
  88   affected_extensions.clear();
  89   affected_extensions.insert(ext2_id);
  90   EXPECT_CALL(observer, ExtensionWarningsChanged(affected_extensions));
  91   std::set<Warning::WarningType> to_clear;
  92   to_clear.insert(warning_2);
  93   warning_service.ClearWarnings(to_clear);
  94   testing::Mock::VerifyAndClearExpectations(&warning_service);
  95
  96   // Check that the correct warnings appear in |warnings|.
  97   std::set<Warning::WarningType> existing_warnings =
  98       warning_service.GetWarningTypesAffectingExtension(ext1_id);
  99   EXPECT_EQ(1u, existing_warnings.size());
 100   existing_warnings =
 101       warning_service.GetWarningTypesAffectingExtension(ext2_id);
 102   EXPECT_EQ(0u, existing_warnings.size());
 103
 104   // Remove the other one warning.
 105   affected_extensions.clear();
 106   affected_extensions.insert(ext1_id);
 107   EXPECT_CALL(observer, ExtensionWarningsChanged(affected_extensions));
 108   to_clear.insert(warning_1);
 109   warning_service.ClearWarnings(to_clear);
 110   testing::Mock::VerifyAndClearExpectations(&warning_service);
 111
 112   // Check that no warnings remain.
 113   existing_warnings =
 114       warning_service.GetWarningTypesAffectingExtension(ext1_id);
 115   EXPECT_EQ(0u, existing_warnings.size());
 116   existing_warnings =
 117       warning_service.GetWarningTypesAffectingExtension(ext2_id);
 118   EXPECT_EQ(0u, existing_warnings.size());
 119
 120   warning_service.RemoveObserver(&observer);
 121 }
 122
 123 }  // namespace extensions