When an extension is uninstalled, close all desktop notifications from that e...

chrome/browser/chromeos/notifications/desktop_notifications_unittest.cc

 // Copyright (c) 2010 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 "chrome/browser/chromeos/notifications/desktop_notifications_unittest.h"
 
 #include "base/stringprintf.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/common/render_messages_params.h"
 
 namespace chromeos {
 
 // static
 std::string DesktopNotificationsTest::log_output_;
 
 class MockNotificationUI : public BalloonCollectionImpl::NotificationUI {
  public:
   virtual void Add(Balloon* balloon) {}
   virtual bool Update(Balloon* balloon) { return false; }
   virtual void Remove(Balloon* balloon) {}
   virtual void Show(Balloon* balloon) {}
   virtual void ResizeNotification(Balloon* balloon,
                                   const gfx::Size& size) {}
   virtual void SetActiveView(BalloonViewImpl* view) {}
 };
 
-MockBalloonCollection::MockBalloonCollection()
-    : log_proxy_(new LoggingNotificationProxy()) {
+MockBalloonCollection::MockBalloonCollection() {
   set_notification_ui(new MockNotificationUI());
 }
 
 void MockBalloonCollection::Add(const Notification& notification,
                                 Profile* profile) {
-  // Swap in the logging proxy for the purpose of logging calls that
+  // Swap in a logging proxy for the purpose of logging calls that
   // would be made into javascript, then pass this down to the
   // balloon collection.
-  Notification test_notification(notification.origin_url(),
-                                 notification.content_url(),
-                                 notification.display_source(),
-                                 string16(), /* replace_id */
-                                 log_proxy_.get());
+  Notification test_notification(
+      notification.origin_url(),
+      notification.content_url(),
+      notification.display_source(),
+      notification.replace_id(),
+      new LoggingNotificationProxy(notification.notification_id()));
   BalloonCollectionImpl::Add(test_notification, profile);
 }
 
-bool MockBalloonCollection::Remove(const Notification& notification) {
-  Notification test_notification(notification.origin_url(),
-                                 notification.content_url(),
-                                 notification.display_source(),
-                                 string16(), /* replace_id */
-                                 log_proxy_.get());
-  return BalloonCollectionImpl::Remove(test_notification);
-}
-
 Balloon* MockBalloonCollection::MakeBalloon(const Notification& notification,
                                             Profile* profile) {
   // Start with a normal balloon but mock out the view.
   Balloon* balloon = BalloonCollectionImpl::MakeBalloon(notification, profile);
   balloon->set_view(new MockBalloonView(balloon));
   balloons_.insert(balloon);
   return balloon;
 }
 
 void MockBalloonCollection::OnBalloonClosed(Balloon* source) {
@@ skipping 204 matching lines @@
 
   MessageLoopForUI::current()->RunAllPending();
   EXPECT_EQ(1, balloon_collection_->count());
   Balloon* balloon = (*balloon_collection_->balloons().begin());
   GURL data_url = balloon->notification().content_url();
   EXPECT_EQ(std::string::npos, data_url.spec().find("<script>"));
   EXPECT_EQ(std::string::npos, data_url.spec().find("<i>"));
 }
 
 }  // namespace chromeos