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

feat: use all current identities for strategy test state transitions #1820

Merged
merged 4 commits into from
May 7, 2024
Merged

feat: use all current identities for strategy test state transitions #1820

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
3d7ac59
feat: use all current identities for ContractCreate operation
pauldelucia Apr 17, 2024
c50b158
Merge branch 'v1.0-dev' into feat/use-all-identities-in-strategies
pauldelucia May 7, 2024
a50f98a
finalize
pauldelucia May 7, 2024
91399c8
Merge branch 'v1.0-dev' into feat/use-all-identities-in-strategies
pauldelucia May 7, 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
41 changes: 27 additions & 14 deletions packages/strategy-tests/src/lib.rs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -42,8 +42,8 @@ use dpp::state_transition::data_contract_update_transition::methods::DataContrac
use operations::{DataContractUpdateAction, DataContractUpdateOp};
use platform_version::TryFromPlatformVersioned;
use rand::prelude::StdRng;
use rand::seq::SliceRandom;
use rand::Rng;
use tracing::error;
use std::collections::{BTreeMap, BTreeSet, HashMap, HashSet};
use bincode::{Decode, Encode};
use dpp::data_contract::document_type::accessors::DocumentTypeV0Getters;
Expand Down Expand Up @@ -413,7 +413,7 @@ impl Strategy {
) {
Ok(transitions) => transitions,
Err(e) => {
error!("identity_state_transitions_for_block error: {}", e);
tracing::error!("identity_state_transitions_for_block error: {}", e);
return (vec![], finalize_block_operations, vec![]);
}
};
Expand Down Expand Up @@ -1202,15 +1202,17 @@ impl Strategy {
) {
Ok(contract_factory) => contract_factory,
Err(e) => {
error!("Failed to get DataContractFactory while creating random contract: {e}");
tracing::error!("Failed to get DataContractFactory while creating random contract: {e}");
continue;
}
};

// Create `count` ContractCreate transitions and push to operations vec
for _ in 0..count {
// Get the contract owner_id from loaded_identity and loaded_identity nonce
let identity = &current_identities[0];
// Get the contract owner_id from a random current_identity and identity nonce
let identity = current_identities
.choose(rng)
.expect("Expected to get an identity for ContractCreate");
let identity_nonce =
identity_nonce_counter.entry(identity.id()).or_default();
*identity_nonce += 1;
Expand Down Expand Up @@ -1242,7 +1244,7 @@ impl Strategy {
))
}
Err(e) => {
error!(
tracing::error!(
"Error generating random document type: {:?}",
e
);
Expand All @@ -1261,7 +1263,7 @@ impl Strategy {
) {
Ok(contract) => contract,
Err(e) => {
error!("Failed to create random data contract: {e}");
tracing::error!("Failed to create random data contract: {e}");
continue;
}
};
Expand All @@ -1271,7 +1273,9 @@ impl Strategy {
{
Ok(transition) => transition,
Err(e) => {
error!("Failed to create ContractCreate transition: {e}");
tracing::error!(
"Failed to create ContractCreate transition: {e}"
);
continue;
}
};
Expand All @@ -1293,7 +1297,7 @@ impl Strategy {
fn(Identifier, String) -> Result<SecurityLevel, ProtocolError>,
>,
) {
error!("Error signing state transition: {:?}", e);
tracing::error!("Error signing state transition: {:?}", e);
}

operations.push(state_transition);
Expand Down Expand Up @@ -1327,7 +1331,13 @@ impl Strategy {
);
contract_ref.increment_version();

let identity = &current_identities[0];
let identity = current_identities
.iter()
.find(|identity| {
identity.id() == contract_ref.owner_id()
})
.or_else(|| current_identities.choose(rng)).expect("Expected to get an identity for DataContractUpdate");

let identity_contract_nonce = contract_nonce_counter
.entry((identity.id(), contract_ref.id()))
.or_default();
Expand All @@ -1336,7 +1346,7 @@ impl Strategy {
// Prepare the DataContractUpdateTransition with the updated contract_ref
match DataContractUpdateTransition::try_from_platform_versioned((DataContract::V0(contract_ref.clone()), *identity_contract_nonce), platform_version) {
Ok(data_contract_update_transition) => {
let identity_public_key = current_identities[0]
let identity_public_key = identity
.get_first_public_key_matching(
Purpose::AUTHENTICATION,
HashSet::from([SecurityLevel::CRITICAL]),
Expand All @@ -1356,15 +1366,18 @@ impl Strategy {

operations.push(state_transition);
},
Err(e) => error!("Error converting data contract to update transition: {:?}", e),
Err(e) => tracing::error!("Error converting data contract to update transition: {:?}", e),
}
}
Err(e) => {
error!("Error generating random document type: {:?}", e)
tracing::error!(
"Error generating random document type: {:?}",
e
)
}
}
} else {
// Handle the case where the contract is not found in known_contracts
tracing::error!("Contract not found in known_contracts");
}
}
}
Expand Down
Loading