diff --git a/contracts/vault/src/test.rs b/contracts/vault/src/test.rs index 376b725a9..8ca2af415 100644 --- a/contracts/vault/src/test.rs +++ b/contracts/vault/src/test.rs @@ -2352,6 +2352,7 @@ fn test_whitelist_consistency_with_set_strategy() { assert_eq!(vault.strategy().unwrap(), benji_strategy); // Removal of the active vault strategy is rejected (#1637) + assert!(vault.try_whitelist_strategy(&benji_strategy, &false).is_err()); assert!(vault .try_whitelist_strategy(&benji_strategy, &false) .is_err()); diff --git a/contracts/vault/src/whitelist.rs b/contracts/vault/src/whitelist.rs index 80bbf90f6..c34964f54 100644 --- a/contracts/vault/src/whitelist.rs +++ b/contracts/vault/src/whitelist.rs @@ -266,6 +266,8 @@ mod tests { ); assert_eq!(read_registration_state(&env, &active), Some(STATE_ACTIVE)); + assert_eq!(SecureWhitelist::add_strategy(&env, &admin, &retired), Ok(())); + assert_eq!(retire_strategy_internal(&env, &retired, None), Ok(STATE_RETIRED)); assert_eq!( SecureWhitelist::add_strategy(&env, &admin, &retired), Ok(())