Custom assertion methods return Result

tests/integration/common/errors.rs

@@ -54,3 +54,5 @@ impl fmt::Display for RpcError {
         write!(f, "{self:?}")
     }
 }
+
+impl std::error::Error for RpcError {}

tests/integration/common/utils.rs

@@ -1,4 +1,4 @@
-use std::fmt::LowerHex;
+use std::fmt::{Debug, LowerHex};
 use std::fs;
 use std::path::Path;
 use std::process::{Child, Command};
@@ -17,7 +17,7 @@ use starknet_rs_core::types::contract::{CompiledClass, SierraClass};
 use starknet_rs_core::types::{
     BlockId, BlockTag, ContractClass, ContractExecutionError, DeployAccountTransactionResult,
     ExecutionResult, FeeEstimate, Felt, FlattenedSierraClass, FunctionCall,
-    InnerContractExecutionError, ResourceBounds, ResourceBoundsMapping,
+    InnerContractExecutionError, ResourceBounds, ResourceBoundsMapping, TransactionReceipt,
 };
 use starknet_rs_core::utils::{
     UdcUniqueSettings, get_selector_from_name, get_udc_deployed_address,
@@ -102,16 +102,20 @@ pub fn get_timestamp_asserter_contract_artifacts() -> SierraWithCasmHash {
     )
 }
 
