Add test for closing native notifications

chrome/browser/notifications/notification_platform_bridge_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>
 #import <objc/runtime.h>
 
 #include "base/mac/scoped_nsobject.h"
 #import "base/mac/scoped_objc_class_swizzler.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/notifications/notification.h"
 #include "chrome/browser/notifications/notification_platform_bridge_mac.h"
 #include "chrome/browser/notifications/notification_test_util.h"
 #include "chrome/browser/ui/cocoa/cocoa_test_helper.h"
 #include "chrome/browser/ui/cocoa/notifications/notification_builder_mac.h"
 #include "chrome/browser/ui/cocoa/notifications/notification_constants_mac.h"
 #include "chrome/browser/ui/cocoa/notifications/notification_response_builder_mac.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/gtest_mac.h"
 #include "url/gurl.h"
 
+namespace {
+
+NSUserNotification* BuildNotification() {
+  base::scoped_nsobject<NotificationBuilder> builder(
+      [[NotificationBuilder alloc] initWithCloseLabel:@"Close"
+                                         optionsLabel:@"Options"
+                                        settingsLabel:@"Settings"]);
+  [builder setTitle:@"Title"];
+  [builder setSubTitle:@"https://www.miguel.com"];
+  [builder setOrigin:@"https://www.miguel.com/"];
+  [builder setContextMessage:@""];
+  [builder setButtons:@"Button1" secondaryButton:@"Button2"];
+  [builder setTag:@"tag1"];
+  [builder setIcon:[NSImage imageNamed:@"NSApplicationIcon"]];
+  [builder setNotificationId:@"notification_id"];
+  [builder setProfileId:@"profile_id"];
+  [builder setIncognito:false];
+  [builder setNotificationType:@(NotificationCommon::PERSISTENT)];
+
+  return [builder buildUserNotification];
+}
+
+}  // namespace
+
 class NotificationPlatformBridgeMacTest : public CocoaTest {
  protected:
   std::unique_ptr<Notification> CreateNotification(const char* title,
                                                    const char* subtitle,
                                                    const char* origin,
                                                    const char* button1,
                                                    const char* button2) {
     message_center::RichNotificationData optional_fields;
     optional_fields.context_message = base::UTF8ToUTF16(origin);
     if (button1) {
@@ skipping 10 matching lines @@
     std::unique_ptr<Notification> notification(new Notification(
         message_center::NOTIFICATION_TYPE_SIMPLE, base::UTF8ToUTF16(title),
         base::UTF8ToUTF16(subtitle), gfx::Image(),
         message_center::NotifierId(url), base::UTF8ToUTF16("Notifier's Name"),
         url, "id1", optional_fields, new MockNotificationDelegate("id1")));
 
     return notification;
   }
 
   NSMutableDictionary* BuildDefaultNotificationResponse() {
-    base::scoped_nsobject<NotificationBuilder> builder(
-        [[NotificationBuilder alloc] initWithCloseLabel:@"Close"
-                                           optionsLabel:@"Options"
-                                          settingsLabel:@"Settings"]);
-    [builder setTitle:@"Title"];
-    [builder setSubTitle:@"https://www.miguel.com"];
-    [builder setOrigin:@"https://www.miguel.com/"];
-    [builder setContextMessage:@""];
-    [builder setButtons:@"Button1" secondaryButton:@"Button2"];
-    [builder setTag:@"tag1"];
-    [builder setIcon:[NSImage imageNamed:@"NSApplicationIcon"]];
-    [builder setNotificationId:@"notificationId"];
-    [builder setProfileId:@"profileId"];
-    [builder setIncognito:false];
-    [builder
-        setNotificationType:[NSNumber
-                                numberWithInt:NotificationCommon::PERSISTENT]];
-
-    NSUserNotification* notification = [builder buildUserNotification];
     return [NSMutableDictionary
         dictionaryWithDictionary:[NotificationResponseBuilder
-                                     buildDictionary:notification]];
+                                     buildDictionary:BuildNotification()]];
   }
 };
 
 TEST_F(NotificationPlatformBridgeMacTest, TestNotificationVerifyValidResponse) {
   NSDictionary* response = BuildDefaultNotificationResponse();
   EXPECT_TRUE(NotificationPlatformBridgeMac::VerifyNotificationData(response));
 }
 
 TEST_F(NotificationPlatformBridgeMacTest, TestNotificationUnknownType) {
   NSMutableDictionary* response = BuildDefaultNotificationResponse();
@@ skipping 86 matching lines @@
 
 // Expectations for notifications with one button.
 - (void)expectationsOneButton:(NSUserNotification*)notification {
   EXPECT_NSEQ(@"Title", [notification title]);
   EXPECT_NSEQ(@"Context", [notification informativeText]);
   EXPECT_NSEQ(@"https://gmail.com", [notification subtitle]);
   EXPECT_NSEQ(@"Close", [notification otherButtonTitle]);
   EXPECT_NSEQ(@"Options", [notification actionButtonTitle]);
 }
 
+- (NSArray*)expectationsDeliveredNotification {
+  return @[ BuildNotification() ];
+}
+
+- (void)expectationsRemoveDeliveredNotification:
+    (NSUserNotification*)notification {
+  EXPECT_NSEQ(@"Title", [notification title]);
+  EXPECT_NSEQ(@"notification_id",
+              [notification.userInfo
+                  objectForKey:notification_constants::kNotificationId]);
+  EXPECT_NSEQ(@"profile_id",
+              [notification.userInfo
+                  objectForKey:notification_constants::kNotificationProfileId]);
+}
+
 @end
 
 TEST_F(NotificationPlatformBridgeMacTest, TestDisplayNoButtons) {
   base::scoped_nsobject<NSUserNotificationCenter> notification_center(
       [NSUserNotificationCenter _centerForIdentifier:@"" type:0x0]);
   base::mac::ScopedObjCClassSwizzler swizzler(
       [notification_center class], @selector(deliverNotification:),
       @selector(expectationsNoButtons:));
 
   std::unique_ptr<Notification> notification = CreateNotification(
@@ skipping 10 matching lines @@
   base::scoped_nsobject<NSUserNotificationCenter> notification_center(
       [NSUserNotificationCenter _centerForIdentifier:@"" type:0x0]);
   base::mac::ScopedObjCClassSwizzler swizzler(
       [notification_center class], @selector(deliverNotification:),
       @selector(expectationsOneButton:));
   std::unique_ptr<NotificationPlatformBridgeMac> bridge(
       new NotificationPlatformBridgeMac(notification_center));
   bridge->Display(NotificationCommon::PERSISTENT, "notification_id",
                   "profile_id", false, *notification);
 }
+
+TEST_F(NotificationPlatformBridgeMacTest, TestCloseNotification) {
+  base::scoped_nsobject<NSUserNotificationCenter> notification_center(
+      [NSUserNotificationCenter _centerForIdentifier:@"" type:0x0]);
+
+  base::mac::ScopedObjCClassSwizzler delivered_notifications_swizzler(
+      [notification_center class], @selector(deliveredNotifications),
+      @selector(expectationsDeliveredNotification));
+  base::mac::ScopedObjCClassSwizzler remove_delivered_notification_swizzler(
+      [notification_center class], @selector(removeDeliveredNotification:),
+      @selector(expectationsRemoveDeliveredNotification:));
+
+  std::unique_ptr<NotificationPlatformBridgeMac> bridge(
+      new NotificationPlatformBridgeMac(notification_center));
+  bridge->Close("profile_id", "notification_id");
+}