diff --git a/planetmint/backend/tarantool/query.py b/planetmint/backend/tarantool/query.py index 23528b3..1f9502d 100644 --- a/planetmint/backend/tarantool/query.py +++ b/planetmint/backend/tarantool/query.py @@ -117,7 +117,7 @@ def get_metadata(connection, transaction_ids: list): @register_query(TarantoolDB) -def store_asset(connection, asset: dict): +def store_asset(connection, asset): space = connection.space("assets") convert = lambda obj: obj if isinstance(obj, tuple) else (obj, obj["id"], obj["id"]) try: diff --git a/tests/tendermint/test_lib.py b/tests/tendermint/test_lib.py index 14dfe5d..da2c1ee 100644 --- a/tests/tendermint/test_lib.py +++ b/tests/tendermint/test_lib.py @@ -188,11 +188,12 @@ def test_store_transaction(mocker, b, signed_create_tx, b.connection, [{'data': signed_create_tx.asset['data'], 'tx_id': signed_create_tx.id, 'asset_id': signed_create_tx.id}] ) + else: + mocked_store_asset.assert_called_once_with( + b.connection, + [(signed_create_tx.asset, signed_create_tx.id, signed_create_tx.id)] + ) - mocked_store_asset.assert_called_once_with( - b.connection, - [{'data': signed_create_tx.asset, 'tx_id': signed_create_tx.id, 'asset_id': signed_create_tx.id}] - ) mocked_store_metadata.assert_called_once_with( b.connection, [{'id': signed_create_tx.id, 'metadata': signed_create_tx.metadata}],