1 files changed, 8 insertions, 0 deletions
diff --git a/chrome/browser/chromeos/notifications/balloon_collection_impl.cc b/chrome/browser/chromeos/notifications/balloon_collection_impl.cc
index f4b28c5..c8415c1 100644
--- a/chrome/browser/chromeos/notifications/balloon_collection_impl.cc
+++ b/chrome/browser/chromeos/notifications/balloon_collection_impl.cc
@@ -47,6 +47,10 @@ void BalloonCollectionImpl::Add(const Notification& notification,
   // There may be no listener in a unit test.
   if (space_change_listener_)
     space_change_listener_->OnBalloonSpaceChanged();
+
+  // This is used only for testing.
+  if (on_collection_changed_callback_.get())
+    on_collection_changed_callback_->Run();
 }
 
 bool BalloonCollectionImpl::AddWebUIMessageCallback(
@@ -131,6 +135,10 @@ void BalloonCollectionImpl::OnBalloonClosed(Balloon* source) {
   // There may be no listener in a unit test.
   if (space_change_listener_)
     space_change_listener_->OnBalloonSpaceChanged();
+
+  // This is used only for testing.
+  if (on_collection_changed_callback_.get())
+    on_collection_changed_callback_->Run();
 }
 
 const BalloonCollectionImpl::Balloons&