Fix signup internal test

ProtonVPN · Jan 15, 2024 · ac9981e · ac9981e
1 parent 9960d0e
commit ac9981e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/app/src/androidTest/java/com/protonvpn/actions/OnboardingRobot.kt b/app/src/androidTest/java/com/protonvpn/actions/OnboardingRobot.kt
@@ -24,6 +24,8 @@ import com.protonvpn.base.BaseRobot
 import com.protonvpn.base.BaseVerify
 import com.protonvpn.data.Timeouts
 import me.proton.test.fusion.Fusion.node
+import okhttp3.internal.wait
+import java.time.Duration
 
 class OnboardingRobot : BaseRobot() {
 
@@ -39,7 +41,7 @@ class OnboardingRobot : BaseRobot() {
 
     class Verify : BaseVerify() {
         fun welcomeScreenIsDisplayed() {
-            node.withText(R.string.onboarding_welcome_title).assertIsDisplayed()
+            node.withText(R.string.onboarding_welcome_title).await(Timeouts.LONG ) { assertIsDisplayed() }
         }
 
         fun onboardingPaymentIdDisplayed() {

diff --git a/app/src/androidTest/java/com/protonvpn/tests/signup/SignUpTests.kt b/app/src/androidTest/java/com/protonvpn/tests/signup/SignUpTests.kt
@@ -40,12 +40,15 @@ import me.proton.core.auth.test.robot.signup.SignUpRobot
 import me.proton.core.auth.test.rule.AcceptExternalRule
 import me.proton.core.network.domain.client.ExtraHeaderProvider
 import me.proton.core.util.kotlin.random
+import me.proton.test.fusion.Fusion
 import me.proton.test.fusion.FusionConfig
 import org.junit.Before
 import org.junit.Rule
 import org.junit.Test
 import org.junit.runner.RunWith
 import javax.inject.Inject
+import kotlin.time.Duration.Companion.milliseconds
+import me.proton.core.auth.presentation.R
 
 @LargeTest
 @RunWith(AndroidJUnit4::class)
@@ -82,6 +85,7 @@ class SignUpTests : MinimalSignUpExternalTests {
     override val isCongratsDisplayed = false
 
     @Test
+    // TODO Migrate this test to core
     override fun signupSwitchToInternalAccountHappyPath() {
         val testUsername = "test-${String.random()}"