ERC20Snapshot changes

contracts/mocks/ERC20SnapshotMock.sol

@@ -3,6 +3,7 @@
 pragma solidity ^0.8.0;
 
 import "../token/ERC20/extensions/ERC20Snapshot.sol";
+import "hardhat/console.sol";
 
 contract ERC20SnapshotMock is ERC20Snapshot {
     constructor(
@@ -11,6 +12,7 @@ contract ERC20SnapshotMock is ERC20Snapshot {
         address initialAccount,
         uint256 initialBalance
     ) ERC20(name, symbol) {
+        console.log("we'll mint");
         _mint(initialAccount, initialBalance);
     }
 

contracts/token/ERC20/extensions/ERC20Snapshot.sol

@@ -6,6 +6,7 @@ pragma solidity ^0.8.0;
 import "../ERC20.sol";
 import "../../../utils/Arrays.sol";
 import "../../../utils/Counters.sol";
+import "hardhat/console.sol";
 
 /**
  * @dev This contract extends an ERC20 token with a snapshot mechanism. When a snapshot is created, the balances and
@@ -127,13 +128,16 @@ abstract contract ERC20Snapshot is ERC20 {
     ) internal virtual override {
         if (from == address(0)) {
             // mint
+            console.log("mint", amount);
             _updateTotalSupplySnapshot();
         } else if (to == address(0)) {
             // burn
+            console.log("burn", amount);
             _updateAccountSnapshot(from);
             _updateTotalSupplySnapshot();
         } else {
             // transfer
+            console.log("transfer", amount);
             _updateAccountSnapshot(from);
             _updateAccountSnapshot(to);
         }

test/token/ERC20/extensions/ERC20Snapshot.test.js

@@ -129,10 +129,13 @@ contract('ERC20Snapshot', function (accounts) {
         });
       });
 
-      context('with balance changes after the snapshot', function () {
+      context.only('with balance changes after the snapshot', function () {
         beforeEach(async function () {
+          console.log('transfer 10');
           await this.token.transfer(recipient, new BN('10'), { from: initialHolder });
+          console.log('mint 50');
           await this.token.mint(other, new BN('50'));
+          console.log('burn 20');
           await this.token.burn(initialHolder, new BN('20'));
         });