added tests

Author: Roee-87

wasm/src/ledger/transition.rs

@@ -207,8 +207,8 @@ impl Transition {
             compute_function_id(&U16Native::new(CurrentNetwork::ID), self.0.program_id(), self.0.function_name())
                 .map_err(|e| e.to_string())?;
 
-        // Create a vector that will be populated with decrypted private inputs and 
-        // non-private inputs.        
+        // Create a vector that will be populated with decrypted private inputs and
+        // non-private inputs.
         let mut decrypted_inputs = Vec::with_capacity(self.0.inputs().len());
 
         // Iterate over the inputs and decrypt if they are private.  Non-private inputs
@@ -229,7 +229,7 @@ impl Transition {
         let outputs = self.0.outputs();
         let num_inputs = self.0.inputs().len();
 
-        // Create a vector that will be populated with decrypted private outputs and 
+        // Create a vector that will be populated with decrypted private outputs and
         // non-private outputs.
         let mut decrypted_outputs = Vec::with_capacity(outputs.len());
 
@@ -259,7 +259,7 @@ impl Transition {
                 *self.0.tcm(),
                 *self.0.scm(),
             )
-            .map_err(||"failed to construct decrypted transition".to_string()),
+            .map_err(|_| "failed to construct decrypted transition".to_string())?,
         ))
     }
 }
@@ -520,4 +520,40 @@ mod tests {
 
         assert_eq!(decrypted_transition.to_string(), TRANSITION_MAINNET_DECRYPTED);
     }
+
+    #[wasm_bindgen_test]
+    #[cfg(feature = "testnet")]
+    fn test_transition_decrypt_testnet_invalid_vk() {
+        // Create a view key from the test private key.
+        let private_key = PrivateKey::from_str(TEST_PRIVATE_KEY).unwrap();
+        let view_key = ViewKey::from_private_key(&private_key);
+
+        // Get a transition with records.
+        let transition = Transition::from_string(TRANSITION_TESTNET).unwrap();
+
+        // Create an invalid transition view key.
+        let invalid_tvk = transition.tvk(&view_key);
+
+        // Attempt to decrypt the transition using the invalid transition view key.
+        let result = transition.decrypt_transition(&invalid_tvk);
+        assert!(result.is_err());
+    }
+
+    #[wasm_bindgen_test]
+    #[cfg(feature = "mainnet")]
+    fn test_transition_decrypt_testnet_invalid_vk() {
+        // Create a view key from the test private key.
+        let private_key = PrivateKey::from_str(TEST_PRIVATE_KEY).unwrap();
+        let view_key = ViewKey::from_private_key(&private_key);
+
+        // Get a transition with records.
+        let transition = Transition::from_string(TRANSITION_MAINNET).unwrap();
+
+        // Create an invalid transition view key.
+        let invalid_tvk = transition.tvk(&view_key);
+
+        // Attempt to decrypt the transition using the invalid transition view key.
+        let result = transition.decrypt_transition(&invalid_tvk);
+        assert!(result.is_err());
+    }
 }

wasm/src/record/record_ciphertext.rs

@@ -23,7 +23,7 @@ use crate::{
     js_array_from_fields,
     to_bits_array_le,
     types::native::{CurrentNetwork, RecordCiphertextNative},
-    utilities::encrypt::EncryptionToolkilt,
+    utilities::encrypt::EncryptionToolkit,
 };
 use snarkvm_console::prelude::{FromBytes, Network, ToBits, ToBytes, ToFields};
 
