XC-229: Fix bitcoin_get_utxos mock network assert

dfinity · Dec 3, 2024 · 38ec6a7 · 38ec6a7
1 parent f55087f
commit 38ec6a7
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/rs/bitcoin/mock/src/main.rs b/rs/bitcoin/mock/src/main.rs
@@ -1,7 +1,7 @@
 use candid::candid_method;
 use ic_btc_interface::{
     Address, GetCurrentFeePercentilesRequest, GetUtxosRequest, GetUtxosResponse,
-    MillisatoshiPerByte, Network, Utxo, UtxosFilterInRequest,
+    MillisatoshiPerByte, Network, NetworkInRequest, Utxo, UtxosFilterInRequest,
 };
 use ic_cdk::api::management_canister::bitcoin::{BitcoinNetwork, SendTransactionRequest};
 use ic_cdk_macros::{init, update};
@@ -87,7 +87,7 @@ fn set_tip_height(tip_height: u32) {
 #[update]
 fn bitcoin_get_utxos(utxos_request: GetUtxosRequest) -> GetUtxosResponse {
     read_state(|s| {
-        assert_eq!(utxos_request.network, s.network.into());
+        assert_network_equivalent(&utxos_request.network, &s.network);
 
         let mut utxos = s
             .address_to_utxos
@@ -238,3 +238,17 @@ fn check_candid_interface_compatibility() {
         candid_parser::utils::CandidSource::File(old_interface.as_path()),
     );
 }
+
+fn assert_network_equivalent(lhs: &NetworkInRequest, rhs: &Network) {
+    match lhs {
+        &NetworkInRequest::Mainnet | &NetworkInRequest::mainnet => {
+            assert_eq!(rhs, &Network::Mainnet)
+        }
+        &NetworkInRequest::Testnet | &NetworkInRequest::testnet => {
+            assert_eq!(rhs, &Network::Testnet)
+        }
+        &NetworkInRequest::Regtest | &NetworkInRequest::regtest => {
+            assert_eq!(rhs, &Network::Regtest)
+        }
+    }
+}