def setUp(self):
# web3.py instance
self.contract_interface = constants.COMPILER.get_contract_interface(
"mycro.sol", "MycroCoin")
self.mycro_contract, self.mycro_address, self.mycro_instance = deploy_contract(
constants.W3, self.contract_interface)
def test_cannot_upgrade_non_mycro_dao(self):
__, __, new_mycro_instance = deploy_contract(self.w3,
self.contract_interface)
with self.assertRaises(TransactionFailed):
new_mycro_instance.upgradeFrom(
self.w3.eth.accounts[1],
transact={'from': self.w3.eth.accounts[0]})
def test_upgrade(self):
# register a project with the original mycro dao
project_address = constants.W3.eth.accounts[1]
self.mycro_instance.registerProject(
project_address, transact={'from': constants.W3.eth.accounts[0]})
# create, propose and vote for an ASC
merge_asc_interface = constants.COMPILER.get_contract_interface(
"merge_asc.sol", "MergeASC")
_, asc_address, _ = deploy_contract(constants.W3, merge_asc_interface,
constants.W3.eth.accounts[0], 15,
constants.PR_ID)
self.mycro_instance.propose(
asc_address, transact={'from': constants.W3.eth.accounts[0]})
# create a new mycro dao
new_mycro_contract, new_mycro_address, new_mycro_instance = deploy_contract(
constants.W3, self.contract_interface)
# we want to check if any RegisterProject events are emitted
event_filter = new_mycro_contract.events.RegisterProject.createFilter(
argument_filters={'filter': {
'event': 'RegisterProject'
}},
fromBlock=0)
# perform the upgrade
new_mycro_instance.upgradeFrom(
self.mycro_address,
transact={'from': constants.W3.eth.accounts[0]})
# want to make sure that BaseDao state is upgraded along the mycro specific state
self.assertEqual(new_mycro_instance.getProposals(),
self.mycro_instance.getProposals())
self.assertEqual(new_mycro_instance.getProjects(),
self.mycro_instance.getProjects())
# Projects which are copied during an upgrade should emit an event
self.assertEqual(1, len(event_filter.get_new_entries()))
def setUp(self):
self.merge_module_interface = constants.COMPILER.get_contract_interface(
"merge_module.sol", "MergeModule")
_, _, self.merge_module_instance = deploy_contract(
constants.W3, self.merge_module_interface)