chrome/browser/extensions/extension_warning_service_unittest.cc - Issue 503033002: Move ExtensionWarningService and ExtensionsWarningSet to extensions.

Side by Side Diff: chrome/browser/extensions/extension_warning_service_unittest.cc

Issue 503033002: Move ExtensionWarningService and ExtensionsWarningSet to extensions. (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@3_web_view_internal

Patch Set: rebase Created 6 years, 3 months ago

Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.

Jump to:

View unified diff | Download patch

OLD	NEW
	(Empty)
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 "chrome/browser/extensions/extension_warning_service.h"

6

7 #include "chrome/browser/profiles/profile.h"

8 #include "chrome/browser/ui/global_error/global_error_service.h"

9 #include "chrome/browser/ui/global_error/global_error_service_factory.h"

10 #include "chrome/test/base/testing_profile.h"

11 #include "testing/gmock/include/gmock/gmock.h"

12 #include "testing/gtest/include/gtest/gtest.h"

13

14 namespace extensions {

15

16 namespace {

17

18 class TestExtensionWarningService : public ExtensionWarningService {

19 public:

20 explicit TestExtensionWarningService(Profile* profile)

21 : ExtensionWarningService(profile) {

22 }

23 virtual ~TestExtensionWarningService() {}

24

25 void AddWarning(const ExtensionWarning& warning) {

26 ExtensionWarningSet warnings;

27 warnings.insert(warning);

28 AddWarnings(warnings);

29 }

30 };

31

32 class MockObserver : public ExtensionWarningService::Observer {

33 public:

34 virtual ~MockObserver() {}

35 MOCK_METHOD0(ExtensionWarningsChanged, void());

36 };

37

38 const char* ext1_id = "extension1";

39 const char* ext2_id = "extension2";

40 const ExtensionWarning::WarningType warning_1 =

41 ExtensionWarning::kNetworkDelay;

42 const ExtensionWarning::WarningType warning_2 =

43 ExtensionWarning::kNetworkConflict;

44

45 } // namespace

46

47 // Check that inserting a warning triggers notifications, whereas inserting

48 // the same warning again is silent.

49 TEST(ExtensionWarningServiceTest, SetWarning) {

50 TestingProfile profile;

51 TestExtensionWarningService warning_service(&profile);

52 MockObserver observer;

53 warning_service.AddObserver(&observer);

54

55 // Insert warning for the first time.

56 EXPECT_CALL(observer, ExtensionWarningsChanged());

57 warning_service.AddWarning(

58 ExtensionWarning::CreateNetworkDelayWarning(ext1_id));

59 testing::Mock::VerifyAndClearExpectations(&warning_service);

60

61 // Second insertion of same warning does not trigger anything.

62 warning_service.AddWarning(

63 ExtensionWarning::CreateNetworkDelayWarning(ext1_id));

64 testing::Mock::VerifyAndClearExpectations(&warning_service);

65

66 warning_service.RemoveObserver(&observer);

67 }

68

69 // Check that ClearWarnings deletes exactly the specified warnings and

70 // triggers notifications where appropriate.

71 TEST(ExtensionWarningServiceTest, ClearWarnings) {

72 TestingProfile profile;

73 TestExtensionWarningService warning_service(&profile);

74 MockObserver observer;

75 warning_service.AddObserver(&observer);

76

77 // Insert two unique warnings in one batch.

78 EXPECT_CALL(observer, ExtensionWarningsChanged());

79 ExtensionWarningSet warning_set;

80 warning_set.insert(ExtensionWarning::CreateNetworkDelayWarning(ext1_id));

81 warning_set.insert(ExtensionWarning::CreateNetworkConflictWarning(ext2_id));

82 warning_service.AddWarnings(warning_set);

83 testing::Mock::VerifyAndClearExpectations(&warning_service);

84

85 // Remove one warning and check that the badge remains.

86 EXPECT_CALL(observer, ExtensionWarningsChanged());

87 std::set<ExtensionWarning::WarningType> to_clear;

88 to_clear.insert(warning_2);

89 warning_service.ClearWarnings(to_clear);

90 testing::Mock::VerifyAndClearExpectations(&warning_service);

91

92 // Check that the correct warnings appear in \|warnings\|.

93 std::set<ExtensionWarning::WarningType> existing_warnings =

94 warning_service.GetWarningTypesAffectingExtension(ext1_id);

95 EXPECT_EQ(1u, existing_warnings.size());

96 existing_warnings =

97 warning_service.GetWarningTypesAffectingExtension(ext2_id);

98 EXPECT_EQ(0u, existing_warnings.size());

99

100 // Remove the other one warning.

101 EXPECT_CALL(observer, ExtensionWarningsChanged());

102 to_clear.insert(warning_1);

103 warning_service.ClearWarnings(to_clear);

104 testing::Mock::VerifyAndClearExpectations(&warning_service);

105

106 // Check that not warnings remain.

107 existing_warnings =

108 warning_service.GetWarningTypesAffectingExtension(ext1_id);

109 EXPECT_EQ(0u, existing_warnings.size());

110 existing_warnings =

111 warning_service.GetWarningTypesAffectingExtension(ext2_id);

112 EXPECT_EQ(0u, existing_warnings.size());

113

114 warning_service.RemoveObserver(&observer);

115 }

116

117 } // namespace extensions

OLD	NEW

« no previous file with comments | « chrome/browser/extensions/extension_warning_service.cc ('k') | chrome/browser/extensions/extension_warning_set.h » ('j') | no next file with comments »