Mirroristas/planetmint
2
0
Fork 0
mirror of https://github.com/planetmint/planetmint.git synced 2025-11-24 22:45:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

store and get utxo fixed

Browse Source
This commit is contained in:
andrei 2022-05-10 14:39:55 +03:00
parent c84acc689a
commit f65cb049e8
4 changed files with 48 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
planetmint/backend/tarantool/init.lua
View File

@ -64,7 +64,7 @@ keys:create_index('txid_search', {type = 'tree', unique=false, parts={'transacti
keys:create_index('output_search', {type = 'tree', unique=false, parts={'output_id'}})
utxos = box.schema.space.create('utxos', {engine = 'memtx' , is_sync = false})
utxos:format({{name='transaction_id' , type='string'}, {name='output_index' , type='integer'}})
utxos:format({{name='transaction_id' , type='string'}, {name='output_index' , type='integer'}, {name='utxo_dict', type='string'}})
utxos:create_index('id_search', {type='hash' , parts={'transaction_id', 'output_index'}})
utxos:create_index('transaction_search', {type='tree', unique=false, parts={'transaction_id'}})
utxos:create_index('index_search', {type='tree', unique=false, parts={'output_index'}})

10
planetmint/backend/tarantool/query.py
View File

@ -9,11 +9,10 @@ from secrets import token_hex
from operator import itemgetter
from planetmint.backend import query
from planetmint.backend.exceptions import DuplicateKeyError
from planetmint.backend.exceptions import OperationError
from planetmint.backend.utils import module_dispatch_registrar
from planetmint.backend.tarantool.connection import TarantoolDB
from planetmint.backend.tarantool.transaction.tools import TransactionCompose, TransactionDecompose
from json import dumps, loads
register_query = module_dispatch_registrar(query)
@ -337,7 +336,7 @@ def store_unspent_outputs(connection, *unspent_outputs: list):
space = connection.space('utxos')
if unspent_outputs:
for utxo in unspent_outputs:
space.insert((utxo['transaction_id'], utxo['output_index']))
space.insert((utxo['transaction_id'], utxo['output_index'], dumps(utxo)))
@register_query(TarantoolDB)
@ -349,9 +348,10 @@ def delete_unspent_outputs(connection, *unspent_outputs: list):
@register_query(TarantoolDB)
def get_unspent_outputs(connection):
def get_unspent_outputs(connection, query=None): # for now we don't have implementation for 'query'.
space = connection.space('utxos')
return space.select()
_utxos = space.select([]).data
return [loads(utx[2]) for utx in _utxos]
@register_query(TarantoolDB)

3
tests/conftest.py
View File

@ -540,8 +540,9 @@ def dummy_unspent_outputs():
@pytest.fixture
def utxoset(dummy_unspent_outputs, utxo_collection):
from json import dumps
for utxo in dummy_unspent_outputs:
res = utxo_collection.insert((utxo["transaction_id"], utxo["output_index"]))
res = utxo_collection.insert((utxo["transaction_id"], utxo["output_index"], dumps(utxo)))
assert res
assert len(utxo_collection.select()) == 3
return dummy_unspent_outputs, utxo_collection

29
tests/tendermint/test_lib.py
View File

@ -7,7 +7,6 @@ from operator import index
import os
from unittest.mock import patch
try:
from hashlib import sha3_256
except ImportError:
@ -45,7 +44,7 @@ def test_asset_is_separated_from_transaciton(b):
tx = Transaction.create([alice.public_key],
[([bob.public_key], 1)],
metadata=None,
asset=asset)\
asset=asset) \
.sign([alice.private_key])
# with store_bulk_transactions we use `insert_many` where PyMongo
@ -89,7 +88,7 @@ def test_validation_error(b):
alice = generate_key_pair()
tx = Transaction.create([alice.public_key],
[([alice.public_key], 1)],
asset=None)\
asset=None) \
.sign([alice.private_key]).to_dict()
tx['metadata'] = ''
@ -105,7 +104,7 @@ def test_write_and_post_transaction(mock_post, b):
alice = generate_key_pair()
tx = Transaction.create([alice.public_key],
[([alice.public_key], 1)],
asset=None)\
asset=None) \
.sign([alice.private_key]).to_dict()
tx = b.validate_transaction(tx)
@ -220,6 +219,7 @@ def test_store_transaction(mocker, b, signed_create_tx,
@pytest.mark.bdb
def test_store_bulk_transaction(mocker, b, signed_create_tx,
signed_transfer_tx, db_context):
from planetmint.backend.tarantool.connection import TarantoolDB
mocked_store_assets = mocker.patch(
'planetmint.backend.query.store_assets')
mocked_store_metadata = mocker.patch(
@ -233,9 +233,15 @@ def test_store_bulk_transaction(mocker, b, signed_create_tx,
# utxo = utxoset.find_one()
# assert utxo['transaction_id'] == signed_create_tx.id
# assert utxo['output_index'] == 0
if isinstance(b.connection, TarantoolDB):
mocked_store_assets.assert_called_once_with(
b.connection,
[ ( signed_create_tx.asset['data'], signed_create_tx.id, signed_create_tx.id )],
b.connection, # signed_create_tx.asset['data'] this was before
[(signed_create_tx.asset, signed_create_tx.id, signed_create_tx.id)],
)
else:
mocked_store_assets.assert_called_once_with(
b.connection, # signed_create_tx.asset['data'] this was before
[(signed_create_tx.asset["data"], signed_create_tx.id, signed_create_tx.id)],
)
mocked_store_metadata.assert_called_once_with(
b.connection,
@ -254,6 +260,7 @@ def test_store_bulk_transaction(mocker, b, signed_create_tx,
# utxo = utxoset.find_one()
# assert utxo['transaction_id'] == signed_transfer_tx.id
# assert utxo['output_index'] == 0
if not isinstance(b.connection, TarantoolDB):
assert not mocked_store_assets.called
mocked_store_metadata.asser_called_once_with(
b.connection,
@ -369,22 +376,22 @@ def test_get_spent_transaction_critical_double_spend(b, alice, bob, carol):
tx = Transaction.create([alice.public_key],
[([alice.public_key], 1)],
asset=asset)\
asset=asset) \
.sign([alice.private_key])
tx_transfer = Transaction.transfer(tx.to_inputs(),
[([bob.public_key], 1)],
asset_id=tx.id)\
asset_id=tx.id) \
.sign([alice.private_key])
double_spend = Transaction.transfer(tx.to_inputs(),
[([carol.public_key], 1)],
asset_id=tx.id)\
asset_id=tx.id) \
.sign([alice.private_key])
same_input_double_spend = Transaction.transfer(tx.to_inputs() + tx.to_inputs(),
[([bob.public_key], 1)],
asset_id=tx.id)\
asset_id=tx.id) \
.sign([alice.private_key])
b.store_bulk_transactions([tx])
@ -475,7 +482,7 @@ def test_get_spent_key_order(b, user_pk, user_sk, user2_pk, user2_sk):
tx1 = Transaction.create([user_pk],
[([alice.public_key], 3), ([user_pk], 2)],
asset=None)\
asset=None) \
.sign([user_sk])
b.store_bulk_transactions([tx1])
assert tx1.validate(b)