Fix MockCryptoClient

Author: fedemkr

BitwardenShared/Core/Auth/Services/TestHelpers/MockCryptoClient.swift

@@ -20,6 +20,22 @@ class MockCryptoClient: CryptoClientProtocol {
     var enrollAdminPasswordPublicKey: String?
     var enrollAdminPasswordResetResult: Result<String, Error> = .success("RESET_PASSWORD_KEY")
 
+    var enrollPinPin: String?
+    var enrollPinResult: Result<EnrollPinResponse, Error> = .success(
+        EnrollPinResponse(
+            pinProtectedUserKeyEnvelope: "pinProtectedUserKeyEnvelope",
+            userKeyEncryptedPin: "userKeyEncryptedPin"
+        )
+    )
+
+    var enrollPinWithEncryptedPinEncryptedPin: String?
+    var enrollPinWithEncryptedPinResult: Result<EnrollPinResponse, Error> = .success(
+        EnrollPinResponse(
+            pinProtectedUserKeyEnvelope: "pinProtectedUserKeyEnvelope",
+            userKeyEncryptedPin: "userKeyEncryptedPin"
+        )
+    )
+
     var getUserEncryptionKeyCalled = false
     var getUserEncryptionKeyResult: Result<String, Error> = .success("USER_ENCRYPTION_KEY")
 
@@ -56,6 +72,16 @@ class MockCryptoClient: CryptoClientProtocol {
         return try enrollAdminPasswordResetResult.get()
     }
 
+    func enrollPin(pin: String) throws -> EnrollPinResponse {
+        enrollPinPin = pin
+        return try enrollPinResult.get()
+    }
+
+    func enrollPinWithEncryptedPin(encryptedPin: EncString) throws -> EnrollPinResponse {
+        enrollPinWithEncryptedPinEncryptedPin = encryptedPin
+        return try enrollPinWithEncryptedPinResult.get()
+    }
+
     func getUserEncryptionKey() async throws -> String {
         getUserEncryptionKeyCalled = true
         return try getUserEncryptionKeyResult.get()