Ensure the xpc transaction is ended

chrome/browser/ui/cocoa/notifications/alert_notification_service.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 "chrome/browser/ui/cocoa/notifications/alert_notification_service.h"
 
 #import "base/mac/scoped_nsobject.h"
 #import "chrome/browser/ui/cocoa/notifications/notification_builder_mac.h"
 #include "chrome/browser/ui/cocoa/notifications/notification_constants_mac.h"
+#import "chrome/browser/ui/cocoa/notifications/xpc_transaction_handler.h"
 
 @class NSUserNotificationCenter;
 
-@implementation AlertNotificationService
+@implementation AlertNotificationService {
+  XPCTransactionHandler* transactionHandler_;
+}
+
+- (instancetype)initWithTransactionHandler:(XPCTransactionHandler*)handler {
+  if ((self = [super init])) {
+    transactionHandler_ = handler;
+  }
+  return self;
+}
 
 - (void)deliverNotification:(NSDictionary*)notificationData {
   base::scoped_nsobject<NotificationBuilder> builder(
       [[NotificationBuilder alloc] initWithDictionary:notificationData]);
 
   NSUserNotification* toast = [builder buildUserNotification];
-
   [[NSUserNotificationCenter defaultUserNotificationCenter]
       deliverNotification:toast];
+  [transactionHandler_ openTransactionIfNeeded];
 }
 
 - (void)closeNotificationWithId:(NSString*)notificationId
                   withProfileId:(NSString*)profileId {
   NSUserNotificationCenter* notificationCenter =
       [NSUserNotificationCenter defaultUserNotificationCenter];
   for (NSUserNotification* candidate in
        [notificationCenter deliveredNotifications]) {
     NSString* candidateId = [candidate.userInfo
         objectForKey:notification_constants::kNotificationId];
 
     NSString* candidateProfileId = [candidate.userInfo
         objectForKey:notification_constants::kNotificationProfileId];
 
     if ([candidateId isEqualToString:notificationId] &&
         [profileId isEqualToString:candidateProfileId]) {
       [notificationCenter removeDeliveredNotification:candidate];
+      [transactionHandler_ closeTransactionIfNeeded];
       break;
     }
   }
 }
 
 - (void)closeAllNotifications {
   [[NSUserNotificationCenter defaultUserNotificationCenter]
       removeAllDeliveredNotifications];
+  [transactionHandler_ closeTransactionIfNeeded];
 }
 
 @end