-
Notifications
You must be signed in to change notification settings - Fork 991
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move helper fns into a separate module
- Loading branch information
Showing
2 changed files
with
179 additions
and
165 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,162 @@ | ||
| //! Helpers for use in multitoken tests. | ||
| use std::path::PathBuf; | ||
|
|
||
| use color_eyre::eyre::Result; | ||
| use namada_core::types::address::Address; | ||
| use namada_core::types::storage; | ||
| use namada_tx_prelude::storage::KeySeg; | ||
| use rand::Rng; | ||
|
|
||
| use super::setup::constants::{wasm_abs_path, NAM, VP_ALWAYS_TRUE_WASM}; | ||
| use super::setup::{Bin, NamadaCmd, Test}; | ||
| use crate::e2e::setup::constants::{ALBERT, TX_WRITE_STORAGE_KEY_WASM}; | ||
| use crate::run; | ||
|
|
||
| const MULTITOKEN_KEY_SEGMENT: &str = "tokens"; | ||
| const BALANCE_KEY_SEGMENT: &str = "balance"; | ||
| const RED_TOKEN_KEY_SEGMENT: &str = "red"; | ||
|
|
||
| const ARBITRARY_SIGNER: &str = ALBERT; | ||
|
|
||
| pub fn get_address_for_alias(test: &Test, alias: &str) -> Result<Address> { | ||
| let wallet_find_address = run!( | ||
| test, | ||
| Bin::Wallet, | ||
| vec!["address", "find", "--alias", alias], | ||
| Some(40) | ||
| )? | ||
| .background(); | ||
| // sleep a couple secs so the command has chance to run? maybe this isn't | ||
| // necessary | ||
| std::thread::sleep(std::time::Duration::from_secs(2)); | ||
| let mut wallet_find_address = wallet_find_address.foreground(); | ||
|
|
||
| let (_, addr) = wallet_find_address.exp_regex(r"atest1\w+")?; | ||
| Ok(Address::decode(addr)?) | ||
| } | ||
|
|
||
| /// Initializes a VP to represent a multitoken account. | ||
| pub fn init_multitoken_vp(test: &Test, rpc_addr: &str) -> Result<String> { | ||
| // we use a VP that always returns true for the multitoken VP here, as we | ||
| // are testing out the VPs of the sender and receiver of multitoken | ||
| // transactions here - not any multitoken VP itself | ||
| let multitoken_vp_wasm_path = wasm_abs_path(VP_ALWAYS_TRUE_WASM) | ||
| .to_string_lossy() | ||
| .to_string(); | ||
| let multitoken_alias = "multitoken"; | ||
|
|
||
| let init_account_args = vec![ | ||
| "init-account", | ||
| "--source", | ||
| ARBITRARY_SIGNER, | ||
| "--public-key", | ||
| // Value obtained from | ||
| // `namada::types::key::ed25519::tests::gen_keypair` | ||
| "001be519a321e29020fa3cbfbfd01bd5e92db134305609270b71dace25b5a21168", | ||
| "--code-path", | ||
| &multitoken_vp_wasm_path, | ||
| "--alias", | ||
| multitoken_alias, | ||
| "--gas-amount", | ||
| "0", | ||
| "--gas-limit", | ||
| "0", | ||
| "--gas-token", | ||
| NAM, | ||
| "--ledger-address", | ||
| rpc_addr, | ||
| ]; | ||
| let mut client_init_account = | ||
| run!(test, Bin::Client, init_account_args, Some(40))?; | ||
| client_init_account.exp_string("Transaction is valid.")?; | ||
| client_init_account.exp_string("Transaction applied")?; | ||
| client_init_account.assert_success(); | ||
| Ok(multitoken_alias.to_string()) | ||
| } | ||
|
|
||
| /// Generates a random path within the `test` directory. | ||
| fn generate_random_test_dir_path(test: &Test) -> PathBuf { | ||
| let rng = rand::thread_rng(); | ||
| let random_string: String = rng | ||
| .sample_iter(&rand::distributions::Alphanumeric) | ||
| .take(24) | ||
| .map(char::from) | ||
| .collect(); | ||
| test.test_dir.path().join(random_string) | ||
| } | ||
|
|
||
| /// Writes `contents` to a random path within the `test` directory, and return | ||
| /// the path. | ||
| pub fn write_test_file( | ||
| test: &Test, | ||
| contents: impl AsRef<[u8]>, | ||
| ) -> Result<PathBuf> { | ||
| let path = generate_random_test_dir_path(test); | ||
| std::fs::write(&path, contents)?; | ||
| Ok(path) | ||
| } | ||
|
|
||
| /// Mint 100 red tokens to the given address. | ||
| pub fn mint_red_tokens( | ||
| test: &Test, | ||
| rpc_addr: &str, | ||
| multitoken: &Address, | ||
| owner: &Address, | ||
| ) -> Result<()> { | ||
| let red_balance_key = storage::Key::from(multitoken.to_db_key()) | ||
| .push(&MULTITOKEN_KEY_SEGMENT.to_owned())? | ||
| .push(&RED_TOKEN_KEY_SEGMENT.to_owned())? | ||
| .push(&BALANCE_KEY_SEGMENT.to_owned())? | ||
| .push(owner)?; | ||
|
|
||
| let tx_code_path = wasm_abs_path(TX_WRITE_STORAGE_KEY_WASM); | ||
| let tx_data_path = write_test_file(test, format!("{red_balance_key}"))?; | ||
|
|
||
| let tx_data_path = tx_data_path.to_string_lossy().to_string(); | ||
| let tx_code_path = tx_code_path.to_string_lossy().to_string(); | ||
| let tx_args = vec![ | ||
| "tx", | ||
| "--signer", | ||
| ARBITRARY_SIGNER, | ||
| "--code-path", | ||
| &tx_code_path, | ||
| "--data-path", | ||
| &tx_data_path, | ||
| "--ledger-address", | ||
| rpc_addr, | ||
| ]; | ||
| let mut client_tx = run!(test, Bin::Client, tx_args, Some(40))?; | ||
| client_tx.exp_string("Transaction is valid.")?; | ||
| client_tx.exp_string("Transaction applied")?; | ||
| client_tx.assert_success(); | ||
| Ok(()) | ||
| } | ||
|
|
||
| pub fn attempt_red_tokens_transfer( | ||
| test: &Test, | ||
| rpc_addr: &str, | ||
| multitoken: &str, | ||
| from: &str, | ||
| to: &str, | ||
| signer: &str, | ||
| ) -> Result<NamadaCmd> { | ||
| let sub_prefix = "tokens/red"; | ||
| let transfer_args = vec![ | ||
| "transfer", | ||
| "--token", | ||
| multitoken, | ||
| "--sub-prefix", | ||
| sub_prefix, | ||
| "--source", | ||
| from, | ||
| "--target", | ||
| to, | ||
| "--signer", | ||
| signer, | ||
| "--amount", | ||
| "10", | ||
| "--ledger-address", | ||
| rpc_addr, | ||
| ]; | ||
| run!(test, Bin::Client, transfer_args, Some(40)) | ||
| } |