Add support for native extension notifications

chrome/browser/ui/cocoa/notifications/notification_builder_mac_unittest.mm

 // Copyright 2016 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.
 
 #import <AppKit/AppKit.h>
 
 #include "base/mac/foundation_util.h"
 #include "base/mac/scoped_nsobject.h"
 #include "base/strings/sys_string_conversions.h"
 #include "chrome/browser/notifications/notification_common.h"
@@ skipping 97 matching lines @@
             base::SysNSStringToUTF8([notification actionButtonTitle]));
   EXPECT_EQ("Close", base::SysNSStringToUTF8([notification otherButtonTitle]));
 
   NSArray* buttons = [notification valueForKey:@"_alternateActionButtonTitles"];
   ASSERT_EQ(3u, buttons.count);
   EXPECT_EQ("Button1", base::SysNSStringToUTF8([buttons objectAtIndex:0]));
   EXPECT_EQ("Button2", base::SysNSStringToUTF8([buttons objectAtIndex:1]));
   EXPECT_EQ("Settings", base::SysNSStringToUTF8([buttons objectAtIndex:2]));
 }
 
+TEST(NotificationBuilderMacTest, TestNotificationExtensionNoButtons) {
+  base::scoped_nsobject<NotificationBuilder> builder(
+      [[NotificationBuilder alloc] initWithCloseLabel:@"Close"
+                                         optionsLabel:@"Options"
+                                        settingsLabel:@"Settings"]);
+  [builder setTitle:@"Title"];
+  [builder setSubTitle:@"https://www.miguel.com"];
+  [builder setContextMessage:@"SubTitle"];
+  [builder setNotificationId:@"notificationId"];
+  [builder setProfileId:@"profileId"];
+  [builder setIncognito:false];
+  [builder
+      setNotificationType:[NSNumber
+                              numberWithInt:NotificationCommon::EXTENSION]];
+
+  NSUserNotification* notification = [builder buildUserNotification];
+
+  EXPECT_FALSE(notification.hasActionButton);
+  EXPECT_EQ("Close", base::SysNSStringToUTF8([notification otherButtonTitle]));
+}
+
+TEST(NotificationBuilderMacTest, TestNotificationExtensionButtons) {
+  base::scoped_nsobject<NotificationBuilder> builder(
+      [[NotificationBuilder alloc] initWithCloseLabel:@"Close"
+                                         optionsLabel:@"Options"
+                                        settingsLabel:@"Settings"]);
+  [builder setTitle:@"Title"];
+  [builder setSubTitle:@"https://www.miguel.com"];
+  [builder setContextMessage:@"SubTitle"];
+  [builder setButtons:@"Button1" secondaryButton:@"Button2"];
+  [builder setNotificationId:@"notificationId"];
+  [builder setProfileId:@"profileId"];
+  [builder setIncognito:false];
+  [builder
+      setNotificationType:[NSNumber
+                              numberWithInt:NotificationCommon::EXTENSION]];
+
+  NSUserNotification* notification = [builder buildUserNotification];
+
+  NSArray* buttons = [notification valueForKey:@"_alternateActionButtonTitles"];
+
+  // No settings button
+  ASSERT_EQ(2u, buttons.count);
+  EXPECT_EQ("Button1", base::SysNSStringToUTF8([buttons objectAtIndex:0]));
+  EXPECT_EQ("Button2", base::SysNSStringToUTF8([buttons objectAtIndex:1]));
+}
+
 TEST(NotificationBuilderMacTest, TestUserInfo) {
   base::scoped_nsobject<NotificationBuilder> builder(
       [[NotificationBuilder alloc] initWithCloseLabel:@"Close"
                                          optionsLabel:@"Options"
                                         settingsLabel:@"Settings"]);
   [builder setTitle:@"Title"];
   [builder setProfileId:@"Profile1"];
   [builder setOrigin:@"https://www.miguel.com"];
   [builder setNotificationId:@"Notification1"];
   [builder setIncognito:true];
@@ skipping 42 matching lines @@
       [[NotificationBuilder alloc] initWithDictionary:notificationData]);
 
   NSUserNotification* notification = [finalBuilder buildUserNotification];
 
   EXPECT_EQ("Title", base::SysNSStringToUTF8([notification title]));
   EXPECT_EQ("SubTitle",
             base::SysNSStringToUTF8([notification informativeText]));
   EXPECT_EQ("https://www.miguel.com",
             base::SysNSStringToUTF8([notification subtitle]));
 }