test: Add sudo to custom example

examples/contracts/custom/src/contract.rs

@@ -1,28 +1,34 @@
 use cosmwasm_std::{CosmosMsg, QueryRequest, Response, StdResult};
-use sylvia::types::{ExecCtx, InstantiateCtx, QueryCtx};
+use cw_storage_plus::Item;
+use sylvia::types::{ExecCtx, InstantiateCtx, QueryCtx, SudoCtx};
 use sylvia::{contract, schemars};
 
 #[cfg(not(feature = "library"))]
 use sylvia::entry_points;
 
 use crate::messages::{CountResponse, CounterMsg, CounterQuery};
 
-pub struct CustomContract;
+pub struct CustomContract {
+    pub(crate) sudo_counter: Item<'static, u64>,
+}
 
 #[cfg_attr(not(feature = "library"), entry_points)]
 #[contract]
 #[messages(cw1 as Cw1: custom(msg, query))]
 #[sv::custom(query=CounterQuery, msg=CounterMsg)]
 impl CustomContract {
     pub const fn new() -> Self {
-        Self
+        Self {
+            sudo_counter: Item::new("sudo_counter"),
+        }
     }
 
     #[msg(instantiate)]
     pub fn instantiate(
         &self,
-        _ctx: InstantiateCtx<CounterQuery>,
+        ctx: InstantiateCtx<CounterQuery>,
     ) -> StdResult<Response<CounterMsg>> {
+        self.sudo_counter.save(ctx.deps.storage, &0)?;
         Ok(Response::default())
     }
 
@@ -42,4 +48,21 @@ impl CustomContract {
 
         Ok(resp)
     }
+
+    #[msg(query)]
+    pub fn sudo_counter(&self, ctx: QueryCtx<CounterQuery>) -> StdResult<CountResponse> {
+        let count = self.sudo_counter.load(ctx.deps.storage)?;
+
+        Ok(CountResponse { count })
+    }
+
+    #[msg(sudo)]
+    pub fn increment_sudo_counter(
+        &self,
+        ctx: SudoCtx<CounterQuery>,
+    ) -> StdResult<Response<CounterMsg>> {
+        self.sudo_counter
+            .update(ctx.deps.storage, |value| -> StdResult<_> { Ok(value + 1) })?;
+        Ok(Response::new())
+    }
 }

examples/contracts/custom/src/multitest/tests.rs

@@ -24,4 +24,9 @@ fn test_custom() {
 
     let count = contract.query_custom().unwrap().count;
     assert_eq!(count, 1);
+
+    contract.increment_sudo_counter().unwrap();
+
+    let count = contract.sudo_counter().unwrap().count;
+    assert_eq!(count, 1);
 }