Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Subaccounts for load tests #552

Merged
merged 3 commits into from
Apr 15, 2024
Merged

Subaccounts for load tests #552

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
61c2fcb
subaccounts for loadtests
volovyks Apr 1, 2024
a3227a4
Merge branch 'develop' into serhii/lt-subaccounts
volovyks Apr 2, 2024
b3827dd
delete subaccount
volovyks Apr 9, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 3 additions & 2 deletions Cargo.lock
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

1 change: 1 addition & 0 deletions load-tests/Cargo.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,3 +22,4 @@ near-jsonrpc-client = "0.6.0"
near-fetch = "0.2.0"
web3 = "0.19.0"
borsh = "0.10.3"
near-workspaces = "0.10.0"
49 changes: 30 additions & 19 deletions load-tests/src/fastauth/mod.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,9 @@ pub mod primitives;
pub mod utils;

use core::panic;
use near_workspaces::{types::NearToken, Account};
use reqwest::{header::CONTENT_TYPE, Body};
use std::{time::Duration, vec};
use std::{str::FromStr, time::Duration, vec};

use constants::VALID_OIDC_PROVIDER_KEY;
use goose::prelude::*;
Expand All @@ -29,14 +30,35 @@ use near_primitives::{
types::AccountId,
};
use primitives::UserSession;
use rand::{distributions::Alphanumeric, Rng};
use utils::build_send_and_check_request;

