test(fdl-ktx): add tests for the destructuring declarations

Author: thatfiredev

firebase-dynamic-links/ktx/src/test/kotlin/com/google/firebase/dynamiclinks/ktx/DynamicLinksTests.kt

@@ -19,13 +19,17 @@ import com.google.common.truth.Truth.assertThat
 import com.google.firebase.FirebaseApp
 import com.google.firebase.FirebaseOptions
 import com.google.firebase.dynamiclinks.FirebaseDynamicLinks
+import com.google.firebase.dynamiclinks.PendingDynamicLinkData
+import com.google.firebase.dynamiclinks.ShortDynamicLink
 import com.google.firebase.ktx.Firebase
 import com.google.firebase.ktx.app
 import com.google.firebase.ktx.initialize
 import org.junit.After
 import org.junit.Before
 import org.junit.Test
 import org.junit.runner.RunWith
+import org.mockito.Mockito.`when`
+import org.mockito.Mockito.mock
 import org.robolectric.RobolectricTestRunner
 import org.robolectric.RuntimeEnvironment
 
@@ -223,4 +227,45 @@ class DynamicLinksTests : BaseTestCase() {
         val efr = Integer.parseInt(dynamicLink.uri.getQueryParameter("efr")!!) == 1
         assertThat(efr).isEqualTo(forcedRedirect)
     }
+
+    @Test
+    fun `ShortDynamicLink destructure declaration works`() {
+        val fakeWarning = object : ShortDynamicLink.Warning {
+            override fun getMessage() = "Warning"
+            override fun getCode() = "warning"
+        }
+
+        val expectedShortLink = Uri.parse("https://example.com")
+        val expectedPreviewLink = Uri.parse("https://example.com/preview")
+        val expectedWarnings = mutableListOf<ShortDynamicLink.Warning>(fakeWarning)
+
+        val mockShortDynamicLink = mock(ShortDynamicLink::class.java)
+        `when`(mockShortDynamicLink.shortLink).thenReturn(expectedShortLink)
+        `when`(mockShortDynamicLink.previewLink).thenReturn(expectedPreviewLink)
+        `when`(mockShortDynamicLink.warnings).thenReturn(expectedWarnings)
+
+        val (shortLink, previewLink, warnings) = mockShortDynamicLink
+
+        assertThat(shortLink).isEqualTo(expectedShortLink)
+        assertThat(previewLink).isEqualTo(expectedPreviewLink)
+        assertThat(warnings).isEqualTo(expectedWarnings)
+    }
+
+    @Test
+    fun `PendingDynamicLinkData destructure declaration works`() {
+        val expectedLink = Uri.parse("https://example.com")
+        val expectedMinAppVersion = 30
+        val expectedTimestamp = 172947600L
+
+        val mockPendingData = mock(PendingDynamicLinkData::class.java)
+        `when`(mockPendingData.link).thenReturn(expectedLink)
+        `when`(mockPendingData.minimumAppVersion).thenReturn(expectedMinAppVersion)
+        `when`(mockPendingData.clickTimestamp).thenReturn(expectedTimestamp)
+
+        val (link, minAppVersion, timestamp) = mockPendingData
+
+        assertThat(link).isEqualTo(expectedLink)
+        assertThat(minAppVersion).isEqualTo(expectedMinAppVersion)
+        assertThat(timestamp).isEqualTo(expectedTimestamp)
+    }
 }