FIRDynamicLinksTest.m 46 KB

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