@0x:contracts-integrations Addressed review feedback

2019-10-04 15:42:36 -07:00
parent 24e4567b25
commit 3c6957095d
18 changed files with 78 additions and 99 deletions
--- a/contracts/staking/contracts/test/TestStaking.sol
+++ b/contracts/staking/contracts/test/TestStaking.sol
@@ -40,6 +40,22 @@ contract TestStaking is
        testZrxVaultAddress = zrxVaultAddress;
    }

+    /// @dev Sets the weth contract address.
+    /// @param wethAddress The address of the weth contract.
+    function setWethContract(address wethAddress)
+        external
+    {
+        testWethAddress = wethAddress;
+    }
+
+    /// @dev Sets the zrx vault address.
+    /// @param zrxVaultAddress The address of a zrx vault.
+    function setZrxVault(address zrxVaultAddress)
+        external
+    {
+        testZrxVaultAddress = zrxVaultAddress;
+    }
+
    /// @dev Overridden to use testWethAddress;
    function getWethContract()
        public