From 4430ca55bde672c27052c1c08986acdbdcdfaf45 Mon Sep 17 00:00:00 2001
From: Said Rehouni <said.rehouni@kape.com>
Date: Fri, 2 Feb 2024 12:44:22 +0100
Subject: [PATCH] PIA-0000: Update vpn profile installing property enabling for
 mock

---
 PIA VPN-tvOS/RootContainer/UI/RootContainerView.swift     | 1 +
 .../InstallVPNConfigurationUseCaseMock.swift              | 5 ++++-
 .../VPNConfigurationInstallingFactory.swift               | 8 +++++++-
 3 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/PIA VPN-tvOS/RootContainer/UI/RootContainerView.swift b/PIA VPN-tvOS/RootContainer/UI/RootContainerView.swift
index 8dcccf3c..77c925c3 100644
--- a/PIA VPN-tvOS/RootContainer/UI/RootContainerView.swift	
+++ b/PIA VPN-tvOS/RootContainer/UI/RootContainerView.swift	
@@ -27,6 +27,7 @@ struct RootContainerView: View {
                         .withOnboardingRoutes()
                 case .activatedNotOnboarded, .activated:
                     UserActivatedContainerFactory.makeUSerActivatedContainerView()
+                        .withOnboardingRoutes()
                 }
         }.onChange(of: scenePhase) { _, newPhase in
             if newPhase == .active {
diff --git a/PIA VPN-tvOS/VPNConfigurationInstalling/CompositionRoot/InstallVPNConfigurationUseCaseMock.swift b/PIA VPN-tvOS/VPNConfigurationInstalling/CompositionRoot/InstallVPNConfigurationUseCaseMock.swift
index 29c1de4a..72130d61 100644
--- a/PIA VPN-tvOS/VPNConfigurationInstalling/CompositionRoot/InstallVPNConfigurationUseCaseMock.swift	
+++ b/PIA VPN-tvOS/VPNConfigurationInstalling/CompositionRoot/InstallVPNConfigurationUseCaseMock.swift	
@@ -10,13 +10,16 @@ import Foundation
 
 class InstallVPNConfigurationUseCaseMock: InstallVPNConfigurationUseCaseType {
     private let error: InstallVPNConfigurationError?
+    var onSuccessAction: (() -> Void)?
     
-    init(error: InstallVPNConfigurationError?) {
+    init(error: InstallVPNConfigurationError?, onSuccessAction: (() -> Void)? = nil) {
         self.error = error
+        self.onSuccessAction = onSuccessAction
     }
     
     func callAsFunction() async throws {
         guard let error = error else {
+            onSuccessAction?()
             return
         }
         
diff --git a/PIA VPN-tvOS/VPNConfigurationInstalling/CompositionRoot/VPNConfigurationInstallingFactory.swift b/PIA VPN-tvOS/VPNConfigurationInstalling/CompositionRoot/VPNConfigurationInstallingFactory.swift
index 48b79716..81d2875f 100644
--- a/PIA VPN-tvOS/VPNConfigurationInstalling/CompositionRoot/VPNConfigurationInstallingFactory.swift	
+++ b/PIA VPN-tvOS/VPNConfigurationInstalling/CompositionRoot/VPNConfigurationInstallingFactory.swift	
@@ -34,7 +34,13 @@ class VPNConfigurationInstallingFactory {
     private static func makeInstallVPNConfigurationUseCase() -> InstallVPNConfigurationUseCaseType {
         
         guard !isSimulator else {
-            return InstallVPNConfigurationUseCaseMock(error: nil)
+            
+            let onSuccessAction = {
+                let vpnConfigurationAvailability = VPNConfigurationAvailability()
+                vpnConfigurationAvailability.set(value: true)
+            }
+            
+            return InstallVPNConfigurationUseCaseMock(error: nil, onSuccessAction: onSuccessAction)
         }
         
         return InstallVpnConfigurationProvider(vpnProvider:  makeVpnConfigurationProvider(),