From 9bc5ad732054ee8a3d1c724abf0b379e1484cc8c Mon Sep 17 00:00:00 2001
From: Mandar <dmandar@users.noreply.github.com>
Date: Thu, 29 Nov 2018 12:25:20 -0800
Subject: [PATCH] Add test for deprecated FDLURLComponents init API.

---
 .../Tests/FDLURLComponentsTests.m             | 34 ++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/Example/DynamicLinks/Tests/FDLURLComponentsTests.m b/Example/DynamicLinks/Tests/FDLURLComponentsTests.m
index d2cd4b8ae9a..80e29827a96 100644
--- a/Example/DynamicLinks/Tests/FDLURLComponentsTests.m
+++ b/Example/DynamicLinks/Tests/FDLURLComponentsTests.m
@@ -511,7 +511,7 @@ - (void)testFDLComponentsCustomDomainWithPath {
   XCTAssertEqualObjects(actualURL, expectedURL);
 }
 
-- (void)testFDLComponentsFailsOnMalformedDomain {
+- (void)testFDLComponentsFailsOnMalformedDomainURIPrefix {
   NSString *linkString = @"https://google.com";
   NSURL *link = [NSURL URLWithString:linkString];
 
@@ -522,6 +522,38 @@ - (void)testFDLComponentsFailsOnMalformedDomain {
   XCTAssertNil(components.url);
 }
 
+- (void)testFDLComponentsNotNilOnDomainWithHTTPScheme {
+  NSString *linkString = @"https://google.com";
+  NSURL *link = [NSURL URLWithString:linkString];
+
+  FIRDynamicLinkComponents *components =
+      [FIRDynamicLinkComponents componentsWithLink:link domain:@"http://xyz.page.link"];
+
+  XCTAssertNotNil(components);
+}
+
+- (void)testFDLComponentsNotNilOnDomainWithHTTPSScheme {
+  NSString *linkString = @"https://google.com";
+  NSURL *link = [NSURL URLWithString:linkString];
+
+  FIRDynamicLinkComponents *components =
+      [FIRDynamicLinkComponents componentsWithLink:link domain:@"https://xyz.page.link"];
+
+  XCTAssertNotNil(components);
+}
+
+- (void)testFDLComponentsFailsOnMalformedDomain {
+  NSString *linkString = @"https://google.com";
+  NSURL *link = [NSURL URLWithString:linkString];
+
+  FIRDynamicLinkComponents *components =
+      [FIRDynamicLinkComponents componentsWithLink:link
+                                            domain:@"this is invalid domain URI Prefix"];
+
+  XCTAssertNotNil(components);
+  XCTAssertNil(components.url);
+}
+
 - (void)testFDLComponentsCreatesFullLinkCorrectly {
   FIRDynamicLinkGoogleAnalyticsParameters *analyticsParams =
       [FIRDynamicLinkGoogleAnalyticsParameters parameters];