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 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "extensions/browser/warning_service.h"
#include "content/public/test/test_browser_context.h"
#include "extensions/browser/extensions_test.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extensions {
namespace {
class TestWarningService : public WarningService {
public:
explicit TestWarningService(content::BrowserContext* browser_context)
: WarningService(browser_context) {
}
~TestWarningService() override {}
void AddWarning(const Warning& warning) {
WarningSet warnings;
warnings.insert(warning);
AddWarnings(warnings);
}
};
class MockObserver : public WarningService::Observer {
public:
virtual ~MockObserver() {}
MOCK_METHOD1(ExtensionWarningsChanged, void(const ExtensionIdSet&));
};
using WarningServiceTest = ExtensionsTest;
const char ext1_id[] = "extension1";
const char ext2_id[] = "extension2";
const Warning::WarningType warning_1 = Warning::kNetworkDelay;
const Warning::WarningType warning_2 = Warning::kRepeatedCacheFlushes;
} // namespace
// Check that inserting a warning triggers notifications, whereas inserting
// the same warning again is silent.
TEST_F(WarningServiceTest, SetWarning) {
content::TestBrowserContext browser_context;
TestWarningService warning_service(&browser_context);
MockObserver observer;
warning_service.AddObserver(&observer);
ExtensionIdSet affected_extensions;
affected_extensions.insert(ext1_id);
// Insert warning for the first time.
EXPECT_CALL(observer, ExtensionWarningsChanged(affected_extensions));
warning_service.AddWarning(
Warning::CreateNetworkDelayWarning(ext1_id));
testing::Mock::VerifyAndClearExpectations(&warning_service);
// Second insertion of same warning does not trigger anything.
warning_service.AddWarning(Warning::CreateNetworkDelayWarning(ext1_id));
testing::Mock::VerifyAndClearExpectations(&warning_service);
warning_service.RemoveObserver(&observer);
}
// Check that ClearWarnings deletes exactly the specified warnings and
// triggers notifications where appropriate.
TEST_F(WarningServiceTest, ClearWarnings) {
content::TestBrowserContext browser_context;
TestWarningService warning_service(&browser_context);
MockObserver observer;
warning_service.AddObserver(&observer);
// Insert two unique warnings in one batch.
std::set<std::string> affected_extensions;
affected_extensions.insert(ext1_id);
affected_extensions.insert(ext2_id);
EXPECT_CALL(observer, ExtensionWarningsChanged(affected_extensions));
WarningSet warning_set;
warning_set.insert(Warning::CreateNetworkDelayWarning(ext1_id));
warning_set.insert(Warning::CreateRepeatedCacheFlushesWarning(ext2_id));
warning_service.AddWarnings(warning_set);
testing::Mock::VerifyAndClearExpectations(&warning_service);
// Remove one warning and check that the badge remains.
affected_extensions.clear();
affected_extensions.insert(ext2_id);
EXPECT_CALL(observer, ExtensionWarningsChanged(affected_extensions));
std::set<Warning::WarningType> to_clear;
to_clear.insert(warning_2);
warning_service.ClearWarnings(to_clear);
testing::Mock::VerifyAndClearExpectations(&warning_service);
// Check that the correct warnings appear in |warnings|.
std::set<Warning::WarningType> existing_warnings =
warning_service.GetWarningTypesAffectingExtension(ext1_id);
EXPECT_EQ(1u, existing_warnings.size());
existing_warnings =
warning_service.GetWarningTypesAffectingExtension(ext2_id);
EXPECT_EQ(0u, existing_warnings.size());
// Remove the other one warning.
affected_extensions.clear();
affected_extensions.insert(ext1_id);
EXPECT_CALL(observer, ExtensionWarningsChanged(affected_extensions));
to_clear.insert(warning_1);
warning_service.ClearWarnings(to_clear);
testing::Mock::VerifyAndClearExpectations(&warning_service);
// Check that no warnings remain.
existing_warnings =
warning_service.GetWarningTypesAffectingExtension(ext1_id);
EXPECT_EQ(0u, existing_warnings.size());
existing_warnings =
warning_service.GetWarningTypesAffectingExtension(ext2_id);
EXPECT_EQ(0u, existing_warnings.size());
warning_service.RemoveObserver(&observer);
}
} // namespace extensions
|