mirror of
https://github.com/bigchaindb/bigchaindb.git
synced 2024-10-13 13:34:05 +00:00
7e33f2bd52
* Allow AssetLinks to be used in place of Assets in the Transaction Model and enforce `Transaction.transfer()` to only take an AssetLink * Remove AssetLink's inheritance from Asset * Remove id from the Asset model * Fix get_txids_by_asset_id query for rethinkdb after removing asset's uuid Because `CREATE` transactions don't have an asset that contains an id anymore, one way to find all the transactions related to an asset is to query the database twice: once for the `CREATE` transaction and another for the `TRANSFER` transactions. * Add TODO notice for vote test utils to be fixtures * Update asset model documentation to reflect usage of transaction id * Fix outdated asset description in transaction schema
216 lines
7.4 KiB
Python
216 lines
7.4 KiB
Python
import pytest
|
|
import random
|
|
from unittest.mock import patch
|
|
|
|
|
|
@pytest.mark.usefixtures('inputs')
|
|
def test_asset_transfer(b, user_pk, user_sk):
|
|
from bigchaindb.common.transaction import AssetLink
|
|
from bigchaindb.models import Transaction
|
|
|
|
tx_input = b.get_owned_ids(user_pk).pop()
|
|
tx_create = b.get_transaction(tx_input.txid)
|
|
|
|
tx_transfer = Transaction.transfer(tx_create.to_inputs(), [([user_pk], 1)],
|
|
AssetLink(tx_create.id))
|
|
tx_transfer_signed = tx_transfer.sign([user_sk])
|
|
|
|
assert tx_transfer_signed.validate(b) == tx_transfer_signed
|
|
assert tx_transfer_signed.asset.id == tx_create.id
|
|
|
|
|
|
def test_validate_bad_asset_creation(b, user_pk):
|
|
from bigchaindb.models import Transaction, Asset
|
|
|
|
# `data` needs to be a dictionary
|
|
tx = Transaction.create([b.me], [([user_pk], 1)])
|
|
tx.asset.data = 'a'
|
|
with patch.object(Asset, 'validate_asset', return_value=None):
|
|
tx_signed = tx.sign([b.me_private])
|
|
with pytest.raises(TypeError):
|
|
b.validate_transaction(tx_signed)
|
|
|
|
|
|
@pytest.mark.usefixtures('inputs')
|
|
def test_validate_transfer_asset_id_mismatch(b, user_pk, user_sk):
|
|
from bigchaindb.common.exceptions import AssetIdMismatch
|
|
from bigchaindb.common.transaction import AssetLink
|
|
from bigchaindb.models import Transaction
|
|
|
|
tx_create = b.get_owned_ids(user_pk).pop()
|
|
tx_create = b.get_transaction(tx_create.txid)
|
|
tx_transfer = Transaction.transfer(tx_create.to_inputs(), [([user_pk], 1)],
|
|
AssetLink(tx_create.id))
|
|
tx_transfer.asset.id = 'aaa'
|
|
tx_transfer_signed = tx_transfer.sign([user_sk])
|
|
with pytest.raises(AssetIdMismatch):
|
|
tx_transfer_signed.validate(b)
|
|
|
|
|
|
def test_get_asset_id_create_transaction(b, user_pk):
|
|
from bigchaindb.models import Transaction, Asset
|
|
|
|
tx_create = Transaction.create([b.me], [([user_pk], 1)])
|
|
asset_id = Asset.get_asset_id(tx_create)
|
|
|
|
assert asset_id == tx_create.id
|
|
|
|
|
|
@pytest.mark.usefixtures('inputs')
|
|
def test_get_asset_id_transfer_transaction(b, user_pk, user_sk):
|
|
from bigchaindb.common.transaction import AssetLink
|
|
from bigchaindb.models import Transaction, Asset
|
|
|
|
tx_create = b.get_owned_ids(user_pk).pop()
|
|
tx_create = b.get_transaction(tx_create.txid)
|
|
# create a transfer transaction
|
|
tx_transfer = Transaction.transfer(tx_create.to_inputs(), [([user_pk], 1)],
|
|
AssetLink(tx_create.id))
|
|
tx_transfer_signed = tx_transfer.sign([user_sk])
|
|
# create a block
|
|
block = b.create_block([tx_transfer_signed])
|
|
b.write_block(block)
|
|
# vote the block valid
|
|
vote = b.vote(block.id, b.get_last_voted_block().id, True)
|
|
b.write_vote(vote)
|
|
asset_id = Asset.get_asset_id(tx_transfer)
|
|
|
|
assert asset_id == tx_transfer.asset.id
|
|
|
|
|
|
def test_asset_id_mismatch(b, user_pk):
|
|
from bigchaindb.models import Transaction, Asset
|
|
from bigchaindb.common.exceptions import AssetIdMismatch
|
|
|
|
tx1 = Transaction.create([b.me], [([user_pk], 1)],
|
|
metadata={'msg': random.random()})
|
|
tx2 = Transaction.create([b.me], [([user_pk], 1)],
|
|
metadata={'msg': random.random()})
|
|
|
|
with pytest.raises(AssetIdMismatch):
|
|
Asset.get_asset_id([tx1, tx2])
|
|
|
|
|
|
@pytest.mark.usefixtures('inputs')
|
|
def test_get_transactions_by_asset_id(b, user_pk, user_sk):
|
|
from bigchaindb.common.transaction import AssetLink
|
|
from bigchaindb.models import Transaction
|
|
|
|
tx_create = b.get_owned_ids(user_pk).pop()
|
|
tx_create = b.get_transaction(tx_create.txid)
|
|
asset_id = tx_create.id
|
|
txs = b.get_transactions_by_asset_id(asset_id)
|
|
|
|
assert len(txs) == 1
|
|
assert txs[0].id == tx_create.id
|
|
assert txs[0].id == asset_id
|
|
|
|
# create a transfer transaction
|
|
tx_transfer = Transaction.transfer(tx_create.to_inputs(), [([user_pk], 1)],
|
|
AssetLink(tx_create.id))
|
|
tx_transfer_signed = tx_transfer.sign([user_sk])
|
|
# create the block
|
|
block = b.create_block([tx_transfer_signed])
|
|
b.write_block(block)
|
|
# vote the block valid
|
|
vote = b.vote(block.id, b.get_last_voted_block().id, True)
|
|
b.write_vote(vote)
|
|
|
|
txs = b.get_transactions_by_asset_id(asset_id)
|
|
|
|
assert len(txs) == 2
|
|
assert tx_create.id in [t.id for t in txs]
|
|
assert tx_transfer.id in [t.id for t in txs]
|
|
# FIXME: can I rely on the ordering here?
|
|
assert asset_id == txs[0].id
|
|
assert asset_id == txs[1].asset.id
|
|
|
|
|
|
@pytest.mark.usefixtures('inputs')
|
|
def test_get_transactions_by_asset_id_with_invalid_block(b, user_pk, user_sk):
|
|
from bigchaindb.common.transaction import AssetLink
|
|
from bigchaindb.models import Transaction
|
|
|
|
tx_create = b.get_owned_ids(user_pk).pop()
|
|
tx_create = b.get_transaction(tx_create.txid)
|
|
asset_id = tx_create.id
|
|
txs = b.get_transactions_by_asset_id(asset_id)
|
|
|
|
assert len(txs) == 1
|
|
assert txs[0].id == tx_create.id
|
|
assert txs[0].id == asset_id
|
|
|
|
# create a transfer transaction
|
|
tx_transfer = Transaction.transfer(tx_create.to_inputs(), [([user_pk], 1)],
|
|
AssetLink(tx_create.id))
|
|
tx_transfer_signed = tx_transfer.sign([user_sk])
|
|
# create the block
|
|
block = b.create_block([tx_transfer_signed])
|
|
b.write_block(block)
|
|
# vote the block invalid
|
|
vote = b.vote(block.id, b.get_last_voted_block().id, False)
|
|
b.write_vote(vote)
|
|
|
|
txs = b.get_transactions_by_asset_id(asset_id)
|
|
|
|
assert len(txs) == 1
|
|
|
|
|
|
@pytest.mark.usefixtures('inputs')
|
|
def test_get_asset_by_id(b, user_pk, user_sk):
|
|
from bigchaindb.common.transaction import AssetLink
|
|
from bigchaindb.models import Asset, Transaction
|
|
|
|
tx_create = b.get_owned_ids(user_pk).pop()
|
|
tx_create = b.get_transaction(tx_create.txid)
|
|
|
|
# create a transfer transaction
|
|
tx_transfer = Transaction.transfer(tx_create.to_inputs(), [([user_pk], 1)],
|
|
AssetLink(tx_create.id))
|
|
tx_transfer_signed = tx_transfer.sign([user_sk])
|
|
# create the block
|
|
block = b.create_block([tx_transfer_signed])
|
|
b.write_block(block)
|
|
# vote the block valid
|
|
vote = b.vote(block.id, b.get_last_voted_block().id, True)
|
|
b.write_vote(vote)
|
|
|
|
asset_id = Asset.get_asset_id([tx_create, tx_transfer])
|
|
txs = b.get_transactions_by_asset_id(asset_id)
|
|
assert len(txs) == 2
|
|
|
|
asset = b.get_asset_by_id(asset_id)
|
|
assert asset == tx_create.asset
|
|
|
|
|
|
def test_create_invalid_divisible_asset(b, user_pk, user_sk):
|
|
from bigchaindb.models import Transaction, Asset
|
|
from bigchaindb.common.exceptions import AmountError
|
|
|
|
# Asset amount must be more than 0
|
|
asset = Asset()
|
|
tx = Transaction.create([user_pk], [([user_pk], 1)], asset=asset)
|
|
tx.conditions[0].amount = 0
|
|
with pytest.raises(AmountError):
|
|
tx.sign([user_sk])
|
|
|
|
# even if a transaction is badly constructed the server should raise the
|
|
# exception
|
|
asset = Asset()
|
|
tx = Transaction.create([user_pk], [([user_pk], 1)], asset=asset)
|
|
tx.conditions[0].amount = 0
|
|
with patch.object(Asset, 'validate_asset', return_value=None):
|
|
tx_signed = tx.sign([user_sk])
|
|
with pytest.raises(AmountError):
|
|
tx_signed.validate(b)
|
|
assert b.is_valid_transaction(tx_signed) is False
|
|
|
|
|
|
def test_create_valid_divisible_asset(b, user_pk, user_sk):
|
|
from bigchaindb.models import Transaction, Asset
|
|
|
|
asset = Asset()
|
|
tx = Transaction.create([user_pk], [([user_pk], 2)], asset=asset)
|
|
tx_signed = tx.sign([user_sk])
|
|
assert b.is_valid_transaction(tx_signed)
|