@@ -2,9 +2,10 @@ use starknet::ContractAddress;
2
2
3
3
use snforge_std :: {declare, ContractClassTrait };
4
4
5
- use cairo_template :: hello_starknet :: {
6
- IHelloStarknetSafeDispatcher , IHelloStarknetSafeDispatcherTrait
7
- };
5
+ use cairo_template :: hello_starknet :: IHelloStarknetSafeDispatcher ;
6
+ use cairo_template :: hello_starknet :: IHelloStarknetSafeDispatcherTrait ;
7
+ use cairo_template :: hello_starknet :: IHelloStarknetDispatcher ;
8
+ use cairo_template :: hello_starknet :: IHelloStarknetDispatcherTrait ;
8
9
9
10
fn deploy_contract (name : felt252 ) -> ContractAddress {
10
11
let contract = declare (name );
@@ -15,14 +16,14 @@ fn deploy_contract(name: felt252) -> ContractAddress {
15
16
fn test_increase_balance () {
16
17
let contract_address = deploy_contract (' HelloStarknet' );
17
18
18
- let safe_dispatcher = IHelloStarknetSafeDispatcher { contract_address };
19
+ let dispatcher = IHelloStarknetDispatcher { contract_address };
19
20
20
- let balance_before = safe_dispatcher . get_balance () . unwrap ();
21
+ let balance_before = dispatcher . get_balance ();
21
22
assert (balance_before == 0 , ' Invalid balance' );
22
23
23
- safe_dispatcher . increase_balance (42 ) . unwrap ( );
24
+ dispatcher . increase_balance (42 );
24
25
25
- let balance_after = safe_dispatcher . get_balance () . unwrap ();
26
+ let balance_after = dispatcher . get_balance ();
26
27
assert (balance_after == 42 , ' Invalid balance' );
27
28
}
28
29
@@ -32,9 +33,11 @@ fn test_cannot_increase_balance_with_zero_value() {
32
33
33
34
let safe_dispatcher = IHelloStarknetSafeDispatcher { contract_address };
34
35
36
+ #[feature(" safe_dispatcher" )]
35
37
let balance_before = safe_dispatcher . get_balance (). unwrap ();
36
38
assert (balance_before == 0 , ' Invalid balance' );
37
39
40
+ #[feature(" safe_dispatcher" )]
38
41
match safe_dispatcher . increase_balance (0 ) {
39
42
Result :: Ok (_ ) => panic_with_felt252 (' Should have panicked' ),
40
43
Result :: Err (panic_data ) => {
0 commit comments