-pub async fn assert_tx_succeeded_accepted<T: Provider>(tx_hash: &Felt, client: &T) {
-    let receipt = client.get_transaction_receipt(tx_hash).await.unwrap().receipt;
+pub async fn assert_tx_succeeded_accepted<T: Provider>(
+    tx_hash: &Felt,
+    client: &T,
+) -> Result<(), anyhow::Error> {
+    let receipt = client.get_transaction_receipt(tx_hash).await?.receipt;
+
     match receipt.execution_result() {
         ExecutionResult::Succeeded => (),
-        other => panic!("Should have succeeded; got: {other:?}"),
+        other => anyhow::bail!("Tx {tx_hash:#x} should have succeeded; got: {other:?}"),
     }
 
     match receipt.finality_status() {
-        starknet_rs_core::types::TransactionFinalityStatus::AcceptedOnL2 => (),
-        other => panic!("Should have been accepted on L2; got: {other:?}"),
+        starknet_rs_core::types::TransactionFinalityStatus::AcceptedOnL2 => Ok(()),
+        other => anyhow::bail!("Tx {tx_hash:#x} should have been accepted on L2; got: {other:?}"),
     }
 }
 
@@ -129,42 +133,44 @@ pub async fn assert_tx_succeeded_pre_confirmed<T: Provider>(_tx_hash: &Felt, _cl
     // }
 }
 
-pub async fn get_contract_balance(devnet: &BackgroundDevnet, contract_address: Felt) -> Felt {
+pub async fn get_contract_balance(
+    devnet: &BackgroundDevnet,
+    contract_address: Felt,
+) -> Result<Felt, anyhow::Error> {
     get_contract_balance_by_block_id(devnet, contract_address, BlockId::Tag(BlockTag::Latest)).await
 }
 
 pub async fn get_contract_balance_by_block_id(
     devnet: &BackgroundDevnet,
     contract_address: Felt,
     block_id: BlockId,
-) -> Felt {
+) -> Result<Felt, anyhow::Error> {
     let contract_call = FunctionCall {
         contract_address,
         entry_point_selector: get_selector_from_name("get_balance").unwrap(),
         calldata: vec![],
     };
     match devnet.json_rpc_client.call(contract_call, block_id).await {
-        Ok(res) => {
-            assert_eq!(res.len(), 1);
-            res[0]
-        }
-        Err(e) => panic!("Call failed: {e}"),
+        Ok(res) if res.len() == 1 => Ok(res[0]),
+        other => anyhow::bail!("Expected a single-felt result; got: {other:?}"),
     }
 }
 
 pub async fn assert_tx_reverted<T: Provider>(
     tx_hash: &Felt,
     client: &T,
     expected_failure_reasons: &[&str],
-) {
-    let receipt = client.get_transaction_receipt(tx_hash).await.unwrap().receipt;
+) -> Result<(), anyhow::Error> {
+    let receipt: TransactionReceipt = client.get_transaction_receipt(tx_hash).await?.receipt;
+
     match receipt.execution_result() {
         ExecutionResult::Reverted { reason } => {
             for expected_reason in expected_failure_reasons {
-                assert_contains(reason, expected_reason);
+                assert_contains(reason, expected_reason)?
             }
+            Ok(())
         }
-        other => panic!("Should have reverted; got: {other:?}; receipt: {receipt:?}"),
+        other => anyhow::bail!("Should have reverted; got: {other:?}; receipt: {receipt:?}"),
     }
 }
 
@@ -215,7 +221,7 @@ async fn send_ctrl_c_signal(process: &Child) {
 
 fn take_abi_from_json(value: &mut serde_json::Value) -> Result<serde_json::Value, anyhow::Error> {
     let abi_jsonified = value["abi"].take();
-    assert_ne!(abi_jsonified, serde_json::json!(null));
+    anyhow::ensure!(abi_jsonified != serde_json::json!(null));
     Ok(serde_json::from_str(abi_jsonified.as_str().unwrap())?)
 }
 
@@ -231,8 +237,8 @@ pub fn assert_cairo1_classes_equal(
     let abi_a = take_abi_from_json(&mut class_a_jsonified)?;
     let abi_b = take_abi_from_json(&mut class_b_jsonified)?;
 
-    assert_eq!(class_a_jsonified, class_b_jsonified);
-    assert_eq!(abi_a, abi_b);
+    anyhow::ensure!(class_a_jsonified == class_b_jsonified);
+    anyhow::ensure!(abi_a == abi_b);
 
     Ok(())
 }
@@ -429,21 +435,26 @@ pub async fn deploy_argent_account(
 
     let account_address = deployment.address();
     devnet.mint(account_address, u128::MAX).await;
-    let deployment_result = deployment.send().await.map_err(|e| anyhow::anyhow!("{e:?}"))?;
+    let deployment_result = deployment.send().await?;
 
     Ok((deployment_result, signer))
 }
 
 /// Assert that the set of elements of `iterable1` is a subset of the elements of `iterable2` and
 /// vice versa.
-pub fn assert_equal_elements<T>(iterable1: &[T], iterable2: &[T])
+pub fn assert_equal_elements<T>(iterable1: &[T], iterable2: &[T]) -> Result<(), anyhow::Error>
 where
-    T: PartialEq,
+    T: PartialEq + Debug,
 {
-    assert_eq!(iterable1.len(), iterable2.len());
+    if iterable1.len() != iterable2.len() {
+        anyhow::bail!("Length mismatch: left = {}, right = {}", iterable1.len(), iterable2.len());
+    }
     for e in iterable1 {
-        assert!(iterable2.contains(e));
+        if !iterable2.contains(e) {
+            anyhow::bail!("Element {:?} from left not found in right", e);
+        }
     }
+    Ok(())
 }
 
 pub fn felt_to_u256(f: Felt) -> U256 {
@@ -474,23 +485,29 @@ pub fn extract_json_rpc_error(error: ProviderError) -> Result<JsonRpcError, anyh
     }
 }
 
-pub fn assert_json_rpc_errors_equal(e1: JsonRpcError, e2: JsonRpcError) {
-    assert_eq!((e1.code, e1.message, e1.data), (e2.code, e2.message, e2.data));
+pub fn assert_json_rpc_errors_equal(
+    e1: JsonRpcError,
+    e2: JsonRpcError,
+) -> Result<(), anyhow::Error> {
+    anyhow::ensure!((e1.code, e1.message, e1.data) == (e2.code, e2.message, e2.data));
+    Ok(())
 }
 
 /// Extract the message that is encapsulated inside the provided error.
-pub fn extract_message_error(error: &ContractExecutionError) -> &String {
+pub fn extract_message_error(error: &ContractExecutionError) -> Result<&String, anyhow::Error> {
     match error {
-        ContractExecutionError::Message(msg) => msg,
-        other => panic!("Unexpected error: {other:?}"),
+        ContractExecutionError::Message(msg) => Ok(msg),
+        other => anyhow::bail!("Unexpected error: {other:?}"),
     }
 }
 
 /// Extract the error that is nested inside the provided `error`.
-pub fn extract_nested_error(error: &ContractExecutionError) -> &InnerContractExecutionError {
+pub fn extract_nested_error(
+    error: &ContractExecutionError,
+) -> Result<&InnerContractExecutionError, anyhow::Error> {
     match error {
-        ContractExecutionError::Nested(nested) => nested,
-        other => panic!("Unexpected error: {other:?}"),
+        ContractExecutionError::Nested(nested) => Ok(nested),
+        other => anyhow::bail!("Unexpected error: {other:?}"),
     }
 }
 
@@ -569,23 +586,25 @@ pub async fn receive_notification(
     expected_subscription_id: SubscriptionId,
 ) -> Result<serde_json::Value, anyhow::Error> {
     let mut notification = receive_rpc_via_ws(ws).await?;
-    assert_eq!(notification["jsonrpc"], "2.0");
-    assert_eq!(notification["method"], method);
-    assert_eq!(
+    anyhow::ensure!(notification["jsonrpc"] == "2.0");
+    anyhow::ensure!(notification["method"] == method);
+    anyhow::ensure!(
         notification["params"]["subscription_id"]
             .as_str()
             .ok_or(anyhow::Error::msg("No Subscription ID in notification"))?
-            .to_string(),
-        expected_subscription_id
+            .to_string()
+            == expected_subscription_id
     );
     Ok(notification["params"]["result"].take())
 }
 
-pub async fn assert_no_notifications(ws: &mut WebSocketStream<MaybeTlsStream<TcpStream>>) {
+pub async fn assert_no_notifications(
+    ws: &mut WebSocketStream<MaybeTlsStream<TcpStream>>,
+) -> Result<(), anyhow::Error> {
     match receive_rpc_via_ws(ws).await {
-        Ok(resp) => panic!("Expected no notifications; found: {resp}"),
-        Err(e) if e.to_string().contains("deadline has elapsed") => { /* expected */ }
-        Err(e) => panic!("Expected to error out due to empty channel; found: {e}"),
+        Ok(resp) => anyhow::bail!("Expected no notifications; found: {resp}"),
+        Err(e) if e.to_string().contains("deadline has elapsed") => Ok(()), /* expected */
+        Err(e) => anyhow::bail!("Expected to error out due to empty channel; found: {e}"),
     }
 }
 
@@ -672,15 +691,16 @@ impl From<FeeEstimate> for LocalFee {
 }
 
 /// Panics if `text` does not contain `pattern`
-pub fn assert_contains(text: &str, pattern: &str) {
+pub fn assert_contains(text: &str, pattern: &str) -> Result<(), anyhow::Error> {
     if !text.contains(pattern) {
-        panic!(
+        anyhow::bail!(
             "Failed content assertion!
     Pattern: '{pattern}'
     not present in
     Text: '{text}'"
         );
     }
+    Ok(())
 }
 
 /// Set time and generate a new block

tests/integration/general_rpc_tests.rs

@@ -88,7 +88,8 @@ async fn storage_proof_request_should_always_return_error() {
         Err(e) => assert_json_rpc_errors_equal(
             extract_json_rpc_error(e).unwrap(),
             JsonRpcError { code: 42, message: devnet_storage_proof_msg.into(), data: None },
-        ),
+        )
+        .unwrap(),
         other => panic!("Unexpected result: {other:?}"),
     }
 }

tests/integration/get_transaction_by_hash.rs

@@ -38,7 +38,9 @@ async fn get_declare_v3_transaction_by_hash_happy_path() {
         .await
         .unwrap();
 
-    assert_tx_succeeded_accepted(&declare_result.transaction_hash, &devnet.json_rpc_client).await;
+    assert_tx_succeeded_accepted(&declare_result.transaction_hash, &devnet.json_rpc_client)
+        .await
+        .unwrap();
 }
 
 #[tokio::test]
@@ -67,7 +69,8 @@ async fn get_deploy_account_transaction_by_hash_happy_path() {
 
     let deploy_account_result = deployment.send().await.unwrap();
     assert_tx_succeeded_accepted(&deploy_account_result.transaction_hash, &devnet.json_rpc_client)
-        .await;
+        .await
+        .unwrap();
 }
 
 #[tokio::test]
@@ -97,7 +100,9 @@ async fn get_invoke_v3_transaction_by_hash_happy_path() {
         .await
         .unwrap();
 
-    assert_tx_succeeded_accepted(&invoke_tx_result.transaction_hash, &devnet.json_rpc_client).await;
+    assert_tx_succeeded_accepted(&invoke_tx_result.transaction_hash, &devnet.json_rpc_client)
+        .await
+        .unwrap();
 }
 
 #[tokio::test]

tests/integration/get_transaction_receipt_by_hash.rs

@@ -242,7 +242,7 @@ async fn reverted_invoke_transaction_receipt() {
         TransactionReceipt::Invoke(receipt) => {
             match receipt.execution_result {
                 starknet_rs_core::types::ExecutionResult::Reverted { reason } => {
-                    assert_contains(&reason, "Insufficient max L2Gas");
+                    assert_contains(&reason, "Insufficient max L2Gas").unwrap();
                 }
                 _ => panic!("Invalid receipt {:?}", receipt),
             }