Skip to content

Commit 27de36e

Browse files
starknet_os_flow_tests: split get_class_info_of_cairo0_contract into function
1 parent 55e910c commit 27de36e

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;
@@ -433,17 +434,21 @@ pub(crate) fn divide_vec_into_n_parts<T>(mut vec: Vec<T>, n: usize) -> Vec<Vec<T
433434
items_per_part
434435
}
435436

437+
pub(crate) fn get_class_info_of_cairo0_contract(
438+
contract_class: DeprecatedContractClass,
439+
) -> ClassInfo {
440+
let abi_length = contract_class.abi.as_ref().unwrap().len();
441+
ClassInfo {
442+
contract_class: ContractClass::V0(contract_class),
443+
sierra_program_length: 0,
444+
abi_length,
445+
sierra_version: SierraVersion::DEPRECATED,
446+
}
447+
}
448+
436449
pub(crate) fn get_class_info_of_feature_contract(feature_contract: FeatureContract) -> ClassInfo {
437450
match feature_contract.get_class() {
438-
ContractClass::V0(contract_class) => {
439-
let abi_length = contract_class.abi.as_ref().unwrap().len();
440-
ClassInfo {
441-
contract_class: ContractClass::V0(contract_class),
442-
sierra_program_length: 0,
443-
abi_length,
444-
sierra_version: SierraVersion::DEPRECATED,
445-
}
446-
}
451+
ContractClass::V0(contract_class) => get_class_info_of_cairo0_contract(contract_class),
447452
ContractClass::V1((contract_class, sierra_version)) => {
448453
let sierra = feature_contract.get_sierra();
449454
ClassInfo {

0 commit comments

Comments
 (0)