Fix failing tests.

ronhombre · Feb 10, 2024 · 57f8ee4 · 57f8ee4
1 parent d0d1b8c
commit 57f8ee4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/commonMain/kotlin/asia/hombre/kyber/KyberAgreement.kt b/src/commonMain/kotlin/asia/hombre/kyber/KyberAgreement.kt
@@ -48,7 +48,7 @@ class KyberAgreement(kemKeyPair: KyberKEMKeyPair) {
          * @param randomness Random 32 Bytes
          * @return KyberCipherText The Cipher Text to send to the second party.
          */
-        private fun toCipherText(encryptionKey: KyberEncryptionKey, plainText: ByteArray, randomness: ByteArray): KyberCipherText {
+        internal fun toCipherText(encryptionKey: KyberEncryptionKey, plainText: ByteArray, randomness: ByteArray): KyberCipherText {
             val parameter = encryptionKey.parameter
             val decodedKey = KyberMath.byteDecode(encryptionKey.keyBytes, 12)
             val nttKeyVector = Array(parameter.K) { ShortArray(KyberConstants.N) }
@@ -135,7 +135,7 @@ class KyberAgreement(kemKeyPair: KyberKEMKeyPair) {
          * @param cipherText Cipher Text from the second party.
          * @return ByteArray The original Plain Text.
          */
-        private fun fromCipherText(decapsulationKey: KyberDecapsulationKey, cipherText: KyberCipherText): ByteArray {
+        internal fun fromCipherText(decapsulationKey: KyberDecapsulationKey, cipherText: KyberCipherText): ByteArray {
             val parameter = cipherText.parameter
             val coefficients = Array(cipherText.parameter.K) { ShortArray(KyberConstants.N) }
 

diff --git a/src/commonTest/kotlin/asia/hombre/kyber/Tests.kt b/src/commonTest/kotlin/asia/hombre/kyber/Tests.kt
@@ -263,7 +263,7 @@ class Tests {
 
             val original = SecureRandom().nextBytesOf(32)
             val cipher = KyberAgreement.encapsulate(keyPairBob.encapsulationKey, original).cipherText
-            val recovered = KyberAgreement.decapsulate(keyPairBob.decapsulationKey, cipher)
+            val recovered = KyberAgreement.fromCipherText(keyPairBob.decapsulationKey, cipher)
 
             assertContentEquals(original, recovered, "PKE Encryption and Decryption for 512 failed at attempt $i!")
         }
@@ -276,7 +276,7 @@ class Tests {
 
             val original = SecureRandom().nextBytesOf(32)
             val cipher = KyberAgreement.encapsulate(keyPairBob.encapsulationKey, original).cipherText
-            val recovered = KyberAgreement.decapsulate(keyPairBob.decapsulationKey, cipher)
+            val recovered = KyberAgreement.fromCipherText(keyPairBob.decapsulationKey, cipher)
 
             assertContentEquals(original, recovered, "PKE Encryption and Decryption for 768 failed at attempt $i!")
         }
@@ -289,7 +289,7 @@ class Tests {
 
             val original = SecureRandom().nextBytesOf(32)
             val cipher = KyberAgreement.encapsulate(keyPairBob.encapsulationKey, original).cipherText
-            val recovered = KyberAgreement.decapsulate(keyPairBob.decapsulationKey, cipher)
+            val recovered = KyberAgreement.fromCipherText(keyPairBob.decapsulationKey, cipher)
 
             assertContentEquals(original, recovered, "PKE Encryption and Decryption for 1024 failed at attempt $i!")
         }