blob: b424c675b6e3adf1658cd415c3a08239e7f7fe75 (
plain)
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
|
// Copyright (c) 2010 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.
#ifndef CHROME_BROWSER_CHROMEOS_NOTIFICATIONS_DESKTOP_NOTIFICATIONS_UNITTEST_H_
#define CHROME_BROWSER_CHROMEOS_NOTIFICATIONS_DESKTOP_NOTIFICATIONS_UNITTEST_H_
#include <set>
#include <string>
#include "base/message_loop.h"
#include "base/string_util.h"
#include "chrome/browser/browser_list.h"
#include "chrome/browser/chrome_thread.h"
#include "chrome/browser/chromeos/notifications/balloon_collection_impl.h"
#include "chrome/browser/notifications/balloon.h"
#include "chrome/browser/notifications/desktop_notification_service.h"
#include "chrome/browser/notifications/notification.h"
#include "chrome/browser/notifications/notification_object_proxy.h"
#include "chrome/browser/notifications/notification_test_util.h"
#include "chrome/browser/notifications/notification_ui_manager.h"
#include "chrome/browser/notifications/notifications_prefs_cache.h"
#include "chrome/common/render_messages.h"
#include "chrome/test/testing_profile.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace chromeos {
class DesktopNotificationsTest;
typedef LoggingNotificationProxyBase<DesktopNotificationsTest>
LoggingNotificationProxy;
// Test version of the balloon collection which counts the number
// of notifications that are added to it.
class MockBalloonCollection : public BalloonCollectionImpl {
public:
MockBalloonCollection();
// BalloonCollectionImpl overrides
virtual void Add(const Notification& notification,
Profile* profile);
virtual bool Remove(const Notification& notification);
virtual Balloon* MakeBalloon(const Notification& notification,
Profile* profile);
virtual void OnBalloonClosed(Balloon* source);
// Number of balloons being shown.
std::set<Balloon*>& balloons() { return balloons_; }
int count() const { return balloons_.size(); }
// Returns the highest y-coordinate of all the balloons in the collection.
int UppermostVerticalPosition();
private:
std::set<Balloon*> balloons_;
scoped_refptr<LoggingNotificationProxy> log_proxy_;
};
class DesktopNotificationsTest : public testing::Test {
public:
DesktopNotificationsTest();
~DesktopNotificationsTest();
static void log(const std::string& message) {
log_output_.append(message);
}
Profile* profile() { return profile_.get(); }
protected:
// testing::Test overrides
virtual void SetUp();
virtual void TearDown();
void AllowOrigin(const GURL& origin) {
service_->GrantPermission(origin);
}
void DenyOrigin(const GURL& origin) {
service_->DenyPermission(origin);
}
int HasPermission(const GURL& origin) {
return service_->prefs_cache()->HasPermission(origin);
}
// Constructs a notification parameter structure for use in tests.
ViewHostMsg_ShowNotification_Params StandardTestNotification();
// Create a message loop to allow notifications code to post tasks,
// and a thread so that notifications code runs on the expected thread.
MessageLoopForUI message_loop_;
ChromeThread ui_thread_;
// Test profile.
scoped_ptr<TestingProfile> profile_;
// Mock balloon collection -- owned by the NotificationUIManager
MockBalloonCollection* balloon_collection_;
// Real UI manager.
scoped_ptr<NotificationUIManager> ui_manager_;
// Real DesktopNotificationService
scoped_ptr<DesktopNotificationService> service_;
// Contains the cumulative output of the unit test.
static std::string log_output_;
};
} // namespace chromeos
#endif // CHROME_BROWSER_CHROMEOS_NOTIFICATIONS_DESKTOP_NOTIFICATIONS_UNITTEST_H_
|
