catching username availabilty exceptions and handling as user facing error

vector/src/main/java/im/vector/app/features/onboarding/OnboardingViewModel.kt

@@ -193,25 +193,32 @@ class OnboardingViewModel @AssistedInject constructor(
     }
 
     private suspend fun checkUserNameAvailability(userName: String) {
-        when (val result = registrationWizard.registrationAvailable(userName)) {
+        runCatching { registrationWizard.registrationAvailable(userName) }.fold(
-            RegistrationAvailability.Available -> {
+                onSuccess = { result ->
-                setState {
+                    when (result) {
-                    copy(
+                        RegistrationAvailability.Available -> {
-                            registrationState = RegistrationState(
+                            setState {
-                                    isUserNameAvailable = true,
+                                copy(
-                                    selectedMatrixId = when {
+                                        registrationState = RegistrationState(
-                                        userName.isMatrixId() -> userName
+                                                isUserNameAvailable = true,
-                                        else -> "@$userName:${selectedHomeserver.userFacingUrl.toReducedUrl()}"
+                                                selectedMatrixId = when {
-                                    },
+                                                    userName.isMatrixId() -> userName
-                            )
+                                                    else -> "@$userName:${selectedHomeserver.userFacingUrl.toReducedUrl()}"
-                    )
+                                                },
-                }
+                                        )
-            }
+                                )
+                            }
+                        }
 
-            is RegistrationAvailability.NotAvailable -> {
+                        is RegistrationAvailability.NotAvailable -> {
-                _viewEvents.post(OnboardingViewEvents.Failure(result.failure))
+                            _viewEvents.post(OnboardingViewEvents.Failure(result.failure))
-            }
+                        }
-        }
+                    }
+                },
+                onFailure = {
+                    _viewEvents.post(OnboardingViewEvents.Failure(it))
+                }
+        )
     }
 
     private fun withAction(action: OnboardingAction, block: (OnboardingAction) -> Unit) {

vector/src/test/java/im/vector/app/features/onboarding/OnboardingViewModelTest.kt

@@ -390,6 +390,20 @@ class OnboardingViewModelTest {
                 .finish()
     }
 
+    @Test
+    fun `given available username throws, when a register username is entered, then emits error`() = runTest {
+        viewModelWith(initialRegistrationState(A_HOMESERVER_URL))
+        fakeAuthenticationService.givenRegistrationWizard(FakeRegistrationWizard().also { it.givenUserNameIsAvailableThrows(A_USERNAME, AN_ERROR) })
+        val test = viewModel.test()
+
+        viewModel.handle(OnboardingAction.UserNameEnteredAction.Registration(A_USERNAME))
+
+        test
+                .assertStates(initialState)
+                .assertEvents(OnboardingViewEvents.Failure(AN_ERROR))
+                .finish()
+    }
+
     @Test
     fun `given available username, when a register username is entered, then emits available registration state`() = runTest {
         viewModelWith(initialRegistrationState(A_HOMESERVER_URL))

vector/src/test/java/im/vector/app/test/fakes/FakeRegistrationWizard.kt

@@ -57,6 +57,10 @@ class FakeRegistrationWizard : RegistrationWizard by mockk(relaxed = false) {
         coEvery { registrationAvailable(userName) } returns RegistrationAvailability.Available
     }
 
+    fun givenUserNameIsAvailableThrows(userName: String, cause: Throwable) {
+        coEvery { registrationAvailable(userName) } throws cause
+    }
+
     fun givenUserNameIsUnavailable(userName: String, failure: Failure.ServerError) {
         coEvery { registrationAvailable(userName) } returns RegistrationAvailability.NotAvailable(failure)
     }