115 lines
4.3 KiB
Python
115 lines
4.3 KiB
Python
from time import sleep
|
|
|
|
from client.client import Client
|
|
from dependency_config import container
|
|
from utils.utils import increaseTime
|
|
|
|
alice = Client(container.get_root('alice'), container.get_token('alice'))
|
|
bob = Client(container.get_root('bob'), container.get_token('bob'))
|
|
charlie = Client(container.get_root('charlie'), container.get_token('charlie'))
|
|
authority = Client(
|
|
container.get_root('authority'), container.get_token('authority')
|
|
)
|
|
w3 = alice.root_chain.w3 # get w3 instance
|
|
|
|
# Give alice 5 tokens
|
|
alice.register()
|
|
|
|
aliceTokensStart = alice.token_contract.balance_of()
|
|
print('Alice has {} tokens'.format(aliceTokensStart))
|
|
assert aliceTokensStart == 5, "START: Alice has incorrect number of tokens"
|
|
bobTokensStart = bob.token_contract.balance_of()
|
|
print('Bob has {} tokens'.format(bobTokensStart))
|
|
assert bobTokensStart == 0, "START: Bob has incorrect number of tokens"
|
|
charlieTokensStart = charlie.token_contract.balance_of()
|
|
print('Charlie has {} tokens'.format(charlieTokensStart))
|
|
assert charlieTokensStart == 0, "START: Charlie has incorrect number of tokens"
|
|
|
|
# Alice deposits 3 of her coins to the plasma contract and gets 3 plasma nft
|
|
# utxos in return
|
|
tokenId = 1
|
|
tx_hash, gas_used = alice.deposit(tokenId)
|
|
event_data = alice.root_chain.get_event_data('Deposit', tx_hash)
|
|
print('ALICE EVENT DATA1', event_data[0]['args'])
|
|
|
|
tx_hash, gas_used = alice.deposit(tokenId + 1)
|
|
event_data = alice.root_chain.get_event_data('Deposit', tx_hash)
|
|
deposit2_utxo = event_data[0]['args']['slot']
|
|
deposit2_block_number = event_data[0]['args']['blockNumber']
|
|
print('ALICE EVENT DATA2', event_data[0]['args'])
|
|
|
|
tx_hash, gas_used = alice.deposit(tokenId + 2)
|
|
event_data = alice.root_chain.get_event_data('Deposit', tx_hash)
|
|
deposit3_utxo = event_data[0]['args']['slot']
|
|
deposit3_block_number = event_data[0]['args']['blockNumber']
|
|
print('ALICE EVENT DATA3', event_data[0]['args'])
|
|
|
|
|
|
# Check that all deposits have registered
|
|
sleep(2)
|
|
registered_deposits = alice.get_all_deposits()
|
|
assert len(registered_deposits) == 3, "Alice has incorrect number of deposits"
|
|
|
|
# Alice to Bob, and Alice to Charlie. We care about the Alice to Bob
|
|
# transaction
|
|
alice_to_bob = alice.send_transaction(
|
|
deposit3_utxo, deposit3_block_number, bob.token_contract.account.address
|
|
)
|
|
random_tx = alice.send_transaction(
|
|
deposit2_utxo,
|
|
deposit2_block_number,
|
|
charlie.token_contract.account.address,
|
|
)
|
|
plasma_block1 = authority.submit_block()
|
|
|
|
# Add an empty block in betweeen (for proof of exclusion reasons)
|
|
authority.submit_block()
|
|
|
|
# Bob to Charlie
|
|
bob_to_charlie = bob.send_transaction(
|
|
deposit3_utxo, plasma_block1, charlie.token_contract.account.address
|
|
)
|
|
|
|
# This is the info that bob is required to send to charlie. This happens on
|
|
# the P2P layer
|
|
incl_proofs, excl_proofs = bob.get_coin_history(deposit3_utxo)
|
|
|
|
# Charlie receives them, verifies the validity. If found invalid, charlie
|
|
# should not accept them and the demo fails (similar to how you shouldn't sell
|
|
# a good when you're given counterfeit currency)
|
|
assert charlie.verify_coin_history(deposit3_utxo, incl_proofs, excl_proofs)
|
|
|
|
plasma_block2 = authority.submit_block()
|
|
|
|
# Block has been submitted, now we start watching for exits of our coin
|
|
charlie.watch_exits(deposit3_utxo)
|
|
|
|
# Charlie should be able to submit an exit by referencing blocks 0 and 1 which
|
|
# included his transaction.
|
|
charlie.start_exit(deposit3_utxo, plasma_block1, plasma_block2)
|
|
|
|
# We exited the coin so we should stop watching
|
|
charlie.stop_watching_exits(deposit3_utxo)
|
|
|
|
# Here we should start watching for challenges
|
|
|
|
# After 8 days pass, charlie's exit should be finalizable
|
|
increaseTime(w3, 8 * 24 * 3600)
|
|
authority.finalize_exits()
|
|
|
|
# Charlie should now be able to withdraw the utxo which included token 2 to his
|
|
# wallet.
|
|
charlie.withdraw(deposit3_utxo)
|
|
|
|
aliceTokensEnd = alice.token_contract.balance_of()
|
|
print('Alice has {} tokens'.format(aliceTokensEnd))
|
|
assert aliceTokensEnd == 2, "END: Alice has incorrect number of tokens"
|
|
bobTokensEnd = bob.token_contract.balance_of()
|
|
print('Bob has {} tokens'.format(bobTokensEnd))
|
|
assert bobTokensEnd == 0, "END: Bob has incorrect number of tokens"
|
|
charlieTokensEnd = charlie.token_contract.balance_of()
|
|
print('Charlie has {} tokens'.format(charlieTokensEnd))
|
|
assert charlieTokensEnd == 1, "END: Charlie has incorrect number of tokens"
|
|
|
|
print('Plasma Cash with ERC721 tokens success :)')
|