src/chrome/browser/notifications/desktop_notifications_unittest.h

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