test: sso devce backup critical flow test(WPB-20253)

tests/testsCore/src/androidTest/kotlin/com/wire/android/tests/core/criticalFlows/AccountManagement.kt

@@ -99,19 +99,19 @@ class AccountManagement : KoinTest {
             assertEmailWelcomePage()
         }
         pages.loginPage.apply {
-            enterPersonalUserLoggingEmail(teamMember.email ?: "")
+            enterTeamMemberLoggingEmail(teamMember.email ?: "")
             clickLoginButton()
-            enterPersonalUserLoginPassword(teamMember.password ?: "")
+            enterTeamMemberLoggingPassword(teamMember.password ?: "")
             clickLoginButton()
         }
         pages.registrationPage.apply {
-            waitUntilLoginFlowIsComplete()
+            waitUntilLoginFlowIsCompleted()
             clickAllowNotificationButton()
             clickDeclineShareDataAlert()
         }
         pages.conversationListPage.apply {
             assertGroupConversationVisible("MyTeam")
-            clickMainMenuButtonOnConversationPage()
+            clickConversationsMenuEntry()
             clickSettingsButtonOnMenuEntry()
             pages.settingsPage.apply {
                 clickDebugSettingsButton()

tests/testsCore/src/androidTest/kotlin/com/wire/android/tests/core/criticalFlows/FileSharing.kt

@@ -29,7 +29,7 @@ import backendUtils.team.deleteTeam
 import com.wire.android.tests.support.UiAutomatorSetup
 import com.wire.android.tests.core.di.testModule
 import com.wire.android.tests.core.pages.AllPages
-import deleteDownloadedFilesContainingFileWord
+import deleteDownloadedFilesContaining
 import kotlinx.coroutines.runBlocking
 import org.junit.After
 import org.junit.Before
@@ -66,7 +66,7 @@ class FileSharing : KoinTest {
         context = InstrumentationRegistry.getInstrumentation().context
         // device = UiAutomatorSetup.start(UiAutomatorSetup.APP_DEV)
         device = UiAutomatorSetup.start(UiAutomatorSetup.APP_INTERNAL)
-       // device = UiAutomatorSetup.start(UiAutomatorSetup.APP_STAGING)
+        // device = UiAutomatorSetup.start(UiAutomatorSetup.APP_STAGING)
         backendClient = BackendClient.loadBackend("STAGING")
         teamHelper = TeamHelper()
     }
@@ -75,9 +75,9 @@ class FileSharing : KoinTest {
     fun tearDown() {
         //  UiAutomatorSetup.stopApp()
         // To delete team
-         teamOwner2?.deleteTeam(backendClient!!)
+        teamOwner2?.deleteTeam(backendClient!!)
         teamOwner1?.deleteTeam(backendClient!!)
-        deleteDownloadedFilesContainingFileWord()
+        deleteDownloadedFilesContaining("File")
     }
 
     @Suppress("CyclomaticComplexMethod", "LongMethod")
@@ -145,7 +145,7 @@ class FileSharing : KoinTest {
         }
         pages.registrationPage.apply {
 
-            waitUntilLoginFlowIsComplete()
+            waitUntilLoginFlowIsCompleted()
             clickAllowNotificationButton()
             clickDeclineShareDataAlert()
             testServiceHelper.apply {

tests/testsCore/src/androidTest/kotlin/com/wire/android/tests/core/criticalFlows/GroupMessaging.kt

@@ -118,7 +118,7 @@ class GroupMessaging : KoinTest {
             clickLoginButton()
         }
         pages.registrationPage.apply {
-            waitUntilLoginFlowIsComplete()
+            waitUntilLoginFlowIsCompleted()
             clickAllowNotificationButton()
             clickDeclineShareDataAlert()
         }
@@ -139,7 +139,7 @@ class GroupMessaging : KoinTest {
             typeMessageInInputField("Hello Team Members")
             clickSendButton()
             assertSentMessageIsVisibleInCurrentConversation("Hello Team Members")
-            tapBackButtonOnConversationViewPage()
+            tapBackButtonToCloseConversationViewPage()
         }
         testServiceHelper.apply {
             addDevice("user2Name", null, "Device1")

tests/testsCore/src/androidTest/kotlin/com/wire/android/tests/core/criticalFlows/NewMemberMessaging.kt

@@ -122,13 +122,13 @@ class NewMemberMessaging : KoinTest {
             clickProceedButtonOnDeeplinkOverlay()
         }
         pages.loginPage.apply {
-            enterTeamOwnerLoggingEmail(member1?.email ?: "")
+            enterTeamMemberLoggingEmail(member1?.email ?: "")
             clickLoginButton()
-            enterTeamOwnerLoggingPassword(member1?.password ?: "")
+            enterTeamMemberLoggingPassword(member1?.password ?: "")
             clickLoginButton()
         }
         pages.registrationPage.apply {
-            waitUntilLoginFlowIsComplete()
+            waitUntilLoginFlowIsCompleted()
             clickAllowNotificationButton()
             clickDeclineShareDataAlert()
         }
@@ -152,7 +152,7 @@ class NewMemberMessaging : KoinTest {
             typeMessageInInputField("Hello Team Owner")
             clickSendButton()
             assertSentMessageIsVisibleInCurrentConversation("Hello Team Owner")
-            tapBackButtonOnConversationViewPage()
+            tapBackButtonToCloseConversationViewPage()
         }
         pages.connectedUserProfilePage.apply {
             tapCloseButtonOnConnectedUserProfilePage()
@@ -182,7 +182,7 @@ class NewMemberMessaging : KoinTest {
         }
 
         pages.conversationViewPage.apply {
-            tapBackButtonOnConversationViewPage()
+            tapBackButtonToCloseConversationViewPage()
         }
 
         pages.conversationListPage.apply {

tests/testsCore/src/androidTest/kotlin/com/wire/android/tests/core/criticalFlows/PersonalAccountLifeCycle.kt

@@ -135,7 +135,7 @@ class PersonalAccountLifeCycle : KoinTest {
             assertUserNameHelpText()
             setUserName(personalUser?.uniqueUsername.orEmpty())
             clickConfirmButton()
-            waitUntilRegistrationFlowIsComplete()
+            waitUntilRegistrationFlowIsCompleted()
             clickAllowNotificationButton()
             clickDeclineShareDataAlert()
             assertConversationPageVisible()
@@ -200,10 +200,10 @@ class PersonalAccountLifeCycle : KoinTest {
                 assertUnblockUserButtonVisible()
                 tapCloseButtonOnConnectedUserProfilePage()
                 pages.conversationViewPage.apply {
-                    tapBackButtonOnConversationViewPage()
+                    tapBackButtonToCloseConversationViewPage()
                 }
                 pages.conversationListPage.apply {
-                    clickMainMenuButtonOnConversationPage()
+                    clickConversationsMenuEntry()
                     clickSettingsButtonOnMenuEntry()
                 }
                 waitFor(1)

tests/testsCore/src/androidTest/kotlin/com/wire/android/tests/core/criticalFlows/SSODeviceBackup.kt

@@ -0,0 +1,241 @@
+/*
+ * Wire
+ * Copyright (C) 2025 Wire Swiss GmbH
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see http://www.gnu.org/licenses/.
+ */
+package com.wire.android.tests.core.criticalFlows
+
+import SSOServiceHelper.thereIsASSOTeamOwnerForOkta
+import SSOServiceHelper.userAddsOktaUser
+import SSOServiceHelper.userXIsMe
+import androidx.test.ext.junit.runners.AndroidJUnit4
+import org.junit.runner.RunWith
+import android.content.Context
+import androidx.test.platform.app.InstrumentationRegistry
+import androidx.test.uiautomator.UiDevice
+import backendUtils.BackendClient
+import backendUtils.team.TeamHelper
+import backendUtils.team.deleteTeam
+import com.wire.android.tests.core.di.testModule
+import com.wire.android.tests.core.pages.AllPages
+import com.wire.android.tests.support.UiAutomatorSetup
+import deleteDownloadedFilesContaining
+import kotlinx.coroutines.runBlocking
+import okta.OktaApiClient
+import org.junit.After
+import org.junit.Before
+import org.junit.Rule
+import org.junit.Test
+import org.koin.test.KoinTest
+import org.koin.test.KoinTestRule
+import org.koin.test.inject
+import service.TestServiceHelper
+import uiautomatorutils.UiWaitUtils.WaitUtils.waitFor
+import user.usermanager.ClientUserManager
+import user.utils.ClientUser
+import kotlin.getValue
+
+@RunWith(AndroidJUnit4::class)
+class SSODeviceBackup : KoinTest {
+
+    @get:Rule
+    val koinTestRule = KoinTestRule.Companion.create {
+        modules(testModule)
+    }
+    private val pages: AllPages by inject()
+    private lateinit var device: UiDevice
+
+    lateinit var context: Context
+    var teamOwner: ClientUser? = null
+    var member1: ClientUser? = null
+    private lateinit var oktaApiClient: OktaApiClient
+
+    var backendClient: BackendClient? = null
+    val teamServiceHelper by lazy {
+        TestServiceHelper()
+    }
+    val teamHelper by lazy {
+        TeamHelper()
+    }
+
+    @Before
+    fun setUp() {
+        context = InstrumentationRegistry.getInstrumentation().context
+        device = UiAutomatorSetup.start(UiAutomatorSetup.APP_INTERNAL)
+        backendClient = BackendClient.loadBackend("STAGING")
+        oktaApiClient = OktaApiClient()
+    }
+
+    @After
+    fun tearDown() {
+        teamOwner?.deleteTeam(backendClient!!)
+        deleteDownloadedFilesContaining("Wire")
+        oktaApiClient.cleanUp()
+    }
+
+    @Suppress("CyclomaticComplexMethod", "LongMethod")
+    @Test
+    fun givenSSOTeamWithOkta_whenSettingUpNewDeviceAndRestoringBackup_thenMessageIsRestored() {
+
+        runBlocking {
+
+            teamServiceHelper.thereIsASSOTeamOwnerForOkta(
+                context,
+                "user1Name",
+                "Messaging",
+                oktaApiClient
+            )
+
+            teamServiceHelper.userAddsOktaUser("user1Name", "user2Name", oktaApiClient)
+
+            teamServiceHelper.userXIsMe("user2Name")
+
+            teamOwner = teamHelper.usersManager.findUserBy(
+                "user1Name",
+                ClientUserManager.FindBy.NAME_ALIAS
+            )
+            member1 = teamHelper.usersManager.findUserBy(
+                "user2Name",
+                ClientUserManager.FindBy.NAME_ALIAS
+            )
+
+            val ssoCode = SSOServiceHelper.getSSOCode()
+             waitFor(20) // Delay added to allow Okta app assignment to fully sync and avoid 403 error
+            pages.registrationPage.apply {
+                assertEmailWelcomePage()
+            }
+            pages.loginPage.apply {
+                clickStagingDeepLink()
+                clickProceedButtonOnDeeplinkOverlay()
+            }
+            pages.loginPage.apply {
+                enterSSOCodeOnSSOLoginTab(ssoCode)
+                clickLoginButton()
+            }
+            pages.ssoPage.apply {
+                enterOktaEmail(member1?.email ?: "")
+                enterOktaPassword(member1?.password ?: "")
+                tapOktaSignIn()
+            }
+            pages.registrationPage.apply {
+                clickAllowNotificationButton()
+                setUserName(member1?.uniqueUsername.orEmpty())
+                clickConfirmButton()
+                waitUntilLoginFlowIsCompleted()
+                clickDeclineShareDataAlert()
+            }
+            pages.conversationListPage.apply {
+                tapStartNewConversationButton()
+            }
+            pages.searchPage.apply {
+                tapSearchPeopleField()
+                typeUserNameInSearchField("user1Name")
+                assertUsernameInSearchResultIs(teamOwner?.name ?: "")
+                tapUsernameInSearchResult(teamOwner?.name ?: "")
+            }
+            pages.connectedUserProfilePage.apply {
+                assertStartConversationButtonVisible()
+                clickStartConversationButton()
+            }
+            pages.conversationViewPage.apply {
+                assertConversationScreenVisible()
+                typeMessageInInputField("Testing of the backup functionality")
+                clickSendButton()
+                assertSentMessageIsVisibleInCurrentConversation("Testing of the backup functionality")
+                tapBackButtonToCloseConversationViewPage()
+            }
+            pages.connectedUserProfilePage.apply {
+                tapCloseButtonOnConnectedUserProfilePage()
+            }
+            pages.conversationListPage.apply {
+                clickCloseButtonOnNewConversationScreen()
+                assertConversationListVisible()
+            }
+            pages.conversationListPage.apply {
+                clickConversationsMenuEntry()
+                clickSettingsButtonOnMenuEntry()
+            }
+            pages.settingsPage.apply {
+                openBackupAndRestoreConversationsMenu()
+                iSeeBackupPageHeading()
+                clickCreateBackupButton()
+                clickBackUpNowButton()
+                iSeeBackupConfirmation("Conversations successfully saved")
+                iTapSaveFileButton()
+                iTapSaveInOSMenuButton()
+                iSeeBackupPageHeading()
+                clickBackButtonOnSettingsPage()
+            }
+            pages.conversationListPage.apply {
+                clickConversationsMenuEntry()
+                clickConversationsButtonOnMenuEntry()
+                clickUserProfileButton()
+            }
+            pages.selfUserProfilePage.apply {
+                iSeeUserProfilePage()
+                tapLogoutButton()
+                iSeeClearDataOnLogOutAlert()
+                iSeeInfoTextCheckbox("Delete all your personal information and conversations on this device")
+                tapInfoTextCheckbox()
+                tapLogoutButton()
+            }
+            pages.registrationPage.apply {
+                assertEmailWelcomePage()
+            }
+            pages.loginPage.apply {
+                clickStagingDeepLink()
+                clickProceedButtonOnDeeplinkOverlay()
+            }
+            pages.loginPage.apply {
+                enterSSOCodeOnSSOLoginTab(ssoCode)
+                clickLoginButton()
+            }
+            pages.registrationPage.apply {
+                waitUntilLoginFlowIsCompleted()
+                clickDeclineShareDataAlert()
+            }
+            pages.conversationListPage.apply {
+                assertConversationIsVisibleWithTeamOwner(teamOwner?.name ?: "")
+                tapConversationNameInConversationList(teamOwner?.name ?: "")
+            }
+            pages.conversationViewPage.apply {
+                assertMessageNotVisible("Testing of the backup functionality")
+                tapBackButtonToCloseConversationViewPage()
+            }
+            pages.conversationListPage.apply {
+
+                clickConversationsMenuEntry()
+                clickSettingsButtonOnMenuEntry()
+            }
+            pages.settingsPage.apply {
+                openBackupAndRestoreConversationsMenu()
+                iSeeBackupPageHeading()
+                clickRestoreBackupButton()
+                clickChooseBackupFileButton()
+                selectBackupFileInDocumentsUI(teamHelper, "user2Name")
+                waitUntilThisTextIsDisplayedOnBackupAlert("Conversations have been restored")
+                clickOkButtonOnBackupAlert()
+            }
+            pages.conversationListPage.apply {
+                assertConversationListVisible()
+                assertConversationIsVisibleWithTeamOwner(teamOwner?.name ?: "")
+                tapConversationNameInConversationList(teamOwner?.name ?: "")
+            }
+            pages.conversationViewPage.apply {
+                assertMessageNotVisible("Testing of the backup functionality")
+            }
+        }
+    }
+}

tests/testsCore/src/androidTest/kotlin/com/wire/android/tests/core/pages/AllPages.kt

@@ -36,4 +36,10 @@ class AllPages(val device: UiDevice) {
     val groupConversationDetailsPage = GroupConversationDetailsPage(device)
 
     val searchPage = SearchPage(device)
+
+    val ssoPage = SSOPage(device)
+
+    val chromePage = ChromePage(device)
+
+    val selfUserProfilePage = SelfUserProfilePage(device)
 }