From dddf55cad180171944ea35ba01842e0da694e343 Mon Sep 17 00:00:00 2001
From: Stylianos Tzouvaras <stylianos.tzouvaras@netcompany.com>
Date: Tue, 3 Dec 2024 15:15:48 +0200
Subject: [PATCH] Fixed error state for Service and Credential Selection when
 RQESService returns error

---
 .../UI/CredentialSelection/CredentialSelectionViewModel.swift | 4 +++-
 .../UI/ServiceSelection/ServiceSelectionViewModel.swift       | 3 ++-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/Sources/Presentation/UI/CredentialSelection/CredentialSelectionViewModel.swift b/Sources/Presentation/UI/CredentialSelection/CredentialSelectionViewModel.swift
index a4afb2d..ad8b256 100644
--- a/Sources/Presentation/UI/CredentialSelection/CredentialSelectionViewModel.swift
+++ b/Sources/Presentation/UI/CredentialSelection/CredentialSelectionViewModel.swift
@@ -71,14 +71,16 @@ final class CredentialSelectionViewModel<Router: RouterGraph>: ViewModel<Router,
   }
   
   func nextStep() {
-    onPause()
     openAuthorization()
   }
   
   func openAuthorization() {
     Task {
       do {
+        
         let authorizationUrl = try await interactor.openCredentialAuthrorizationURL()
+        self.onPause()
+        
         await UIApplication.shared.openURLIfPossible(authorizationUrl) {
           self.setErrorState {
             self.setState { $0.copy(error: nil) }
diff --git a/Sources/Presentation/UI/ServiceSelection/ServiceSelectionViewModel.swift b/Sources/Presentation/UI/ServiceSelection/ServiceSelectionViewModel.swift
index 0d1fcc3..2a698f5 100644
--- a/Sources/Presentation/UI/ServiceSelection/ServiceSelectionViewModel.swift
+++ b/Sources/Presentation/UI/ServiceSelection/ServiceSelectionViewModel.swift
@@ -68,7 +68,6 @@ class ServiceSelectionViewModel<Router: RouterGraph>: ViewModel<Router, ServiceS
   }
   
   func nextStep() {
-    onPause()
     openAuthorization()
   }
   
@@ -82,7 +81,9 @@ class ServiceSelectionViewModel<Router: RouterGraph>: ViewModel<Router, ServiceS
             self.setState { $0.copy(error: nil) }
           }
         }
+        
         let authorizationUrl = try await interactor.openAuthrorizationURL()
+        self.onPause()
         
         await UIApplication.shared.openURLIfPossible(authorizationUrl) {
           self.setErrorState {