Add test case for s2k_fo SRP autentication

sebsto · Nov 6, 2024 · b015d79 · b015d79
1 parent be0df53
commit b015d79
Showing 1 changed file with 14 additions and 4 deletions.
diff --git 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)
         }
     }