PIA-1299: Add missing stubs on unit tests (#84)

pia-foss · Feb 15, 2024 · f7a2a92 · f7a2a92
1 parent a206e9f
commit f7a2a92
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/PIA VPN-tvOSTests/Common/Mocks/AccountProviderTypeMock.swift b/PIA VPN-tvOSTests/Common/Mocks/AccountProviderTypeMock.swift
@@ -8,8 +8,13 @@
 
 import Foundation
 @testable import PIA_VPN_tvOS
+import PIALibrary
 
 class AccountProviderTypeMock: AccountProviderType {
+    var publicUsername: String? = nil
+
+    var currentUser: PIALibrary.UserAccount? = nil
+
     var isLoggedIn: Bool = false
 
     private(set) var logoutCalled = false

diff --git a/PIA VPN-tvOSTests/VPNConfigurationInstalling/Mocks/VpnConfigurationProviderTypeMock.swift b/PIA VPN-tvOSTests/VPNConfigurationInstalling/Mocks/VpnConfigurationProviderTypeMock.swift
@@ -11,6 +11,7 @@ import PIALibrary
 @testable import PIA_VPN_tvOS
 
 class VpnConfigurationProviderTypeMock: VpnConfigurationProviderType {
+
     private let error: Error?
 
     init(error: Error?) {
@@ -20,4 +21,9 @@ class VpnConfigurationProviderTypeMock: VpnConfigurationProviderType {
     func install(force forceInstall: Bool, _ callback: SuccessLibraryCallback?) {
         callback?(error)
     }
+
+    func uninstall(_ callback: PIALibrary.SuccessLibraryCallback?) {
+        callback?(error)
+    }
+
 }