FIRDynamicLinksTest.m 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222
  1. /*
  2. * Copyright 2018 Google
  3. *
  4. * Licensed under the Apache License, Version 2.0 (the "License");
  5. * you may not use this file except in compliance with the License.
  6. * You may obtain a copy of the License at
  7. *
  8. * http://www.apache.org/licenses/LICENSE-2.0
  9. *
  10. * Unless required by applicable law or agreed to in writing, software
  11. * distributed under the License is distributed on an "AS IS" BASIS,
  12. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13. * See the License for the specific language governing permissions and
  14. * limitations under the License.
  15. */
  16. #import <XCTest/XCTest.h>
  17. #import <FirebaseAnalyticsInterop/FIRAnalyticsInterop.h>
  18. #import <FirebaseCore/FIRApp.h>
  19. #import <FirebaseCore/FIROptions.h>
  20. #import <GoogleUtilities/GULSwizzler+Unswizzle.h>
  21. #import <GoogleUtilities/GULSwizzler.h>
  22. #import "DynamicLinks/FIRDLRetrievalProcessFactory.h"
  23. #import "DynamicLinks/FIRDLRetrievalProcessResult+Private.h"
  24. #import "DynamicLinks/FIRDynamicLink+Private.h"
  25. #import "DynamicLinks/FIRDynamicLinkNetworking+Private.h"
  26. #import "DynamicLinks/FIRDynamicLinks+FirstParty.h"
  27. #import "DynamicLinks/FIRDynamicLinks+Private.h"
  28. #import "DynamicLinks/Utilities/FDLUtilities.h"
  29. #import "OCMock.h"
  30. static NSString *const kAPIKey = @"myAPIKey";
  31. static NSString *const kClientID = @"myClientID.apps.googleusercontent.com";
  32. static NSString *const kStructuredLinkFmtFreeform = @"%@://google/link/%@";
  33. static NSString *const kStructuredLinkFmtDeepLink = @"%@://google/link/?deep_link_id=%@";
  34. static NSString *const kStructuredLinkFmtInvitation = @"%@://google/link/?invitation_id=%@";
  35. static NSString *const kStructuredLinkFmtInvitationWeak =
  36. @"%@://google/link/?invitation_id=%@&match_type=weak";
  37. static NSString *const kStructuredLinkFmtDeepLinkAndInvitation =
  38. @"%@://google/link/?deep_link_id=%@&invitation_id=%@";
  39. static NSString *const kStructuredUniversalLinkFmtFreeForm = @"https://goo.gl/app/sample%@";
  40. static NSString *const kStructuredUniversalLinkFmtDeepLink =
  41. @"https://goo.gl/app/sample?link=%@&isi=585027354";
  42. static NSString *const kStructuredUniversalLinkFmtSubdomain = @"https://sample.page.link%@";
  43. static NSString *const kStructuredUniversalLinkFmtSubdomainDeepLink =
  44. @"https://sample.page.link?link=%@&isi=585027354";
  45. static NSString *const kURLScheme = @"gindeeplinkurl";
  46. static const NSTimeInterval kAsyncTestTimout = 0.5;
  47. /**
  48. * This string was generated by percent-encoding the Tactile URL for the Tokyo American Club in
  49. * Tokyo, and then replacing a '%2B' with a '+' to verify that the '+' does not cause our parsing to
  50. * fail and double-encoding one value to verify that only one decoding pass is run.
  51. */
  52. NSString *kEncodedComplicatedURLString =
  53. @"https%3A%2F%2Fwww.google.com%252Fmaps%2Fplace%2FTokyo+Am"
  54. @"erican%2BClub%2F%4035.658578%2C139.741588%2C3a%2C75y%2C90t%2Fdata%3D%213m8%211e2%213m6%211s42"
  55. @"66698%212e1%213e10%216s%252F%252Fstorage.googleapis.com%252Fstatic.panoramio.com%252Fphotos%2"
  56. @"52Fmedium%252F4266698.jpg%217i640%218i480%214m2%213m1%211s0x0000000000000000%3A0x1b8b8130c791"
  57. @"48e1%216m1%211e1";
  58. /** This string was generated by percent-decoding kEncodedComplicatedURLString. */
  59. NSString *kDecodedComplicatedURLString =
  60. @"https://www.google.com%2Fmaps/place/Tokyo+American+Club/"
  61. @"@35.658578,139.741588,3a,75y,90t/data=!3m8!1e2!3m6!1s4266698!2e1!3e10!6s%2F%2Fstorage.googlea"
  62. @"pis.com%2Fstatic.panoramio.com%2Fphotos%2Fmedium%2F4266698.jpg!7i640!8i480!4m2!3m1!1s0x000000"
  63. @"0000000000:0x1b8b8130c79148e1!6m1!1e1";
  64. static void *kOpenURLHandlerKey = &kOpenURLHandlerKey;
  65. typedef NSURL * (^FakeShortLinkResolverHandler)(NSURL *shortLink);
  66. @interface FIRDynamicLinks (FIRApp)
  67. - (void)configureDynamicLinks:(FIRApp *)app;
  68. - (BOOL)setUpWithLaunchOptions:(nullable NSDictionary *)launchOptions
  69. apiKey:(NSString *)apiKey
  70. clientID:(NSString *)clientID
  71. urlScheme:(nullable NSString *)urlScheme
  72. userDefaults:(nullable NSUserDefaults *)userDefaults;
  73. - (BOOL)canParseUniversalLinkURL:(nullable NSURL *)url;
  74. @end
  75. @interface FakeShortLinkResolver : FIRDynamicLinkNetworking
  76. + (instancetype)resolverWithBlock:(FakeShortLinkResolverHandler)resolverHandler;
  77. @end
  78. @implementation FakeShortLinkResolver {
  79. FakeShortLinkResolverHandler _resolverHandler;
  80. }
  81. + (instancetype)resolverWithBlock:(FakeShortLinkResolverHandler)resolverHandler {
  82. FakeShortLinkResolver *resolver = [[self alloc] init];
  83. resolver->_resolverHandler = [resolverHandler copy];
  84. return resolver;
  85. }
  86. - (void)resolveShortLink:(NSURL *)url
  87. FDLSDKVersion:(NSString *)FDLSDKVersion
  88. completion:(FIRDynamicLinkResolverHandler)completion {
  89. if (_resolverHandler && completion) {
  90. NSURL *resolvedLink = _resolverHandler(url);
  91. completion(resolvedLink, nil);
  92. }
  93. }
  94. @end
  95. // dummy protocol to prevent compile warning
  96. @protocol DummyProtocol <NSObject>
  97. @property(atomic, assign) BOOL retrievingPendingDynamicLink;
  98. @property(nonatomic, readonly) FIRDynamicLinkNetworking *dynamicLinkNetworking;
  99. - (void)handlePendingDynamicLinkRetrievalFailureWithErrorCode:(NSInteger)errorCode
  100. errorDescription:(NSString *)errorDescription
  101. underlyingError:(nullable NSError *)underlyingError;
  102. @end
  103. // Swizzle DynamicLinks.dynamicLinkNetworking property to return fake resolver.
  104. static void SwizzleDynamicLinkNetworking(id linkResolver) {
  105. id (^dynamicLinkNetworkingBlock)(void) = ^id(void) {
  106. return linkResolver;
  107. };
  108. [GULSwizzler swizzleClass:[FIRDynamicLinks class]
  109. selector:@selector(dynamicLinkNetworking)
  110. isClassSelector:NO
  111. withBlock:dynamicLinkNetworkingBlock];
  112. }
  113. static void SwizzleDynamicLinkNetworkingWithMock() {
  114. id linkResolver = OCMPartialMock([[FIRDynamicLinkNetworking alloc] initWithAPIKey:kAPIKey
  115. clientID:kClientID
  116. URLScheme:kURLScheme]);
  117. [[linkResolver stub] resolveShortLink:OCMOCK_ANY FDLSDKVersion:@"1.0.0" completion:OCMOCK_ANY];
  118. SwizzleDynamicLinkNetworking(linkResolver);
  119. }
  120. static void UnswizzleDynamicLinkNetworking() {
  121. [GULSwizzler unswizzleClass:[FIRDynamicLinks class]
  122. selector:@selector(dynamicLinkNetworking)
  123. isClassSelector:NO];
  124. }
  125. @interface FIRDynamicLinksTest : XCTestCase
  126. // An instance of |GINDurableDeepLinkService| used for testing.
  127. @property(nonatomic, strong) FIRDynamicLinks *service;
  128. // An instance of |NSUserDefaults| that have all default values removed.
  129. @property(nonatomic, strong) NSUserDefaults *userDefaults;
  130. // FIRAnalytics mock. Necessary because we don't call [FIRAPP configure].
  131. @property(nonatomic, strong) id analytics;
  132. @end
  133. @implementation FIRDynamicLinksTest
  134. // Disable deprecated warning for internal methods.
  135. #pragma clang diagnostic push
  136. #pragma clang diagnostic ignored "-Wdeprecated-declarations"
  137. #pragma mark - Test lifecycle
  138. - (void)setUp {
  139. [super setUp];
  140. if (!(FIRApp.defaultApp)) {
  141. [FIRApp configure];
  142. }
  143. self.service = [[FIRDynamicLinks alloc] init];
  144. self.userDefaults = [[NSUserDefaults alloc] init];
  145. [self.userDefaults removePersistentDomainForName:[[NSBundle mainBundle] bundleIdentifier]];
  146. self.analytics = OCMProtocolMock(@protocol(FIRAnalyticsInterop));
  147. }
  148. - (void)tearDown {
  149. self.service = nil;
  150. self.userDefaults = nil;
  151. [self.analytics stopMocking];
  152. self.analytics = nil;
  153. [super tearDown];
  154. }
  155. #pragma mark - Set Up.
  156. - (void)testURLScheme_AllNil {
  157. #pragma clang diagnostic push
  158. #pragma clang diagnostic ignored "-Wnonnull"
  159. BOOL setUpSucceed = [self.service setUpWithLaunchOptions:nil
  160. apiKey:nil
  161. clientID:nil
  162. urlScheme:nil
  163. userDefaults:nil];
  164. #pragma clang diagnostic pop
  165. XCTAssertFalse(setUpSucceed, @"Should fail when apiKey and clientID are both nil.");
  166. }
  167. - (void)testURLScheme_NoApiKey {
  168. #pragma clang diagnostic push
  169. #pragma clang diagnostic ignored "-Wnonnull"
  170. BOOL setUpSucceed = [self.service setUpWithLaunchOptions:nil
  171. apiKey:nil
  172. clientID:kClientID
  173. urlScheme:nil
  174. userDefaults:nil];
  175. #pragma clang diagnostic pop
  176. XCTAssertFalse(setUpSucceed, @"Should fail when apiKey is nil.");
  177. }
  178. - (void)testURLScheme_NoClientID {
  179. #pragma clang diagnostic push
  180. #pragma clang diagnostic ignored "-Wnonnull"
  181. BOOL setUpSucceed = [self.service setUpWithLaunchOptions:nil
  182. apiKey:kAPIKey
  183. clientID:nil
  184. urlScheme:nil
  185. userDefaults:nil];
  186. #pragma clang diagnostic pop
  187. XCTAssertFalse(setUpSucceed, @"Should fail when clientID is nil.");
  188. }
  189. - (void)testURLScheme_MinimumParameters {
  190. BOOL setUpSucceed = [self.service setUpWithLaunchOptions:nil
  191. apiKey:kAPIKey
  192. clientID:kClientID
  193. urlScheme:nil
  194. userDefaults:nil];
  195. XCTAssertTrue(setUpSucceed, @"Should not fail when both apiKey and clientID are set.");
  196. }
  197. - (void)testFactoryMethodReturnsProperClassObject {
  198. id service = [FIRDynamicLinks dynamicLinks];
  199. XCTAssertNotNil(service, @"Factory method returned nil");
  200. XCTAssertEqualObjects([service class], [FIRDynamicLinks class],
  201. @"Factory returned incorrect class object");
  202. }
  203. - (void)testURLScheme_LaunchOptionsWithCustomSchemeURL {
  204. NSString *deepLinkString =
  205. [NSString stringWithFormat:kStructuredLinkFmtDeepLink,
  206. [[NSBundle mainBundle] bundleIdentifier], @"abc123"];
  207. NSDictionary *launchOptions =
  208. @{UIApplicationLaunchOptionsURLKey : [NSURL URLWithString:deepLinkString]};
  209. [self.userDefaults setBool:NO forKey:kFIRDLReadDeepLinkAfterInstallKey];
  210. [self.service setUpWithLaunchOptions:launchOptions
  211. apiKey:kAPIKey
  212. clientID:kClientID
  213. urlScheme:nil
  214. userDefaults:self.userDefaults];
  215. XCTAssertTrue([self.userDefaults boolForKey:kFIRDLReadDeepLinkAfterInstallKey]);
  216. }
  217. - (void)testURLScheme_LaunchOptionsWithUniversalLinkURL {
  218. NSString *deepLinkString =
  219. [NSString stringWithFormat:kStructuredUniversalLinkFmtDeepLink, @"abc123"];
  220. NSDictionary *launchOptions =
  221. @{UIApplicationLaunchOptionsURLKey : [NSURL URLWithString:deepLinkString]};
  222. [self.userDefaults setBool:NO forKey:kFIRDLReadDeepLinkAfterInstallKey];
  223. [self.service setUpWithLaunchOptions:launchOptions
  224. apiKey:kAPIKey
  225. clientID:kClientID
  226. urlScheme:nil
  227. userDefaults:self.userDefaults];
  228. XCTAssertTrue([self.userDefaults boolForKey:kFIRDLReadDeepLinkAfterInstallKey]);
  229. }
  230. - (void)testURLScheme_LaunchOptionsWithInvalidURLWillNotResetUserDefaultsFlag {
  231. NSDictionary *launchOptions =
  232. @{UIApplicationLaunchOptionsURLKey : [NSURL URLWithString:@"https://www.google.com"]};
  233. [self.userDefaults setBool:YES forKey:kFIRDLReadDeepLinkAfterInstallKey];
  234. [self.service setUpWithLaunchOptions:launchOptions
  235. apiKey:kAPIKey
  236. clientID:kClientID
  237. urlScheme:nil
  238. userDefaults:self.userDefaults];
  239. XCTAssertTrue([self.userDefaults boolForKey:kFIRDLReadDeepLinkAfterInstallKey]);
  240. }
  241. - (void)testURLScheme_Nil {
  242. BOOL setUpSucceed = [self.service setUpWithLaunchOptions:nil
  243. apiKey:kAPIKey
  244. clientID:kClientID
  245. urlScheme:nil
  246. userDefaults:nil];
  247. XCTAssertTrue(setUpSucceed);
  248. XCTAssertEqualObjects(self.service.URLScheme, [NSBundle mainBundle].bundleIdentifier);
  249. }
  250. - (void)testURLScheme_EmptyString {
  251. BOOL setUpSucceed = [self.service setUpWithLaunchOptions:nil
  252. apiKey:kAPIKey
  253. clientID:kClientID
  254. urlScheme:@""
  255. userDefaults:nil];
  256. XCTAssertTrue(setUpSucceed);
  257. XCTAssertEqualObjects(self.service.URLScheme, [NSBundle mainBundle].bundleIdentifier);
  258. }
  259. - (void)testURLScheme_NonNil {
  260. [self.service setUpWithLaunchOptions:nil
  261. apiKey:kAPIKey
  262. clientID:kClientID
  263. urlScheme:kURLScheme
  264. userDefaults:self.userDefaults];
  265. XCTAssertEqualObjects(self.service.URLScheme, kURLScheme,
  266. @"URLScheme should be the same as in argument");
  267. }
  268. // TODO (b/63079414) Re-enable this
  269. - (void)DISABLED_testConfigNamedFIRApp {
  270. [self removeAllFIRApps];
  271. id deepLinkServicePartialMock = OCMPartialMock([FIRDynamicLinks dynamicLinks]);
  272. [[deepLinkServicePartialMock reject] configureDynamicLinks:[OCMArg any]];
  273. [FIRApp configureWithName:@"NonDefaultName" options:[FIROptions defaultOptions]];
  274. [deepLinkServicePartialMock stopMocking];
  275. }
  276. // TODO (b/37855379) re-enable the test
  277. - (void)DISABLED_testConfigForFIRApp {
  278. [self removeAllFIRApps];
  279. id deepLinkServicePartialMock = OCMPartialMock([FIRDynamicLinks dynamicLinks]);
  280. [FIRApp configure];
  281. OCMVerify([deepLinkServicePartialMock configureDynamicLinks:[OCMArg any]]);
  282. [deepLinkServicePartialMock stopMocking];
  283. }
  284. #pragma mark - dynamicLinkFromCustomSchemeURL
  285. - (void)testCustomScheme_NoDeepLink {
  286. NSString *urlString =
  287. [NSString stringWithFormat:kStructuredLinkFmtFreeform, kURLScheme, @"abc123xyz"];
  288. NSURL *url = [NSURL URLWithString:urlString];
  289. [self.service setUpWithLaunchOptions:nil
  290. apiKey:kAPIKey
  291. clientID:kClientID
  292. urlScheme:kURLScheme
  293. userDefaults:self.userDefaults];
  294. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  295. XCTAssertNil(dynamicLink, @"invite should be nil since there is no parameter.");
  296. }
  297. - (void)testCustomScheme_DeepLinkOnly {
  298. NSString *deepLinkString = @"https://developers.google.com/products/";
  299. NSString *urlString =
  300. [NSString stringWithFormat:kStructuredLinkFmtDeepLink, kURLScheme, deepLinkString];
  301. NSURL *url = [NSURL URLWithString:urlString];
  302. [self.service setUpWithLaunchOptions:nil
  303. apiKey:kAPIKey
  304. clientID:kClientID
  305. urlScheme:kURLScheme
  306. userDefaults:self.userDefaults];
  307. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  308. XCTAssertEqualObjects(dynamicLink.url.absoluteString, deepLinkString);
  309. XCTAssertNil(dynamicLink.inviteId);
  310. }
  311. - (void)testCustomScheme_InvitationOnly {
  312. NSString *invitationId = @"213920940217491274389172947";
  313. NSString *urlString =
  314. [NSString stringWithFormat:kStructuredLinkFmtInvitation, kURLScheme, invitationId];
  315. NSURL *url = [NSURL URLWithString:urlString];
  316. [self.service setUpWithLaunchOptions:nil
  317. apiKey:kAPIKey
  318. clientID:kClientID
  319. urlScheme:kURLScheme
  320. userDefaults:self.userDefaults];
  321. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  322. XCTAssertNil(dynamicLink.url);
  323. XCTAssertEqualObjects(dynamicLink.inviteId, invitationId);
  324. }
  325. - (void)testCustomScheme_DeepLinkAndInvitation {
  326. NSString *deepLinkString = @"https://developers.google.com/products/";
  327. NSString *invitationId = @"21392094021749127-4389172947";
  328. NSString *urlString = [NSString stringWithFormat:kStructuredLinkFmtDeepLinkAndInvitation,
  329. [[NSBundle mainBundle] bundleIdentifier],
  330. deepLinkString, invitationId];
  331. NSURL *url = [NSURL URLWithString:urlString];
  332. [self.service setUpWithLaunchOptions:nil
  333. apiKey:kAPIKey
  334. clientID:kClientID
  335. urlScheme:kURLScheme
  336. userDefaults:self.userDefaults];
  337. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  338. XCTAssertEqualObjects(dynamicLink.url.absoluteString, deepLinkString);
  339. XCTAssertEqualObjects(dynamicLink.inviteId, invitationId);
  340. }
  341. - (void)testCustomScheme_FirstTimeOpenedWithCustomSchemeShouldGetStrongMatch {
  342. NSString *invitationId = @"21392094021749127-4389172947";
  343. NSString *urlString =
  344. [NSString stringWithFormat:kStructuredLinkFmtInvitation, kURLScheme, invitationId];
  345. NSURL *url = [NSURL URLWithString:urlString];
  346. [self.service setUpWithLaunchOptions:nil
  347. apiKey:kAPIKey
  348. clientID:kClientID
  349. urlScheme:kURLScheme
  350. userDefaults:self.userDefaults];
  351. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  352. XCTAssertEqual(dynamicLink.matchConfidence, FIRDynamicLinkMatchConfidenceStrong,
  353. @"matchConfidence should be strong when app is first opened with custom scheme.");
  354. XCTAssertNil(dynamicLink.url);
  355. XCTAssertEqualObjects(dynamicLink.inviteId, invitationId);
  356. }
  357. - (void)testCustomScheme_FirstTimeOpenedFromFingerprintCodepathShouldGetWeakMatch {
  358. NSString *invitationId = @"21392094021749127-4389172947";
  359. NSString *urlString =
  360. [NSString stringWithFormat:kStructuredLinkFmtInvitationWeak, kURLScheme, invitationId];
  361. NSURL *url = [NSURL URLWithString:urlString];
  362. [self.service setUpWithLaunchOptions:nil
  363. apiKey:kAPIKey
  364. clientID:kClientID
  365. urlScheme:kURLScheme
  366. userDefaults:self.userDefaults];
  367. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  368. XCTAssertEqual(
  369. dynamicLink.matchConfidence, FIRDynamicLinkMatchConfidenceWeak,
  370. @"matchConfidence should be weak when app is first opened from fingerprint codepath.");
  371. XCTAssertNil(dynamicLink.url);
  372. XCTAssertEqualObjects(dynamicLink.inviteId, invitationId);
  373. }
  374. - (void)testCustomScheme_StrongMatch {
  375. NSString *invitationId = @"21392094021749127-4389172947";
  376. NSString *urlString =
  377. [NSString stringWithFormat:kStructuredLinkFmtInvitation, kURLScheme, invitationId];
  378. NSURL *url = [NSURL URLWithString:urlString];
  379. // Simulate opening the app.
  380. [[NSUserDefaults standardUserDefaults] setBool:YES forKey:kFIRDLReadDeepLinkAfterInstallKey];
  381. [self.service setUpWithLaunchOptions:nil
  382. apiKey:kAPIKey
  383. clientID:kClientID
  384. urlScheme:kURLScheme
  385. userDefaults:self.userDefaults];
  386. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  387. XCTAssertEqual(dynamicLink.matchConfidence, FIRDynamicLinkMatchConfidenceStrong,
  388. @"matchConfidence should be strong when opening an url after app is installed.");
  389. XCTAssertNil(dynamicLink.url);
  390. XCTAssertEqualObjects(dynamicLink.inviteId, invitationId);
  391. }
  392. - (void)testLinkParamWithPlus {
  393. [self.service setUpWithLaunchOptions:nil
  394. apiKey:kAPIKey
  395. clientID:kClientID
  396. urlScheme:kURLScheme
  397. userDefaults:self.userDefaults];
  398. NSString *durableDeepLinkString =
  399. [NSString stringWithFormat:@"gindeeplinkurl://google/link?deep_link_id=%@",
  400. kEncodedComplicatedURLString];
  401. NSURL *durabledeepLinkURL = [NSURL URLWithString:durableDeepLinkString];
  402. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:durabledeepLinkURL];
  403. NSString *deepLinkURLString = dynamicLink.url.absoluteString;
  404. XCTAssertEqualObjects(kDecodedComplicatedURLString, deepLinkURLString,
  405. @"ddl url parameter and deep link url should be the same");
  406. }
  407. #pragma mark - dynamicLinkFromUniversalLinkURL
  408. - (void)testUniversalLink_NoDeepLink {
  409. NSString *webPageURLString =
  410. [NSString stringWithFormat:kStructuredUniversalLinkFmtFreeForm, @"/abc123"];
  411. NSURL *url = [NSURL URLWithString:webPageURLString];
  412. NSUserActivity *activity =
  413. [[NSUserActivity alloc] initWithActivityType:NSUserActivityTypeBrowsingWeb];
  414. activity.webpageURL = [NSURL URLWithString:webPageURLString];
  415. [self.service setUpWithLaunchOptions:nil
  416. apiKey:kAPIKey
  417. clientID:kClientID
  418. urlScheme:kURLScheme
  419. userDefaults:self.userDefaults];
  420. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  421. XCTAssertNil(dynamicLink, @"invite should be nil since there is no parameter.");
  422. }
  423. - (void)testDynamicLinkFromUniversalLinkURLWithSpecialCharacters {
  424. NSString *durableDeepLinkString =
  425. [NSString stringWithFormat:@"https://xyz.page.link/?link=%@", kEncodedComplicatedURLString];
  426. NSURL *durabledeepLinkURL = [NSURL URLWithString:durableDeepLinkString];
  427. SwizzleDynamicLinkNetworkingWithMock();
  428. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromUniversalLinkURL:durabledeepLinkURL];
  429. NSString *deepLinkURLString = dynamicLink.url.absoluteString;
  430. XCTAssertEqualObjects(kDecodedComplicatedURLString, deepLinkURLString,
  431. @"ddl url parameter and deep link url should be the same");
  432. UnswizzleDynamicLinkNetworking();
  433. }
  434. - (void)testDynamicLinkFromUniversalLinkURLWithEncodedCharacters {
  435. NSString *durableDeepLinkString =
  436. [NSString stringWithFormat:@"https://xyz.page.link/?link=%@", kEncodedComplicatedURLString];
  437. NSURL *durabledeepLinkURL = [NSURL URLWithString:durableDeepLinkString];
  438. SwizzleDynamicLinkNetworkingWithMock();
  439. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromUniversalLinkURL:durabledeepLinkURL];
  440. NSString *deepLinkURLString = dynamicLink.url.absoluteString;
  441. XCTAssertEqualObjects(kDecodedComplicatedURLString, deepLinkURLString,
  442. @"ddl url parameter and deep link url should be the same");
  443. UnswizzleDynamicLinkNetworking();
  444. }
  445. - (void)testUniversalLink_DeepLink {
  446. NSString *deepLinkString = @"https://www.google.com/maps/place/Minneapolis";
  447. NSString *webPageURLString =
  448. [NSString stringWithFormat:kStructuredUniversalLinkFmtDeepLink, deepLinkString];
  449. NSURL *url = [NSURL URLWithString:webPageURLString];
  450. [self.service setUpWithLaunchOptions:nil
  451. apiKey:kAPIKey
  452. clientID:kClientID
  453. urlScheme:kURLScheme
  454. userDefaults:self.userDefaults];
  455. SwizzleDynamicLinkNetworkingWithMock();
  456. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromUniversalLinkURL:url];
  457. XCTAssertEqual(dynamicLink.matchConfidence, FIRDynamicLinkMatchConfidenceStrong);
  458. XCTAssertEqualObjects(dynamicLink.url.absoluteString, deepLinkString);
  459. UnswizzleDynamicLinkNetworking();
  460. }
  461. - (void)testUniversalLink_DeepLinkWithParameters {
  462. NSString *deepLinkString = @"https://www.google.com?key1%3Dvalue1%26key2%3Dvalue2";
  463. NSString *parsedDeepLinkString = @"https://www.google.com?key1=value1&key2=value2";
  464. NSString *webPageURLString =
  465. [NSString stringWithFormat:kStructuredUniversalLinkFmtDeepLink, deepLinkString];
  466. NSURL *url = [NSURL URLWithString:webPageURLString];
  467. [self.service setUpWithLaunchOptions:nil
  468. apiKey:kAPIKey
  469. clientID:kClientID
  470. urlScheme:kURLScheme
  471. userDefaults:self.userDefaults];
  472. SwizzleDynamicLinkNetworkingWithMock();
  473. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromUniversalLinkURL:url];
  474. XCTAssertEqual(dynamicLink.matchConfidence, FIRDynamicLinkMatchConfidenceStrong);
  475. XCTAssertEqualObjects(dynamicLink.url.absoluteString, parsedDeepLinkString);
  476. UnswizzleDynamicLinkNetworking();
  477. }
  478. - (void)testResolveLinkReturnsDLWithNilMinAppVersionWhenNotPresent {
  479. [self.service setUpWithLaunchOptions:nil
  480. apiKey:kAPIKey
  481. clientID:kClientID
  482. urlScheme:kURLScheme
  483. userDefaults:self.userDefaults];
  484. NSString *urlString = @"http://reinl.page.link/t4ionvr";
  485. NSURL *url = [NSURL URLWithString:urlString];
  486. void (^executeRequestBlock)(id, NSDictionary *, NSString *, FIRNetworkRequestCompletionHandler) =
  487. ^(id p1, NSDictionary *requestBody, NSString *requestURLString,
  488. FIRNetworkRequestCompletionHandler handler) {
  489. NSDictionary *dictionary = @{kFDLResolvedLinkDeepLinkURLKey : kEncodedComplicatedURLString};
  490. NSData *data = FIRDataWithDictionary(dictionary, nil);
  491. handler(data, nil);
  492. };
  493. SEL executeRequestSelector = @selector(executeOnePlatformRequest:forURL:completionHandler:);
  494. [GULSwizzler swizzleClass:[FIRDynamicLinkNetworking class]
  495. selector:executeRequestSelector
  496. isClassSelector:NO
  497. withBlock:executeRequestBlock];
  498. XCTestExpectation *expectation = [self expectationWithDescription:@"handler called"];
  499. [self.service
  500. handleUniversalLink:url
  501. completion:^(FIRDynamicLink *_Nullable dynamicLink, NSError *_Nullable error) {
  502. XCTAssertNil(dynamicLink.minimumAppVersion,
  503. @"Min app version not nil when not present.");
  504. [expectation fulfill];
  505. }];
  506. [self waitForExpectationsWithTimeout:kAsyncTestTimout handler:nil];
  507. }
  508. - (void)testResolveLinkReturnsDLWithMinAppVersionWhenPresent {
  509. NSString *expectedMinVersion = @"8g5u3e";
  510. [self.service setUpWithLaunchOptions:nil
  511. apiKey:kAPIKey
  512. clientID:kClientID
  513. urlScheme:kURLScheme
  514. userDefaults:self.userDefaults];
  515. NSString *urlString = @"http://reinl.page.link/t4ionvr";
  516. NSURL *url = [NSURL URLWithString:urlString];
  517. void (^executeRequestBlock)(id, NSDictionary *, NSString *, FIRNetworkRequestCompletionHandler) =
  518. ^(id p1, NSDictionary *requestBody, NSString *requestURLString,
  519. FIRNetworkRequestCompletionHandler handler) {
  520. NSDictionary *dictionary = @{
  521. kFDLResolvedLinkDeepLinkURLKey : kEncodedComplicatedURLString,
  522. kFDLResolvedLinkMinAppVersionKey : expectedMinVersion,
  523. };
  524. NSData *data = FIRDataWithDictionary(dictionary, nil);
  525. handler(data, nil);
  526. };
  527. SEL executeRequestSelector = @selector(executeOnePlatformRequest:forURL:completionHandler:);
  528. [GULSwizzler swizzleClass:[FIRDynamicLinkNetworking class]
  529. selector:executeRequestSelector
  530. isClassSelector:NO
  531. withBlock:executeRequestBlock];
  532. XCTestExpectation *expectation = [self expectationWithDescription:@"handler called"];
  533. [self.service
  534. handleUniversalLink:url
  535. completion:^(FIRDynamicLink *_Nullable dynamicLink, NSError *_Nullable error) {
  536. XCTAssertEqualObjects(expectedMinVersion, dynamicLink.minimumAppVersion,
  537. @"min app version did not match imv parameter.");
  538. [expectation fulfill];
  539. }];
  540. [self waitForExpectationsWithTimeout:kAsyncTestTimout handler:nil];
  541. }
  542. - (void)testUniversalLinkWithSubdomain_NoDeepLink {
  543. NSString *webPageURLString =
  544. [NSString stringWithFormat:kStructuredUniversalLinkFmtSubdomain, @"/abc123"];
  545. NSURL *url = [NSURL URLWithString:webPageURLString];
  546. [self.service setUpWithLaunchOptions:nil
  547. apiKey:kAPIKey
  548. clientID:kClientID
  549. urlScheme:kURLScheme
  550. userDefaults:self.userDefaults];
  551. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  552. XCTAssertNil(dynamicLink, @"deepLink should be nil since there is no parameter.");
  553. }
  554. - (void)testDynamicLinkFromCustomSchemeURLReturnsDLWithNilMinimumVersion {
  555. [self.service setUpWithLaunchOptions:nil
  556. apiKey:kAPIKey
  557. clientID:kClientID
  558. urlScheme:kURLScheme
  559. userDefaults:self.userDefaults];
  560. NSURL *url = FIRDLDeepLinkURLWithInviteID(nil, kEncodedComplicatedURLString, nil, nil, nil, NO,
  561. nil, nil, kURLScheme, nil);
  562. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  563. XCTAssertNil(dynamicLink.minimumAppVersion, @"Min app version was not nil when not set.");
  564. }
  565. - (void)testDynamicLinkFromCustomSchemeURLReturnsDLMinimumVersion {
  566. NSString *expectedMinVersion = @"03-9g03hfd";
  567. [self.service setUpWithLaunchOptions:nil
  568. apiKey:kAPIKey
  569. clientID:kClientID
  570. urlScheme:kURLScheme
  571. userDefaults:self.userDefaults];
  572. NSURL *url = FIRDLDeepLinkURLWithInviteID(nil, kEncodedComplicatedURLString, nil, nil, nil, NO,
  573. nil, expectedMinVersion, kURLScheme, nil);
  574. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromCustomSchemeURL:url];
  575. NSString *minVersion = dynamicLink.minimumAppVersion;
  576. XCTAssertEqualObjects(expectedMinVersion, minVersion,
  577. @"Min version didn't match the min app version parameter");
  578. }
  579. - (void)testDynamicLinkFromUniversalLinkURLReturnsDLWithNilMinimumVersion {
  580. [self.service setUpWithLaunchOptions:nil
  581. apiKey:kAPIKey
  582. clientID:kClientID
  583. urlScheme:kURLScheme
  584. userDefaults:self.userDefaults];
  585. NSURL *url = FIRDLDeepLinkURLWithInviteID(nil, kEncodedComplicatedURLString, nil, nil, nil, NO,
  586. nil, nil, kURLScheme, nil);
  587. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromUniversalLinkURL:url];
  588. NSString *minVersion = dynamicLink.minimumAppVersion;
  589. XCTAssertNil(minVersion, @"Min app version was not nil when not set.");
  590. }
  591. - (void)testDynamicLinkFromUniversalLinkURLReturnsDLMinimumVersion {
  592. NSString *expectedMinVersion = @"03-9g03hfd";
  593. NSString *urlSuffix =
  594. [NSString stringWithFormat:@"%@&imv=%@", kEncodedComplicatedURLString, expectedMinVersion];
  595. NSString *urlString =
  596. [NSString stringWithFormat:kStructuredUniversalLinkFmtSubdomainDeepLink, urlSuffix];
  597. NSURL *url = [NSURL URLWithString:urlString];
  598. [self.service setUpWithLaunchOptions:nil
  599. apiKey:kAPIKey
  600. clientID:kClientID
  601. urlScheme:kURLScheme
  602. userDefaults:self.userDefaults];
  603. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromUniversalLinkURL:url];
  604. NSString *minVersion = dynamicLink.minimumAppVersion;
  605. XCTAssertEqualObjects(expectedMinVersion, minVersion, @"Min version didn't match imv= parameter");
  606. }
  607. - (void)testUniversalLinkWithSubdomain_DeepLink {
  608. NSString *deepLinkString = @"https://www.google.com/maps/place/Minneapolis";
  609. NSString *webPageURLString =
  610. [NSString stringWithFormat:kStructuredUniversalLinkFmtSubdomainDeepLink, deepLinkString];
  611. NSURL *url = [NSURL URLWithString:webPageURLString];
  612. [self.service setUpWithLaunchOptions:nil
  613. apiKey:kAPIKey
  614. clientID:kClientID
  615. urlScheme:kURLScheme
  616. userDefaults:self.userDefaults];
  617. SwizzleDynamicLinkNetworkingWithMock();
  618. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromUniversalLinkURL:url];
  619. XCTAssertEqual(dynamicLink.matchConfidence, FIRDynamicLinkMatchConfidenceStrong);
  620. XCTAssertEqualObjects(dynamicLink.url.absoluteString, deepLinkString);
  621. UnswizzleDynamicLinkNetworking();
  622. }
  623. - (void)testUniversalLinkWithSubdomain_DeepLinkWithParameters {
  624. NSString *deepLinkString = @"https://www.google.com?key1%3Dvalue1%26key2%3Dvalue2";
  625. NSString *parsedDeepLinkString = @"https://www.google.com?key1=value1&key2=value2";
  626. NSString *webPageURLString =
  627. [NSString stringWithFormat:kStructuredUniversalLinkFmtSubdomainDeepLink, deepLinkString];
  628. NSURL *url = [NSURL URLWithString:webPageURLString];
  629. [self.service setUpWithLaunchOptions:nil
  630. apiKey:kAPIKey
  631. clientID:kClientID
  632. urlScheme:kURLScheme
  633. userDefaults:self.userDefaults];
  634. FIRDynamicLink *dynamicLink = [self.service dynamicLinkFromUniversalLinkURL:url];
  635. XCTAssertEqual(dynamicLink.matchConfidence, FIRDynamicLinkMatchConfidenceStrong);
  636. XCTAssertEqualObjects(dynamicLink.url.absoluteString, parsedDeepLinkString);
  637. }
  638. - (void)testMatchesShortLinkFormat {
  639. NSArray<NSString *> *urlStrings =
  640. @[ @"https://test.app.goo.gl/xyz", @"https://test.app.goo.gl/xyz?link=" ];
  641. for (NSString *urlString in urlStrings) {
  642. NSURL *url = [NSURL URLWithString:urlString];
  643. BOOL matchesShortLinkFormat = [self.service matchesShortLinkFormat:url];
  644. XCTAssertTrue(matchesShortLinkFormat,
  645. @"Non-DDL domain URL matched short link format with URL: %@", url);
  646. }
  647. }
  648. // Custom domain entries in plist file:
  649. // https://google.com
  650. // https://google.com/one
  651. // https://a.firebase.com/mypath
  652. - (void)testFailMatchesShortLinkFormatForCustomDomains {
  653. NSArray<NSString *> *urlStrings = @[
  654. @"https://google.com",
  655. @"https://google.com?link=",
  656. @"https://a.firebase.com",
  657. @"https://a.firebase.com/mypath?link=",
  658. ];
  659. for (NSString *urlString in urlStrings) {
  660. NSURL *url = [NSURL URLWithString:urlString];
  661. BOOL matchesShortLinkFormat = [self.service matchesShortLinkFormat:url];
  662. XCTAssertFalse(matchesShortLinkFormat,
  663. @"Non-DDL domain URL matched short link format with URL: %@", url);
  664. }
  665. }
  666. // Custom domain entries in plist file:
  667. // https://google.com
  668. // https://google.com/one
  669. // https://a.firebase.com/mypath
  670. - (void)testPassMatchesShortLinkFormatForCustomDomains {
  671. NSArray<NSString *> *urlStrings = @[
  672. @"https://google.com/xyz", @"https://google.com/xyz/?link=", @"https://google.com/xyz?link=",
  673. @"https://google.com/one/xyz", @"https://google.com/one/xyz?link=",
  674. @"https://google.com/one?utm_campaignlink=", @"https://google.com/mylink",
  675. @"https://google.com/one/mylink", @"https://a.firebase.com/mypath/mylink"
  676. ];
  677. for (NSString *urlString in urlStrings) {
  678. NSURL *url = [NSURL URLWithString:urlString];
  679. BOOL matchesShortLinkFormat = [self.service matchesShortLinkFormat:url];
  680. XCTAssertTrue(matchesShortLinkFormat,
  681. @"Non-DDL domain URL matched short link format with URL: %@", url);
  682. }
  683. }
  684. - (void)testPassMatchesShortLinkFormat {
  685. NSArray<NSString *> *urlStrings = @[
  686. @"https://test.app.goo.gl/xyz",
  687. @"https://test.app.goo.gl/xyz?link=",
  688. ];
  689. for (NSString *urlString in urlStrings) {
  690. NSURL *url = [NSURL URLWithString:urlString];
  691. BOOL matchesShortLinkFormat = [self.service matchesShortLinkFormat:url];
  692. XCTAssertTrue(matchesShortLinkFormat,
  693. @"Non-DDL domain URL matched short link format with URL: %@", url);
  694. }
  695. }
  696. - (void)testFailMatchesShortLinkFormat {
  697. NSArray<NSString *> *urlStrings = @[
  698. @"https://test.app.goo.gl", @"https://test.app.goo.gl?link=", @"https://test.app.goo.gl/",
  699. @"https://sample.page.link?link=https://google.com/test&ibi=com.google.sample&ius=79306483",
  700. @"https://sample.page.link/?link=https://google.com/test&ibi=com.google.sample&ius=79306483"
  701. ];
  702. for (NSString *urlString in urlStrings) {
  703. NSURL *url = [NSURL URLWithString:urlString];
  704. BOOL matchesShortLinkFormat = [self.service matchesShortLinkFormat:url];
  705. XCTAssertFalse(matchesShortLinkFormat,
  706. @"Non-DDL domain URL matched short link format with URL: %@", url);
  707. }
  708. }
  709. - (void)testMatchesUnversalLinkWithShortDurableLink {
  710. NSString *urlString = @"https://sample.page.link/79g49s";
  711. NSURL *url = [NSURL URLWithString:urlString];
  712. BOOL matchesShort = [self.service matchesShortLinkFormat:url];
  713. XCTAssertTrue(matchesShort, @"Short Durable Link didn't match short link");
  714. }
  715. - (void)testMatchesUnversalLinkWithAppInvite {
  716. NSString *urlString = @"https://sample.page.link/i/79g49s";
  717. NSURL *url = [NSURL URLWithString:urlString];
  718. BOOL matchesShort = [self.service matchesShortLinkFormat:url];
  719. XCTAssertTrue(matchesShort, @"AppInvite didn't match short link");
  720. }
  721. - (void)testDoesNotMatchesShortLinkFormatWithNonDDLDomains {
  722. NSArray<NSString *> *urlStrings = @[
  723. @"https://www.google.com/maps/place/@1,1/My+Home/", @"https://mydomain.com/t439gfde",
  724. @"https://goo.gl/309dht4", @"https://59eh.goo.gl/309dht4", @"https://app.59eh.goo.gl/309dht4",
  725. @"https://goo.gl/i/309dht4", @"https://page.link/i/309dht4", @"https://fjo3eh.goo.gl/i/309dht4",
  726. @"https://app.fjo3eh.goo.gl/i/309dht4", @"https://1234.page.link/link/dismiss"
  727. ];
  728. for (NSString *urlString in urlStrings) {
  729. NSURL *url = [NSURL URLWithString:urlString];
  730. BOOL matchesShortLinkFormat = [self.service matchesShortLinkFormat:url];
  731. XCTAssertFalse(matchesShortLinkFormat,
  732. @"Non-DDL domain URL matched short link format with URL: %@", url);
  733. }
  734. }
  735. - (void)testHandleUniversalLinkWithShortLink {
  736. NSString *shortLinkString = @"https://sample.page.link/549igo";
  737. NSString *bundleID = [NSBundle mainBundle].bundleIdentifier;
  738. NSString *customSchemeURLString = [NSString
  739. stringWithFormat:kStructuredLinkFmtDeepLink, bundleID, kEncodedComplicatedURLString];
  740. XCTestExpectation *handleLinkCompletionExpectation =
  741. [self expectationWithDescription:@"handleLink"];
  742. XCTestExpectation *linkResolverCompletionExpectation =
  743. [self expectationWithDescription:@"linkResolver"];
  744. FakeShortLinkResolver *resolver =
  745. [FakeShortLinkResolver resolverWithBlock:^NSURL *(NSURL *shortLink) {
  746. [linkResolverCompletionExpectation fulfill];
  747. return [NSURL URLWithString:customSchemeURLString];
  748. }];
  749. SwizzleDynamicLinkNetworking(resolver);
  750. [self.service
  751. handleUniversalLink:[NSURL URLWithString:shortLinkString]
  752. completion:^(FIRDynamicLink *_Nonnull dynamicLink, NSError *_Nullable error) {
  753. NSString *returnedURLString = dynamicLink.url.absoluteString;
  754. XCTAssertEqualObjects(kDecodedComplicatedURLString, returnedURLString,
  755. @"Handle universal link returned unexpected link");
  756. [handleLinkCompletionExpectation fulfill];
  757. }];
  758. [self waitForExpectationsWithTimeout:kAsyncTestTimout handler:nil];
  759. UnswizzleDynamicLinkNetworking();
  760. }
  761. - (void)testHandleUniversalLinkWithLongLink {
  762. NSString *longLinkString = [NSString
  763. stringWithFormat:@"https://sample.page.link?link=%@&ibi=com.google.sample&ius=79306483",
  764. kEncodedComplicatedURLString];
  765. XCTestExpectation *handleLinkCompletionExpectation =
  766. [self expectationWithDescription:@"handleLink"];
  767. __block NSUInteger resolverInvocationsCount = 0;
  768. // should not be used.
  769. FakeShortLinkResolver *resolver =
  770. [FakeShortLinkResolver resolverWithBlock:^NSURL *(NSURL *shortLink) {
  771. resolverInvocationsCount++;
  772. return [NSURL URLWithString:kDecodedComplicatedURLString];
  773. }];
  774. id handleUniversalLinkBlock = ^(FIRDynamicLink *_Nonnull dynamicLink, NSError *_Nullable error) {
  775. [handleLinkCompletionExpectation fulfill];
  776. NSURL *expectedResolvedLink = [NSURL URLWithString:kDecodedComplicatedURLString];
  777. XCTAssertEqualObjects(expectedResolvedLink, dynamicLink.url,
  778. @"Resolve short link returned unexpected link");
  779. };
  780. SwizzleDynamicLinkNetworking(resolver);
  781. [self.service handleUniversalLink:[NSURL URLWithString:longLinkString]
  782. completion:handleUniversalLinkBlock];
  783. [self waitForExpectationsWithTimeout:kAsyncTestTimout handler:nil];
  784. // It is expected to call resolveLink once for logging.
  785. XCTAssertEqual(resolverInvocationsCount, 1,
  786. @"handleUniversalLink called resolveLink more than once");
  787. UnswizzleDynamicLinkNetworking();
  788. }
  789. - (void)testHandleUniversalLinkCallsHandleUniversalLinkResolver {
  790. XCTestExpectation *handleLinkCompletionExpectation =
  791. [self expectationWithDescription:@"handleLink"];
  792. void (^replacementBlock)(void) = ^{
  793. [handleLinkCompletionExpectation fulfill];
  794. };
  795. SEL selectorToSwizzle = @selector(handleUniversalLink:completion:);
  796. [GULSwizzler swizzleClass:[FIRDynamicLinks class]
  797. selector:selectorToSwizzle
  798. isClassSelector:NO
  799. withBlock:replacementBlock];
  800. NSURL *url = [NSURL URLWithString:@"https://google.com"];
  801. [self.service
  802. handleUniversalLink:url
  803. completion:^(FIRDynamicLink *_Nullable dynamicLink, NSError *_Nullable error){
  804. }];
  805. [self waitForExpectationsWithTimeout:kAsyncTestTimout handler:nil];
  806. [GULSwizzler unswizzleClass:[FIRDynamicLinks class]
  807. selector:selectorToSwizzle
  808. isClassSelector:NO];
  809. }
  810. - (void)testHandleUniversalLinkCompletionReturnsNoForNonDDL {
  811. NSArray<NSString *> *urlStrings = @[
  812. @"https://www.google.com/maps/place/@1,1/My+Home/", @"https://mydomain.com/t439gfde",
  813. @"https://goo.gl/309dht4", @"https://59eh.goo.gl/309dht4", @"https://app.59eh.goo.gl/309dht4",
  814. @"https://goo.gl/i/309dht4", @"https://page.link/i/309dht4", @"https://fjo3eh.goo.gl/i/309dht4",
  815. @"https://app.fjo3eh.goo.gl/i/309dht4", @"https://1234.page.link/link/dismiss"
  816. ];
  817. [urlStrings enumerateObjectsUsingBlock:^(NSString *_Nonnull urlString, NSUInteger idx,
  818. BOOL *_Nonnull stop) {
  819. NSURL *url = [NSURL URLWithString:urlString];
  820. BOOL handled = [self.service
  821. handleUniversalLink:url
  822. completion:^(FIRDynamicLink *_Nullable dynamicLink, NSError *_Nullable error) {
  823. XCTAssertNil(dynamicLink, @"Non DDL returned FIRDynamicLink");
  824. }];
  825. XCTAssertFalse(handled, @"Non DDL Universal Link was handled");
  826. }];
  827. }
  828. - (void)test_ensureInternalMethodsNotRenamed {
  829. // sanity check to ensure these methods has not been renamed
  830. // we relaying on these to be the same for tests to work properly
  831. // we are not exposing these methods in internal headers as we do not have "internal headers"
  832. // 1P apps can import all of our headers
  833. XCTAssertTrue([self.service
  834. respondsToSelector:@selector
  835. (handlePendingDynamicLinkRetrievalFailureWithErrorCode:errorDescription:underlyingError:)]);
  836. XCTAssertTrue([self.service respondsToSelector:@selector(setRetrievingPendingDynamicLink:)]);
  837. XCTAssertTrue([self.service respondsToSelector:@selector(dynamicLinkNetworking)]);
  838. }
  839. - (void)testCheckForPendingDynamicLinkReturnsImmediatelyIfAlreadyRead {
  840. id mockService = OCMPartialMock(self.service);
  841. [[mockService expect] handlePendingDynamicLinkRetrievalFailureWithErrorCode:-1
  842. errorDescription:[OCMArg any]
  843. underlyingError:[OCMArg any]];
  844. [[mockService reject] setRetrievingPendingDynamicLink:YES];
  845. [mockService setUpWithLaunchOptions:nil
  846. apiKey:kAPIKey
  847. clientID:kClientID
  848. urlScheme:nil
  849. userDefaults:[NSUserDefaults standardUserDefaults]];
  850. [[NSUserDefaults standardUserDefaults] setBool:YES forKey:kFIRDLReadDeepLinkAfterInstallKey];
  851. [mockService checkForPendingDynamicLink];
  852. [mockService verify];
  853. [mockService stopMocking];
  854. }
  855. - (void)testRetrievalProcessResultURLContainsAllParametersPassedToDynamicLinkInitializer {
  856. NSDictionary<NSString *, NSString *> *linkParameters = @{
  857. @"deep_link_id" : @"https://mmaksym.com/test-app1",
  858. @"match_message" : @"Link is uniquely matched for this device.",
  859. @"match_type" : @"unique",
  860. @"utm_campaign" : @"Maksym M Test",
  861. @"utm_medium" : @"test_medium",
  862. @"utm_source" : @"test_source",
  863. @"a_parameter" : @"a_value"
  864. };
  865. FIRDynamicLink *dynamicLink =
  866. [[FIRDynamicLink alloc] initWithParametersDictionary:linkParameters];
  867. FIRDLRetrievalProcessResult *result =
  868. [[FIRDLRetrievalProcessResult alloc] initWithDynamicLink:dynamicLink
  869. error:nil
  870. message:nil
  871. matchSource:nil];
  872. NSURL *customSchemeURL = [result URLWithCustomURLScheme:@"scheme"];
  873. XCTAssertNotNil(customSchemeURL);
  874. // Validate URL parameters
  875. NSURLComponents *urlComponents = [NSURLComponents componentsWithURL:customSchemeURL
  876. resolvingAgainstBaseURL:NO];
  877. XCTAssertNotNil(urlComponents);
  878. XCTAssertEqualObjects(urlComponents.scheme, @"scheme");
  879. NSMutableDictionary<NSString *, NSString *> *notEncodedParameters = [linkParameters mutableCopy];
  880. for (NSURLQueryItem *queryItem in urlComponents.queryItems) {
  881. NSString *expectedValue = notEncodedParameters[queryItem.name];
  882. XCTAssertNotNil(expectedValue, @"Extra parameter encoded: %@ = %@", queryItem.name,
  883. queryItem.value);
  884. XCTAssertEqualObjects(queryItem.value, expectedValue);
  885. [notEncodedParameters removeObjectForKey:queryItem.name];
  886. }
  887. XCTAssertEqual(notEncodedParameters.count, 0, @"The parameters must have been encoded: %@",
  888. notEncodedParameters);
  889. }
  890. - (void)test_multipleRequestsToRetrievePendingDeepLinkShouldNotCrash {
  891. id mockService = OCMPartialMock(self.service);
  892. [[mockService expect] handlePendingDynamicLinkRetrievalFailureWithErrorCode:-1
  893. errorDescription:[OCMArg any]
  894. underlyingError:[OCMArg any]];
  895. // swizzle method to prevent actual retrieval, this will ensure that first pending link
  896. // retrieval will stuck and second will fail with error
  897. id (^replacementBlock)(void) = (id) ^ { return nil; };
  898. SEL selectorToSwizzle = @selector(automaticRetrievalProcess);
  899. [GULSwizzler swizzleClass:[FIRDLRetrievalProcessFactory class]
  900. selector:selectorToSwizzle
  901. isClassSelector:NO
  902. withBlock:replacementBlock];
  903. [mockService setUpWithLaunchOptions:nil
  904. apiKey:kAPIKey
  905. clientID:kClientID
  906. urlScheme:kURLScheme
  907. userDefaults:self.userDefaults];
  908. [mockService checkForPendingDynamicLink];
  909. // we should not crash here
  910. [mockService checkForPendingDynamicLink];
  911. [mockService verify];
  912. [mockService stopMocking];
  913. [GULSwizzler unswizzleClass:[FIRDLRetrievalProcessFactory class]
  914. selector:selectorToSwizzle
  915. isClassSelector:NO];
  916. }
  917. #pragma mark - Self-diagnose tests
  918. - (void)testSelfDiagnoseWithNilCompletion {
  919. [FIRDynamicLinks performDiagnosticsWithCompletion:nil];
  920. }
  921. - (void)testSelfDiagnoseCompletionCalled {
  922. XCTestExpectation *expectation =
  923. [self expectationWithDescription:@"Self diagnose completion block must be called"];
  924. [FIRDynamicLinks
  925. performDiagnosticsWithCompletion:^(NSString *_Nonnull diagnosticOutput, BOOL hasErrors) {
  926. XCTAssert(diagnosticOutput.length > 0, @"Diagnostic expected to provide output");
  927. [expectation fulfill];
  928. }];
  929. [self waitForExpectationsWithTimeout:2.0 handler:nil];
  930. }
  931. #pragma mark - Custom domain tests
  932. - (void)testValidCustomDomainNames {
  933. // Entries in plist file:
  934. // https://google.com
  935. // https://google.com/one
  936. // https://a.firebase.com/mypath
  937. NSArray<NSString *> *urlStrings = @[
  938. @"https://google.com/mylink", // Short FDL starting with 'https://google.com'
  939. @"https://google.com/one", // Short FDL starting with 'https://google.com'
  940. @"https://google.com/one/mylink", // Short FDL starting with 'https://google.com/one'
  941. @"https://a.firebase.com/mypath/mylink", // Short FDL starting https://a.firebase.com/mypath
  942. ];
  943. NSArray<NSString *> *longFDLURLStrings = @[
  944. @"https://a.firebase.com/mypath/?link=abcd&test=1", // Long FDL starting with
  945. // https://a.firebase.com/mypath
  946. @"https://google.com/?link=abcd", // Long FDL starting with 'https://google.com'
  947. ];
  948. for (NSString *urlString in urlStrings) {
  949. NSURL *url = [NSURL URLWithString:urlString];
  950. BOOL matchesShortLinkFormat = [self.service matchesShortLinkFormat:url];
  951. XCTAssertTrue(matchesShortLinkFormat, @"URL did not validate as short link: %@", url);
  952. }
  953. for (NSString *urlString in longFDLURLStrings) {
  954. NSURL *url = [NSURL URLWithString:urlString];
  955. BOOL matchesLongLinkFormat = [self.service canParseUniversalLinkURL:url];
  956. XCTAssertTrue(matchesLongLinkFormat, @"URL did not validate as long link: %@", url);
  957. }
  958. }
  959. - (void)testInvalidCustomDomainNames {
  960. // Entries in plist file:
  961. // https://google.com
  962. // https://google.com/one
  963. // https://a.firebase.com/mypath
  964. NSArray<NSString *> *urlStrings = @[
  965. @"google.com", // Valid domain. No scheme.
  966. @"https://google.com", // Valid domain. No path after domainURIPrefix.
  967. @"https://google.com/", // Valid domain. No path after domainURIPrefix.
  968. @"https://google.com/one/", // Valid domain. No path after domainURIPrefix.
  969. @"https://google.com/one/two/mylink", // domainURIPrefix not exact match.
  970. @"https://google.co.in/mylink", // No matching domainURIPrefix.
  971. @"https://firebase.com/mypath", // No matching domainURIPrefix: Invalid (sub)domain.
  972. @"https://b.firebase.com/mypath", // No matching domainURIPrefix: Invalid subdomain.
  973. @"https://a.firebase.com/mypathabc", // No matching domainURIPrefix: Invalid subdomain.
  974. @"mydomain.com", // https scheme not specified for domainURIPrefix.
  975. @"http://mydomain", // Domain not in plist. No path after domainURIPrefix.
  976. ];
  977. for (NSString *urlString in urlStrings) {
  978. NSURL *url = [NSURL URLWithString:urlString];
  979. BOOL matchesShortLinkFormat = [self.service matchesShortLinkFormat:url];
  980. XCTAssertFalse(matchesShortLinkFormat,
  981. @"Non-DDL domain URL matched short link format with URL: %@", url);
  982. }
  983. }
  984. #pragma mark - Private Helpers
  985. - (void)removeAllFIRApps {
  986. NSDictionary *apps = [FIRApp allApps];
  987. for (FIRApp *app in apps.allValues) {
  988. [app deleteApp:^(BOOL succes) {
  989. if (!succes) {
  990. NSLog(@"Error deleting FIRApp before tests - config tests may fail.");
  991. }
  992. }];
  993. }
  994. }
  995. #pragma clang pop
  996. @end