def test_can_add_node_declaration_block(
forced_memory_blockchain: MemoryBlockchain,
user_account_key_pair: KeyPair,
):
blockchain = forced_memory_blockchain
user_account = user_account_key_pair.public
request0 = NodeDeclarationSignedChangeRequest.create(
identifier=hexstr('abcd'),
network_addresses=['127.0.0.1'],
fee_amount=3,
signing_key=user_account_key_pair.private)
block0 = Block.create_from_signed_change_request(blockchain, request0)
blockchain.add_block(block0)
assert blockchain.get_current_node(user_account) == request0.message.node
blockchain.snapshot_blockchain_state()
assert blockchain.blockchain_states[-1].get_node(
user_account) == request0.message.node
request1 = NodeDeclarationSignedChangeRequest.create(
identifier=hexstr('abcd'),
network_addresses=['127.0.0.2', '192.168.0.34'],
fee_amount=3,
signing_key=user_account_key_pair.private)
block1 = Block.create_from_signed_change_request(blockchain, request1)
blockchain.add_block(block1)
assert blockchain.get_current_node(user_account) == request1.message.node
blockchain.snapshot_blockchain_state()
assert blockchain.blockchain_states[-1].get_node(
user_account) == request1.message.node
def test_can_get_primary_validator_node(forced_memory_blockchain, api_client):
blockchain = forced_memory_blockchain
signing_key = get_node_signing_key()
pv_node_declaration_request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['http://my.domain.com/'],
fee_amount=3,
signing_key=signing_key,
)
block = Block.create_from_signed_change_request(
blockchain,
signed_change_request=pv_node_declaration_request,
pv_signing_key=signing_key,
)
blockchain.add_block(block)
pv_node = pv_node_declaration_request.message.node
pv_schedule_request = PrimaryValidatorScheduleSignedChangeRequest.create(
0, 99, signing_key=signing_key)
block = Block.create_from_signed_change_request(
blockchain,
signed_change_request=pv_schedule_request,
pv_signing_key=signing_key,
)
blockchain.add_block(block)
response = api_client.get(f'{API_V1_NODES_PREFIX}/pv/')
assert response.status_code == 200
data = response.json()
assert data['identifier'] == pv_node.identifier
assert data['fee_amount'] == 3
def test_can_serialize_deserialize(forced_mock_blockchain, sample_signed_change_request):
block = Block.create_from_signed_change_request(
forced_mock_blockchain, sample_signed_change_request, get_node_signing_key()
)
serialized_dict = block.serialize_to_dict()
deserialized_block = Block.deserialize_from_dict(serialized_dict)
assert deserialized_block == block
assert deserialized_block is not block
def test_can_duplicate_recipients(
forced_mock_blockchain: MockBlockchain, treasury_account_key_pair: KeyPair, user_account_key_pair: KeyPair
):
def get_account_balance(self, account, on_block_number):
return 430 if account == treasury_account_key_pair.public else 10
sender = treasury_account_key_pair.public
recipient = user_account_key_pair.public
message = CoinTransferSignedChangeRequestMessage(
balance_lock=forced_mock_blockchain.get_account_current_balance_lock(sender),
txs=[
CoinTransferTransaction(recipient=recipient, amount=3),
CoinTransferTransaction(recipient=recipient, amount=5),
]
)
request = CoinTransferSignedChangeRequest.create_from_signed_change_request_message(
message, treasury_account_key_pair.private
)
with patch.object(MockBlockchain, 'get_account_balance', new=get_account_balance):
block = Block.create_from_signed_change_request(forced_mock_blockchain, request)
updated_account_states = block.message.updated_account_states
assert len(updated_account_states) == 2
sender_account_state = block.message.get_account_state(treasury_account_key_pair.public)
assert sender_account_state
assert sender_account_state.balance == 430 - 3 - 5
assert sender_account_state.balance_lock
recipient_account_state = block.message.get_account_state(user_account_key_pair.public)
assert recipient_account_state
assert recipient_account_state.balance == 10 + 3 + 5
def test_can_get_nodes_blocks_node_overrides_genesis_state_node(
blockchain_directory, blockchain_genesis_state, user_account_key_pair
):
blockchain = FileBlockchain(base_directory=blockchain_directory)
account_number = user_account_key_pair.public
blockchain_state_node = baker.make(
Node, network_addresses=['https://192.168.0.32:8555/'], identifier=account_number
)
blockchain_genesis_state.account_states[account_number] = AccountState(node=blockchain_state_node)
blockchain.add_blockchain_state(blockchain_genesis_state)
blockchain.validate()
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://127.0.0.1:8555/'], fee_amount=3, signing_key=user_account_key_pair.private
)
blocks_node = request.message.node
assert blocks_node.identifier
block = Block.create_from_signed_change_request(blockchain, request, get_node_signing_key())
blockchain.add_block(block)
assert blocks_node != blockchain_state_node
assert list(blockchain.yield_nodes()) == [blocks_node]
def test_can_create_block_from_signed_change_request(
forced_mock_blockchain, sample_signed_change_request: CoinTransferSignedChangeRequest
):
sender = sample_signed_change_request.signer
assert sender
def get_account_balance(self, account, on_block_number):
return 450 if account == sender else 0
with patch.object(MockBlockchain, 'get_account_balance', new=get_account_balance):
block = Block.create_from_signed_change_request(
forced_mock_blockchain, sample_signed_change_request, get_node_signing_key()
)
assert block.message
assert block.hash
assert block.signature
block.validate_signature()
assert block.signer
assert block.signer == derive_public_key(get_node_signing_key())
block_message = block.message
signed_change_request = block_message.signed_change_request
assert signed_change_request == sample_signed_change_request
assert signed_change_request is not sample_signed_change_request # test that a copy of it was made
assert isinstance(block_message.timestamp, datetime)
assert block_message.timestamp.tzinfo is None
assert block_message.timestamp - datetime.utcnow() < timedelta(seconds=1)
assert block_message.block_number == 0
assert block_message.block_identifier == 'next-block-identifier'
updated_account_states = block_message.updated_account_states
assert isinstance(updated_account_states, dict)
assert len(updated_account_states) == 4
assert updated_account_states[sender].balance == 450 - 425 - 4 - 1
assert updated_account_states[sender].balance_lock
assert updated_account_states['484b3176c63d5f37d808404af1a12c4b9649cd6f6769f35bdf5a816133623fbc'].balance == 425
assert updated_account_states['484b3176c63d5f37d808404af1a12c4b9649cd6f6769f35bdf5a816133623fbc'
].balance_lock is None
assert updated_account_states['ad1f8845c6a1abb6011a2a434a079a087c460657aad54329a84b406dce8bf314'].balance == 4
assert updated_account_states['ad1f8845c6a1abb6011a2a434a079a087c460657aad54329a84b406dce8bf314'
].balance_lock is None
assert updated_account_states['5e12967707909e62b2bb2036c209085a784fabbc3deccefee70052b6181c8ed8'].balance == 1
assert updated_account_states['5e12967707909e62b2bb2036c209085a784fabbc3deccefee70052b6181c8ed8'
].balance_lock is None
def test_node_serializes_without_identifier_in_block(memory_blockchain,
user_account_key_pair):
account_number = user_account_key_pair.public
node = baker.make(Node, identifier=account_number)
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.34:8555/'],
fee_amount=3,
signing_key=user_account_key_pair.private)
block = Block.create_from_signed_change_request(memory_blockchain, request,
get_node_signing_key())
serialized = block.serialize_to_dict()
assert 'identifier' not in serialized['message']['updated_account_states'][
account_number]['node']
deserialized = Block.deserialize_from_dict(serialized)
deserialized_node = deserialized.message.updated_account_states[
account_number].node
assert deserialized_node is not node
assert deserialized_node.identifier == account_number
def test_node_identifier_is_removed_when_node_declaration_signed_change_request_is_serialized(
memory_blockchain, user_account_key_pair):
request = NodeDeclarationSignedChangeRequest.create(
identifier=hexstr('abcd'),
network_addresses=['127.0.0.1'],
fee_amount=3,
fee_account=hexstr('dcba'),
signing_key=user_account_key_pair.private)
block = Block.create_from_signed_change_request(memory_blockchain, request)
compact_dict = block.to_compact_dict()
assert ck('identifier') not in compact_dict[ck('message')][ck(
'signed_change_request')][ck('message')][ck('node')]
def test_can_get_nodes_from_different_block_numbers(
blockchain_directory, blockchain_genesis_state, user_account_key_pair
):
account_number = user_account_key_pair.public
blockchain = FileBlockchain(base_directory=blockchain_directory)
blockchain_state_node = baker.make(
Node, network_addresses=['https://192.168.0.29:8555/'], identifier=account_number
)
blockchain_genesis_state.account_states[account_number] = AccountState(node=blockchain_state_node)
blockchain.add_blockchain_state(blockchain_genesis_state)
blockchain.validate()
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.30:8555/'], fee_amount=3, signing_key=user_account_key_pair.private
)
node0 = request.message.node
blockchain.add_block(Block.create_from_signed_change_request(blockchain, request, get_node_signing_key()))
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.31:8555/'], fee_amount=3, signing_key=user_account_key_pair.private
)
node1 = request.message.node
blockchain.add_block(Block.create_from_signed_change_request(blockchain, request, get_node_signing_key()))
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.32:8555/'], fee_amount=3, signing_key=user_account_key_pair.private
)
node2 = request.message.node
blockchain.add_block(Block.create_from_signed_change_request(blockchain, request, get_node_signing_key()))
assert list(blockchain.yield_nodes()) == [node2]
assert list(blockchain.yield_nodes(block_number=2)) == [node2]
assert list(blockchain.yield_nodes(block_number=1)) == [node1]
assert list(blockchain.yield_nodes(block_number=0)) == [node0]
assert list(blockchain.yield_nodes(block_number=-1)) == [blockchain_state_node]
def test_can_get_nodes_single_node_from_blocks(blockchain_directory, blockchain_genesis_state, user_account_key_pair):
blockchain = FileBlockchain(base_directory=blockchain_directory)
blockchain.add_blockchain_state(blockchain_genesis_state)
blockchain.validate()
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://127.0.0.1:8555/'], fee_amount=3, signing_key=user_account_key_pair.private
)
node = request.message.node
assert node.identifier
block = Block.create_from_signed_change_request(blockchain, request, get_node_signing_key())
blockchain.add_block(block)
assert list(blockchain.yield_nodes()) == [node]
def test_normalized_block_message(forced_mock_blockchain, sample_signed_change_request):
expected_message_template = (
'{'
'"block_identifier":"next-block-identifier",'
'"block_number":0,'
'"block_type":"ct",'
'"signed_change_request":'
'{"message":{"balance_lock":'
'"4d3cf1d9e4547d324de2084b568f807ef12045075a7a01b8bec1e7f013fc3732",'
'"txs":'
'[{"amount":425,"recipient":"484b3176c63d5f37d808404af1a12c4b9649cd6f6769f35bdf5a816133623fbc"},'
'{"amount":1,"is_fee":true,"recipient":"5e12967707909e62b2bb2036c209085a784fabbc3deccefee70052b6181c8ed8"},'
'{"amount":4,"is_fee":true,"recipient":'
'"ad1f8845c6a1abb6011a2a434a079a087c460657aad54329a84b406dce8bf314"}]},'
'"signature":"362dc47191d5d1a33308de1f036a5e93fbaf0b05fa971d9537f954f13cd22b5ed9bee56f4701bd'
'af9b995c47271806ba73e75d63f46084f5830cec5f5b7e9600",'
'"signer":"4d3cf1d9e4547d324de2084b568f807ef12045075a7a01b8bec1e7f013fc3732"},'
'"timestamp":"<replace-with-timestamp>",'
'"updated_account_states":{'
'"484b3176c63d5f37d808404af1a12c4b9649cd6f6769f35bdf5a816133623fbc":{"balance":425},'
'"4d3cf1d9e4547d324de2084b568f807ef12045075a7a01b8bec1e7f013fc3732":'
'{'
'"balance":20,'
'"balance_lock":"ff3127bdb408e5f3f4f07dd364ce719b2854dc28ee66aa7af839e46468761885"'
'},'
'"5e12967707909e62b2bb2036c209085a784fabbc3deccefee70052b6181c8ed8":{"balance":1},'
'"ad1f8845c6a1abb6011a2a434a079a087c460657aad54329a84b406dce8bf314":{"balance":4}'
'}'
'}'
)
def get_account_balance(self, account, on_block_number):
return 450 if account == sample_signed_change_request.signer else 0
with patch.object(MockBlockchain, 'get_account_balance', new=get_account_balance):
block = Block.create_from_signed_change_request(
forced_mock_blockchain, sample_signed_change_request, get_node_signing_key()
)
expected_message = expected_message_template.replace(
'<replace-with-timestamp>', block.message.timestamp.isoformat()
).encode('utf-8')
assert block.message.get_normalized() == expected_message
def test_create_node_declaration_block(memory_blockchain,
user_account_key_pair):
request = NodeDeclarationSignedChangeRequest.create(
identifier=hexstr('abcd'),
network_addresses=['127.0.0.1'],
fee_amount=3,
fee_account=hexstr('dcba'),
signing_key=user_account_key_pair.private)
block = Block.create_from_signed_change_request(memory_blockchain, request)
assert block
assert block.message.signed_change_request
assert block.message.signed_change_request.signer
assert block.message.signed_change_request.signature
assert block.message.signed_change_request.message
assert block.message.signed_change_request.message.node.network_addresses == [
'127.0.0.1'
]
assert block.message.signed_change_request.message.node.fee_amount == 3
assert block.message.signed_change_request.message.node.fee_account == hexstr(
'dcba')
def test_normalized_block_message(forced_mock_blockchain, sample_signed_change_request):
expected_message_template = (
'{'
'"block_identifier":"next-block-identifier",'
'"block_number":0,'
'"block_type":"ct",'
'"signed_change_request":'
'{"message":{"balance_lock":'
'"4d3cf1d9e4547d324de2084b568f807ef12045075a7a01b8bec1e7f013fc3732",'
'"txs":'
'[{"amount":425,"recipient":"484b3176c63d5f37d808404af1a12c4b9649cd6f6769f35bdf5a816133623fbc"},'
'{"amount":1,"fee":true,"recipient":"5e12967707909e62b2bb2036c209085a784fabbc3deccefee70052b6181c8ed8"},'
'{"amount":4,"fee":true,"recipient":'
'"ad1f8845c6a1abb6011a2a434a079a087c460657aad54329a84b406dce8bf314"}]},'
'"signature":"8c1b5719745cdc81e71905e874c1f1fb938d941dd6d03ddc6dc39fc60ca42dcb8a17bb2e721c3f2a'
'128a2dff35a3b0f843efe78893adde78a27192ca54212a08",'
'"signer":"4d3cf1d9e4547d324de2084b568f807ef12045075a7a01b8bec1e7f013fc3732"},'
'"timestamp":"<replace-with-timestamp>",'
'"updated_account_states":{'
'"484b3176c63d5f37d808404af1a12c4b9649cd6f6769f35bdf5a816133623fbc":{"balance":425},'
'"4d3cf1d9e4547d324de2084b568f807ef12045075a7a01b8bec1e7f013fc3732":'
'{'
'"balance":20,'
'"balance_lock":"ae4116766c916e761c5ab7590e2426f9c391078519d8cef8673ee3fe0cdb75ad"'
'},'
'"5e12967707909e62b2bb2036c209085a784fabbc3deccefee70052b6181c8ed8":{"balance":1},'
'"ad1f8845c6a1abb6011a2a434a079a087c460657aad54329a84b406dce8bf314":{"balance":4}'
'}'
'}'
)
def get_account_balance(self, account, on_block_number):
return 450 if account == sample_signed_change_request.signer else 0
with patch.object(MockBlockchain, 'get_account_balance', new=get_account_balance):
block = Block.create_from_signed_change_request(forced_mock_blockchain, sample_signed_change_request)
expected_message = expected_message_template.replace(
'<replace-with-timestamp>', block.message.timestamp.isoformat()
).encode('utf-8')
assert block.message.get_normalized() == expected_message
def test_can_get_node(forced_memory_blockchain, api_client,
user_account_key_pair):
blockchain = forced_memory_blockchain
node_declaration_request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['http://my.domain.com/'],
fee_amount=3,
signing_key=user_account_key_pair.private,
)
block = Block.create_from_signed_change_request(
blockchain,
signed_change_request=node_declaration_request,
pv_signing_key=user_account_key_pair.private,
)
blockchain.add_block(block)
node = node_declaration_request.message.node
response = api_client.get(f'{API_V1_NODES_PREFIX}/{node.identifier}/')
assert response.status_code == 200
data = response.json()
assert data['identifier'] == node.identifier
assert data['fee_amount'] == 3
def test_can_get_self_node(forced_memory_blockchain, api_client):
blockchain = forced_memory_blockchain
signing_key = get_node_signing_key()
node_id = derive_public_key(signing_key)
node_declaration_request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['http://my.domain.com/'],
fee_amount=3,
signing_key=signing_key,
)
block = Block.create_from_signed_change_request(
blockchain,
signed_change_request=node_declaration_request,
pv_signing_key=signing_key,
)
blockchain.add_block(block)
response = api_client.get(f'{API_V1_NODES_PREFIX}/self/')
assert response.status_code == 200
data = response.json()
assert data['identifier'] == node_id
assert data['fee_amount'] == 3
def test_can_add_coin_transfer_block(
forced_memory_blockchain: MemoryBlockchain,
treasury_account_key_pair: KeyPair,
user_account_key_pair: KeyPair,
primary_validator_key_pair: KeyPair,
node_key_pair: KeyPair,
):
blockchain = forced_memory_blockchain
treasury_account = treasury_account_key_pair.public
treasury_initial_balance = blockchain.get_account_current_balance(
treasury_account)
assert treasury_initial_balance is not None
user_account = user_account_key_pair.public
pv_account = primary_validator_key_pair.public
node_account = node_key_pair.public
total_fees = 1 + 4
block0 = Block.create_from_main_transaction(
blockchain,
user_account,
30,
signing_key=treasury_account_key_pair.private)
blockchain.add_block(block0)
assert blockchain.get_account_current_balance(user_account) == 30
assert blockchain.get_account_current_balance(
treasury_account) == treasury_initial_balance - 30 - total_fees
assert blockchain.get_account_current_balance(node_account) == 1
assert blockchain.get_account_current_balance(pv_account) == 4
with pytest.raises(
ValidationError,
match='Block number must be equal to next block number.*'):
blockchain.add_block(block0)
block1 = Block.create_from_main_transaction(
blockchain,
user_account,
10,
signing_key=treasury_account_key_pair.private)
blockchain.add_block(block1)
assert blockchain.get_account_current_balance(user_account) == 40
assert blockchain.get_account_current_balance(
treasury_account
) == treasury_initial_balance - 30 - 10 - 2 * total_fees
assert blockchain.get_account_current_balance(node_account) == 1 * 2
assert blockchain.get_account_current_balance(pv_account) == 4 * 2
block2 = Block.create_from_main_transaction(
blockchain,
treasury_account,
5,
signing_key=user_account_key_pair.private)
blockchain.add_block(block2)
assert blockchain.get_account_current_balance(
user_account) == 40 - 5 - total_fees
assert blockchain.get_account_current_balance(
treasury_account
) == treasury_initial_balance - 30 - 10 + 5 - 2 * total_fees
assert blockchain.get_account_current_balance(node_account) == 1 * 3
assert blockchain.get_account_current_balance(pv_account) == 4 * 3
def test_can_get_nodes_from_complex_blockchain(blockchain_directory, blockchain_genesis_state):
key_pair1 = generate_key_pair()
key_pair2 = generate_key_pair()
key_pair3 = generate_key_pair()
key_pair4 = generate_key_pair()
key_pair5 = generate_key_pair()
key_pair6 = generate_key_pair()
assert len({
key_pair1.public, key_pair2.public, key_pair3.public, key_pair4.public, key_pair5.public, key_pair6.public
}) == 6
blockchain = FileBlockchain(base_directory=blockchain_directory)
node1 = baker.make(Node, network_addresses=['https://192.168.0.29:8555/'], identifier=key_pair1.public)
blockchain_genesis_state.account_states[node1.identifier] = AccountState(node=node1)
blockchain.add_blockchain_state(blockchain_genesis_state)
blockchain.validate()
# Block 0
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.30:8555/'], fee_amount=3, signing_key=key_pair2.private
)
node2 = request.message.node
blockchain.add_block(Block.create_from_signed_change_request(blockchain, request, get_node_signing_key()))
# Block 1
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.31:8555/'], fee_amount=3, signing_key=key_pair3.private
)
node3_old = request.message.node
blockchain.add_block(Block.create_from_signed_change_request(blockchain, request, get_node_signing_key()))
# Block 2
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.32:8555/'], fee_amount=3, signing_key=key_pair4.private
)
node4 = request.message.node
blockchain.add_block(Block.create_from_signed_change_request(blockchain, request, get_node_signing_key()))
def sort_key(value):
return value.network_addresses
def sort_me(list_):
return sorted(list_, key=sort_key)
assert sort_me(blockchain.yield_nodes(block_number=2)) == sort_me([node4, node3_old, node2, node1])
blockchain.snapshot_blockchain_state()
# Block 3
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.33:8555/'], fee_amount=3, signing_key=key_pair3.private
)
node3 = request.message.node
blockchain.add_block(Block.create_from_signed_change_request(blockchain, request, get_node_signing_key()))
# Block 4
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.34:8555/'], fee_amount=3, signing_key=key_pair5.private
)
node5 = request.message.node
blockchain.add_block(Block.create_from_signed_change_request(blockchain, request, get_node_signing_key()))
# Block 5
request = NodeDeclarationSignedChangeRequest.create(
network_addresses=['https://192.168.0.35:8555/'], fee_amount=3, signing_key=key_pair6.private
)
node6 = request.message.node
blockchain.add_block(Block.create_from_signed_change_request(blockchain, request, get_node_signing_key()))
assert sort_me(blockchain.yield_nodes()) == sort_me([node6, node5, node3, node4, node2, node1])
assert sort_me(blockchain.yield_nodes(block_number=5)) == sort_me([node6, node5, node3, node4, node2, node1])
assert sort_me(blockchain.yield_nodes(block_number=4)) == sort_me([node5, node3, node4, node2, node1])
assert sort_me(blockchain.yield_nodes(block_number=3)) == sort_me([node3, node4, node2, node1])
assert sort_me(blockchain.yield_nodes(block_number=2)) == sort_me([node4, node3_old, node2, node1])
assert sort_me(blockchain.yield_nodes(block_number=1)) == sort_me([node3_old, node2, node1])
assert sort_me(blockchain.yield_nodes(block_number=0)) == sort_me([node2, node1])
assert sort_me(blockchain.yield_nodes(block_number=-1)) == sort_me([node1])
def test_can_create_block_from_main_transaction(
forced_mock_blockchain, treasury_account_key_pair: KeyPair, user_account_key_pair: KeyPair,
primary_validator_key_pair: KeyPair, node_key_pair: KeyPair
):
def get_account_balance(self, account, on_block_number):
return 430 if account == treasury_account_key_pair.public else 0
with patch.object(MockBlockchain, 'get_account_balance', new=get_account_balance):
block = Block.create_from_main_transaction(
forced_mock_blockchain, user_account_key_pair.public, 20, signing_key=treasury_account_key_pair.private
)
# Assert block
assert block.message
assert block.hash
assert block.signature
block.validate_signature()
assert block.signer
assert block.signer == derive_verify_key(get_node_signing_key())
# Assert block.message
block_message = block.message
assert block_message
assert isinstance(block_message.timestamp, datetime)
assert block_message.timestamp.tzinfo is None
assert block_message.timestamp - datetime.utcnow() < timedelta(seconds=1)
assert block_message.block_number == 0
assert block_message.block_identifier == 'next-block-identifier'
updated_account_states = block_message.updated_account_states
assert isinstance(updated_account_states, dict)
assert len(updated_account_states) == 4
assert updated_account_states[treasury_account_key_pair.public].balance == 430 - 25
assert updated_account_states[treasury_account_key_pair.public].balance_lock
assert updated_account_states[user_account_key_pair.public].balance == 20
assert updated_account_states[user_account_key_pair.public].balance_lock is None
assert updated_account_states[primary_validator_key_pair.public].balance == 4
assert updated_account_states[primary_validator_key_pair.public].balance_lock is None
assert updated_account_states[node_key_pair.public].balance == 1
assert updated_account_states[node_key_pair.public].balance_lock is None
# Assert block_message.signed_change_request
signed_change_request = block_message.signed_change_request
assert signed_change_request.signer == treasury_account_key_pair.public
assert signed_change_request.signature
# Assert block_message.signed_change_request.message
coin_transfer_signed_request_message = signed_change_request.message
assert isinstance(coin_transfer_signed_request_message, CoinTransferSignedChangeRequestMessage)
assert coin_transfer_signed_request_message.balance_lock
assert len(coin_transfer_signed_request_message.txs) == 3
txs_dict = {tx.recipient: tx for tx in coin_transfer_signed_request_message.txs}
assert len(txs_dict) == 3
assert txs_dict[user_account_key_pair.public].amount == 20
assert txs_dict[user_account_key_pair.public].is_fee is False
assert txs_dict[primary_validator_key_pair.public].amount == 4
assert txs_dict[primary_validator_key_pair.public].is_fee
assert txs_dict[node_key_pair.public].amount == 1
assert txs_dict[node_key_pair.public].is_fee
assert coin_transfer_signed_request_message.get_total_amount() == 25
def test_get_nested_models():
assert set(Block.get_nested_models(include_self=True)) == {
Block, BlockMessage, SignedChangeRequest, AccountState, Node, PrimaryValidatorSchedule,
SignedChangeRequestMessage
}
