2222#import " FIRMessaging.h"
2323#import " FIRMessagingExtensionHelper.h"
2424
25+ API_AVAILABLE (ios(10.0 ))
2526typedef void (^FIRMessagingContentHandler)(UNNotificationContent *content);
2627
28+ #if TARGET_OS_IOS
2729static NSString *const kFCMPayloadOptionsName = @" fcm_options"
2830static NSString *const kFCMPayloadOptionsImageURLName = @" image"
2931static NSString *const kValidImageURL =
3032 @" https://firebasestorage.googleapis.com/v0/b/fcm-ios-f7f9c.appspot.com/o/"
3133 @" chubbyBunny.jpg?alt=media&token=d6c56a57-c007-4b27-b20f-f267cc83e9e5"
34+ #endif
3235
3336@interface FIRMessagingExtensionHelper (ExposedForTest)
3437#if TARGET_OS_IOS && __IPHONE_OS_VERSION_MAX_ALLOWED >= __IPHONE_10_0
@@ -46,67 +49,72 @@ @implementation FIRMessagingExtensionHelperTest
4649
4750- (void )setUp {
4851 [super setUp ];
49- FIRMessagingExtensionHelper *extensionHelper = [FIRMessaging extensionHelper ];
50- _mockExtensionHelper = OCMPartialMock (extensionHelper);
52+ if (@available (iOS 10.0 , *)) {
53+ FIRMessagingExtensionHelper *extensionHelper = [FIRMessaging extensionHelper ];
54+ _mockExtensionHelper = OCMPartialMock (extensionHelper);
55+ } else {
56+ // Fallback on earlier versions
57+ }
5158}
5259
5360- (void )tearDown {
5461 [_mockExtensionHelper stopMocking ];
5562}
5663
57- #if TARGET_OS_IOS && __IPHONE_OS_VERSION_MAX_ALLOWED >= __IPHONE_10_0
64+ #if TARGET_OS_IOS
5865#ifdef COCOAPODS
5966// This test requires internet access.
6067- (void )testModifyNotificationWithValidPayloadData {
61- XCTestExpectation *validPayloadExpectation =
62- [ self expectationWithDescription: @" Test payload is valid. " ];
63-
64- UNMutableNotificationContent *content = [[UNMutableNotificationContent alloc ] init ];
65- content.userInfo = @{kFCMPayloadOptionsName : @{kFCMPayloadOptionsImageURLName : kValidImageURL }};
66- FIRMessagingContentHandler handler = ^(UNNotificationContent *content) {
67- [validPayloadExpectation fulfill ];
68- };
69- [_mockExtensionHelper populateNotificationContent: content withContentHandler: handler];
70-
71- OCMVerify ([_mockExtensionHelper loadAttachmentForURL :any ]
72- completionHandler: [OCMArg any ]]) ;
73- [ self waitForExpectationsWithTimeout: 1.0 handler: nil ];
68+ if (@ available (iOS 10.0 , *)) {
69+ XCTestExpectation *validPayloadExpectation =
70+ [ self expectationWithDescription: @" Test payload is valid. " ];
71+ UNMutableNotificationContent *content = [[UNMutableNotificationContent alloc ] init ];
72+ content.userInfo = @{kFCMPayloadOptionsName : @{kFCMPayloadOptionsImageURLName : kValidImageURL }};
73+ FIRMessagingContentHandler handler = ^(UNNotificationContent *content) {
74+ [validPayloadExpectation fulfill ];
75+ };
76+ [_mockExtensionHelper populateNotificationContent: content withContentHandler: handler];
77+ OCMVerify ([_mockExtensionHelper loadAttachmentForURL: [OCMArg any ]
78+ completionHandler :any ]]);
79+ [ self waitForExpectationsWithTimeout: 1.0 handler: nil ] ;
80+ }
7481}
7582#endif
7683
7784- (void )testModifyNotificationWithInvalidPayloadData {
78- XCTestExpectation *validPayloadExpectation =
79- [self expectationWithDescription: @" Test payload is valid."
80-
81- UNMutableNotificationContent *content = [[UNMutableNotificationContent alloc ] init ];
82- content.userInfo =
83- @{kFCMPayloadOptionsName : @{kFCMPayloadOptionsImageURLName : @" a invalid URL"
84- FIRMessagingContentHandler handler = ^(UNNotificationContent *content) {
85- [validPayloadExpectation fulfill ];
86- };
87- [_mockExtensionHelper populateNotificationContent: content withContentHandler: handler];
88-
89- OCMReject ([_mockExtensionHelper loadAttachmentForURL: [OCMArg any ]
90- completionHandler: [OCMArg any ]]);
91- [self waitForExpectationsWithTimeout: 1.0 handler: nil ];
85+ if (@available (iOS 10.0 , *)) {
86+ XCTestExpectation *validPayloadExpectation =
87+ [self expectationWithDescription: @" Test payload is valid."
88+ UNMutableNotificationContent *content = [[UNMutableNotificationContent alloc ] init ];
89+ content.userInfo =
90+ @{kFCMPayloadOptionsName : @{kFCMPayloadOptionsImageURLName : @" a invalid URL"
91+ FIRMessagingContentHandler handler = ^(UNNotificationContent *content) {
92+ [validPayloadExpectation fulfill ];
93+ };
94+ [_mockExtensionHelper populateNotificationContent: content withContentHandler: handler];
95+
96+ OCMReject ([_mockExtensionHelper loadAttachmentForURL: [OCMArg any ]
97+ completionHandler: [OCMArg any ]]);
98+ [self waitForExpectationsWithTimeout: 1.0 handler: nil ];
99+ }
92100}
93101
94102- (void )testModifyNotificationWithEmptyPayloadData {
95- XCTestExpectation *validPayloadExpectation =
96- [ self expectationWithDescription: @" Test payload is valid. " ];
97-
98- UNMutableNotificationContent *content = [[UNMutableNotificationContent alloc ] init ];
99- content.userInfo =
100- @{kFCMPayloadOptionsName : @{kFCMPayloadOptionsImageURLName : @" a invalid URL"
101- FIRMessagingContentHandler handler = ^(UNNotificationContent *content) {
102- [validPayloadExpectation fulfill ];
103- };
104- [_mockExtensionHelper populateNotificationContent: content withContentHandler: handler];
105-
106- OCMReject ([_mockExtensionHelper loadAttachmentForURL :any ]
107- completionHandler: [OCMArg any ]]) ;
108- [ self waitForExpectationsWithTimeout: 1.0 handler: nil ];
103+ if (@ available (iOS 10.0 , *)) {
104+ XCTestExpectation *validPayloadExpectation =
105+ [ self expectationWithDescription: @" Test payload is valid. " ];
106+ UNMutableNotificationContent *content = [[UNMutableNotificationContent alloc ] init ];
107+ content.userInfo =
108+ @{kFCMPayloadOptionsName : @{kFCMPayloadOptionsImageURLName : @" a invalid URL"
109+ FIRMessagingContentHandler handler = ^(UNNotificationContent *content) {
110+ [validPayloadExpectation fulfill ];
111+ };
112+ [_mockExtensionHelper populateNotificationContent: content withContentHandler: handler];
113+ OCMReject ([_mockExtensionHelper loadAttachmentForURL: [OCMArg any ]
114+ completionHandler :any ]]);
115+ [ self waitForExpectationsWithTimeout: 1.0 handler: nil ] ;
116+ }
109117}
110- #endif
118+ #endif // TARGET_OS_IOS
111119
112120@end
0 commit comments