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_NOTIFICATION_TEST_UTIL_H_ 6 #define CHROME_BROWSER_NOTIFICATIONS_NOTIFICATION_TEST_UTIL_H_ 7 8 #include <string> 9 10 #include "chrome/browser/notifications/notification_object_proxy.h" 11 #include "chrome/browser/notifications/balloon.h" 12 #include "ui/gfx/size.h" 13 14 // NotificationDelegate which does nothing, useful for testing when 15 // the notification events are not important. 16 class MockNotificationDelegate : public NotificationDelegate { 17 public: 18 explicit MockNotificationDelegate(const std::string& id); 19 20 // NotificationDelegate interface. Display()21 virtual void Display() OVERRIDE {} Error()22 virtual void Error() OVERRIDE {} Close(bool by_user)23 virtual void Close(bool by_user) OVERRIDE {} Click()24 virtual void Click() OVERRIDE {} 25 virtual std::string id() const OVERRIDE; 26 virtual content::RenderViewHost* GetRenderViewHost() const OVERRIDE; 27 28 private: 29 virtual ~MockNotificationDelegate(); 30 31 std::string id_; 32 33 DISALLOW_COPY_AND_ASSIGN(MockNotificationDelegate); 34 }; 35 36 // Mock implementation of Javascript object proxy which logs events that 37 // would have been fired on it. Useful for tests where the sequence of 38 // notification events needs to be verified. 39 // 40 // |Logger| class provided in template must implement method 41 // static void log(string); 42 template<class Logger> 43 class LoggingNotificationDelegate : public NotificationDelegate { 44 public: LoggingNotificationDelegate(std::string id)45 explicit LoggingNotificationDelegate(std::string id) 46 : notification_id_(id) { 47 } 48 49 // NotificationObjectProxy override Display()50 virtual void Display() OVERRIDE { 51 Logger::log("notification displayed\n"); 52 } Error()53 virtual void Error() OVERRIDE { 54 Logger::log("notification error\n"); 55 } Click()56 virtual void Click() OVERRIDE { 57 Logger::log("notification clicked\n"); 58 } ButtonClick(int index)59 virtual void ButtonClick(int index) OVERRIDE { 60 Logger::log("notification button clicked\n"); 61 } Close(bool by_user)62 virtual void Close(bool by_user) OVERRIDE { 63 if (by_user) 64 Logger::log("notification closed by user\n"); 65 else 66 Logger::log("notification closed by script\n"); 67 } id()68 virtual std::string id() const OVERRIDE { 69 return notification_id_; 70 } GetRenderViewHost()71 virtual content::RenderViewHost* GetRenderViewHost() const OVERRIDE { 72 return NULL; 73 } 74 75 private: 76 std::string notification_id_; 77 78 DISALLOW_COPY_AND_ASSIGN(LoggingNotificationDelegate); 79 }; 80 81 #endif // CHROME_BROWSER_NOTIFICATIONS_NOTIFICATION_TEST_UTIL_H_ 82