Skip to content

Commit 48ce9b6

Browse files
committed
chore: add declare_tx fixture to compilation tests
1 parent fc5813a commit 48ce9b6

File tree

1 file changed

+23
-16
lines changed

1 file changed

+23
-16
lines changed

crates/gateway/src/compilation_test.rs

+23-16
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
use assert_matches::assert_matches;
22
use blockifier::execution::contract_class::ContractClass;
33
use cairo_lang_starknet_classes::allowed_libfuncs::AllowedLibfuncsError;
4-
use mempool_test_utils::starknet_api_test_utils::declare_tx;
4+
use mempool_test_utils::starknet_api_test_utils::declare_tx as rpc_declare_tx;
55
use rstest::{fixture, rstest};
66
use starknet_api::core::CompiledClassHash;
77
use starknet_api::rpc_transaction::{RPCDeclareTransaction, RPCTransaction};
@@ -15,13 +15,21 @@ fn gateway_compiler() -> GatewayCompiler {
1515
GatewayCompiler { config: Default::default() }
1616
}
1717

18+
#[fixture]
19+
fn declare_tx() -> RPCDeclareTransaction {
20+
assert_matches!(
21+
rpc_declare_tx(),
22+
RPCTransaction::Declare(declare_tx) => declare_tx
23+
)
24+
}
25+
1826
// TODO(Arni): Redesign this test once the compiler is passed with dependancy injection.
1927
#[rstest]
20-
fn test_compile_contract_class_compiled_class_hash_mismatch(gateway_compiler: GatewayCompiler) {
21-
let mut tx = assert_matches!(
22-
declare_tx(),
23-
RPCTransaction::Declare(RPCDeclareTransaction::V3(tx)) => tx
24-
);
28+
fn test_compile_contract_class_compiled_class_hash_mismatch(
29+
gateway_compiler: GatewayCompiler,
30+
declare_tx: RPCDeclareTransaction,
31+
) {
32+
let RPCDeclareTransaction::V3(mut tx) = declare_tx;
2533
let expected_hash = tx.compiled_class_hash;
2634
let wrong_supplied_hash = CompiledClassHash::default();
2735
tx.compiled_class_hash = wrong_supplied_hash;
@@ -36,11 +44,11 @@ fn test_compile_contract_class_compiled_class_hash_mismatch(gateway_compiler: Ga
3644
}
3745

3846
#[rstest]
39-
fn test_compile_contract_class_bad_sierra(gateway_compiler: GatewayCompiler) {
40-
let mut tx = assert_matches!(
41-
declare_tx(),
42-
RPCTransaction::Declare(RPCDeclareTransaction::V3(tx)) => tx
43-
);
47+
fn test_compile_contract_class_bad_sierra(
48+
gateway_compiler: GatewayCompiler,
49+
declare_tx: RPCDeclareTransaction,
50+
) {
51+
let RPCDeclareTransaction::V3(mut tx) = declare_tx;
4452
// Truncate the sierra program to trigger an error.
4553
tx.contract_class.sierra_program = tx.contract_class.sierra_program[..100].to_vec();
4654
let declare_tx = RPCDeclareTransaction::V3(tx);
@@ -55,11 +63,10 @@ fn test_compile_contract_class_bad_sierra(gateway_compiler: GatewayCompiler) {
5563
}
5664

5765
#[rstest]
58-
fn test_process_declare_tx_success(gateway_compiler: GatewayCompiler) {
59-
let declare_tx = assert_matches!(
60-
declare_tx(),
61-
RPCTransaction::Declare(declare_tx) => declare_tx
62-
);
66+
fn test_process_declare_tx_success(
67+
gateway_compiler: GatewayCompiler,
68+
declare_tx: RPCDeclareTransaction,
69+
) {
6370
let RPCDeclareTransaction::V3(declare_tx_v3) = &declare_tx;
6471
let contract_class = &declare_tx_v3.contract_class;
6572

0 commit comments

Comments
 (0)