Problem: No test for checking mixed spent outputs

Solution: Add test for `get_spending_transactions` to check that correct
matching is done when querying documents with multiple inputs
This commit is contained in:
Vanshdeep Singh 2018-09-24 13:12:26 +02:00
parent 9929d8a28a
commit 731be170b8
2 changed files with 42 additions and 0 deletions

View File

@ -234,6 +234,47 @@ def test_get_spending_transactions(user_pk, user_sk):
assert txns == [tx2.to_dict(), tx4.to_dict()] assert txns == [tx2.to_dict(), tx4.to_dict()]
def test_get_spending_transactions_multiple_inputs():
from bigchaindb.backend import connect, query
from bigchaindb.models import Transaction
from bigchaindb.common.crypto import generate_key_pair
conn = connect()
(alice_sk, alice_pk) = generate_key_pair()
(bob_sk, bob_pk) = generate_key_pair()
(carol_sk, carol_pk) = generate_key_pair()
out = [([alice_pk], 9)]
tx1 = Transaction.create([alice_pk], out).sign([alice_sk])
inputs1 = tx1.to_inputs()
tx2 = Transaction.transfer([inputs1[0]],
[([alice_pk], 6), ([bob_pk], 3)],
tx1.id).sign([alice_sk])
inputs2 = tx2.to_inputs()
tx3 = Transaction.transfer([inputs2[0]],
[([bob_pk], 3), ([carol_pk], 3)],
tx1.id).sign([alice_sk])
inputs3 = tx3.to_inputs()
tx4 = Transaction.transfer([inputs2[1], inputs3[0]],
[([carol_pk], 6)],
tx1.id).sign([bob_sk])
txns = [deepcopy(tx.to_dict()) for tx in [tx1, tx2, tx3, tx4]]
conn.db.transactions.insert_many(txns)
links = [
({'transaction_id': tx2.id, 'output_index': 0}, 1),
({'transaction_id': tx2.id, 'output_index': 1}, 1),
({'transaction_id': tx3.id, 'output_index': 0}, 1),
({'transaction_id': tx3.id, 'output_index': 1}, 0),
]
for l, num in links:
txns = list(query.get_spending_transactions(conn, [l]))
assert len(txns) == num
def test_store_block(): def test_store_block():
from bigchaindb.backend import connect, query from bigchaindb.backend import connect, query
from bigchaindb.lib import Block from bigchaindb.lib import Block

View File

@ -64,6 +64,7 @@ def test_bigchain_class_initialization_with_parameters():
assert bigchain.validation == BaseValidationRules assert bigchain.validation == BaseValidationRules
@pytest.mark.bdb
def test_get_spent_issue_1271(b, alice, bob, carol): def test_get_spent_issue_1271(b, alice, bob, carol):
from bigchaindb.models import Transaction from bigchaindb.models import Transaction