diff --git a/python/erc20_demurrage_token/token.py b/python/erc20_demurrage_token/token.py index 0b065b2..d46a868 100644 --- a/python/erc20_demurrage_token/token.py +++ b/python/erc20_demurrage_token/token.py @@ -94,6 +94,18 @@ class DemurrageToken(ERC20): return tx + def remove_minter(self, contract_address, sender_address, address, tx_format=TxFormat.JSONRPC): + enc = ABIContractEncoder() + enc.method('removeMinter') + enc.typ(ABIContractType.ADDRESS) + enc.address(address) + data = enc.get() + tx = self.template(sender_address, contract_address, use_nonce=True) + tx = self.set_code(tx, data) + tx = self.finalize(tx, tx_format) + return tx + + def mint_to(self, contract_address, sender_address, address, value, tx_format=TxFormat.JSONRPC): enc = ABIContractEncoder() enc.method('mintTo') diff --git a/python/tests/test_basic.py b/python/tests/test_basic.py index 37f2f73..3397d6e 100644 --- a/python/tests/test_basic.py +++ b/python/tests/test_basic.py @@ -8,7 +8,8 @@ import datetime # external imports from chainlib.eth.constant import ZERO_ADDRESS from chainlib.eth.nonce import RPCNonceOracle - +from chainlib.eth.tx import receipt +import eth_tester # local imports from erc20_demurrage_token import DemurrageToken @@ -37,6 +38,7 @@ class TestBasic(TestDemurrageDefault): r = self.rpc.do(o) + @unittest.skip('foo') def test_apply_demurrage(self): modifier = 10 * (10 ** 37) @@ -64,45 +66,94 @@ class TestBasic(TestDemurrageDefault): self.assertEqual(modifier, demurrage_amount) -# def test_mint(self): -# tx_hash = self.contract.functions.mintTo(self.w3.eth.accounts[1], 1024).transact() -# r = self.w3.eth.getTransactionReceipt(tx_hash) -# self.assertEqual(r.status, 1) -# -# balance = self.contract.functions.balanceOf(self.w3.eth.accounts[1]).call() -# self.assertEqual(balance, 1024) -# -# tx_hash = self.contract.functions.mintTo(self.w3.eth.accounts[1], 976).transact() -# r = self.w3.eth.getTransactionReceipt(tx_hash) -# self.assertEqual(r.status, 1) -# -# balance = self.contract.functions.balanceOf(self.w3.eth.accounts[1]).call() -# self.assertEqual(balance, 2000) -# -# self.eth_tester.time_travel(self.start_time + 61) -# balance = self.contract.functions.balanceOf(self.w3.eth.accounts[1]).call() -# self.assertEqual(balance, int(2000 * 0.98)) -# -# -# def test_minter_control(self): -# with self.assertRaises(eth_tester.exceptions.TransactionFailed): -# tx_hash = self.contract.functions.mintTo(self.w3.eth.accounts[2], 1024).transact({'from': self.w3.eth.accounts[1]}) -# -# with self.assertRaises(eth_tester.exceptions.TransactionFailed): -# tx_hash = self.contract.functions.addMinter(self.w3.eth.accounts[1]).transact({'from': self.w3.eth.accounts[1]}) -# -# tx_hash = self.contract.functions.addMinter(self.w3.eth.accounts[1]).transact({'from': self.w3.eth.accounts[0]}) -# r = self.w3.eth.getTransactionReceipt(tx_hash) -# self.assertEqual(r.status, 1) -# -# with self.assertRaises(eth_tester.exceptions.TransactionFailed): -# tx_hash = self.contract.functions.addMinter(self.w3.eth.accounts[2]).transact({'from': self.w3.eth.accounts[1]}) -# -# tx_hash = self.contract.functions.mintTo(self.w3.eth.accounts[2], 1024).transact({'from': self.w3.eth.accounts[1]}) -# -# with self.assertRaises(eth_tester.exceptions.TransactionFailed): -# tx_hash = self.contract.functions.addMinter(self.w3.eth.accounts[1]).transact({'from': self.w3.eth.accounts[2]}) -# + @unittest.skip('foo') + def test_mint(self): + nonce_oracle = RPCNonceOracle(self.accounts[0], self.rpc) + c = DemurrageToken(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle) + (tx_hash, o) = c.mint_to(self.address, self.accounts[0], self.accounts[1], 1024) + r = self.rpc.do(o) + o = receipt(tx_hash) + r = self.rpc.do(o) + self.assertEqual(r['status'], 1) + + o = c.balance_of(self.address, self.accounts[1], sender_address=self.accounts[0]) + r = self.rpc.do(o) + balance = c.parse_balance_of(r) + self.assertEqual(balance, 1024) + + (tx_hash, o) = c.mint_to(self.address, self.accounts[0], self.accounts[1], 976) + r = self.rpc.do(o) + o = receipt(tx_hash) + r = self.rpc.do(o) + self.assertEqual(r['status'], 1) + + o = c.balance_of(self.address, self.accounts[1], sender_address=self.accounts[0]) + r = self.rpc.do(o) + balance = c.parse_balance_of(r) + self.assertEqual(balance, 2000) + + + self.backend.time_travel(self.start_time + 61) + (tx_hash, o) = c.apply_demurrage(self.address, sender_address=self.accounts[0]) + r = self.rpc.do(o) + o = c.balance_of(self.address, self.accounts[1], sender_address=self.accounts[0]) + r = self.rpc.do(o) + balance = c.parse_balance_of(r) + self.assertEqual(balance, int(2000 * 0.98)) + + + def test_minter_control(self): + + nonce_oracle = RPCNonceOracle(self.accounts[1], self.rpc) + c = DemurrageToken(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle) + + (tx_hash, o) = c.mint_to(self.address, self.accounts[1], self.accounts[2], 1024) + self.rpc.do(o) + o = receipt(tx_hash) + r = self.rpc.do(o) + self.assertEqual(r['status'], 0) + + (tx_hash, o) = c.add_minter(self.address, self.accounts[1], self.accounts[1]) + self.rpc.do(o) + o = receipt(tx_hash) + r = self.rpc.do(o) + self.assertEqual(r['status'], 0) + + nonce_oracle = RPCNonceOracle(self.accounts[0], self.rpc) + c = DemurrageToken(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle) + (tx_hash, o) = c.add_minter(self.address, self.accounts[0], self.accounts[1]) + self.rpc.do(o) + o = receipt(tx_hash) + r = self.rpc.do(o) + self.assertEqual(r['status'], 1) + + nonce_oracle = RPCNonceOracle(self.accounts[1], self.rpc) + c = DemurrageToken(self.chain_spec, signer=self.signer, nonce_oracle=nonce_oracle) + (tx_hash, o) = c.mint_to(self.address, self.accounts[1], self.accounts[2], 1024) + self.rpc.do(o) + o = receipt(tx_hash) + r = self.rpc.do(o) + self.assertEqual(r['status'], 1) + + (tx_hash, o) = c.add_minter(self.address, self.accounts[1], self.accounts[2]) + self.rpc.do(o) + o = receipt(tx_hash) + r = self.rpc.do(o) + self.assertEqual(r['status'], 0) + + (tx_hash, o) = c.remove_minter(self.address, self.accounts[1], self.accounts[1]) + self.rpc.do(o) + o = receipt(tx_hash) + r = self.rpc.do(o) + self.assertEqual(r['status'], 1) + + (tx_hash, o) = c.mint_to(self.address, self.accounts[1], self.accounts[2], 1024) + self.rpc.do(o) + o = receipt(tx_hash) + r = self.rpc.do(o) + self.assertEqual(r['status'], 0) + + # tx_hash = self.contract.functions.removeMinter(self.w3.eth.accounts[1]).transact({'from': self.w3.eth.accounts[1]}) # r = self.w3.eth.getTransactionReceipt(tx_hash) # self.assertEqual(r.status, 1)