feat: update new_register_balances_query_msg

neutron-org · Feb 22, 2024 · a1dcc8c · a1dcc8c
1 parent d9a6078
commit a1dcc8c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/contracts/neutron_interchain_queries/src/contract.rs b/contracts/neutron_interchain_queries/src/contract.rs
@@ -41,7 +41,7 @@ use neutron_sdk::interchain_queries::v045::queries::{
 };
 use neutron_sdk::interchain_queries::v045::types::{COSMOS_SDK_TRANSFER_MSG_URL, RECIPIENT_FIELD};
 use neutron_sdk::interchain_queries::v045::{
-    new_register_balance_query_msg, new_register_bank_total_supply_query_msg,
+    new_register_balances_query_msg, new_register_bank_total_supply_query_msg,
     new_register_delegator_delegations_query_msg, new_register_distribution_fee_pool_query_msg,
     new_register_gov_proposal_query_msg, new_register_staking_validators_query_msg,
     new_register_transfers_query_msg,
@@ -141,7 +141,7 @@ pub fn register_balance_query(
     denom: String,
     update_period: u64,
 ) -> NeutronResult<Response<NeutronMsg>> {
-    let msg = new_register_balance_query_msg(connection_id, addr, denom, update_period)?;
+    let msg = new_register_balances_query_msg(connection_id, addr, vec![denom], update_period)?;
 
     Ok(Response::new().add_message(msg))
 }

diff --git a/contracts/neutron_validator_test/src/contract.rs b/contracts/neutron_validator_test/src/contract.rs
@@ -32,7 +32,6 @@ use schemars::JsonSchema;
 use serde::{Deserialize, Serialize};
 
 use crate::msg::{ExecuteMsg, InstantiateMsg, QueryMsg};
-use neutron_sdk::bindings::msg::{IbcFee, MsgSubmitTxResponse, NeutronMsg};
 use neutron_sdk::bindings::query::{
     NeutronQuery, QueryInterchainAccountAddressResponse, QueryRegisteredQueryResponse,
 };
@@ -41,13 +40,15 @@ use neutron_sdk::interchain_queries::queries::get_registered_query;
 use neutron_sdk::interchain_queries::types::{
     TransactionFilterItem, TransactionFilterOp, TransactionFilterValue,
 };
+use neutron_sdk::interchain_queries::v045::new_register_transfers_query_msg;
 use neutron_sdk::interchain_queries::v045::queries::query_balance;
 use neutron_sdk::interchain_queries::v045::types::{COSMOS_SDK_TRANSFER_MSG_URL, RECIPIENT_FIELD};
-use neutron_sdk::interchain_queries::v045::{
-    new_register_balance_query_msg, new_register_transfers_query_msg,
-};
 use neutron_sdk::interchain_txs::helpers::{decode_acknowledgement_response, get_port_id};
 use neutron_sdk::sudo::msg::{RequestPacket, SudoMsg};
+use neutron_sdk::{
+    bindings::msg::{IbcFee, MsgSubmitTxResponse, NeutronMsg},
+    interchain_queries::v045::new_register_balances_query_msg,
+};
 use neutron_sdk::{NeutronError, NeutronResult};
 
 use crate::storage::{
@@ -412,7 +413,7 @@ pub fn register_balance_query(
     denom: String,
     update_period: u64,
 ) -> NeutronResult<Response<NeutronMsg>> {
-    let msg = new_register_balance_query_msg(connection_id, addr, denom, update_period)?;
+    let msg = new_register_balances_query_msg(connection_id, addr, vec![denom], update_period)?;
 
     Ok(Response::new().add_message(msg))
 }