fix(PushNotifications): Adding missing escape hatch to the plugin.

aws-amplify · Sep 1, 2023 · 2b0bdef · 2b0bdef
1 parent 3fdd8cb
commit 2b0bdef
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/...SPinpointPushNotificationsPlugin/AWSPinpointPushNotificationsPlugin+ClientBehaviour.swift b/...SPinpointPushNotificationsPlugin/AWSPinpointPushNotificationsPlugin+ClientBehaviour.swift
@@ -6,6 +6,7 @@
 //
 
 import Amplify
+import AWSPinpoint
 import Foundation
 @_spi(InternalAWSPinpoint) import InternalAWSPinpoint
 import UserNotifications
@@ -59,6 +60,13 @@ extension AWSPinpointPushNotificationsPlugin {
     }
 #endif
 
+    /// Retrieves the escape hatch to perform actions directly on PinpointClient.
+    ///
+    /// - Returns: PinpointClientProtocol instance
+    public func getEscapeHatch() -> PinpointClientProtocol {
+        pinpoint.pinpointClient
+    }
+
     private func recordNotification(_ userInfo: [String: Any],
                                     applicationState: ApplicationState,
                                     action: PushNotification.Action) async {

diff --git a/...NotificationsPluginUnitTests/AWSPinpointPushNotificationsPluginClientBehaviourTests.swift b/...NotificationsPluginUnitTests/AWSPinpointPushNotificationsPluginClientBehaviourTests.swift
@@ -6,6 +6,7 @@
 //
 
 import Amplify
+import AWSPinpoint
 @_spi(InternalAWSPinpoint) @testable import InternalAWSPinpoint
 @testable import AWSPinpointPushNotificationsPlugin
 import UserNotifications
@@ -206,6 +207,16 @@ class AWSPinpointPushNotificationsPluginClientBehaviourTests: AWSPinpointPushNot
     }
 #endif
 
+    // - MARK: Escape Hatch tests
+    func testGetEscapeHatch() {
+        guard let escapeHatch = plugin.getEscapeHatch() as? PinpointClient else {
+            XCTFail("Unable to retrieve PinpointClient")
+            return
+        }
+
+        XCTAssertTrue(escapeHatch === (mockPinpoint.pinpointClient as? PinpointClient))
+    }
+
     private func createUserInfo(for source: PushNotification.Source) -> Notifications.Push.UserInfo {
         return [
             "data": [