use startSession

stripe · Jan 29, 2025 · 7b949db · 7b949db
1 parent 2ee9a21
commit 7b949db
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 14 deletions.
diff --git a/paymentsheet/src/main/java/com/stripe/android/link/account/DefaultLinkAuth.kt b/paymentsheet/src/main/java/com/stripe/android/link/account/DefaultLinkAuth.kt
@@ -44,16 +44,21 @@ internal class DefaultLinkAuth @Inject constructor(
         return signupResult.toLinkAuthResult()
     }
 
-    override suspend fun lookUp(email: String, emailSource: EmailSource): LinkAuthResult {
+    override suspend fun lookUp(
+        email: String,
+        emailSource: EmailSource,
+        startSession: Boolean
+    ): LinkAuthResult {
         val lookupResult = if (linkGate.useAttestationEndpoints) {
             mobileLookUp(
                 email = email,
-                emailSource = emailSource
+                emailSource = emailSource,
+                startSession = startSession
             )
         } else {
             linkAccountManager.lookupConsumer(
                 email = email,
-                startSession = true
+                startSession = startSession
             )
         }
         return lookupResult.toLinkAuthResult()
@@ -82,7 +87,8 @@ internal class DefaultLinkAuth @Inject constructor(
 
     private suspend fun mobileLookUp(
         email: String,
-        emailSource: EmailSource
+        emailSource: EmailSource,
+        startSession: Boolean
     ): Result<LinkAccount?> {
         return runCatching {
             val verificationToken = integrityRequestManager.requestToken().getOrThrow()
@@ -91,7 +97,7 @@ internal class DefaultLinkAuth @Inject constructor(
                 emailSource = emailSource,
                 verificationToken = verificationToken,
                 appId = applicationId,
-                startSession = true
+                startSession = startSession
             ).getOrThrow()
         }
     }

diff --git a/paymentsheet/src/main/java/com/stripe/android/link/account/LinkAuth.kt b/paymentsheet/src/main/java/com/stripe/android/link/account/LinkAuth.kt
@@ -15,5 +15,6 @@ internal interface LinkAuth {
     suspend fun lookUp(
         email: String,
         emailSource: EmailSource,
+        startSession: Boolean
     ): LinkAuthResult
 }
diff --git a/paymentsheet/src/test/java/com/stripe/android/link/account/DefaultLinkAuthTest.kt b/paymentsheet/src/test/java/com/stripe/android/link/account/DefaultLinkAuthTest.kt
@@ -242,7 +242,8 @@ internal class DefaultLinkAuthTest {
 
         val result = linkAuth.lookUp(
             email = TestFactory.CUSTOMER_EMAIL,
-            emailSource = TestFactory.EMAIL_SOURCE
+            emailSource = TestFactory.EMAIL_SOURCE,
+            startSession = true
         )
 
         val accountManagerCall = linkAccountManager.awaitMobileLookupCall()
@@ -279,7 +280,8 @@ internal class DefaultLinkAuthTest {
 
         val result = linkAuth.lookUp(
             email = TestFactory.CUSTOMER_EMAIL,
-            emailSource = TestFactory.EMAIL_SOURCE
+            emailSource = TestFactory.EMAIL_SOURCE,
+            startSession = false
         )
 
         integrityRequestManager.awaitRequestTokenCall()
@@ -308,7 +310,8 @@ internal class DefaultLinkAuthTest {
 
         val result = linkAuth.lookUp(
             email = TestFactory.CUSTOMER_EMAIL,
-            emailSource = TestFactory.EMAIL_SOURCE
+            emailSource = TestFactory.EMAIL_SOURCE,
+            startSession = false
         )
 
         integrityRequestManager.awaitRequestTokenCall()
@@ -334,7 +337,8 @@ internal class DefaultLinkAuthTest {
 
         val result = linkAuth.lookUp(
             email = TestFactory.CUSTOMER_EMAIL,
-            emailSource = TestFactory.EMAIL_SOURCE
+            emailSource = TestFactory.EMAIL_SOURCE,
+            startSession = false
         )
 
         integrityRequestManager.awaitRequestTokenCall()
@@ -358,13 +362,14 @@ internal class DefaultLinkAuthTest {
 
         val result = linkAuth.lookUp(
             email = TestFactory.CUSTOMER_EMAIL,
-            emailSource = TestFactory.EMAIL_SOURCE
+            emailSource = TestFactory.EMAIL_SOURCE,
+            startSession = false
         )
 
         val accountManagerCall = linkAccountManager.awaitLookupCall()
 
         assertThat(accountManagerCall.email).isEqualTo(TestFactory.CUSTOMER_EMAIL)
-        assertThat(accountManagerCall.startSession).isTrue()
+        assertThat(accountManagerCall.startSession).isFalse()
 
         assertThat(result).isEqualTo(LinkAuthResult.Success(TestFactory.LINK_ACCOUNT))
 
@@ -388,7 +393,8 @@ internal class DefaultLinkAuthTest {
 
         val result = linkAuth.lookUp(
             email = TestFactory.CUSTOMER_EMAIL,
-            emailSource = TestFactory.EMAIL_SOURCE
+            emailSource = TestFactory.EMAIL_SOURCE,
+            startSession = true
         )
 
         val accountManagerCall = linkAccountManager.awaitLookupCall()
@@ -416,12 +422,14 @@ internal class DefaultLinkAuthTest {
 
         val result = linkAuth.lookUp(
             email = TestFactory.CUSTOMER_EMAIL,
-            emailSource = TestFactory.EMAIL_SOURCE
+            emailSource = TestFactory.EMAIL_SOURCE,
+            startSession = false
         )
 
-        linkAccountManager.awaitMobileLookupCall()
+        val accountManagerCall = linkAccountManager.awaitMobileLookupCall()
         integrityRequestManager.awaitRequestTokenCall()
 
+        assertThat(accountManagerCall.startSession).isFalse()
         assertThat(result).isEqualTo(LinkAuthResult.NoLinkAccountFound)
 
         linkAccountManager.ensureAllEventsConsumed()