Function test_ciphertext_decryption_method

wasm/src/programs/data/ciphertext.rs:323–369

()

    #[wasm_bindgen_test]
    fn test_ciphertext_decryption_method() {
        // Construct the private key and view key associated with the transition.
        let private_key = PrivateKey::from_string(PRIVATE_KEY).unwrap();
        let view_key = ViewKey::from_private_key(&private_key);

        // Construct the ciphertext to be decrypted and the transition it is a part of.
        let transition = Transition::from_string(TRANSITION).unwrap();
        let ciphertext = Ciphertext::from_string(CIPHERTEXT.to_string()).unwrap();

        // Compute the tpk and tvk.
        let tpk = transition.tpk();
        let tvk = tpk.scalar_multiply(&view_key.to_scalar()).to_x_coordinate();

        // Decrypt the ciphertext using the transition view key.
        if CurrentNetwork::ID == 1 {
            let decrypted_plaintext_1 = ciphertext
                .decrypt_with_transition_info(view_key, tpk, transition.program_id(), transition.function_name(), 1)
                .unwrap();
            let decrypted_plaintext_2 = ciphertext
                .decrypt_with_transition_view_key(tvk, transition.program_id(), transition.function_name(), 1)
                .unwrap();

            // Ensure the decrypted plaintexts are correct.
            assert_eq!(decrypted_plaintext_1.to_string(), "2u32");
            assert_eq!(decrypted_plaintext_2.to_string(), "2u32");
        } else {
            let private_key = PrivateKey::from_string(&get_env("PUZZLE_PK")).unwrap();
            let ciphertext = Ciphertext::from_string(MAINNET_CIPHERTEXT.to_string()).unwrap();
            let view_key = ViewKey::from_private_key(&private_key);
            let tpk =
                Group::from_string("4836727228112195507899959946473951966332108644709301619354443293014225839809group")
                    .unwrap();
            let decrypted_plaintext = ciphertext
                .decrypt_with_transition_info(
                    view_key,
                    tpk,
                    "puzzle_spinner_v002.aleo".to_string(),
                    "spin".to_string(),
                    2,
                )
                .unwrap();
            assert_eq!(
                decrypted_plaintext.to_string(),
                crate::utilities::test::PUZZLE_SPINNER_V002_INPUT_2.to_string()
            );
        }
    }
}