Update test

Author: imzyy

packages/firebase_dynamic_links/android/src/main/java/io/flutter/plugins/firebasedynamiclinks/FirebaseDynamicLinksPlugin.java

@@ -117,19 +117,16 @@ private Map<String, Object> getMapFromPendingDynamicLinkData(
     return dynamicLink;
   }
 
-  private void handleGetDynamicLink(final Result result, Uri uri) {
-    FirebaseDynamicLinks.getInstance()
-        .getDynamicLink(uri)
-        .addOnSuccessListener(
+  private void addDynamicLinkListener(Task<PendingDynamicLinkData> task, Result result) {
+    task.addOnSuccessListener(
             new OnSuccessListener<PendingDynamicLinkData>() {
               @Override
               public void onSuccess(PendingDynamicLinkData pendingDynamicLinkData) {
                 if (pendingDynamicLinkData != null) {
-                  Uri link = pendingDynamicLinkData.getLink();
-                  if (link != null) {
-                    result.success(link.toString());
-                    return;
-                  }
+                  Map<String, Object> dynamicLink =
+                      getMapFromPendingDynamicLinkData(pendingDynamicLinkData);
+                  result.success(dynamicLink);
+                  return;
                 }
                 result.success(null);
               }
@@ -143,37 +140,20 @@ public void onFailure(@NonNull Exception e) {
             });
   }
 
+  private void handleGetDynamicLink(final Result result, Uri uri) {
+    addDynamicLinkListener(FirebaseDynamicLinks.getInstance().getDynamicLink(uri), result);
+  }
+
   private void handleGetInitialDynamicLink(final Result result) {
     // If there's no activity, then there's no initial dynamic link.
     if (registrar.activity() == null) {
       result.success(null);
       return;
     }
 
-    FirebaseDynamicLinks.getInstance()
-        .getDynamicLink(registrar.activity().getIntent())
-        .addOnSuccessListener(
-            registrar.activity(),
-            new OnSuccessListener<PendingDynamicLinkData>() {
-              @Override
-              public void onSuccess(PendingDynamicLinkData pendingDynamicLinkData) {
-                if (pendingDynamicLinkData != null) {
-                  Map<String, Object> dynamicLink =
-                      getMapFromPendingDynamicLinkData(pendingDynamicLinkData);
-                  result.success(dynamicLink);
-                  return;
-                }
-                result.success(null);
-              }
-            })
-        .addOnFailureListener(
-            registrar.activity(),
-            new OnFailureListener() {
-              @Override
-              public void onFailure(@NonNull Exception e) {
-                result.error(e.getClass().getSimpleName(), e.getMessage(), null);
-              }
-            });
+    addDynamicLinkListener(
+        FirebaseDynamicLinks.getInstance().getDynamicLink(registrar.activity().getIntent()),
+        result);
   }
 
   private OnCompleteListener<ShortDynamicLink> createShortLinkListener(final Result result) {

packages/firebase_dynamic_links/lib/src/firebase_dynamic_links.dart

@@ -39,7 +39,7 @@ class FirebaseDynamicLinks {
 
   Future<PendingDynamicLinkData> getDynamicLink(Uri url) async {
     final Map<String, dynamic> linkData = await FirebaseDynamicLinks.channel
-        .invokeMethod<String>('FirebaseDynamicLinks#getDynamicLink',
+        .invokeMapMethod<String, dynamic>('FirebaseDynamicLinks#getDynamicLink',
             <String, dynamic>{'url': url.toString()});
     return getPendingDynamicLinkDataFromMap(linkData);
   }

packages/firebase_dynamic_links/test/firebase_dynamic_links_test.dart

@@ -40,7 +40,9 @@ void main() {
               },
             };
           case 'FirebaseDynamicLinks#getDynamicLink':
-            return 'https://google.com';
+            return <dynamic, dynamic>{
+              'link': 'https://google.com',
+            };
           default:
             return null;
         }
@@ -69,10 +71,10 @@ void main() {
 
     test('getDynamicLink', () async {
       final Uri argument = Uri.parse('short-link');
-      final Uri url =
+      final PendingDynamicLinkData data =
           await FirebaseDynamicLinks.instance.getDynamicLink(argument);
 
-      expect(url.host, 'google.com');
+      expect(data.link.host, 'google.com');
 
       expect(log, <Matcher>[
         isMethodCall('FirebaseDynamicLinks#getDynamicLink',