| // Copyright 2013 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_EXTENSIONS_EXTENSION_NOTIFICATION_OBSERVER_H_ |
| #define CHROME_BROWSER_EXTENSIONS_EXTENSION_NOTIFICATION_OBSERVER_H_ |
| |
| #include <set> |
| #include <string> |
| |
| #include "base/compiler_specific.h" |
| #include "chrome/browser/chrome_notification_types.h" |
| #include "content/public/browser/notification_details.h" |
| #include "content/public/browser/notification_observer.h" |
| #include "content/public/browser/notification_registrar.h" |
| #include "content/public/browser/notification_service.h" |
| #include "content/public/browser/notification_source.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace extensions { |
| |
| // Records LOADED, INSTALLED, and UNLOADED notifications for extensions with |
| // specific IDs. Use in tests only. |
| class ExtensionNotificationObserver : public content::NotificationObserver { |
| public: |
| ExtensionNotificationObserver(content::NotificationSource source, |
| const std::set<std::string>& extension_ids); |
| |
| virtual ~ExtensionNotificationObserver(); |
| |
| // Each of these methods returns a testing::AssertionSuccess if exactly those |
| // notifications occurred for any extensions in |extension_ids|, and no more, |
| // since the last time any of these methods were called. |
| testing::AssertionResult CheckNotifications() WARN_UNUSED_RESULT; |
| testing::AssertionResult CheckNotifications( |
| chrome::NotificationType type) WARN_UNUSED_RESULT; |
| testing::AssertionResult CheckNotifications( |
| chrome::NotificationType t1, |
| chrome::NotificationType t2) WARN_UNUSED_RESULT; |
| testing::AssertionResult CheckNotifications( |
| chrome::NotificationType t1, |
| chrome::NotificationType t2, |
| chrome::NotificationType t3) WARN_UNUSED_RESULT; |
| testing::AssertionResult CheckNotifications( |
| chrome::NotificationType t1, |
| chrome::NotificationType t2, |
| chrome::NotificationType t3, |
| chrome::NotificationType t4, |
| chrome::NotificationType t5, |
| chrome::NotificationType t6) WARN_UNUSED_RESULT; |
| |
| private: |
| // content::NotificationObserver implementation. |
| virtual void Observe(int type, |
| const content::NotificationSource& source, |
| const content::NotificationDetails& details) OVERRIDE; |
| |
| // Checks then clears notifications for our extensions. |
| testing::AssertionResult CheckNotifications( |
| const std::vector<chrome::NotificationType>& types); |
| |
| const std::set<std::string> extension_ids_; |
| std::vector<chrome::NotificationType> notifications_; |
| content::NotificationRegistrar registrar_; |
| }; |
| |
| } // namespace extensions |
| |
| #endif // CHROME_BROWSER_EXTENSIONS_EXTENSION_NOTIFICATION_OBSERVER_H_ |