From 57f8ee4f1eb5ae8150296e2ef31e5d50fe995c2b Mon Sep 17 00:00:00 2001
From: Ron Lauren Hombre <118486316+ronhombre@users.noreply.github.com>
Date: Sat, 10 Feb 2024 20:42:46 +0800
Subject: [PATCH] Fix failing tests.

---
 src/commonMain/kotlin/asia/hombre/kyber/KyberAgreement.kt | 4 ++--
 src/commonTest/kotlin/asia/hombre/kyber/Tests.kt          | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/commonMain/kotlin/asia/hombre/kyber/KyberAgreement.kt b/src/commonMain/kotlin/asia/hombre/kyber/KyberAgreement.kt
index 32f48c4..7348988 100644
--- 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
index 3700d6c..f08db18 100644
--- 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!")
         }