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

chrome/browser/cocoa/notifications/balloon_controller_unittest.mm

 // 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 "base/scoped_nsobject.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/cocoa/browser_test_helper.h"
 #include "chrome/browser/cocoa/cocoa_test_helper.h"
 #include "chrome/browser/cocoa/notifications/balloon_controller.h"
 #include "chrome/browser/notifications/balloon.h"
@@ skipping 12 matching lines @@
 @implementation TestBalloonController
 - (void)initializeHost {}
 @end
 
 namespace {
 
 // Use a dummy balloon collection for testing.
 class MockBalloonCollection : public BalloonCollection {
   virtual void Add(const Notification& notification,
                    Profile* profile) {}
-  virtual bool Remove(const Notification& notification) { return false; }
+  virtual bool RemoveById(const std::string& id) { return false; }
+  virtual bool RemoveBySourceOrigin(const GURL& origin) { return false; }
   virtual bool HasSpace() const { return true; }
   virtual void ResizeBalloon(Balloon* balloon, const gfx::Size& size) {};
   virtual void DisplayChanged() {}
   virtual void OnBalloonClosed(Balloon* source) {};
   virtual const Balloons& GetActiveBalloons() {
     NOTREACHED();
     return balloons_;
   }
  private:
   Balloons balloons_;
@@ skipping 58 matching lines @@
 
   [controller showWindow:nil];
 
   EXPECT_TRUE([controller desiredTotalWidth] > 100);
   EXPECT_TRUE([controller desiredTotalHeight] > 100);
 
   [controller closeBalloon:YES];
 }
 
 }