diff --git a/tests/db/test_planetmint_api.py b/tests/db/test_planetmint_api.py index 019ce77..8fa45c1 100644 --- a/tests/db/test_planetmint_api.py +++ b/tests/db/test_planetmint_api.py @@ -64,11 +64,11 @@ class TestBigchainApi(object): # create the transactions tx1 = Create.generate([alice.public_key], [([alice.public_key], 1)], - assets=asset1).sign([alice.private_key]) + assets=[asset1]).sign([alice.private_key]) tx2 = Create.generate([alice.public_key], [([alice.public_key], 1)], - assets=asset2).sign([alice.private_key]) + assets=[asset2]).sign([alice.private_key]) tx3 = Create.generate([alice.public_key], [([alice.public_key], 1)], - assets=asset3).sign([alice.private_key]) + assets=[asset3]).sign([alice.private_key]) # write the transactions to the DB b.store_bulk_transactions([tx1, tx2, tx3]) @@ -97,7 +97,7 @@ class TestBigchainApi(object): asset1 = {'msg': 'Planetmint 1'} tx = Create.generate([alice.public_key], [([alice.public_key], 1)], - assets=asset1).sign([alice.private_key]) + assets=[asset1]).sign([alice.private_key]) b.store_bulk_transactions([tx]) tx_from_db = b.get_transaction(tx.id) @@ -105,9 +105,9 @@ class TestBigchainApi(object): before = tx.to_dict() after = tx_from_db.to_dict() - assert before['asset']['data'] == after['asset']['data'] - before.pop('asset', None) - after.pop('asset', None) + assert before['assets'][0]['data'] == after['assets'][0]['data'] + before.pop('assets', None) + after.pop('assets', None) assert before == after