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
124
125
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/basictypes.h"
#include "base/command_line.h"
#include "base/utf_string_conversions.h"
#include "chrome/browser/background_page_tracker.h"
#include "chrome/test/testing_browser_process.h"
#include "chrome/test/testing_browser_process_test.h"
#include "chrome/test/testing_pref_service.h"
#include "content/common/notification_service.h"
#include "content/common/notification_type.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/platform_test.h"
class MockBackgroundPageTracker : public BackgroundPageTracker {
public:
MockBackgroundPageTracker() {
BackgroundPageTracker::RegisterPrefs(&prefs_);
}
~MockBackgroundPageTracker() {}
// Overridden from BackgroundPageTracker to mock out functionality.
virtual bool IsEnabled() { return true; }
virtual PrefService* GetPrefService() { return &prefs_; }
private:
TestingPrefService prefs_;
};
class BackgroundPageTrackerTest : public TestingBrowserProcessTest {
};
TEST_F(BackgroundPageTrackerTest, Create) {
MockBackgroundPageTracker tracker;
EXPECT_EQ(0, tracker.GetBackgroundPageCount());
EXPECT_EQ(0, tracker.GetUnacknowledgedBackgroundPageCount());
}
TEST_F(BackgroundPageTrackerTest, OnBackgroundPageLoaded) {
MockBackgroundPageTracker tracker;
EXPECT_EQ(0, tracker.GetBackgroundPageCount());
EXPECT_EQ(0, tracker.GetUnacknowledgedBackgroundPageCount());
std::string app1 = "app_id_1";
std::string app2 = "app_id_2";
tracker.OnBackgroundPageLoaded(app1);
EXPECT_EQ(1, tracker.GetBackgroundPageCount());
EXPECT_EQ(1, tracker.GetUnacknowledgedBackgroundPageCount());
tracker.OnBackgroundPageLoaded(app1);
EXPECT_EQ(1, tracker.GetBackgroundPageCount());
EXPECT_EQ(1, tracker.GetUnacknowledgedBackgroundPageCount());
tracker.OnBackgroundPageLoaded(app2);
EXPECT_EQ(2, tracker.GetBackgroundPageCount());
EXPECT_EQ(2, tracker.GetUnacknowledgedBackgroundPageCount());
tracker.OnExtensionUnloaded(app1);
EXPECT_EQ(1, tracker.GetBackgroundPageCount());
EXPECT_EQ(1, tracker.GetUnacknowledgedBackgroundPageCount());
tracker.OnExtensionUnloaded(app2);
EXPECT_EQ(0, tracker.GetBackgroundPageCount());
EXPECT_EQ(0, tracker.GetUnacknowledgedBackgroundPageCount());
}
TEST_F(BackgroundPageTrackerTest, AcknowledgeBackgroundPages) {
MockBackgroundPageTracker tracker;
EXPECT_EQ(0, tracker.GetBackgroundPageCount());
EXPECT_EQ(0, tracker.GetUnacknowledgedBackgroundPageCount());
std::string app1 = "app_id_1";
tracker.OnBackgroundPageLoaded(app1);
EXPECT_EQ(1, tracker.GetBackgroundPageCount());
EXPECT_EQ(1, tracker.GetUnacknowledgedBackgroundPageCount());
tracker.AcknowledgeBackgroundPages();
EXPECT_EQ(1, tracker.GetBackgroundPageCount());
EXPECT_EQ(0, tracker.GetUnacknowledgedBackgroundPageCount());
tracker.OnBackgroundPageLoaded(app1);
EXPECT_EQ(1, tracker.GetBackgroundPageCount());
EXPECT_EQ(0, tracker.GetUnacknowledgedBackgroundPageCount());
}
class BadgeChangedNotificationCounter : public NotificationObserver {
public:
BadgeChangedNotificationCounter()
: count_(0) {
registrar_.Add(this,
NotificationType::BACKGROUND_PAGE_TRACKER_CHANGED,
NotificationService::AllSources());
}
// # notifications received.
int count() { return count_; }
// NotificationObserver implementation.
virtual void Observe(NotificationType type,
const NotificationSource& source,
const NotificationDetails& details) {
EXPECT_EQ(type.value,
NotificationType::BACKGROUND_PAGE_TRACKER_CHANGED);
count_++;
}
private:
int count_;
NotificationRegistrar registrar_;
};
TEST_F(BackgroundPageTrackerTest, TestTrackerChangedNotifications) {
MockBackgroundPageTracker tracker;
BadgeChangedNotificationCounter counter;
std::string app1 = "app_id_1";
std::string app2 = "app_id_2";
std::string app3 = "app_id_3";
// New extension should generate notification
tracker.OnBackgroundPageLoaded(app1);
EXPECT_EQ(1, counter.count());
// Same extension should not generate notification
tracker.OnBackgroundPageLoaded(app1);
EXPECT_EQ(1, counter.count());
// New extension should generate notification
tracker.OnBackgroundPageLoaded(app2);
EXPECT_EQ(2, counter.count());
// Acknowledging pages should generate notification.
tracker.AcknowledgeBackgroundPages();
EXPECT_EQ(3, counter.count());
tracker.OnBackgroundPageLoaded(app1);
EXPECT_EQ(3, counter.count());
tracker.OnBackgroundPageLoaded(app3);
EXPECT_EQ(4, counter.count());
}
|
