130 lines
4.0 KiB
Python
130 lines
4.0 KiB
Python
import pytest
|
|
|
|
from bigchaindb.common.transaction import TransactionLink
|
|
from bigchaindb.models import Block, Transaction
|
|
|
|
pytestmark = pytest.mark.bdb
|
|
|
|
|
|
@pytest.fixture
|
|
def blockdata(b, user_pk, user2_pk):
|
|
txs = [Transaction.create([user_pk], [([user2_pk], 1)]),
|
|
Transaction.create([user2_pk], [([user_pk], 1)]),
|
|
Transaction.create([user_pk], [([user_pk], 1), ([user2_pk], 1)])]
|
|
blocks = []
|
|
for i in range(3):
|
|
block = Block([txs[i]])
|
|
b.write_block(block)
|
|
blocks.append(block.to_dict())
|
|
b.write_vote(b.vote(blocks[1]['id'], '', True))
|
|
b.write_vote(b.vote(blocks[2]['id'], '', False))
|
|
return blocks, [b['id'] for b in blocks]
|
|
|
|
|
|
def test_filter_valid_block_ids_with_undecided(b, blockdata):
|
|
blocks, block_ids = blockdata
|
|
valid_ids = b.fastquery.filter_valid_block_ids(block_ids, include_undecided=True)
|
|
assert set(valid_ids) == {blocks[0]['id'], blocks[1]['id']}
|
|
|
|
|
|
def test_filter_valid_block_ids_only_valid(b, blockdata):
|
|
blocks, block_ids = blockdata
|
|
valid_ids = b.fastquery.filter_valid_block_ids(block_ids)
|
|
assert set(valid_ids) == {blocks[1]['id']}
|
|
|
|
|
|
def test_filter_valid_items(b, blockdata):
|
|
blocks, _ = blockdata
|
|
assert (b.fastquery.filter_valid_items(blocks, block_id_key=lambda b: b['id'])
|
|
== [blocks[0], blocks[1]])
|
|
|
|
|
|
def test_get_outputs_by_public_key(b, user_pk, user2_pk, blockdata):
|
|
blocks, _ = blockdata
|
|
assert b.fastquery.get_outputs_by_public_key(user_pk) == [
|
|
TransactionLink(blocks[1]['block']['transactions'][0]['id'], 0)
|
|
]
|
|
assert b.fastquery.get_outputs_by_public_key(user2_pk) == [
|
|
TransactionLink(blocks[0]['block']['transactions'][0]['id'], 0)
|
|
]
|
|
|
|
|
|
def test_filter_spent_outputs(b, user_pk, user_sk):
|
|
out = [([user_pk], 1)]
|
|
tx1 = Transaction.create([user_pk], out * 3)
|
|
tx1.sign([user_sk])
|
|
# There are 3 inputs
|
|
inputs = tx1.to_inputs()
|
|
|
|
# Each spent individually
|
|
tx2 = Transaction.transfer([inputs[0]], out, tx1.id)
|
|
tx2.sign([user_sk])
|
|
tx3 = Transaction.transfer([inputs[1]], out, tx1.id)
|
|
tx3.sign([user_sk])
|
|
tx4 = Transaction.transfer([inputs[2]], out, tx1.id)
|
|
tx4.sign([user_sk])
|
|
|
|
# The CREATE and first TRANSFER are valid. tx2 produces a new unspent.
|
|
for tx in [tx1, tx2]:
|
|
block = Block([tx])
|
|
b.write_block(block)
|
|
b.write_vote(b.vote(block.id, '', True))
|
|
|
|
# The second TRANSFER is invalid. inputs[1] remains unspent.
|
|
block = Block([tx3])
|
|
b.write_block(block)
|
|
b.write_vote(b.vote(block.id, '', False))
|
|
|
|
# The third TRANSFER is undecided. It procuces a new unspent.
|
|
block = Block([tx4])
|
|
b.write_block(block)
|
|
|
|
outputs = b.fastquery.get_outputs_by_public_key(user_pk)
|
|
unspents = b.fastquery.filter_spent_outputs(outputs)
|
|
|
|
assert set(unspents) == {
|
|
inputs[1].fulfills,
|
|
tx2.to_inputs()[0].fulfills,
|
|
tx4.to_inputs()[0].fulfills
|
|
}
|
|
|
|
|
|
def test_filter_unspent_outputs(b, user_pk, user_sk):
|
|
out = [([user_pk], 1)]
|
|
tx1 = Transaction.create([user_pk], out * 3)
|
|
tx1.sign([user_sk])
|
|
|
|
# There are 3 inputs
|
|
inputs = tx1.to_inputs()
|
|
|
|
# Each spent individually
|
|
tx2 = Transaction.transfer([inputs[0]], out, tx1.id)
|
|
tx2.sign([user_sk])
|
|
tx3 = Transaction.transfer([inputs[1]], out, tx1.id)
|
|
tx3.sign([user_sk])
|
|
tx4 = Transaction.transfer([inputs[2]], out, tx1.id)
|
|
tx4.sign([user_sk])
|
|
|
|
# The CREATE and first TRANSFER are valid. tx2 produces a new unspent.
|
|
for tx in [tx1, tx2]:
|
|
block = Block([tx])
|
|
b.write_block(block)
|
|
b.write_vote(b.vote(block.id, '', True))
|
|
|
|
# The second TRANSFER is invalid. inputs[1] remains unspent.
|
|
block = Block([tx3])
|
|
b.write_block(block)
|
|
b.write_vote(b.vote(block.id, '', False))
|
|
|
|
# The third TRANSFER is undecided. It procuces a new unspent.
|
|
block = Block([tx4])
|
|
b.write_block(block)
|
|
|
|
outputs = b.fastquery.get_outputs_by_public_key(user_pk)
|
|
spents = b.fastquery.filter_unspent_outputs(outputs)
|
|
|
|
assert set(spents) == {
|
|
inputs[0].fulfills,
|
|
inputs[2].fulfills
|
|
}
|