From 4599ecd2e9c789bd9a651c43f1375a9b70540c46 Mon Sep 17 00:00:00 2001 From: Adam Dossa Date: Thu, 13 Jun 2019 20:03:06 -0400 Subject: [PATCH] Fix contract size issue. --- contracts/mocks/SecurityTokenRegistryMock.sol | 13 ++++--------- test/t_security_token_registry_proxy.js | 8 +++++--- 2 files changed, 9 insertions(+), 12 deletions(-) diff --git a/contracts/mocks/SecurityTokenRegistryMock.sol b/contracts/mocks/SecurityTokenRegistryMock.sol index 0c6bc85ff..6715ed03f 100644 --- a/contracts/mocks/SecurityTokenRegistryMock.sol +++ b/contracts/mocks/SecurityTokenRegistryMock.sol @@ -9,18 +9,13 @@ contract SecurityTokenRegistryMock is SecurityTokenRegistry { /// @notice It is a dummy function /// Alert! Alert! Do NOT use it for the mainnet release - modifier onlyOwnerOrSelf() { - require(msg.sender == owner() || msg.sender == address(this), "Only owner or self"); - _; - } - uint256 public someValue; - - function changeTheDeployedAddress(string memory _ticker, address _newSecurityTokenAddress) public { - set(Encoder.getKey("tickerToSecurityToken", _ticker), _newSecurityTokenAddress); + + function changeTheFee(uint256 _newFee) public { + set(STLAUNCHFEE, _newFee); } - function configure(uint256 _someValue) public onlyOwnerOrSelf { + function configure(uint256 _someValue) public { someValue = _someValue; } diff --git a/test/t_security_token_registry_proxy.js b/test/t_security_token_registry_proxy.js index 5219b877c..68069d0f5 100644 --- a/test/t_security_token_registry_proxy.js +++ b/test/t_security_token_registry_proxy.js @@ -274,9 +274,11 @@ contract("SecurityTokenRegistryProxy", async (accounts) => { }); it("Should alter the old storage", async () => { - await I_STRProxied.changeTheDeployedAddress(symbol, account_temp, { from: account_polymath }); - let _tokenAddress = await I_Getter.getSecurityTokenAddress.call(symbol); - assert.equal(_tokenAddress, account_temp, "Should match with the changed address"); + await I_STRProxied.changeTheFee(0, { from: account_polymath }); + let feesToken = await I_STRProxied.getFees.call("0xd677304bb45536bb7fdfa6b9e47a3c58fe413f9e8f01474b0a4b9c6e0275baf2"); + console.log(feesToken); + // assert.equal(feesToken[0].toString(), origPriceUSD.toString()); + // assert.equal(feesToken[1].toString(), origPricePOLY.toString()); }); });