| Index: ui/message_center/cocoa/notification_controller_unittest.mm
 | 
| diff --git a/ui/message_center/cocoa/notification_controller_unittest.mm b/ui/message_center/cocoa/notification_controller_unittest.mm
 | 
| index 59046aaf01bd0cb8eae40fbfe6c03bad27e1ce35..e01745a024520aefa99729f86ddbabd9efce9a91 100644
 | 
| --- a/ui/message_center/cocoa/notification_controller_unittest.mm
 | 
| +++ b/ui/message_center/cocoa/notification_controller_unittest.mm
 | 
| @@ -65,6 +65,10 @@ class MockMessageCenter : public message_center::FakeMessageCenter {
 | 
|    return closeButton_.get();
 | 
|  }
 | 
|  
 | 
| +- (NSImageView*)smallImageView {
 | 
| +  return smallImage_.get();
 | 
| +}
 | 
| +
 | 
|  - (NSButton*)secondButton {
 | 
|    // The buttons are in Cocoa-y-order, so the 2nd button is first.
 | 
|    NSView* view = [[bottomView_ subviews] objectAtIndex:0];
 | 
| @@ -122,7 +126,9 @@ TEST_F(NotificationControllerTest, BasicLayout) {
 | 
|            DummyNotifierId(),
 | 
|            message_center::RichNotificationData(),
 | 
|            NULL));
 | 
| -  notification->set_icon(gfx::Image([TestIcon() retain]));
 | 
| +  gfx::Image testIcon([TestIcon() retain]);
 | 
| +  notification->set_icon(testIcon);
 | 
| +  notification->set_small_image(testIcon);
 | 
|  
 | 
|    base::scoped_nsobject<MCNotificationController> controller(
 | 
|        [[MCNotificationController alloc] initWithNotification:notification.get()
 | 
| @@ -130,6 +136,7 @@ TEST_F(NotificationControllerTest, BasicLayout) {
 | 
|    [controller view];
 | 
|  
 | 
|    EXPECT_EQ(TestIcon(), [[controller iconView] image]);
 | 
| +  EXPECT_EQ(TestIcon(), [[controller smallImageView] image]);
 | 
|    EXPECT_EQ(base::SysNSStringToUTF16([[controller titleView] string]),
 | 
|              notification->title());
 | 
|    EXPECT_EQ(base::SysNSStringToUTF16([[controller messageView] string]),
 | 
| @@ -209,11 +216,15 @@ TEST_F(NotificationControllerTest, Update) {
 | 
|    EXPECT_EQ(NSHeight([[controller view] frame]),
 | 
|              message_center::kNotificationIconSize);
 | 
|    EXPECT_FALSE([[controller iconView] image]);
 | 
| +  EXPECT_FALSE([[controller smallImageView] image]);
 | 
|  
 | 
|    // Update the icon.
 | 
| -  notification->set_icon(gfx::Image([TestIcon() retain]));
 | 
| +  gfx::Image testIcon([TestIcon() retain]);
 | 
| +  notification->set_icon(testIcon);
 | 
| +  notification->set_small_image(testIcon);
 | 
|    [controller updateNotification:notification.get()];
 | 
|    EXPECT_EQ(TestIcon(), [[controller iconView] image]);
 | 
| +  EXPECT_EQ(TestIcon(), [[controller smallImageView] image]);
 | 
|    EXPECT_EQ(NSHeight([[controller view] frame]),
 | 
|              message_center::kNotificationIconSize);
 | 
|  }
 | 
| 
 |