Use scoped_ptr<> to store ConfirmBubbleModel.

chrome/browser/ui/cocoa/confirm_bubble_controller_unittest.mm

 // Copyright (c) 2012 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/compiler_specific.h"
 #include "base/memory/scoped_nsobject.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/string16.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/ui/cocoa/cocoa_test_helper.h"
 #import "chrome/browser/ui/cocoa/confirm_bubble_controller.h"
 #import "chrome/browser/ui/cocoa/confirm_bubble_view.h"
 #include "chrome/browser/ui/confirm_bubble_model.h"
 #include "grit/theme_resources.h"
 #import "testing/gtest_mac.h"
 #include "ui/base/resource/resource_bundle.h"
 #import "ui/gfx/point.h"
 
 namespace {
 
 // The model object used in this test. This model implements all methods and
 // updates its status when ConfirmBubbleController calls its methods.
 class TestConfirmBubbleModel : public ConfirmBubbleModel {
  public:
+  TestConfirmBubbleModel(bool* model_deleted,
+                         bool* accept_clicked,
+                         bool* cancel_clicked,
+                         bool* link_clicked);
   TestConfirmBubbleModel();
   virtual ~TestConfirmBubbleModel() OVERRIDE;
   virtual string16 GetTitle() const OVERRIDE;
   virtual string16 GetMessageText() const OVERRIDE;
   virtual gfx::Image* GetIcon() const OVERRIDE;
   virtual int GetButtons() const OVERRIDE;
   virtual string16 GetButtonLabel(BubbleButton button) const OVERRIDE;
   virtual void Accept() OVERRIDE;
   virtual void Cancel() OVERRIDE;
   virtual string16 GetLinkText() const OVERRIDE;
   virtual void LinkClicked() OVERRIDE;
 
-  bool accept_clicked() const { return accept_clicked_; }
-  bool cancel_clicked() const { return cancel_clicked_; }
-  bool link_clicked() const { return link_clicked_; }
-
  private:
-  bool accept_clicked_;
-  bool cancel_clicked_;
-  bool link_clicked_;
+  bool* model_deleted_;
+  bool* accept_clicked_;
+  bool* cancel_clicked_;
+  bool* link_clicked_;
 };
 
-TestConfirmBubbleModel::TestConfirmBubbleModel() :
-  accept_clicked_(false),
-  cancel_clicked_(false),
-  link_clicked_(false) {
+TestConfirmBubbleModel::TestConfirmBubbleModel(bool* model_deleted,
+                                               bool* accept_clicked,
+                                               bool* cancel_clicked,
+                                               bool* link_clicked)
+    : model_deleted_(model_deleted),
+      accept_clicked_(accept_clicked),
+      cancel_clicked_(cancel_clicked),
+      link_clicked_(link_clicked) {
 }
 
 TestConfirmBubbleModel::~TestConfirmBubbleModel() {
+  *model_deleted_ = true;
 }
 
 string16 TestConfirmBubbleModel::GetTitle() const {
   return ASCIIToUTF16("Test");
 }
 
 string16 TestConfirmBubbleModel::GetMessageText() const {
   return ASCIIToUTF16("Test Message");
 }
 
 gfx::Image* TestConfirmBubbleModel::GetIcon() const {
   return &ResourceBundle::GetSharedInstance().GetImageNamed(
       IDR_PRODUCT_LOGO_16);
 }
 
 int TestConfirmBubbleModel::GetButtons() const {
   return BUTTON_OK | BUTTON_CANCEL;
 }
 
 string16 TestConfirmBubbleModel::GetButtonLabel(BubbleButton button) const {
   return button == BUTTON_OK ? ASCIIToUTF16("OK") : ASCIIToUTF16("Cancel");
 }
 
 void TestConfirmBubbleModel::Accept() {
-  accept_clicked_ = true;
+  *accept_clicked_ = true;
 }
 
 void TestConfirmBubbleModel::Cancel() {
-  cancel_clicked_ = true;
+  *cancel_clicked_ = true;
 }
 
 string16 TestConfirmBubbleModel::GetLinkText() const {
   return ASCIIToUTF16("Link");
 }
 
 void TestConfirmBubbleModel::LinkClicked() {
-  link_clicked_ = true;
+  *link_clicked_ = true;
 }
 
 }  // namespace
 
 class ConfirmBubbleControllerTest : public CocoaTest {
  public:
-  ConfirmBubbleControllerTest() {
+  ConfirmBubbleControllerTest()
+      : model_deleted_(false),
+        accept_clicked_(false),
+        cancel_clicked_(false),
+        link_clicked_(false) {
     NSView* view = [test_window() contentView];
-    model_.reset(new TestConfirmBubbleModel);
+    // This model is owned by the controller created below.
+    model_ = new TestConfirmBubbleModel(&model_deleted_,
+                                        &accept_clicked_,
+                                        &cancel_clicked_,
+                                        &link_clicked_);
     gfx::Point origin(0, 0);
     controller_ =
         [[ConfirmBubbleController alloc] initWithParent:view
                                                  origin:origin.ToCGPoint()
-                                                  model:model_.get()];
+                                                  model:model_];
     [view addSubview:[controller_ view]
           positioned:NSWindowAbove
           relativeTo:nil];
   }
 
   ConfirmBubbleView* GetBubbleView() const {
     return (ConfirmBubbleView*)[controller_ view];
   }
 
-  TestConfirmBubbleModel* model() const {
-    return model_.get();
-  }
+  TestConfirmBubbleModel* model() const { return model_; }
+  bool model_deleted() const { return model_deleted_; }
+  bool accept_clicked() const { return accept_clicked_; }
+  bool cancel_clicked() const { return cancel_clicked_; }
+  bool link_clicked() const { return link_clicked_; }
 
  private:
   ConfirmBubbleController* controller_;  // weak; owns self
-  scoped_ptr<TestConfirmBubbleModel> model_;
+  TestConfirmBubbleModel* model_;  // weak
+  bool model_deleted_;
+  bool accept_clicked_;
+  bool cancel_clicked_;
+  bool link_clicked_;
 };
 
 // Verify clicking a button or a link removes the ConfirmBubbleView object and
 // calls an appropriate model method.
 TEST_F(ConfirmBubbleControllerTest, ClickOk) {
   NSView* view = [test_window() contentView];
   ConfirmBubbleView* bubble_view = GetBubbleView();
   bool contains_bubble_view = [[view subviews] containsObject:bubble_view];
   EXPECT_TRUE(contains_bubble_view);
 
   // Click its OK button and verify this view has been removed from the test
   // window. Also verify TestConfirmBubbleModel::Accept() has been called.
   [bubble_view clickOk];
 
   contains_bubble_view = [[view subviews] containsObject:bubble_view];
   EXPECT_FALSE(contains_bubble_view);
-  EXPECT_TRUE(model()->accept_clicked());
-  EXPECT_FALSE(model()->cancel_clicked());
-  EXPECT_FALSE(model()->link_clicked());
+  EXPECT_TRUE(accept_clicked());
+  EXPECT_FALSE(cancel_clicked());
+  EXPECT_FALSE(link_clicked());
 }
 
 TEST_F(ConfirmBubbleControllerTest, ClickCancel) {
   NSView* view = [test_window() contentView];
   ConfirmBubbleView* bubble_view = GetBubbleView();
   bool contains_bubble_view = [[view subviews] containsObject:bubble_view];
   EXPECT_TRUE(contains_bubble_view);
 
   // Click its cancel button and verify this view has been removed from the test
   // window. Also verify TestConfirmBubbleModel::Cancel() has been called.
   [bubble_view clickCancel];
 
   contains_bubble_view = [[view subviews] containsObject:bubble_view];
   EXPECT_FALSE(contains_bubble_view);
-  EXPECT_FALSE(model()->accept_clicked());
-  EXPECT_TRUE(model()->cancel_clicked());
-  EXPECT_FALSE(model()->link_clicked());
+  EXPECT_FALSE(accept_clicked());
+  EXPECT_TRUE(cancel_clicked());
+  EXPECT_FALSE(link_clicked());
 }
 
 TEST_F(ConfirmBubbleControllerTest, ClickLink) {
   NSView* view = [test_window() contentView];
   ConfirmBubbleView* bubble_view = GetBubbleView();
   bool contains_bubble_view = [[view subviews] containsObject:bubble_view];
   EXPECT_TRUE(contains_bubble_view);
 
   // Click its link and verify this view has been removed from the test window.
   // Also verify TestConfirmBubbleModel::LinkClicked() has been called.
   [bubble_view clickLink];
 
   contains_bubble_view = [[view subviews] containsObject:bubble_view];
   EXPECT_FALSE(contains_bubble_view);
-  EXPECT_FALSE(model()->accept_clicked());
-  EXPECT_FALSE(model()->cancel_clicked());
-  EXPECT_TRUE(model()->link_clicked());
+  EXPECT_FALSE(accept_clicked());
+  EXPECT_FALSE(cancel_clicked());
+  EXPECT_TRUE(link_clicked());
 }