ERC20 totalSupply changed from public property to a function

Fixes #434
OpenZeppelin · Jan 11, 2018 · 1972eab · 1972eab
1 parent 9469772
commit 1972eab
Show file tree

Hide file tree

Showing 11 changed files with 26 additions and 9 deletions.
diff --git a/contracts/examples/SimpleToken.sol b/contracts/examples/SimpleToken.sol
@@ -22,7 +22,7 @@ contract SimpleToken is StandardToken {
    * @dev Constructor that gives msg.sender all of existing tokens.
    */
   function SimpleToken() public {
-    totalSupply = INITIAL_SUPPLY;
+    totalSupply_ = INITIAL_SUPPLY;
     balances[msg.sender] = INITIAL_SUPPLY;
     Transfer(0x0, msg.sender, INITIAL_SUPPLY);
   }

diff --git a/contracts/mocks/BasicTokenMock.sol b/contracts/mocks/BasicTokenMock.sol
@@ -9,7 +9,7 @@ contract BasicTokenMock is BasicToken {
 
   function BasicTokenMock(address initialAccount, uint256 initialBalance) public {
     balances[initialAccount] = initialBalance;
-    totalSupply = initialBalance;
+    totalSupply_ = initialBalance;
   }
 
 }
diff --git a/contracts/mocks/BurnableTokenMock.sol b/contracts/mocks/BurnableTokenMock.sol
@@ -6,7 +6,7 @@ contract BurnableTokenMock is BurnableToken {
 
   function BurnableTokenMock(address initialAccount, uint initialBalance) public {
     balances[initialAccount] = initialBalance;
-    totalSupply = initialBalance;
+    totalSupply_ = initialBalance;
   }
 
 }
diff --git a/contracts/mocks/ERC23TokenMock.sol b/contracts/mocks/ERC23TokenMock.sol
@@ -12,7 +12,7 @@ contract ERC23TokenMock is BasicToken {
 
   function ERC23TokenMock(address initialAccount, uint256 initialBalance) public {
     balances[initialAccount] = initialBalance;
-    totalSupply = initialBalance;
+    totalSupply_ = initialBalance;
   }
 
   // ERC23 compatible transfer function (except the name)

diff --git a/contracts/mocks/SafeERC20Helper.sol b/contracts/mocks/SafeERC20Helper.sol
@@ -4,6 +4,10 @@ import '../token/ERC20.sol';
 import '../token/SafeERC20.sol';
 
 contract ERC20FailingMock is ERC20 {
+  function totalSupply() public view returns (uint256) {
+    return 0;
+  }
+
   function transfer(address, uint256) public returns (bool) {
     return false;
   }
@@ -26,6 +30,10 @@ contract ERC20FailingMock is ERC20 {
 }
 
 contract ERC20SucceedingMock is ERC20 {
+  function totalSupply() public view returns (uint256) {
+    return 0;
+  }
+
   function transfer(address, uint256) public returns (bool) {
     return true;
   }

diff --git a/contracts/mocks/StandardTokenMock.sol b/contracts/mocks/StandardTokenMock.sol
@@ -9,7 +9,7 @@ contract StandardTokenMock is StandardToken {
 
   function StandardTokenMock(address initialAccount, uint256 initialBalance) public {
     balances[initialAccount] = initialBalance;
-    totalSupply = initialBalance;
+    totalSupply_ = initialBalance;
   }
 
 }
diff --git a/contracts/token/BasicToken.sol b/contracts/token/BasicToken.sol
@@ -14,6 +14,15 @@ contract BasicToken is ERC20Basic {
 
   mapping(address => uint256) balances;
 
+  uint256 totalSupply_;
+
+  /**
+  * @dev total number of tokens in existence
+  */
+  function totalSupply() public view returns (uint256) {
+    return totalSupply_;
+  }
+
   /**
   * @dev transfer token for a specified address
   * @param _to The address to transfer to.

diff --git a/contracts/token/BurnableToken.sol b/contracts/token/BurnableToken.sol
@@ -21,7 +21,7 @@ contract BurnableToken is BasicToken {
 
         address burner = msg.sender;
         balances[burner] = balances[burner].sub(_value);
-        totalSupply = totalSupply.sub(_value);
+        totalSupply_ = totalSupply_.sub(_value);
         Burn(burner, _value);
     }
 }
diff --git a/contracts/token/CappedToken.sol b/contracts/token/CappedToken.sol
@@ -23,7 +23,7 @@ contract CappedToken is MintableToken {
    * @return A boolean that indicates if the operation was successful.
    */
   function mint(address _to, uint256 _amount) onlyOwner canMint public returns (bool) {
-    require(totalSupply.add(_amount) <= cap);
+    require(totalSupply_.add(_amount) <= cap);
 
     return super.mint(_to, _amount);
   }

diff --git a/contracts/token/ERC20Basic.sol b/contracts/token/ERC20Basic.sol
@@ -7,7 +7,7 @@ pragma solidity ^0.4.18;
  * @dev see https://github.com/ethereum/EIPs/issues/179
  */
 contract ERC20Basic {
-  uint256 public totalSupply;
+  function totalSupply() public view returns (uint256);
   function balanceOf(address who) public view returns (uint256);
   function transfer(address to, uint256 value) public returns (bool);
   event Transfer(address indexed from, address indexed to, uint256 value);

diff --git a/contracts/token/MintableToken.sol b/contracts/token/MintableToken.sol
@@ -32,7 +32,7 @@ contract MintableToken is StandardToken, Ownable {
    * @return A boolean that indicates if the operation was successful.
    */
   function mint(address _to, uint256 _amount) onlyOwner canMint public returns (bool) {
-    totalSupply = totalSupply.add(_amount);
+    totalSupply_ = totalSupply_.add(_amount);
     balances[_to] = balances[_to].add(_amount);
     Mint(_to, _amount);
     Transfer(address(0), _to, _amount);