Add tests for library call

Author: cptartur

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

@@ -8,13 +8,17 @@ pub trait IHelloStarknet<TContractState> {
 #[starknet::interface]
 pub trait IHelloStarknetProxy<TContractState> {
     fn get_block_number(self: @TContractState) -> u64;
+    fn get_block_number_library_call(self: @TContractState) -> u64;
 }
 
 #[starknet::contract]
 pub mod HelloStarknetProxy {
     use starknet::ContractAddress;
     use starknet::storage::{StoragePointerReadAccess, StoragePointerWriteAccess};
-    use crate::{IHelloStarknetDispatcher, IHelloStarknetDispatcherTrait};
+    use starknet::syscalls::get_class_hash_at_syscall;
+    use crate::{
+        IHelloStarknetDispatcher, IHelloStarknetDispatcherTrait, IHelloStarknetLibraryDispatcher,
+    };
 
     #[storage]
     struct Storage {
@@ -33,6 +37,13 @@ pub mod HelloStarknetProxy {
             let proxied = IHelloStarknetDispatcher { contract_address: address };
             proxied.get_block_number()
         }
+
+        fn get_block_number_library_call(self: @ContractState) -> u64 {
+            let address = self.address.read();
+            let class_hash = get_class_hash_at_syscall(address).unwrap();
+            let library_dispatcher = IHelloStarknetLibraryDispatcher { class_hash };
+            library_dispatcher.get_block_number()
+        }
     }
 }
 

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

@@ -46,3 +46,24 @@ fn call_and_invoke_proxy() {
     let block_number = dispatcher.get_block_number();
     assert(block_number == 123, 'block_number == 123');
 }
+
+#[test]
+fn call_and_invoke_library_call() {
+    let contract = declare("HelloStarknet").unwrap().contract_class();
+    let constructor_calldata = @ArrayTrait::new();
+    let (contract_address, _) = contract.deploy(constructor_calldata).unwrap();
+
+    let proxy_contract = declare("HelloStarknetProxy").unwrap().contract_class();
+    let mut constructor_calldata = ArrayTrait::new();
+    contract_address.serialize(ref constructor_calldata);
+    let (proxy_contract_address, _) = proxy_contract.deploy(@constructor_calldata).unwrap();
+    let dispatcher = IHelloStarknetProxyDispatcher { contract_address: proxy_contract_address };
+
+    let block_number = dispatcher.get_block_number_library_call();
+    assert(block_number == 2000, 'block_number == 2000');
+
+    start_cheat_block_number_global(123);
+
+    let block_number = dispatcher.get_block_number_library_call();
+    assert(block_number == 123, 'block_number == 123');
+}

crates/forge/tests/e2e/running.rs

@@ -63,12 +63,13 @@ fn simple_package_with_cheats() {
     [..]Finished[..]
 
 
-    Collected 2 test(s) from simple_package_with_cheats package
+    Collected 3 test(s) from simple_package_with_cheats package
     Running 0 test(s) from src/
-    Running 2 test(s) from tests/
+    Running 3 test(s) from tests/
     [PASS] simple_package_with_cheats_integrationtest::contract::call_and_invoke [..]
     [PASS] simple_package_with_cheats_integrationtest::contract::call_and_invoke_proxy [..]
-    Tests: 2 passed, 0 failed, 0 ignored, 0 filtered out
+    [PASS] simple_package_with_cheats_integrationtest::contract::call_and_invoke_library_call [..]
+    Tests: 3 passed, 0 failed, 0 ignored, 0 filtered out
     "},
     );
 }