diff --git a/planetmint/backend/tarantool/query.py b/planetmint/backend/tarantool/query.py index 7923df8..c3c5fd0 100644 --- a/planetmint/backend/tarantool/query.py +++ b/planetmint/backend/tarantool/query.py @@ -255,7 +255,7 @@ def _remove_text_score(asset): # @register_query(LocalMongoDBConnection) -def get_owned_ids(connection, owner: str): +def get_owned_ids(connection, owner: str): # FIXME LAST HERE space = connection.space("keys") _keys = space.select(owner, index="keys_search", limit=1) if len(_keys.data) == 0: diff --git a/tests/backend/tarantool/test_queries.py b/tests/backend/tarantool/test_queries.py index 794717d..1017849 100644 --- a/tests/backend/tarantool/test_queries.py +++ b/tests/backend/tarantool/test_queries.py @@ -211,13 +211,14 @@ def test_get_metadata(): def test_get_owned_ids(signed_create_tx, user_pk): - from planetmint.backend import connect, query - conn = connect() + from planetmint.backend import connect + from planetmint.backend.tarantool import query + conn = connect().get_connection() # insert a transaction - conn.db.transactions.insert_one(deepcopy(signed_create_tx.to_dict())) + query.store_transactions(connection=conn, signed_transactions=[signed_create_tx.to_dict()]) - txns = list(query.get_owned_ids(conn, user_pk)) + txns = list(query.get_owned_ids(connection=conn, owner=user_pk)) assert txns[0] == signed_create_tx.to_dict()