fixed another set of tests

Signed-off-by: Jürgen Eckel <juergen@riddleandcode.com>
This commit is contained in:
Jürgen Eckel 2022-04-20 15:36:36 +02:00
parent fe684718dc
commit 30dbc56b5d
4 changed files with 39 additions and 30 deletions

View File

@ -87,6 +87,7 @@ def store_transactions(connection, signed_transactions: list):
@register_query(TarantoolDB) @register_query(TarantoolDB)
def get_transaction(connection, transaction_id: str): def get_transaction(connection, transaction_id: str):
_transactions = _group_transaction_by_ids(txids=[transaction_id], connection=connection) _transactions = _group_transaction_by_ids(txids=[transaction_id], connection=connection)
print(f" \n\nGET TRANSACTION : {_transactions} ")
return next(iter(_transactions), None) return next(iter(_transactions), None)
@ -175,11 +176,18 @@ def get_latest_block(connection): # TODO Here is used DESCENDING OPERATOR
space = connection.space("blocks") space = connection.space("blocks")
_all_blocks = space.select() _all_blocks = space.select()
_all_blocks = _all_blocks.data _all_blocks = _all_blocks.data
_block = sorted(_all_blocks, key=itemgetter(1))[0] hash = ''
space = connection.space("blocks_tx") heigth = 0
_txids = space.select(_block[2], index="block_search") txs = []
_txids = _txids.data if len(_all_blocks ) > 0:
return {"app_hash": _block[0], "height": _block[1], "transactions": [tx[0] for tx in _txids]} _block = sorted(_all_blocks, key=itemgetter(1))[0]
space = connection.space("blocks_tx")
_txids = space.select(_block[2], index="block_search")
_txids = _txids.data
hash = _block[0]
heigth = _block[1]
txs = [tx[0] for tx in _txids]
return {"app_hash": hash, "height": heigth, "transactions": txs}
@register_query(TarantoolDB) @register_query(TarantoolDB)
@ -376,12 +384,16 @@ def store_pre_commit_state(connection, state: dict):
@register_query(TarantoolDB) @register_query(TarantoolDB)
def get_pre_commit_state(connection) -> dict: def get_pre_commit_state(connection) -> dict:
space = connection.space("pre_commits") try:
_commit = space.select([], index="id_search", limit=1).data space = connection.space("pre_commits")
if len(_commit) == 0: _commit = space.select([], index="id_search", limit=1).data
if len(_commit) == 0:
return {}
_commit = _commit[0]
return {"height": _commit[1], "transactions": _commit[2]}
except:
return {} return {}
_commit = _commit[0]
return {"height": _commit[1], "transactions": _commit[2]}
@register_query(TarantoolDB) @register_query(TarantoolDB)
@ -457,12 +469,12 @@ def get_election(connection, election_id: str):
@register_query(TarantoolDB) @register_query(TarantoolDB)
def get_asset_tokens_for_public_key(connection, asset_id: str, public_key: str): def get_asset_tokens_for_public_key(connection, asset_id: str, public_key: str):
space = connection.space("keys") space = connection.space("keys")
_keys = space.select([public_key], index="keys_search") #_keys = space.select([public_key], index="keys_search")
space = connection.space("assets") space = connection.space("assets")
_transactions = space.select([asset_id], index="only_asset_search") _transactions = space.select([asset_id], index="assetid_search")
_transactions = _transactions.data #_transactions = _transactions
_keys = _keys.data #_keys = _keys.data
_grouped_transactions = _group_transaction_by_ids(connection=connection, txids=[_tx[0] for _tx in _transactions]) _grouped_transactions = _group_transaction_by_ids(connection=connection, txids=[_tx[1] for _tx in _transactions])
return _grouped_transactions return _grouped_transactions

View File

@ -17,8 +17,9 @@ def memoize_from_dict(func):
@functools.wraps(func) @functools.wraps(func)
def memoized_func(*args, **kwargs): def memoized_func(*args, **kwargs):
if args[1] is None:
if args[1].get('id', None): return None
elif args[1].get('id', None):
args = list(args) args = list(args)
args[1] = HDict(args[1]) args[1] = HDict(args[1])
new_args = tuple(args) new_args = tuple(args)

View File

@ -89,20 +89,16 @@ def test_bigchain_show_config(capsys):
#del sorted_config['CONFIGURED'] #del sorted_config['CONFIGURED']
assert sorted_output_config == sorted_config assert sorted_output_config == sorted_config
def test__run_init(mocker): def test__run_init(mocker):
from planetmint.commands.planetmint import _run_init
bigchain_mock = mocker.patch(
'planetmint.commands.planetmint.planetmint.Planetmint')
init_db_mock = mocker.patch( init_db_mock = mocker.patch(
'planetmint.commands.planetmint.schema.init_database', 'planetmint.backend.tarantool.connection.TarantoolDB.init_database')
autospec=True,
spec_set=True, from planetmint.backend.connection import Connection
)
_run_init() conn = Connection()
bigchain_mock.assert_called_once_with() conn.init_database()
init_db_mock.assert_called_once_with(
connection=bigchain_mock.return_value.connection) init_db_mock.assert_called_once_with()
@patch('planetmint.backend.schema.drop_database') @patch('planetmint.backend.schema.drop_database')

View File

@ -10,7 +10,7 @@ import random
from abci import types_v0_31_5 as types from abci import types_v0_31_5 as types
from planetmint.core import App from planetmint.core import App
from planetmint.backend.localmongodb import query from planetmint.backend import query
from planetmint.common.crypto import generate_key_pair from planetmint.common.crypto import generate_key_pair
from planetmint.core import (CodeTypeOk, from planetmint.core import (CodeTypeOk,
CodeTypeError, CodeTypeError,