blob: 4bd844ab894bb7a221687dd7b7f5ce51d3638375 (
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
114
115
116
117
|
// 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.
#ifndef CHROME_BROWSER_CHROMEOS_NOTIFICATIONS_DESKTOP_NOTIFICATIONS_UNITTEST_H_
#define CHROME_BROWSER_CHROMEOS_NOTIFICATIONS_DESKTOP_NOTIFICATIONS_UNITTEST_H_
#pragma once
#include <set>
#include <string>
#include "base/message_loop.h"
#include "base/string_util.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_test_util.h"
#include "chrome/browser/notifications/notification_ui_manager.h"
#include "chrome/browser/notifications/notifications_prefs_cache.h"
#include "chrome/browser/ui/browser_list.h"
#include "chrome/test/base/testing_profile.h"
#include "chrome/test/testing_pref_service.h"
#include "content/browser/browser_thread.h"
#include "testing/gtest/include/gtest/gtest.h"
struct DesktopNotificationHostMsg_Show_Params;
namespace chromeos {
class DesktopNotificationsTest;
typedef LoggingNotificationDelegate<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();
virtual ~MockBalloonCollection();
// BalloonCollectionImpl overrides
virtual void Add(const Notification& notification,
Profile* profile);
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_;
};
class DesktopNotificationsTest : public testing::Test {
public:
DesktopNotificationsTest();
virtual ~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.
DesktopNotificationHostMsg_Show_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_;
BrowserThread ui_thread_;
// Mock local state.
TestingPrefService local_state_;
// 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_
|
