dynamic gas price: Add integration test (#3788)

* Dynamic gas fee test

* Add dynamic gas fee test to CI

* Fix issue with nested calls to `block_on` (#3789)

* Fix test for dynamic gas price

---------

Co-authored-by: Luca Joss <luca@informal.systems>

.github/workflows/integration.yaml

@@ -79,7 +79,7 @@ jobs:
             command: osmosisd
             account_prefix: osmo
             native_token: stake
-            features: ''
+            features: dynamic-gas-fee
           - package: juno
             command: junod
             account_prefix: juno

crates/relayer/src/chain/cosmos.rs

@@ -109,6 +109,7 @@ use crate::{chain::client::ClientSettings, config::Error as ConfigError};
 
 use self::types::app_state::GenesisAppState;
 use self::version::Specs;
+use self::{gas::dynamic_gas_price, types::gas::GasConfig};
 
 pub mod batch;
 pub mod client;
@@ -492,6 +493,13 @@ impl CosmosSdkChain {
         Ok(min_gas_price)
     }
 
+    pub fn dynamic_gas_price(&self) -> GasPrice {
+        let gas_config = GasConfig::from(self.config());
+
+        self.rt
+            .block_on(dynamic_gas_price(&gas_config, &self.config.rpc_addr))
+    }
+
     /// The unbonding period of this chain
     pub fn unbonding_period(&self) -> Result<Duration, Error> {
         crate::time!(

crates/relayer/src/chain/cosmos/batch.rs

@@ -102,7 +102,7 @@ pub async fn send_batched_messages_and_wait_check_tx(
         return Ok(Vec::new());
     }
 
-    let batches = batch_messages(config, key_pair, account, tx_memo, messages)?;
+    let batches = batch_messages(config, key_pair, account, tx_memo, messages).await?;
 
     let mut responses = Vec::new();
 
@@ -132,7 +132,7 @@ async fn send_messages_as_batches(
 
     let message_count = messages.len();
 
-    let batches = batch_messages(config, key_pair, account, tx_memo, messages)?;
+    let batches = batch_messages(config, key_pair, account, tx_memo, messages).await?;
 
     debug!(
         "sending {} messages as {} batches to chain {} in parallel",
@@ -173,7 +173,7 @@ async fn sequential_send_messages_as_batches(
 
     let message_count = messages.len();
 
-    let batches = batch_messages(config, key_pair, account, tx_memo, messages)?;
+    let batches = batch_messages(config, key_pair, account, tx_memo, messages).await?;
 
     debug!(
         "sending {} messages as {} batches to chain {} in serial",
@@ -238,7 +238,7 @@ fn response_to_tx_sync_result(
     }
 }
 
-fn batch_messages(
+async fn batch_messages(
     config: &TxConfig,
     key_pair: &Secp256k1KeyPair,
     account: &Account,
@@ -257,7 +257,9 @@ fn batch_messages(
         &config.gas_config,
         config.gas_config.max_gas,
         &config.rpc_address,
-    );
+    )
+    .await;
+
     let tx_metrics = encoded_tx_metrics(config, key_pair, account, tx_memo, &[], &max_fee)?;
     let tx_envelope_len = tx_metrics.envelope_len;
     let empty_body_len = tx_metrics.body_bytes_len;
@@ -368,14 +370,15 @@ mod tests {
         (tx_config, key_pair, account)
     }
 
-    #[test]
-    fn batch_does_not_exceed_max_tx_size() {
+    #[tokio::test]
+    async fn batch_does_not_exceed_max_tx_size() {
         let (config, key_pair, account) = test_fixture();
         let max_fee = gas_amount_to_fee(
             &config.gas_config,
             config.gas_config.max_gas,
             &config.rpc_address,
-        );
+        )
+        .await;
         let mut messages = vec![Any {
             type_url: "/example.Baz".into(),
             value: vec![0; 2],
@@ -412,6 +415,7 @@ mod tests {
                 &memo,
                 messages.clone(),
             )
+            .await
             .unwrap();
 
             assert_eq!(batches.len(), 2);
@@ -428,8 +432,8 @@ mod tests {
         }
     }
 
-    #[test]
-    fn batch_error_on_oversized_message() {
+    #[tokio::test]
+    async fn batch_error_on_oversized_message() {
         const MAX_TX_SIZE: usize = 203;
 
         let (config, key_pair, account) = test_fixture();
@@ -450,6 +454,7 @@ mod tests {
             &memo,
             messages.clone(),
         )
+        .await
         .unwrap();
 
         assert_eq!(batches.len(), 1);
@@ -459,20 +464,21 @@ mod tests {
             &config.gas_config,
             config.gas_config.max_gas,
             &config.rpc_address,
-        );
+        )
+        .await;
         let tx_bytes =
             sign_and_encode_tx(&config, &key_pair, &account, &memo, &batches[0], &max_fee).unwrap();
         assert_eq!(tx_bytes.len(), MAX_TX_SIZE);
 
         limited_config.max_tx_size = MaxTxSize::new(MAX_TX_SIZE - 1).unwrap();
 
-        let res = batch_messages(&limited_config, &key_pair, &account, &memo, messages);
+        let res = batch_messages(&limited_config, &key_pair, &account, &memo, messages).await;
 
         assert!(res.is_err());
     }
 
-    #[test]
-    fn test_batches_are_structured_appropriately_per_max_msg_num() {
+    #[tokio::test]
+    async fn test_batches_are_structured_appropriately_per_max_msg_num() {
         let (config, key_pair, account) = test_fixture();
 
         // Ensure that when MaxMsgNum is 1, the resulting batch
@@ -511,6 +517,7 @@ mod tests {
             &Memo::new("").unwrap(),
             messages.clone(),
         )
+        .await
         .unwrap();
 
         assert_eq!(batches.len(), 5);
@@ -529,14 +536,15 @@ mod tests {
             &Memo::new("").unwrap(),
             messages,
         )
+        .await
         .unwrap();
 
         assert_eq!(batches.len(), 1);
         assert_eq!(batches[0].len(), 5);
     }
 
-    #[test]
-    fn test_batches_are_structured_appropriately_per_max_tx_size() {
+    #[tokio::test]
+    async fn test_batches_are_structured_appropriately_per_max_tx_size() {
         const MAX_TX_SIZE: usize = 198;
 
         let (config, key_pair, account) = test_fixture();
@@ -577,6 +585,7 @@ mod tests {
             &memo,
             messages.clone(),
         )
+        .await
         .unwrap();
 
         assert_eq!(batches.len(), 5);
@@ -585,7 +594,8 @@ mod tests {
             &config.gas_config,
             config.gas_config.max_gas,
             &config.rpc_address,
-        );
+        )
+        .await;
 
         for batch in batches {
             assert_eq!(batch.len(), 1);
@@ -605,15 +615,16 @@ mod tests {
             &Memo::new("").unwrap(),
             messages,
         )
+        .await
         .unwrap();
 
         assert_eq!(batches.len(), 1);
         assert_eq!(batches[0].len(), 5);
     }
 
-    #[test]
+    #[tokio::test]
     #[should_panic(expected = "`max_msg_num` must be greater than or equal to 1, found 0")]
-    fn test_max_msg_num_of_zero_panics() {
+    async fn test_max_msg_num_of_zero_panics() {
         let (mut config, key_pair, account) = test_fixture();
         config.max_msg_num = MaxMsgNum::new(0).unwrap();
         let _batches = batch_messages(
@@ -622,6 +633,7 @@ mod tests {
             &account,
             &Memo::new("").unwrap(),
             vec![],
-        );
+        )
+        .await;
     }
 }

crates/relayer/src/chain/cosmos/estimate.rs

@@ -88,7 +88,7 @@ async fn estimate_fee_with_tx(
         ));
     }
 
-    let adjusted_fee = gas_amount_to_fee(gas_config, estimated_gas, rpc_address);
+    let adjusted_fee = gas_amount_to_fee(gas_config, estimated_gas, rpc_address).await;
 
     debug!(
         id = %chain_id,

crates/relayer/src/chain/cosmos/gas.rs

@@ -5,21 +5,19 @@ use num_bigint::BigInt;
 use num_rational::BigRational;
 use tendermint_rpc::Url;
 
+use super::query_eip_base_fee;
 use crate::chain::cosmos::types::gas::GasConfig;
 use crate::config::GasPrice;
-use crate::util::block_on;
-
-use super::query_eip_base_fee;
 
-pub fn gas_amount_to_fee(config: &GasConfig, gas_amount: u64, rpc_address: &Url) -> Fee {
+pub async fn gas_amount_to_fee(config: &GasConfig, gas_amount: u64, rpc_address: &Url) -> Fee {
     let adjusted_gas_limit = adjust_estimated_gas(AdjustGas {
         gas_multiplier: config.gas_multiplier,
         max_gas: config.max_gas,
         gas_amount,
     });
 
     // The fee in coins based on gas amount
-    let dynamic_gas_price = dynamic_gas_price(config, rpc_address);
+    let dynamic_gas_price = dynamic_gas_price(config, rpc_address).await;
     let amount = calculate_fee(adjusted_gas_limit, &dynamic_gas_price);
 
     Fee {
@@ -29,15 +27,17 @@ pub fn gas_amount_to_fee(config: &GasConfig, gas_amount: u64, rpc_address: &Url)
         granter: config.fee_granter.clone(),
     }
 }
-pub fn dynamic_gas_price(config: &GasConfig, rpc_address: &Url) -> GasPrice {
-    if let Some(dynamic_gas_price_multiplier) = config.dynamic_gas_price_multiplier {
-        let new_price = block_on(query_eip_base_fee(&rpc_address.to_string())).unwrap()
-            * dynamic_gas_price_multiplier;
 
-        GasPrice {
-            price: new_price,
-            denom: config.gas_price.denom.clone(),
-        }
+pub async fn dynamic_gas_price(config: &GasConfig, rpc_address: &Url) -> GasPrice {
+    if let Some(dynamic_gas_price_multiplier) = config.dynamic_gas_price_multiplier {
+        query_eip_base_fee(&rpc_address.to_string())
+            .await
+            .map(|base_fee| base_fee * dynamic_gas_price_multiplier)
+            .map(|new_price| GasPrice {
+                price: new_price,
+                denom: config.gas_price.denom.clone(),
+            })
+            .unwrap_or_else(|_| config.gas_price.clone())
     } else {
         config.gas_price.clone()
     }

crates/relayer/src/chain/cosmos/types/gas.rs

@@ -29,7 +29,7 @@ impl<'a> From<&'a CosmosSdkConfig> for GasConfig {
             gas_price: config.gas_price.clone(),
             max_fee: max_fee_from_config(config),
             fee_granter: fee_granter_from_config(config),
-            dynamic_gas_price_multiplier: None,
+            dynamic_gas_price_multiplier: config.dynamic_gas.dynamic_gas_price(),
         }
     }
 }

crates/relayer/src/lib.rs

@@ -1,6 +1,5 @@
 #![forbid(unsafe_code)]
 #![deny(
-    warnings,
     trivial_casts,
     trivial_numeric_casts,
     unused_import_braces,

tools/integration-test/Cargo.toml

@@ -42,6 +42,7 @@ interchain-security = []
 celestia = []
 async-icq = []
 juno = []
+dynamic-gas-fee = []
 
 [[bin]]
 name = "test_setup_with_binary_channel"