From 9b33890b10dfb6e2538709a5fcb36f8f9a0658d2 Mon Sep 17 00:00:00 2001
From: netkas <netkas@n64.cc>
Date: Fri, 3 Jan 2025 13:26:17 -0500
Subject: [PATCH] Update Cryptography tests for algorithm-specific handling

---
 tests/Socialbox/Classes/CryptographyTest.php | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/tests/Socialbox/Classes/CryptographyTest.php b/tests/Socialbox/Classes/CryptographyTest.php
index 882787d..425af15 100644
--- a/tests/Socialbox/Classes/CryptographyTest.php
+++ b/tests/Socialbox/Classes/CryptographyTest.php
@@ -322,7 +322,6 @@
             $invalidPublicKey = "invalid_public_key";
 
             $this->expectException(CryptographyException::class);
-            $this->expectExceptionMessage("Failed to verify signature");
 
             Cryptography::verifyMessage($message, $signature, $invalidPublicKey);
         }
@@ -347,7 +346,7 @@
          */
         public function testGenerateTransportKeyCreatesValidKeyForDefaultAlgo(): void
         {
-            $transportKey = Cryptography::generateEncryptionKey();
+            $transportKey = Cryptography::generateEncryptionKey('xchacha20');
             $decodedKey = sodium_base642bin($transportKey, SODIUM_BASE64_VARIANT_URLSAFE_NO_PADDING, true);
 
             $this->assertNotEmpty($transportKey);
@@ -380,7 +379,6 @@
         public function testGenerateTransportKeyThrowsExceptionForInvalidAlgorithm(): void
         {
             $this->expectException(CryptographyException::class);
-            $this->expectExceptionMessage("Unsupported algorithm");
 
             Cryptography::generateEncryptionKey("invalid_algorithm");
         }
@@ -411,7 +409,6 @@
         public function testGenerateTransportKeyThrowsExceptionForUnsupportedAlgorithm(): void
         {
             $this->expectException(CryptographyException::class);
-            $this->expectExceptionMessage("Unsupported algorithm");
 
             Cryptography::generateEncryptionKey('invalid_algo');
         }
@@ -430,8 +427,8 @@
                 $this->assertEquals($keyLength, strlen(sodium_base642bin($transportKey, SODIUM_BASE64_VARIANT_URLSAFE_NO_PADDING, true)));
                 $message = "Test message";
 
-                $encryptedMessage = Cryptography::encryptMessage($message, $transportKey);
-                $decryptedMessage = Cryptography::decryptMessage($encryptedMessage, $transportKey);
+                $encryptedMessage = Cryptography::encryptMessage($message, $transportKey, $algorithm);
+                $decryptedMessage = Cryptography::decryptMessage($encryptedMessage, $transportKey, $algorithm);
 
                 $this->assertEquals($message, $decryptedMessage);
             }