From c6feac5754077f4d34e16f5d92e8e0cd9c4fab27 Mon Sep 17 00:00:00 2001
From: andrei <deffuls@debian-BULLSEYE-live-builder-AMD64>
Date: Fri, 18 Feb 2022 16:36:04 +0200
Subject: [PATCH] test_get_block PASSED

---
 planetmint/backend/tarantool/query.py   |  1 +
 tests/backend/tarantool/test_queries.py | 10 ++++++----
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/planetmint/backend/tarantool/query.py b/planetmint/backend/tarantool/query.py
index 6162526..af95b86 100644
--- a/planetmint/backend/tarantool/query.py
+++ b/planetmint/backend/tarantool/query.py
@@ -312,6 +312,7 @@ def get_block(block_id=[], connection=None):
     space = connection.space("blocks")
     _block = space.select(block_id, index="block_search", limit=1)
     _block = _block.data[0]
+    space = connection.space("blocks_tx")
     _txblock = space.select(_block[2], index="block_search")
     _txblock = _txblock.data
     return {"app_hash": _block[0], "height": _block[1], "transactions": [_tx[0] for _tx in _txblock]}
diff --git a/tests/backend/tarantool/test_queries.py b/tests/backend/tarantool/test_queries.py
index fa5d772..89fac0e 100644
--- a/tests/backend/tarantool/test_queries.py
+++ b/tests/backend/tarantool/test_queries.py
@@ -310,17 +310,19 @@ def test_store_block():
 
 
 def test_get_block():
-    from planetmint.backend import connect, query
     from planetmint.lib import Block
-    conn = connect()
+    from planetmint.backend import connect
+    from planetmint.backend.tarantool import query
+
+    conn = connect().get_connection()
 
     block = Block(app_hash='random_utxo',
                   height=3,
                   transactions=[])
 
-    conn.db.blocks.insert_one(block._asdict())
+    query.store_block(connection=conn, block=block._asdict())
 
-    block = dict(query.get_block(conn, 3))
+    block = dict(query.get_block(connection=conn, block_id=3))
     assert block['height'] == 3