pub async fn prepare_user_credentials(user: &mut GooseUser) -> TransactionResult {
tracing::info!("prepare_user_credentials");
// Generate 2 key pairs
let fa_sk = SecretKey::from_random(near_crypto::KeyType::ED25519);
let la_sk = SecretKey::from_random(near_crypto::KeyType::ED25519);

let worker = near_workspaces::testnet().await.unwrap();

let root_account = Account::from_secret_key(
near_workspaces::types::AccountId::try_from("dev-1660670387515-45063246810397".to_string()).unwrap(),
near_workspaces::types::SecretKey::from_str(
"ed25519:4hc3qA3nTE8M63DB8jEZx9ZbHVUPdkMjUAoa11m4xtET7F6w4bk51TwQ3RzEcFhBtXvF6NYzFdiJduaGdJUvynAi"
).unwrap(),
&worker
);
Comment on lines +40 to +46
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

the issue with this is that somebody can easily disrupt the load test by getting this key. We wouldn't notice this either until we run it locally manually. We should be creating a dev account each time and delete the account at the end of the test

Copy link
Collaborator Author

@volovyks volovyks Apr 2, 2024 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yep, I know. But dev_create_account does not work for me. And there is no Error handling. Maybe the faucet is broken.

Copy link
Collaborator Author

@volovyks volovyks Apr 2, 2024 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Tested, near-cli can create accounts using the faucet. Helper and URL parameters looks the same for me.


let subaccount = root_account
.create_subaccount(&format!("user-{}", rand::random::<u64>()))
.initial_balance(NearToken::from_yoctonear(200000000000000000000000u128))
.transact()
.await
.unwrap()
.into_result()
.unwrap();

tracing::info!(
"Created user accountId: {}, pk: {}",
subaccount.id(),
subaccount.secret_key().public_key()
);

// Create JWT with random sub (usually done by OIDC Provider)
let oidc_token = OidcToken::new(&utils::create_jwt_token(
Expand All @@ -46,22 +68,11 @@ pub async fn prepare_user_credentials(user: &mut GooseUser) -> TransactionResult
None,
));

// Generate random near account id
let account_id_rand: String = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(10)
.map(char::from)
.collect();

let near_account_id: AccountId = format!("acc-{}.near", account_id_rand.to_lowercase())
.try_into()
.expect("Failed to generate random account Id");

let session = UserSession {
jwt_token: oidc_token,
near_account_id,
fa_sk,
la_sk,
near_account_id: AccountId::try_from(subaccount.id().to_string()).unwrap(),
fa_sk: SecretKey::from_str(&subaccount.secret_key().to_string()).unwrap(),
la_sk: SecretKey::from_random(near_crypto::KeyType::ED25519), // no need to actually add it ATM
recovery_pk: None,
};

Expand Down
12 changes: 11 additions & 1 deletion load-tests/src/main.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,17 @@ async fn main() -> Result<(), GooseError> {
GooseAttack::initialize()?
.register_scenario(
scenario!("multichainSign")
.register_transaction(transaction!(multichain_sign).set_sequence(2)),
.register_transaction(transaction!(prepare_user_credentials).set_sequence(1))
.register_transaction(transaction!(multichain_sign).set_sequence(2))
.register_transaction(transaction!(multichain_sign).set_sequence(3))
.register_transaction(transaction!(multichain_sign).set_sequence(4))
.register_transaction(transaction!(multichain_sign).set_sequence(5))
.register_transaction(transaction!(multichain_sign).set_sequence(6))
.register_transaction(transaction!(multichain_sign).set_sequence(7))
.register_transaction(transaction!(multichain_sign).set_sequence(8))
.register_transaction(transaction!(multichain_sign).set_sequence(9))
.register_transaction(transaction!(multichain_sign).set_sequence(10))
.register_transaction(transaction!(multichain_sign).set_sequence(11)),
)
.register_scenario(
scenario!("fastAuthRegistration")
Expand Down
30 changes: 18 additions & 12 deletions load-tests/src/multichain/mod.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
use std::{str::FromStr, time::Duration};
use std::time::Duration;

use goose::goose::{GooseMethod, GooseRequest, GooseUser, TransactionResult};
use goose_eggs::{validate_and_load_static_assets, Validate};
use near_crypto::{InMemorySigner, SecretKey};
use near_crypto::InMemorySigner;
use near_jsonrpc_client::JsonRpcClient;
use near_primitives::{
transaction::{Action, FunctionCallAction, Transaction},
Expand All @@ -11,20 +11,23 @@ use near_primitives::{
use rand::Rng;
use reqwest::header::CONTENT_TYPE;

use crate::fastauth::primitives::UserSession;

pub async fn multichain_sign(user: &mut GooseUser) -> TransactionResult {
tracing::info!("multichain_sign");

// TODO: for better experience we can create real account in prepare_user_credentials and then get it from session
let account_id = AccountId::try_from("dev-1660670387515-45063246810397".to_string()).unwrap();
let secret_key = SecretKey::from_str("ed25519:4hc3qA3nTE8M63DB8jEZx9ZbHVUPdkMjUAoa11m4xtET7F6w4bk51TwQ3RzEcFhBtXvF6NYzFdiJduaGdJUvynAi").unwrap();
let public_key = secret_key.public_key();
let multichain_contract_id = AccountId::try_from("multichain0.testnet".to_string()).unwrap(); // TODO: pass in parameters
let sesion = user
.get_session_data::<UserSession>()
.expect("Session Data must be set");

let multichain_contract_id =
AccountId::try_from("v2.multichain-mpc.testnet".to_string()).unwrap();
let testnet_rpc_url = "https://rpc.testnet.near.org".to_string();

let signer = InMemorySigner {
account_id: account_id.clone(),
public_key: public_key.clone(),
secret_key,
account_id: sesion.near_account_id.clone(),
public_key: sesion.fa_sk.public_key(),
secret_key: sesion.fa_sk.clone(),
};

let connector = JsonRpcClient::new_client();
Expand All @@ -40,8 +43,8 @@ pub async fn multichain_sign(user: &mut GooseUser) -> TransactionResult {
tracing::info!("requesting signature for: {:?}", payload_hashed);

let transaction = Transaction {
signer_id: account_id.clone(),
public_key,
signer_id: sesion.near_account_id.clone(),
public_key: sesion.fa_sk.public_key(),
nonce,
receiver_id: multichain_contract_id,
block_hash,
Expand Down Expand Up @@ -85,6 +88,9 @@ pub async fn multichain_sign(user: &mut GooseUser) -> TransactionResult {

let goose_response = user.request(goose_request).await?;

// let text = goose_response.response.unwrap().text().await.unwrap();
// tracing::info!("goose_response: {:?}", text);

let rsp = goose_response.response.as_ref().unwrap();

tracing::info!("goose_response: {:?}", rsp);
Expand Down
2 changes: 1 addition & 1 deletion load-tests/src/multichain/test_plans/short.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1 +1 @@
0,0s;1,120s;0,0s
0,0s;5,30s;5,20m;0,0s
Loading