planetmint/tests/web/test_block_tendermint.py

# Copyright © 2020 Interplanetary Database Association e.V.,
# Planetmint and IPDB software contributors.
# SPDX-License-Identifier: (Apache-2.0 AND CC-BY-4.0)
# Code is Apache-2.0 and docs are CC-BY-4.0

import pytest

from transactions.types.assets.create import Create
from planetmint.lib import Block
from ipld import marshal, multihash

BLOCKS_ENDPOINT = "/api/v1/blocks/"


@pytest.mark.bdb
@pytest.mark.usefixtures("inputs")
def test_get_block_endpoint(b, client, alice):
    import copy

    tx = Create.generate(
        [alice.public_key], [([alice.public_key], 1)], assets=[{"data": multihash(marshal({"cycle": "hero"}))}]
    )
    tx = tx.sign([alice.private_key])

    # with store_bulk_transactions we use `insert_many` where PyMongo
    # automatically adds an `_id` field to the tx, therefore we need the
    # deepcopy, for more info see:
    # https://api.mongodb.com/python/current/faq.html#writes-and-ids
    tx_dict = copy.deepcopy(tx.to_dict())
    b.store_bulk_transactions([tx])

    block = Block(app_hash="random_utxo", height=31, transactions=[tx.id])
    b.store_block(block._asdict())

    res = client.get(BLOCKS_ENDPOINT + str(block.height))
    expected_response = {"height": block.height, "transactions": [tx_dict]}
    assert res.json == expected_response
    assert res.status_code == 200


@pytest.mark.bdb
@pytest.mark.usefixtures("inputs")
def test_get_block_returns_404_if_not_found(client):
    res = client.get(BLOCKS_ENDPOINT + "123")
    assert res.status_code == 404

    res = client.get(BLOCKS_ENDPOINT + "123/")
    assert res.status_code == 404


@pytest.mark.bdb
def test_get_block_containing_transaction(b, client, alice):
    tx = Create.generate(
        [alice.public_key], [([alice.public_key], 1)], assets=[{"data": multihash(marshal({"cycle": "hero"}))}]
    )
    tx = tx.sign([alice.private_key])
    b.store_bulk_transactions([tx])

    block = Block(app_hash="random_utxo", height=13, transactions=[tx.id])
    b.store_block(block._asdict())
    res = client.get("{}?transaction_id={}".format(BLOCKS_ENDPOINT, tx.id))
    expected_response = [block.height]
    assert res.json == expected_response
    assert res.status_code == 200


@pytest.mark.bdb
def test_get_blocks_by_txid_endpoint_returns_empty_list_not_found(client):
    res = client.get(BLOCKS_ENDPOINT + "?transaction_id=")
    assert res.status_code == 200
    assert len(res.json) == 0

    res = client.get(BLOCKS_ENDPOINT + "?transaction_id=123")
    assert res.status_code == 200
    assert len(res.json) == 0