From 962f8a9d0f79f14ad9bd7a564e950a0cab6f1c95 Mon Sep 17 00:00:00 2001 From: Joseph Izang Date: Thu, 4 May 2023 00:13:39 +0100 Subject: [PATCH] add 3 test cases for contract flattener --- .../src/tests/contract_flattener.test.ts | 212 +++++++++++++++++- 1 file changed, 211 insertions(+), 1 deletion(-) diff --git a/apps/remix-ide-e2e/src/tests/contract_flattener.test.ts b/apps/remix-ide-e2e/src/tests/contract_flattener.test.ts index c0e18bdbea..54a251b82a 100644 --- a/apps/remix-ide-e2e/src/tests/contract_flattener.test.ts +++ b/apps/remix-ide-e2e/src/tests/contract_flattener.test.ts @@ -5,5 +5,215 @@ import init from '../helpers/init' module.exports = { before: function (browser: NightwatchBrowser, done: VoidFunction) { init(browser, done) + }, + '@sources': () => sources, + 'Should flatten contract after creation': function (browser: NightwatchBrowser) { + browser.addFile('TestContract.sol', sources[0]['TestContract.sol']) + .waitForElementVisible('*[data-id="treeViewLitreeViewItemTestContract.sol"]') + .pause(7000) + .click('*[data-id="treeViewLitreeViewItemTestContract.sol"]') + .rightClick('*[data-id="treeViewLitreeViewItemTestContract.sol"]') + .click('*[id="menuitemflattenacontract"]') + .waitForElementVisible('*[data-id="treeViewLitreeViewItemTestContract_flattened.sol"]') + }, + 'Should not be able to flatten contract without imports': function (browser: NightwatchBrowser) { + browser.click('*[data-id="treeViewLitreeViewItemcontracts"]') + .waitForElementVisible('*[data-id="treeViewLitreeViewItemcontracts/3_Ballot.sol"]') + .pause(1000) + .click('*[data-id="treeViewLitreeViewItemcontracts/3_Ballot.sol"]') + .rightClick('*[data-id="treeViewLitreeViewItemcontracts/3_Ballot.sol"]') + .click('*[id="menuitemflattenacontract"]') + .waitForElementNotPresent('*[data-id="treeViewLitreeViewItemcontracts/3_Ballot_flattened.sol"]') + }, + 'Should not flatten contract with errors in syntax': function (browser: NightwatchBrowser) { + browser.addFile('samplecontract.sol', { content }) + .waitForElementVisible('*[data-id="treeViewLitreeViewItemsamplecontract.sol"]') + .click('*[data-id="treeViewLitreeViewItemsamplecontract.sol"]') + .pause(1000) + .rightClick('*[data-id="treeViewLitreeViewItemsamplecontract.sol"]') + .click('*[id="menuitemflattenacontract"]') + .waitForElementNotPresent('*[data-id="treeViewLitreeViewItemsamplecontract_flattened.sol"]') } -} \ No newline at end of file +} + +const sources = [ + { + 'TestContract.sol': { + content: ` + // SPDX-License-Identifier: MIT + pragma solidity ^0.8.9; + + import "@openzeppelin/contracts-upgradeable/token/ERC721/ERC721Upgradeable.sol"; + import "@openzeppelin/contracts-upgradeable/token/ERC721/extensions/ERC721BurnableUpgradeable.sol"; + import "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol"; + import "@openzeppelin/contracts-upgradeable/proxy/utils/Initializable.sol"; + import "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol"; + + contract MyToken is Initializable, ERC721Upgradeable, ERC721BurnableUpgradeable, OwnableUpgradeable, UUPSUpgradeable { + /// @custom:oz-upgrades-unsafe-allow constructor + constructor() { + _disableInitializers(); + } + + function initialize() initializer public { + __ERC721_init("MyToken", "MTK"); + __ERC721Burnable_init(); + __Ownable_init(); + __UUPSUpgradeable_init(); + } + + function safeMint(address to, uint256 tokenId) public onlyOwner { + _safeMint(to, tokenId); + } + + function _authorizeUpgrade(address newImplementation) + internal + onlyOwner + override + {} + } + ` + }, +} +] + +const content = ` + // SPDX-License-Identifier: GPL-3.0 + + pragma solidity >=0.7.0 <0.9.0; + + /** + * @title Ballot + * @dev Implements voting process along with vote delegation + */ + contract Voting { + + struct Voter { + uint weight // weight is accumulated by delegation <-- missing semicolon + bool voted // if true, that person already voted <-- missing semicolon + address delegate; // person delegated to + uint vote; // index of the voted proposal + } + + struct Proposal { + // If you can limit the length to a certain number of bytes, + // always use one of bytes1 to bytes32 because they are much cheaper + bytes32 name; // short name (up to 32 bytes) + uint voteCount; // number of accumulated votes + } + + address public chairperson; + + mapping(address => Voter) public voters; + + Proposal[] public proposals; + + /** + * @dev Create a new ballot to choose one of 'proposalNames'. + * @param proposalNames names of proposals + */ + constructor(bytes32[] memory proposalNames) { + chairperson = msg.sender; + voters[chairperson].weight = 1; + + for (uint i = 0; i < proposalNames.length; i++) { + // 'Proposal({...})' creates a temporary + // Proposal object and 'proposals.push(...)' + // appends it to the end of 'proposals'. + proposals.push(Proposal({ + name: proposalNames[i], + voteCount: 0 + })); + } + } + + /** + * @dev Give 'voter' the right to vote on this ballot. May only be called by 'chairperson'. + * @param voter address of voter + */ + function giveRightToVote(address voter) public { + require( + msg.sender == chairperson, + "Only chairperson can give right to vote." + ); + require( + !voters[voter].voted, + "The voter already voted." + ); + require(voters[voter].weight == 0); + voters[voter].weight = 1; + } + + /** + * @dev Delegate your vote to the voter 'to'. + * @param to address to which vote is delegated + */ + function delegate(address to) public { + Voter storage sender = voters[msg.sender]; + require(!sender.voted, "You already voted."); + require(to != msg.sender, "Self-delegation is disallowed."); + + while (voters[to].delegate != address(0)) { + to = voters[to].delegate; + + // We found a loop in the delegation, not allowed. + require(to != msg.sender, "Found loop in delegation."); + } + sender.voted = true; + sender.delegate = to; + Voter storage delegate_ = voters[to]; + if (delegate_.voted) { + // If the delegate already voted, + // directly add to the number of votes + proposals[delegate_.vote].voteCount += sender.weight; + } else { + // If the delegate did not vote yet, + // add to her weight. + delegate_.weight += sender.weight; + } + } + + /** + * @dev Give your vote (including votes delegated to you) to proposal 'proposals[proposal].name'. + * @param proposal index of proposal in the proposals array + */ + function vote(uint proposal) public { + Voter storage sender = voters[msg.sender]; + require(sender.weight != 0, "Has no right to vote"); + require(!sender.voted, "Already voted."); + sender.voted = true; + sender.vote = proposal; + + // If 'proposal' is out of the range of the array, + // this will throw automatically and revert all + // changes. + proposals[proposal].voteCount += sender.weight; + } + + /** + * @dev Computes the winning proposal taking all previous votes into account. + * @return winningProposal_ index of winning proposal in the proposals array + */ + function winningProposal() public view + returns (uint winningProposal_) + { + uint winningVoteCount = 0; + for (uint p = 0; p < proposals.length; p++) { + if (proposals[p].voteCount > winningVoteCount) { + winningVoteCount = proposals[p].voteCount; + winningProposal_ = p; + } + } + } + + /** + * @dev Calls winningProposal() function to get the index of the winner contained in the proposals array and then + * @return winnerName_ the name of the winner + */ + function winnerName() public view + returns (bytes32 winnerName_) + { + winnerName_ = proposals[winningProposal()].name; + } + } + ` \ No newline at end of file