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#ifndef CHROME_BROWSER_NOTIFICATIONS_DESKTOP_NOTIFICATIONS_UNITTEST_H_
6#define CHROME_BROWSER_NOTIFICATIONS_DESKTOP_NOTIFICATIONS_UNITTEST_H_
7
8#include <deque>
9#include <string>
10
11#include "base/message_loop/message_loop.h"
12#include "base/prefs/testing_pref_service.h"
13#include "chrome/browser/notifications/balloon_collection_impl.h"
14#include "chrome/browser/notifications/balloon_notification_ui_manager.h"
15#include "chrome/browser/notifications/desktop_notification_service.h"
16#include "chrome/browser/notifications/notification.h"
17#include "chrome/browser/notifications/notification_test_util.h"
18#include "chrome/test/base/testing_browser_process.h"
19#include "chrome/test/base/testing_profile.h"
20#include "content/public/test/render_view_test.h"
21#include "content/public/test/test_browser_thread.h"
22#include "testing/gtest/include/gtest/gtest.h"
23
24class ActiveDesktopMonitor;
25class DesktopNotificationsTest;
26typedef LoggingNotificationDelegate<DesktopNotificationsTest>
27    LoggingNotificationProxy;
28
29// Test version of the balloon collection which counts the number
30// of notifications that are added to it.
31class MockBalloonCollection : public BalloonCollectionImpl {
32 public:
33  MockBalloonCollection();
34  virtual ~MockBalloonCollection();
35
36  // Our mock collection has an area large enough for a fixed number
37  // of balloons.
38  static const int kMockBalloonSpace;
39  int max_balloon_count() const { return kMockBalloonSpace; }
40
41  // BalloonCollectionImpl overrides
42  virtual void Add(const Notification& notification,
43                   Profile* profile) OVERRIDE;
44  virtual bool HasSpace() const OVERRIDE;
45  virtual Balloon* MakeBalloon(const Notification& notification,
46                               Profile* profile) OVERRIDE;
47  virtual void DisplayChanged() OVERRIDE {}
48  virtual void OnBalloonClosed(Balloon* source) OVERRIDE;
49  virtual const BalloonCollection::Balloons& GetActiveBalloons() OVERRIDE;
50
51  // Number of balloons being shown.
52  std::deque<Balloon*>& balloons() { return balloons_; }
53  int count() const { return balloons_.size(); }
54
55  // Returns the highest y-coordinate of all the balloons in the collection.
56  int UppermostVerticalPosition();
57
58  // Returns the height bounds of a balloon.
59  int MinHeight() { return Layout::min_balloon_height(); }
60  int MaxHeight() { return Layout::max_balloon_height(); }
61
62  // Returns the bounding box.
63  gfx::Rect GetBalloonsBoundingBox() {
64    return BalloonCollectionImpl::GetBalloonsBoundingBox();
65  }
66
67 private:
68  std::deque<Balloon*> balloons_;
69};
70
71class DesktopNotificationsTest : public testing::Test {
72 public:
73  DesktopNotificationsTest();
74  virtual ~DesktopNotificationsTest();
75
76  static void log(const std::string& message) {
77    log_output_.append(message);
78  }
79
80  Profile* profile() { return profile_.get(); }
81
82 protected:
83  // testing::Test overrides
84  virtual void SetUp() OVERRIDE;
85  virtual void TearDown() OVERRIDE;
86
87  void AllowOrigin(const GURL& origin) {
88    service_->GrantPermission(origin);
89  }
90
91  void DenyOrigin(const GURL& origin) {
92    service_->DenyPermission(origin);
93  }
94
95  // Constructs a notification parameter structure for use in tests.
96  content::ShowDesktopNotificationHostMsgParams StandardTestNotification();
97
98  // Must be first member. Because we're running a unit test in browser_tests
99  // we need to handle TestingBrowserProcess initialization ourselves.
100  TestingBrowserProcessInitializer initializer_;
101
102  // Create a message loop to allow notifications code to post tasks,
103  // and a thread so that notifications code runs on the expected thread.
104  base::MessageLoopForUI message_loop_;
105  content::TestBrowserThread ui_thread_;
106
107  // Local state mock.
108  TestingPrefServiceSimple local_state_;
109
110  // Test profile.
111  scoped_ptr<TestingProfile> profile_;
112
113  // Mock balloon collection -- owned by the NotificationUIManager
114  MockBalloonCollection* balloon_collection_;
115
116  // Real UI manager.
117  scoped_ptr<BalloonNotificationUIManager> ui_manager_;
118
119  // Real DesktopNotificationService
120  scoped_ptr<DesktopNotificationService> service_;
121
122  // Contains the cumulative output of the unit test.
123  static std::string log_output_;
124};
125
126#endif  // CHROME_BROWSER_NOTIFICATIONS_DESKTOP_NOTIFICATIONS_UNITTEST_H_
127