iOS Push Notification Action Button Doesn't Appear - ios

Question

I'm trying to add action button for push notification. Actually, It is adding successfully when I build the app and give permission for notifications first time.
The problem begins after when I killed the app and trying to send notification again. Notification is receiving but action button doesn't appear. I think I don't define setNotificationCategories() method in right place but I can't find right solution for this issue. Here is my code: (Thanks for any help)
AppDelegate:
import UIKit
import UserNotifications
#UIApplicationMain
class AppDelegate: UIResponder, UIApplicationDelegate{
var window: UIWindow?
func application(_ application: UIApplication, didFinishLaunchingWithOptions launchOptions: [UIApplicationLaunchOptionsKey: Any]?) -> Bool {
UNUserNotificationCenter.current().requestAuthorization(options: [.alert, .sound]) {(granted, error) in
if !granted {
print("Notification access denied.")
}
setNotificationCategory()
}
if #available(iOS 10.0, *) {
setNotificationCategory()
UNUserNotificationCenter.current().delegate = self
}
return true
}
func applicationWillResignActive(_ application: UIApplication) {
}
func applicationDidEnterBackground(_ application: UIApplication) {
}
func applicationWillEnterForeground(_ application: UIApplication) {
}
func applicationDidBecomeActive(_ application: UIApplication) {
}
func applicationWillTerminate(_ application: UIApplication) {
}
}
#available(iOS 10.0, *)
extension AppDelegate: UNUserNotificationCenterDelegate{
func userNotificationCenter(_ center: UNUserNotificationCenter, didReceive response: UNNotificationResponse, withCompletionHandler completionHandler: #escaping () -> Void) {
if response.notification.request.content.categoryIdentifier == Constants.Notification.Category.general{
if response.actionIdentifier == Constants.Notification.Action.likeAction{
if let notificationID = response.notification.request.content.userInfo["notification_id"]{
NotificationActionHandler.likeActionHandler(notificationID: notificationID as! Int)
}
}
}
completionHandler()
}
func setNotificationCategory(){
let likeAction = UNNotificationAction(identifier: Constants.Notification.Action.likeAction, title: "❤️", options: [])
let generalCategory = UNNotificationCategory(identifier: Constants.Notification.Category.general, actions: [likeAction], intentIdentifiers: [], options: .customDismissAction)
let center = UNUserNotificationCenter.current()
center.setNotificationCategories([generalCategory])
}
}
NotificationActionHandler:
import Foundation
class NotificationActionHandler{
class func likeActionHandler(notificationID: Int){
print("liked! ❤️ \(notificationID)")
}
}