From b015d7939a2fc12ed34afed6058868f6f15857ab Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?S=C3=A9bastien=20Stormacq?= <sebastien.stormacq@gmail.com>
Date: Wed, 6 Nov 2024 10:21:14 +0100
Subject: [PATCH] Add test case for s2k_fo SRP autentication

---
 Tests/xcodeinstallTests/API/SRPTest.swift | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/Tests/xcodeinstallTests/API/SRPTest.swift b/Tests/xcodeinstallTests/API/SRPTest.swift
index e4edfc1..93d4753 100644
--- a/Tests/xcodeinstallTests/API/SRPTest.swift
+++ b/Tests/xcodeinstallTests/API/SRPTest.swift
@@ -78,7 +78,10 @@ struct SRPKeysTestCase {
 
     }
 
-    @Test func pbkdf2() throws {
+    @Test("PBKDF2 for S2K and S2K_FO authentication protocol", arguments: [
+        SRPProtocol.s2k, SRPProtocol.s2k_fo
+    ])
+    func pbkdf2(srpProtocol: SRPProtocol) throws {
 
         // given
         let password = "password"
@@ -96,13 +99,20 @@ struct SRPKeysTestCase {
                 salt: [UInt8](saltData),
                 iterations: iterations,
                 keyLength: keyLength,
-                srpProtocol: .s2k
+                srpProtocol: srpProtocol
             )
             // print(derivedKey.hexdigest().lowercased())
 
             // then
-            let fastlaneHexResult = "d7ff78163a0183db1e635ba5beaf4a45f7984b00aafec95e6a044fda331bbd45"
-            #expect(derivedKey.hexdigest().lowercased() == fastlaneHexResult)
+            let hexResult: String
+            switch srpProtocol {
+            case .s2k:
+                hexResult = "d7ff78163a0183db1e635ba5beaf4a45f7984b00aafec95e6a044fda331bbd45"
+            case .s2k_fo:
+                // TODO: verify this result is correct 
+                hexResult = "858f8ba24e48af6f9cd5c8d4738827eb91340b6901fc5e47ee0d73e3346b502a"
+            }
+            #expect(derivedKey.hexdigest().lowercased() == hexResult)
         }
     }