Save mint lock state on collection creation (#56)

KompleTeam · Nov 22, 2022 · 63f7456 · 63f7456
1 parent 2af5817
commit 63f7456
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 14 deletions.
diff --git a/contracts/modules/mint/src/contract.rs b/contracts/modules/mint/src/contract.rs
@@ -266,6 +266,8 @@ pub fn execute_create_collection(
 
     COLLECTION_INFO.save(deps.storage, collection_id, &collection_info)?;
 
+    MINT_LOCKS.save(deps.storage, collection_id, &false)?;
+
     let cw20_address = match fund_info.cw20_address {
         Some(addr) => Some(deps.api.addr_validate(&addr)?),
         None => None,
@@ -414,8 +416,8 @@ fn execute_mint(
     collection_id: u32,
     metadata_id: Option<u32>,
 ) -> Result<Response, ContractError> {
-    let mint_lock = MINT_LOCKS.may_load(deps.storage, collection_id)?;
-    if let Some(_mint_lock) = mint_lock {
+    let mint_lock = MINT_LOCKS.load(deps.storage, collection_id)?;
+    if mint_lock {
         return Err(ContractError::LockedMint {});
     }
 
@@ -713,8 +715,8 @@ fn execute_receive(
             collection_id,
             metadata_id,
         } => {
-            let mint_lock = MINT_LOCKS.may_load(deps.storage, collection_id)?;
-            if let Some(_mint_lock) = mint_lock {
+            let mint_lock = MINT_LOCKS.load(deps.storage, collection_id)?;
+            if mint_lock {
                 return Err(ContractError::LockedMint {});
             }
 
@@ -1002,11 +1004,8 @@ fn query_creators(deps: Deps) -> StdResult<ResponseWrapper<Vec<String>>> {
 }
 
 fn query_mint_lock(deps: Deps, collection_id: u32) -> StdResult<ResponseWrapper<bool>> {
-    let mint_lock = MINT_LOCKS.may_load(deps.storage, collection_id)?;
-    Ok(ResponseWrapper::new(
-        "mint_lock",
-        mint_lock.unwrap_or(false),
-    ))
+    let mint_lock = MINT_LOCKS.load(deps.storage, collection_id)?;
+    Ok(ResponseWrapper::new("mint_lock", mint_lock))
 }
 
 #[cfg_attr(not(feature = "library"), entry_point)]

diff --git a/contracts/modules/mint/tests/integration.rs b/contracts/modules/mint/tests/integration.rs
@@ -256,11 +256,6 @@ mod actions {
                 .query_wasm_smart(minter_addr.clone(), &msg)
                 .unwrap();
             assert_eq!(response.data, true);
-
-            let msg = QueryMsg::MintLock { collection_id: 2 };
-            let response: ResponseWrapper<bool> =
-                app.wrap().query_wasm_smart(minter_addr, &msg).unwrap();
-            assert_eq!(response.data, false);
         }
     }