Skip to content

Commit d3f102a

Browse files
starknet_os_flow_tests: split get_class_info_of_cairo0_contract into function
1 parent e7a7b06 commit d3f102a

File tree

1 file changed

+14
-9
lines changed
  • crates/starknet_os_flow_tests/src

1 file changed

+14
-9
lines changed

crates/starknet_os_flow_tests/src/utils.rs

Lines changed: 14 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ use starknet_api::core::{
2727
GLOBAL_STATE_VERSION,
2828
};
2929
use starknet_api::declare_tx_args;
30+
use starknet_api::deprecated_contract_class::ContractClass as DeprecatedContractClass;
3031
use starknet_api::executable_transaction::{AccountTransaction, DeclareTransaction};
3132
use starknet_api::state::StorageKey;
3233
use starknet_api::test_utils::declare::declare_tx;
@@ -407,18 +408,22 @@ pub(crate) fn divide_vec_into_n_parts<T>(mut vec: Vec<T>, n: usize) -> Vec<Vec<T
407408
items_per_part
408409
}
409410

411+
pub(crate) fn get_class_info_of_cairo0_contract(
412+
contract_class: DeprecatedContractClass,
413+
) -> ClassInfo {
414+
let abi_length = contract_class.abi.as_ref().unwrap().len();
415+
ClassInfo {
416+
contract_class: ContractClass::V0(contract_class),
417+
sierra_program_length: 0,
418+
abi_length,
419+
sierra_version: SierraVersion::DEPRECATED,
420+
}
421+
}
422+
410423
// TODO(Nimrod): Consider moving it to a method of `FeatureContract`.
411424
pub(crate) fn get_class_info_of_feature_contract(feature_contract: FeatureContract) -> ClassInfo {
412425
match feature_contract.get_class() {
413-
ContractClass::V0(contract_class) => {
414-
let abi_length = contract_class.abi.as_ref().unwrap().len();
415-
ClassInfo {
416-
contract_class: ContractClass::V0(contract_class),
417-
sierra_program_length: 0,
418-
abi_length,
419-
sierra_version: SierraVersion::DEPRECATED,
420-
}
421-
}
426+
ContractClass::V0(contract_class) => get_class_info_of_cairo0_contract(contract_class),
422427
ContractClass::V1((contract_class, sierra_version)) => {
423428
let sierra = feature_contract.get_sierra();
424429
ClassInfo {

0 commit comments

Comments
 (0)