FIRDynamicLinksTest.m 52 KB

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