Add ability for a BubbleReference to update the BubbleUI.

components/bubble/bubble_manager_unittest.cc

 // Copyright 2015 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 "components/bubble/bubble_manager.h"
 
 #include "components/bubble/bubble_controller.h"
 #include "components/bubble/bubble_delegate.h"
+#include "components/bubble/bubble_reference.h"
 #include "components/bubble/bubble_ui.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 class MockBubbleUI : public BubbleUI {
  public:
   MockBubbleUI() {}
   ~MockBubbleUI() override { Destroyed(); }
@@ skipping 18 matching lines @@
   static scoped_ptr<MockBubbleDelegate> Stubborn();
 
   MOCK_METHOD1(ShouldClose, bool(BubbleCloseReason reason));
 
   // A scoped_ptr can't be returned in MOCK_METHOD.
   MOCK_METHOD0(BuildBubbleUIMock, BubbleUI*());
   scoped_ptr<BubbleUI> BuildBubbleUI() override {
     return make_scoped_ptr(BuildBubbleUIMock());
   }
 
+  MOCK_METHOD1(UpdateBubbleUI, bool(BubbleUI*));
+
   // To verify destructor call.
   MOCK_METHOD0(Destroyed, void());
 };
 
 // static
 scoped_ptr<MockBubbleDelegate> MockBubbleDelegate::Default() {
   MockBubbleDelegate* delegate = new MockBubbleDelegate;
   EXPECT_CALL(*delegate, BuildBubbleUIMock())
       .WillOnce(testing::Return(new MockBubbleUI));
   EXPECT_CALL(*delegate, ShouldClose(testing::_))
@@ skipping 280 matching lines @@
                                    &chain_helper, &DelegateChainHelper::Chain),
                                testing::Return(true)));
 
   manager_->ShowBubble(make_scoped_ptr(delegate));
   manager_.reset();
 
   // The manager will take the bubble, but not show it.
   ASSERT_TRUE(chain_helper.BubbleWasTaken());
 }
 
+TEST_F(BubbleManagerTest, BubbleUpdatesTrue) {
+  MockBubbleUI* bubble_ui = new MockBubbleUI;
+  MockBubbleDelegate* delegate = new MockBubbleDelegate;
+  EXPECT_CALL(*delegate, BuildBubbleUIMock())
+      .WillOnce(testing::Return(bubble_ui));
+  EXPECT_CALL(*delegate, ShouldClose(testing::_))
+      .WillOnce(testing::Return(true));
+  EXPECT_CALL(*delegate, UpdateBubbleUI(bubble_ui))
+      .WillOnce(testing::Return(true));
+
+  BubbleReference ref = manager_->ShowBubble(make_scoped_ptr(delegate));
+  ASSERT_TRUE(ref->UpdateBubbleUI());
+}
+
+TEST_F(BubbleManagerTest, BubbleUpdatesFalse) {
+  MockBubbleUI* bubble_ui = new MockBubbleUI;
+  MockBubbleDelegate* delegate = new MockBubbleDelegate;
+  EXPECT_CALL(*delegate, BuildBubbleUIMock())
+      .WillOnce(testing::Return(bubble_ui));
+  EXPECT_CALL(*delegate, ShouldClose(testing::_))
+      .WillOnce(testing::Return(true));
+  EXPECT_CALL(*delegate, UpdateBubbleUI(bubble_ui))
+      .WillOnce(testing::Return(false));
+
+  BubbleReference ref = manager_->ShowBubble(make_scoped_ptr(delegate));
+  ASSERT_FALSE(ref->UpdateBubbleUI());
+}
+
 }  // namespace