fix(gateway): set contract version in migrate endpoint (#706)

axelarnetwork · Nov 19, 2024 · 25c3edf · 25c3edf
1 parent fd509f3
commit 25c3edf
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/contracts/gateway/src/contract.rs b/contracts/gateway/src/contract.rs
@@ -17,6 +17,7 @@ mod query;
 
 const CONTRACT_NAME: &str = env!("CARGO_PKG_NAME");
 const CONTRACT_VERSION: &str = env!("CARGO_PKG_VERSION");
+const BASE_VERSION: &str = "1.0.0";
 
 #[derive(thiserror::Error, Debug)]
 pub enum Error {
@@ -40,10 +41,13 @@ pub enum Error {
 
 #[cfg_attr(not(feature = "library"), entry_point)]
 pub fn migrate(
-    _deps: DepsMut,
+    deps: DepsMut,
     _env: Env,
     _msg: Empty,
 ) -> Result<Response, axelar_wasm_std::error::ContractError> {
+    cw2::assert_contract_version(deps.storage, CONTRACT_NAME, BASE_VERSION)?;
+    cw2::set_contract_version(deps.storage, CONTRACT_NAME, CONTRACT_VERSION)?;
+
     Ok(Response::default())
 }