Mac InfoBar: Use cross platform infobar classes

chrome/browser/ui/cocoa/infobars/confirm_infobar_controller_unittest.mm

Index: chrome/browser/ui/cocoa/infobars/confirm_infobar_controller_unittest.mm
diff --git a/chrome/browser/ui/cocoa/infobars/confirm_infobar_controller_unittest.mm b/chrome/browser/ui/cocoa/infobars/confirm_infobar_controller_unittest.mm
index 2ef14045d11ac90580994cacee2fcaa4e418734a..044c59912b7a56ea0d13ba7da85b9206857c1188 100644
--- a/chrome/browser/ui/cocoa/infobars/confirm_infobar_controller_unittest.mm
+++ b/chrome/browser/ui/cocoa/infobars/confirm_infobar_controller_unittest.mm
@@ -10,6 +10,7 @@
 #include "chrome/browser/infobars/confirm_infobar_delegate.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/browser/ui/cocoa/cocoa_profile_test.h"
+#import "chrome/browser/ui/cocoa/infobars/infobar_cocoa.h"
 #import "chrome/browser/ui/cocoa/infobars/infobar_container_controller.h"
 #include "chrome/browser/ui/cocoa/infobars/mock_confirm_infobar_delegate.h"
 #include "chrome/browser/ui/cocoa/run_loop_testing.h"
@@ -35,15 +36,16 @@ using content::WebContents;
 @end
 
 
-@interface InfoBarContainerTest : NSObject<InfoBarContainer> {
+@interface InfoBarContainerTest : NSObject<InfoBarContainerControllerBase> {
   InfoBarController* controller_;
 }
+
 - (id)initWithController:(InfoBarController*)controller;
-- (void)willRemoveController:(InfoBarController*)controller;
-- (void)removeController:(InfoBarController*)controller;
+
 @end
 
 @implementation InfoBarContainerTest
+
 - (id)initWithController:(InfoBarController*)controller {
   if ((self = [super init])) {
     controller_ = controller;
@@ -51,17 +53,19 @@ using content::WebContents;
   return self;
 }
 
-- (void)willRemoveController:(InfoBarController*)controller {
+- (BrowserWindowController*)browserWindowController {
+  return nil;
 }
 

[Robert Sesek, 2013/08/26 14:06:56]

nit: double blank line

[sail, 2013/08/26 18:32:23]

Done.

-- (void)removeController:(InfoBarController*)controller {
-  DCHECK(controller_ == controller);
-  controller_ = nil;
+
+- (BOOL)shouldSuppressTopInfoBarTip {
+  return NO;
 }
 
-- (BrowserWindowController*)browserWindowController {
-  return nil;
+- (CGFloat)infobarArrowX {
+  return 0;
 }
+
 @end
 
 @interface TestConfirmInfoBarController : ConfirmInfoBarController
@@ -70,7 +74,9 @@ using content::WebContents;
 
 @implementation TestConfirmInfoBarController
 - (void)removeSelf {
-  [self close];
+  [self infobarWillClose];
+  if ([self infobar])
+    [self infobar]->CloseSoon();
 }
 @end
 
@@ -88,8 +94,12 @@ class ConfirmInfoBarControllerTest : public CocoaProfileTest,
     InfoBarService* infobar_service =
         InfoBarService::FromWebContents(web_contents_.get());
     delegate_ = new MockConfirmInfoBarDelegate(this);
+    infobar_.reset(new InfoBarCocoa(infobar_service, delegate_));
+
     controller_.reset([[TestConfirmInfoBarController alloc]
-        initWithDelegate:delegate_ owner:infobar_service]);
+        initWithInfoBar:infobar_.get()]);
+    infobar_->set_controller(controller_);
+
     container_.reset(
         [[InfoBarContainerTest alloc] initWithController:controller_]);
     [controller_ setContainerController:container_];
@@ -100,6 +110,7 @@ class ConfirmInfoBarControllerTest : public CocoaProfileTest,
   }
 
   virtual void TearDown() {
+    [controller_ removeSelf];
     if (delegate_)
       delete delegate_;
     CocoaProfileTest::TearDown();
@@ -125,6 +136,7 @@ class ConfirmInfoBarControllerTest : public CocoaProfileTest,
   }
 
   scoped_ptr<WebContents> web_contents_;
+  scoped_ptr<InfoBarCocoa> infobar_;
 };