Add simple test with cheat

Author: cptartur

crates/forge/tests/data/simple_package_with_cheats/Scarb.toml

@@ -0,0 +1,15 @@
+[package]
+name = "simple_package_with_cheats"
+version = "0.1.0"
+edition = "2024_07"
+
+[dependencies]
+starknet = "2.12.0"
+
+[dev-dependencies]
+snforge_std = { path = "../../../../../snforge_std" }
+
+[[target.starknet-contract]]
+
+[tool.snforge]
+exit_first = false

crates/forge/tests/data/simple_package_with_cheats/src/lib.cairo

@@ -0,0 +1,35 @@
+#[starknet::interface]
+pub trait IHelloStarknet<TContractState> {
+    fn increase_balance(ref self: TContractState, amount: felt252);
+    fn get_balance(self: @TContractState) -> felt252;
+    fn get_block_number(self: @TContractState) -> u64;
+}
+
+#[starknet::contract]
+pub mod HelloStarknet {
+    use core::array::ArrayTrait;
+    use starknet::get_block_number;
+    use starknet::storage::{StoragePointerReadAccess, StoragePointerWriteAccess};
+
+    #[storage]
+    struct Storage {
+        balance: felt252,
+    }
+
+    #[abi(embed_v0)]
+    impl IHelloStarknetImpl of super::IHelloStarknet<ContractState> {
+        // Increases the balance by the given amount
+        fn increase_balance(ref self: ContractState, amount: felt252) {
+            self.balance.write(self.balance.read() + amount);
+        }
+
+        // Returns the current balance
+        fn get_balance(self: @ContractState) -> felt252 {
+            self.balance.read()
+        }
+
+        fn get_block_number(self: @ContractState) -> u64 {
+            get_block_number()
+        }
+    }
+}

crates/forge/tests/data/simple_package_with_cheats/tests/contract.cairo

@@ -0,0 +1,23 @@
+use core::array::ArrayTrait;
+use core::result::ResultTrait;
+use simple_package_with_cheats::{IHelloStarknetDispatcher, IHelloStarknetDispatcherTrait};
+use snforge_std::cheatcodes::contract_class::DeclareResultTrait;
+use snforge_std::{ContractClassTrait, declare, start_cheat_block_number_global};
+
+#[test]
+fn call_and_invoke() {
+    let contract = declare("HelloStarknet").unwrap().contract_class();
+    let constructor_calldata = @ArrayTrait::new();
+    let (contract_address, _) = contract.deploy(constructor_calldata).unwrap();
+    let dispatcher = IHelloStarknetDispatcher { contract_address };
+
+    let block_number = dispatcher.get_block_number();
+    println!("block number {}", block_number);
+    // TODO investigate why the default is 2000
+    assert(block_number == 2000, 'block_info == 2000');
+
+    start_cheat_block_number_global(123);
+
+    let block_number = dispatcher.get_block_number();
+    assert(block_number == 123, 'block_info == 123');
+}

crates/forge/tests/e2e/running.rs

@@ -51,6 +51,27 @@ fn simple_package() {
     );
 }
 
+#[test]
+fn simple_package_with_cheats() {
+    let temp = setup_package("simple_package_with_cheats");
+    let output = test_runner(&temp).assert().code(0);
+
+    assert_stdout_contains(
+        output,
+        indoc! {r"
+    [..]Compiling[..]
+    [..]Finished[..]
+
+
+    Collected 1 test(s) from simple_package_with_cheats package
+    Running 0 test(s) from src/
+    Running 1 test(s) from tests/
+    [PASS] simple_package_with_cheats_integrationtest::contract::call_and_invoke [..]
+    "},
+    );
+}
+
+
 #[test]
 fn simple_package_with_git_dependency() {
     let temp = setup_package("simple_package");