test(erc20): fix tests for ERC20 after refactor

OpenZeppelin · Apr 4, 2024 · dc04ea3 · dc04ea3
1 parent a4b80d4
commit dc04ea3
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/contracts/src/erc20/extensions/burnable.rs b/contracts/src/erc20/extensions/burnable.rs
@@ -24,6 +24,8 @@ impl ERC20Burnable {
     ///
     /// * `value` - Amount to be burnt.
     ///
+    /// # Errors
+    ///
     /// If the `from` address doesn't have enough tokens, then the error
     /// [`Error::InsufficientBalance`] is returned.
     pub fn burn(&mut self, value: U256) -> Result<(), Error> {
@@ -40,6 +42,8 @@ impl ERC20Burnable {
     /// * `account` - Owner's address.
     /// * `value` - Amount to be burnt.
     ///
+    /// # Errors
+    ///
     /// If not enough allowance is available, then the error
     /// [`Error::InsufficientAllowance`] is returned.
     /// * If the `from` address is `Address::ZERO`, then the error

diff --git a/contracts/src/erc20/mod.rs b/contracts/src/erc20/mod.rs
@@ -530,6 +530,11 @@ mod tests {
     fn transfer_from_errors_when_invalid_sender(contract: ERC20) {
         let alice = address!("A11CEacF9aa32246d767FCCD72e02d6bCbcC375d");
         let one = U256::from(1);
+        contract
+            ._allowances
+            .setter(Address::ZERO)
+            .setter(msg::sender())
+            .set(one);
         let result = contract.transfer_from(Address::ZERO, alice, one);
         assert!(matches!(result, Err(Error::InvalidSender(_))));
     }
@@ -538,6 +543,7 @@ mod tests {
     fn transfer_from_errors_when_invalid_receiver(contract: ERC20) {
         let alice = address!("A11CEacF9aa32246d767FCCD72e02d6bCbcC375d");
         let one = U256::from(1);
+        contract._allowances.setter(alice).setter(msg::sender()).set(one);
         let result = contract.transfer_from(alice, Address::ZERO, one);
         assert!(matches!(result, Err(Error::InvalidReceiver(_))));
     }