Reland 201932: Add API function chrome.notifications.getAll

chrome/browser/notifications/sync_notifier/synced_notification_unittest.cc

 // Copyright (c) 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.
 
 #include <string>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/notifications/notification.h"
 #include "chrome/browser/notifications/notification_ui_manager.h"
 #include "chrome/browser/notifications/sync_notifier/synced_notification.h"
+#include "chrome/browser/profiles/profile.h"
 #include "sync/api/sync_data.h"
 #include "sync/protocol/sync.pb.h"
 #include "sync/protocol/synced_notification_specifics.pb.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/message_center/message_center_util.h"
 #include "ui/message_center/notification_types.h"
 
 using syncer::SyncData;
 using notifier::SyncedNotification;
 using sync_pb::EntitySpecifics;
@@ skipping 58 matching lines @@
  public:
   StubNotificationUIManager()
       : notification_(GURL(), GURL(), string16(), string16(), NULL) {}
   virtual ~StubNotificationUIManager() {}
 
   // Adds a notification to be displayed. Virtual for unit test override.
   virtual void Add(const Notification& notification, Profile* profile)
       OVERRIDE {
     // Make a deep copy of the notification that we can inspect.
     notification_ = notification;
+    profile_ = profile;
   }
 
   // Returns true if any notifications match the supplied ID, either currently
   // displayed or in the queue.
   virtual bool DoesIdExist(const std::string& id) OVERRIDE {
     return true;
   }
 
   // Removes any notifications matching the supplied ID, either currently
   // displayed or in the queue.  Returns true if anything was removed.
   virtual bool CancelById(const std::string& notification_id) OVERRIDE {
     return false;
   }
 
+  virtual std::set<std::string> GetAllIdsByProfileAndSourceOrigin(
+      Profile* profile,
+      const GURL& source) OVERRIDE {
+    std::set<std::string> notification_ids;
+    if (source == notification_.origin_url() &&
+        profile->IsSameProfile(profile_))
+      notification_ids.insert(notification_.notification_id());
+    return notification_ids;
+  }
+
   // Removes notifications matching the |source_origin| (which could be an
   // extension ID). Returns true if anything was removed.
   virtual bool CancelAllBySourceOrigin(const GURL& source_origin) OVERRIDE {
     return false;
   }
 
   // Removes notifications matching |profile|. Returns true if any were removed.
   virtual bool CancelAllByProfile(Profile* profile) OVERRIDE {
     return false;
   }
 
   // Cancels all pending notifications and closes anything currently showing.
   // Used when the app is terminating.
   virtual void CancelAll() OVERRIDE {}
 
   // Test hook to get the notification so we can check it
   const Notification& notification() const { return notification_; }
 
  private:
   DISALLOW_COPY_AND_ASSIGN(StubNotificationUIManager);
   Notification notification_;
+  Profile* profile_;
 };
 
 class SyncedNotificationTest : public testing::Test {
  public:
   SyncedNotificationTest() {}
   virtual ~SyncedNotificationTest() {}
 
   // Methods from testing::Test.
 
   virtual void SetUp() OVERRIDE {
@@ skipping 484 matching lines @@
   items->Append(item3);
   expected_fields.Set(message_center::kItemsKey, items);
 
   EXPECT_TRUE(expected_fields.Equals(actual_fields))
       << "Expected: " << expected_fields
       << ", but actual: " << *actual_fields;
 
 }
 
 // TODO(petewil): Add a test for a notification being read and or deleted.