From f629d596c1337fb640e6d4ff9859beefd3565c96 Mon Sep 17 00:00:00 2001
From: samkim-crypto <skim13@cs.stanford.edu>
Date: Mon, 22 Apr 2024 16:36:59 +0900
Subject: [PATCH] add test

---
 .../instruction/transfer/without_fee_split.rs | 50 +++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/zk-token-sdk/src/instruction/transfer/without_fee_split.rs b/zk-token-sdk/src/instruction/transfer/without_fee_split.rs
index e2e16fc73760c3..7895840803de7a 100644
--- a/zk-token-sdk/src/instruction/transfer/without_fee_split.rs
+++ b/zk-token-sdk/src/instruction/transfer/without_fee_split.rs
@@ -255,3 +255,53 @@ impl TransferProofContext {
         })
     }
 }
+
+#[cfg(test)]
+mod test {
+    use {
+        super::*,
+        crate::{encryption::auth_encryption::AeKey, instruction::ZkProofData},
+    };
+
+    #[test]
+    fn test_transfer_with_split_proof_correctness() {
+        let source_keypair = ElGamalKeypair::new_rand();
+
+        let destination_keypair = ElGamalKeypair::new_rand();
+        let destination_pubkey = destination_keypair.pubkey();
+
+        let auditor_keypair = ElGamalKeypair::new_rand();
+        let auditor_pubkey = auditor_keypair.pubkey();
+
+        let aes_key = AeKey::new_rand();
+        let remaining_balance = 100;
+        let remaining_balance_elgamal_ciphertext =
+            source_keypair.pubkey().encrypt(remaining_balance);
+        let remaining_balance_decryptable_ciphertext = aes_key.encrypt(remaining_balance);
+
+        let transfer_amount = 10;
+
+        let (equality_proof_data, validity_proof_data, range_proof_data) =
+            transfer_split_proof_data(
+                &remaining_balance_elgamal_ciphertext,
+                &remaining_balance_decryptable_ciphertext,
+                transfer_amount,
+                &source_keypair,
+                &aes_key,
+                destination_pubkey,
+                &Some(auditor_pubkey),
+            )
+            .unwrap();
+
+        equality_proof_data.verify_proof().unwrap();
+        validity_proof_data.verify_proof().unwrap();
+        range_proof_data.verify_proof().unwrap();
+
+        TransferProofContext::new_from_split_proofs(
+            equality_proof_data.context_data(),
+            validity_proof_data.context_data(),
+            range_proof_data.context_data(),
+        )
+        .unwrap();
+    }
+}