@@ -233,7 +233,8 @@ mod tests {
         let view_key = ViewKey::from_string(OWNER_VIEW_KEY);
         let graph_key = GraphKey::from_view_key(&view_key);
         let plaintext = record.decrypt(&view_key).unwrap();
-        assert_eq!(plaintext.to_string(), OWNER_PLAINTEXT);
+        let owner_plaintext = RecordPlaintext::from_string(OWNER_PLAINTEXT).unwrap();
+        assert_eq!(plaintext.to_string(), owner_plaintext.to_string());
         let incorrect_view_key = ViewKey::from_string(NON_OWNER_VIEW_KEY);
         assert!(record.decrypt(&incorrect_view_key).is_err());
 
@@ -270,4 +271,12 @@ mod tests {
         let decrypted_plaintext = record.decrypt_with_record_view_key(record_vk).unwrap();
         assert_eq!(record_plaintext.to_string(), decrypted_plaintext.to_string());
     }
+
+    #[wasm_bindgen_test]
+    fn test_decrypt_with_record_vk_invalid() {
+        let record = RecordCiphertext::from_string(OWNER_CIPHERTEXT).unwrap();
+        let view_key_non_owner = ViewKey::from_string(NON_OWNER_VIEW_KEY);
+        let invalid_record_vk = record.record_view_key(&view_key_non_owner);
+        assert!(record.decrypt_with_record_view_key(invalid_record_vk).is_err());
+    }
 }

wasm/src/utilities/encrypt.rs

@@ -186,7 +186,7 @@ mod tests {
     use super::*;
 
     use std::str::FromStr;
-    use wasm_bindgen_test::wasm_bindgen_test;
+    use wasm_bindgen_test::{console_log, wasm_bindgen_test};
 
     const NON_OWNER_VIEW_KEY: &str = "AViewKey1e2WyreaH5H4RBcioLL2GnxvHk5Ud46EtwycnhTdXLmXp";
     const OWNER_CIPHERTEXT: &str = "record1qyqsqpe2szk2wwwq56akkwx586hkndl3r8vzdwve32lm7elvphh37rsyqyxx66trwfhkxun9v35hguerqqpqzqrtjzeu6vah9x2me2exkgege824sd8x2379scspmrmtvczs0d93qttl7y92ga0k0rsexu409hu3vlehe3yxjhmey3frh2z5pxm5cmxsv4un97q";
@@ -196,7 +196,37 @@ mod tests {
   _nonce: 3077450429259593211617823051143573281856129402760267155982965992208217472983group.public
 }";
     const OWNER_VIEW_KEY: &str = "AViewKey1ccEt8A2Ryva5rxnKcAbn7wgTaTsb79tzkKHFpeKsm9NX";
+    const RECORD_VIEW_KEY: &str = "4445718830394614891114647247073357094867447866913203502139893824059966201724field";
     const RECORD_TAG: &str = "1796466189545157638691489609907096471289658804813960182690905095269699169603field";
+    const TRANSITION_PUBLIC_KEY: &str =
+        "7532444547840484531569841377269810017844130178606467837628364672670182422388group";
+    const TRANSITION_VIEW_KEY: &str =
+        "5089075468761042335883809641276568724009791331127957254389204093712358605127field";
+
+    #[wasm_bindgen_test]
+    fn test_record_view_key_generation() {
+        let owner_view_key = ViewKey::from_str(OWNER_VIEW_KEY).unwrap();
+        let owner_ciphertext = RecordCiphertext::from_str(OWNER_CIPHERTEXT).unwrap();
+
+        // Generate the record view key
+        let record_vk = EncryptionToolkit::generate_record_view_key(&owner_view_key, &owner_ciphertext).unwrap();
+
+        // Verify the record view key matches the expected value
+        assert_eq!(record_vk.to_string(), RECORD_VIEW_KEY);
+    }
+
+    #[wasm_bindgen_test]
+    fn test_tvk_generation() {
+        let view_key = ViewKey::from_str(OWNER_VIEW_KEY).unwrap();
+        let tpk = Group::from_string(TRANSITION_PUBLIC_KEY).unwrap();
+
+        // Generate the transition view key
+        let tvk = EncryptionToolkit::generate_tvk(&view_key, &tpk);
+
+        // Verify the transition view key matches the expected value
+        // console_log!("Transition View Key: {}", tvk.to_string());
+        assert_eq!(tvk.to_string(), TRANSITION_VIEW_KEY);
+    }
 
     #[wasm_bindgen_test]
     fn test_decrypt_record_symmetric() {