added delegation module query to tests

DA0-DA0 · Feb 13, 2025 · 60e1b71 · 60e1b71
1 parent c2c7160
commit 60e1b71
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/contracts/delegation/dao-vote-delegation/Cargo.toml b/contracts/delegation/dao-vote-delegation/Cargo.toml
@@ -46,3 +46,4 @@ dao-voting-token-staked = { workspace = true, features = ["library"] }
 dao-voting-cw721-staked = { workspace = true, features = ["library"] }
 dao-testing = { workspace = true }
 dao-proposal-sudo = { workspace = true }
+dao-proposal-single = { workspace = true }
diff --git a/contracts/delegation/dao-vote-delegation/src/testing/suite/cw4.rs b/contracts/delegation/dao-vote-delegation/src/testing/suite/cw4.rs
@@ -1,6 +1,6 @@
 use std::ops::{Deref, DerefMut};
 
-use cosmwasm_std::Decimal;
+use cosmwasm_std::{Addr, Decimal};
 use dao_testing::{Cw4TestDao, DaoTestingSuite};
 
 use super::base::DaoVoteDelegationTestingSuiteBase;
@@ -110,5 +110,19 @@ impl Cw4DaoVoteDelegationTestingSuite {
 
         // set the delegation module for all proposal modules
         self.set_delegation_module(&dao, &delegation_addr);
+
+        // ensure delegation modules are set
+        dao.proposal_modules.iter().for_each(|(_, module)| {
+            let delegation_module = self
+                .querier()
+                .query_wasm_smart::<Option<Addr>>(
+                    module,
+                    &dao_proposal_single::msg::QueryMsg::DelegationModule {},
+                )
+                .unwrap()
+                .unwrap();
+
+            assert_eq!(delegation_module, Addr::unchecked(delegation_addr.clone()));
+        });
     }
 }